package org.jzy3d.plot3d.primitives.axes;

import java.awt.image.BufferedImage;
import java.util.ArrayList;
import java.util.List;

import org.apache.log4j.Logger;
import org.jzy3d.colors.Color;
import org.jzy3d.maths.BoundingBox3d;
import org.jzy3d.maths.Coord2d;
import org.jzy3d.maths.Coord3d;
import org.jzy3d.maths.PlaneAxis;
import org.jzy3d.maths.Range;
import org.jzy3d.plot3d.primitives.LineStrip;
import org.jzy3d.plot3d.primitives.axes.layout.IAxeLayout;
import org.jzy3d.plot3d.primitives.contour.ContourLevel;
import org.jzy3d.plot3d.primitives.contour.ContourMesh;
import org.jzy3d.plot3d.primitives.textured.DrawableTexture;
import org.jzy3d.plot3d.rendering.textures.BufferedImageTexture;
import org.jzy3d.plot3d.rendering.view.Camera;
import org.jzy3d.plot3d.text.align.Halign;
import org.jzy3d.plot3d.text.align.Valign;

import com.jogamp.opengl.GL;
import com.jogamp.opengl.glu.GLU;


/**
 * An axe box for displaying a contour map that
 * remains on bottom of the box, whatever the box scale is.
 * 
 * @author Martin Pernollet
 */
public class ContourAxeBox extends AxeBox {
    // static Logger logger = Logger.getLogger(ContourAxeBox.class);
    
	public ContourAxeBox(BoundingBox3d bbox, IAxeLayout layout) {
		super(bbox, layout);
	}

	public ContourAxeBox(BoundingBox3d bbox) {
		super(bbox);
	}
	
	/****************/

	public BufferedImage getContourImage() {
		return contourImg;
	}

	public void setContourImg(BufferedImage contourImage, Range xrange, Range yrange) {
		this.contourImg = contourImage;
		rebuildTexture();
	}
	
	
	
	public ContourMesh getContourMesh() {
		return mesh;
	}

	public void setContourMesh(ContourMesh contour) {
		this.mesh = contour;
	}

	/****************/

	protected void rebuildTexture(){
        BufferedImageTexture contourResource = new BufferedImageTexture(contourImg);
        float zrange = zmax - zmin; // avoids rendering problems with the tick lines on the bottom part of the cube
        this.contourTexture = new DrawableTexture(contourResource, PlaneAxis.Z, zmin - zrange/1000, getDefaultTextureMapping());        
    }

	protected List getDefaultTextureMapping(){
        List mapping = new ArrayList(4);
        mapping.add( new Coord2d(xmin, ymin) );
        mapping.add( new Coord2d(xmax, ymin) );
        mapping.add( new Coord2d(xmax, ymax) );
        mapping.add( new Coord2d(xmin, ymax) );
        return mapping;
    }
	
	/****************/
	
	@Override
    protected void setAxeBox(float xmin, float xmax, float ymin, float ymax, float zmin, float zmax){
		super.setAxeBox(xmin, xmax, ymin, ymax, zmin, zmax);
		this.xmin = xmin;
		this.xmax = xmax;
		this.ymin = ymin;
		this.ymax = ymax;
		this.zmin = zmin;
		this.zmax = zmax;
		
		if( contourImg != null )
			rebuildTexture(); 
		// TODO: could avoid rebuilding if the texture object was able
		// to reset its axisValue
		
		if( mesh != null )
			for (ContourLevel line: mesh.lines.getContourLevels()) {
				line.fixZ(zmin);
			}
	}
	
	@Override
	public void draw(GL gl, GLU glu, Camera camera){
		super.draw(gl, glu, camera);
		
		//gl.glHint(GL2.GL_PERSPECTIVE_CORRECTION_HINT, GL2.GL_NICEST);
		
		// Render the contour texture if available
		if( contourTexture != null ){
			contourTexture.draw(gl, glu, camera);
		}
		
		if( mesh != null ){
			drawMesh(gl, glu, camera, mesh);
		}
		
	}
	
	public void drawMesh(GL gl, GLU glu, Camera camera, ContourMesh mesh){
		// draw contour lines
		for (ContourLevel line: mesh.lines.getContourLevels()) {
			line.draw(gl, glu, camera);
			// logger.info("Contour level '" + line.getValue() + "' has " + line.getLines());
                        // System.out.println("Contour level '" + line.getValue() + "' has " + line.getLines());
		}
		
		// draw text
		for(Double level: mesh.getLevels()){
			String label = mesh.getLevelLabel(level);
			ContourLevel contour = mesh.lines.getContourLevel(level);
			if(label!=null && level!=null){
				for(LineStrip strip: contour.getLines()){
					Coord3d position = strip.get( strip.size()/2 ).xyz;
					txt.drawText(gl, glu, camera, label, position, Halign.CENTER, Valign.CENTER, Color.BLACK,0);
				}
			}
		}
	}
	
	/****************/

	protected float xmin;
	protected float xmax;
	protected float ymin;
	protected float ymax;
	protected float zmin;
	protected float zmax;
	protected BufferedImage contourImg;
	protected DrawableTexture contourTexture;
	protected ContourMesh mesh;
}