MergedPlot.java

Go to the documentation of this file.

package cedar.jetweb.model.plots;

import cedar.jetweb.JetWebException;
import cedar.jetweb.model.MCProcessType;
import cedar.jetweb.db.DBManager;

import java.util.HashMap;
import java.util.Vector;
import java.util.Collection;

public class MergedPlot extends DataPlot{

    private Integer runSeriesCollectionId_=-1;

    //The predicted plots, indexed by their runseries id
    private HashMap<Integer, PredictedPlot> predictedPlots_ = 
    new HashMap<Integer, PredictedPlot>();
    
    private boolean mergePts_ = true;

    public MergedPlot(){

    }

    public MergedPlot(Integer runSeriesCollectionId){

    runSeriesCollectionId_ = runSeriesCollectionId;

    }

    public MergedPlot(PredictedPlot plot){
    try{
        addPlot(plot);
    }catch(JetWebException err){
        System.out.println
        ("MergedPlot: cannot instantiate MergedPlot " + 
         "from PredictedPlot");
        System.out.println(err.getMessage());
    }
    }

    public MCProcessType getMCProcessType(){

    return new MCProcessType();
    }

    public int getProcIdWanted(){

    return -1;
    }

    public Integer getRunSeriesCollectionId(){
    return runSeriesCollectionId_;
    }
    

    public MergedPlot addPlot(PredictedPlot plot) 
    throws JetWebException{
    
    addPlot(plot, true);

    return this;
    }
    public MergedPlot addPlot(PredictedPlot plot, boolean merge)
    throws JetWebException{
        if(predictedPlots_.size()!=0 && plot.getId()!= getId()){
        throw new JetWebException
        ("MergedPlot: Cannot add Predicted plot " + plot.getId() +  
         "that doesn't match existing plots " + getId(), plot.toString());
    }

    if(predictedPlots_.size()==0){
        setId(plot.getId());
        setNumber(plot.getNumber());
        setCollisionId(plot.getCollisionId());
        setLogarithmic(plot.isLogarithmic());
        setTitle(plot.getTitle());
        setYLabel(plot.getYLabel());
        setXLabel(plot.getXLabel());
        setShape(plot.isShape());
        setPaperId(plot.getPaperId());
        proc = plot.getMCProcessType();

        if(cuts_==null){
        setCutCollection(plot.getCutCollection());
        }else if(!cuts_.equals(plot.getCutCollection())){
        setCutCollection(plot.getCutCollection().union(cuts_));
        }

    }

    predictedPlots_.put(plot.getRunSeriesId(), plot);
    if(merge) mergePoints();

    return this;
    }

    public MergedPlot add(MergedPlot plot)
    throws JetWebException{

    if(plot.getId()>0 && getId() > 0 &&
       plot.getId()!= getId()){
        System.out.println
        ("MergedPlot: Cannot add two merged plots with different ids");
    }

    for(PredictedPlot pPlot: plot.getPredictedPlots()){
        addPlot(pPlot, false);
    }

    mergePoints();
    return this;
    }
    public Collection<PredictedPlot> getPredictedPlots(){
    return predictedPlots_.values();
    }

    public PredictedPlot getPredictedPlot(Integer runSeriesId){
    return predictedPlots_.get(runSeriesId);
    }

    public MergedPlot mergePoints()throws JetWebException{

    HashMap<Integer, DataPoint> foundPts = 
        new HashMap<Integer, DataPoint>();

    for(PredictedPlot plot: getPredictedPlots()){
        for(DataPoint pt: plot.getDataPoints()){
        DataPoint existingPt = foundPts.get(pt.getNumber());
        if(existingPt!=null){
            existingPt.add(pt);
            foundPts.put(pt.getNumber(), existingPt);
        }else{
            foundPts.put(pt.getNumber(), pt);
        }
        }
        setShape(plot.isShape());
    }

    dataPoints = new Vector<DataPoint>(foundPts.values());

    Vector<Integer> runseriesIds = 
        new Vector<Integer>(predictedPlots_.keySet());

    DBManager.getRunSeriesCollectionId(runseriesIds);
    setHasData(true);
    

    return this;
    }
}

Generated Wed Jan 17 09:14:27 GMT 2007