package org.jzy3d.plot2d.primitive;

import java.awt.Graphics2D;
import java.awt.image.BufferedImage;

import org.jzy3d.colors.Color;
import org.jzy3d.colors.ColorAWT;
import org.jzy3d.colors.ColorMapper;
import org.jzy3d.colors.colormaps.IColorMap;
import org.jzy3d.plot3d.primitives.axes.layout.providers.ITickProvider;
import org.jzy3d.plot3d.primitives.axes.layout.renderers.ITickRenderer;


/** 
 * @author Martin Pernollet
 */
public class AWTColorbarImageGenerator extends AWTAbstractImageGenerator implements AWTImageGenerator{
	public AWTColorbarImageGenerator(IColorMap map, float min, float max, ITickProvider provider, ITickRenderer renderer){
		this.mapper = new ColorMapper(map, min, max);
		this.min = min;
		this.max = max;
		this.provider = provider;
		this.renderer = renderer;
	}
	
	public AWTColorbarImageGenerator(ColorMapper mapper, ITickProvider provider, ITickRenderer renderer){
	    this.mapper = mapper;
        this.provider = provider;
        this.renderer = renderer;
        this.min = mapper.getMin();
        this.max = mapper.getMax();
        this.textSize = 12;
        this.font = new java.awt.Font("Arial",0,textSize);
	}

	@Override
    public BufferedImage toImage(int width, int height) {
        return toImage(width, height, 20);
    }
    
	/** Renders the {@link AWTColorbarImageGenerator} to an image. */
	public BufferedImage toImage(int width, int height, int barWidth){
		if(barWidth>width) 
			return null;
		BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
		Graphics2D graphic = image.createGraphics();
		configureText(graphic);
		drawBackground(width, height, graphic);
		drawBarColors(height, barWidth, graphic);
		drawBarContour(height, barWidth, graphic);
		drawTextAnnotations(height, barWidth, graphic);
		return image;
	}

    public void drawBarContour(int height, int barWidth, Graphics2D graphic) {
        graphic.setColor(ColorAWT.toAWT(foregroundColor));
		graphic.drawRect(0, textSize/2, barWidth, height-textSize);
    }

    public void drawBarColors(int height, int barWidth, Graphics2D graphic) {
        for(int h=textSize/2; h<=(height-textSize/2); h++){
			// Compute value & color
			double v = min + (max-min) * (h)/(height-textSize);
			Color c = mapper.getColor(v);  //To allow the Color to be a variable independent of the coordinates
			
			// Draw line
			graphic.setColor(ColorAWT.toAWT(c));
			graphic.drawLine(0, height-h, barWidth, height-h);
		}
    }

    public void drawTextAnnotations(int height, int barWidth, Graphics2D graphic) {
        if(provider!=null){
    		double[] ticks = provider.generateTicks(min, max);
    		int ypos;
    		String txt;
    		for(int t=0; t