MultiColorScatter.java :  » Java-3D » jzy3d » org » jzy3d » plot3d » primitives » Java Open Source

package org.jzy3d.plot3d.primitives;

import javax.media.opengl.GL2;
import javax.media.opengl.glu.GLU;

import org.jzy3d.colors.Color;
import org.jzy3d.colors.ColorMapper;
import org.jzy3d.colors.IMultiColorable;
import org.jzy3d.events.DrawableChangedEvent;
import org.jzy3d.maths.BoundingBox3d;
import org.jzy3d.maths.Coord3d;
import org.jzy3d.plot3d.primitives.axes.layout.providers.ITickProvider;
import org.jzy3d.plot3d.primitives.axes.layout.renderers.ITickRenderer;
import org.jzy3d.plot3d.rendering.legends.colorbars.ColorbarLegend;
import org.jzy3d.plot3d.rendering.view.Camera;



/** 
 * A scatter plot supporting a colormap for shading each dot color and alpha.
 * 
 * @author Martin Pernollet
 *
 */
public class MultiColorScatter extends AbstractDrawable implements IMultiColorable{
  public MultiColorScatter(Coord3d[] coordinates, Color[] colors, ColorMapper mapper){
        this(coordinates, colors, mapper, 1.0f);
    }
  public MultiColorScatter(Coord3d[] coordinates, ColorMapper mapper){
        this(coordinates, null, mapper, 1.0f);
    }
    public MultiColorScatter(Coord3d[] coordinates, Color[] colors, ColorMapper mapper, float width){
        bbox = new BoundingBox3d();
        setData(coordinates);
    setColors(colors);
        setWidth(width);
        setColorMapper(mapper);
    }
      
    public void clear(){
        coordinates = null;
        bbox.reset();
    }
    
    public void enableColorBar(ITickProvider provider, ITickRenderer renderer){
        setLegend( new ColorbarLegend(this, provider, renderer) );
        setLegendDisplayed(true);
    }
        
    /**********************************************************************/
    
    public void draw(GL2 gl, GLU glu, Camera cam){
        if(transform!=null)
            transform.execute(gl);
        
        gl.glPointSize(width);      
        gl.glBegin(GL2.GL_POINTS);

        if(coordinates!=null){
            for(Coord3d coord: coordinates){
                Color color = mapper.getColor(coord); // TODO: should store result in the point color
                gl.glColor4f(color.r, color.g, color.b, color.a);
                gl.glVertex3f(coord.x, coord.y, coord.z);
            }
        }
        gl.glEnd();
    }

    /*********************************************************************/
    
    /** 
     * Set the coordinates of the point.
     * @param xyz point's coordinates
     */
    public void setData(Coord3d[] coordinates){
        this.coordinates = coordinates;
        
        bbox.reset();
        for(Coord3d c: coordinates)
            bbox.add(c);
    }    
    public Coord3d[] getData(){
        return coordinates;
    }
  public void setColors(Color[] colors){
    this.colors = colors;
    
    fireDrawableChanged(new DrawableChangedEvent(this, DrawableChangedEvent.FIELD_COLOR));
  }    
    @Override
    public ColorMapper getColorMapper() {
        return mapper;
    }

    @Override
    public void setColorMapper(ColorMapper mapper) {
        this.mapper = mapper;       
    }
        
    /**
     * Set the width of the point.
     * @param width point's width
     */
    public void setWidth(float width){
        this.width = width;
    }
    
    /**********************************************************************/
    
    protected Coord3d[] coordinates;
  protected Color[]   colors;   
  protected float     width;
    protected ColorMapper mapper;
    
}