JetWebHist.java

Go to the documentation of this file.

package cedar.jetweb.hist;

import hep.aida.*;
import hep.aida.ref.plotter.PlotterUtilities;

import java.io.PrintWriter;
import java.io.FileWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.BufferedOutputStream;
import java.io.OutputStream;
import java.io.BufferedWriter;
import java.io.FileNotFoundException;
import java.io.IOException;

import java.util.Enumeration;
import java.util.Iterator;
import java.util.Vector;
import java.util.Properties;

import java.awt.Dimension;
import java.awt.Image;
import java.awt.Graphics;
import java.awt.image.ImageProducer;
import java.awt.Color;

//import javax.swing.JComponent;
//import javax.swing.JFrame;
//import javax.swing.SwingUtilities;

import cedar.jetweb.JetWebConfig;
import cedar.jetweb.model.paper.Paper;
import cedar.jetweb.model.plots.DataPlot;
import cedar.jetweb.model.plots.DataPoint;
import cedar.jetweb.model.plots.FittedPlot;
import cedar.jetweb.model.plots.PredictedPlot;

//import org.apache.commons.io.FileUtils;

public abstract class JetWebHist {
    
    public static void writeGraphics(Paper paper, String path){

    String plotPath = path+"/"+paper.getDirName();

        for (DataPlot plot : paper.getPlots()){
            writeGraphics(plot,plotPath);
        }
    }
    
    protected static IDataPointSet buildHist(DataPlot thisPlot, String path, String title){
        
        IDataPointSet data = null;
        try{
            System.setProperty("java.awt.headless", "true");
            IAnalysisFactory af = IAnalysisFactory.create();
            ITree tree = null;
            if(path!=null) {
                tree = af.createTreeFactory().create(path+"/plot"+thisPlot.getNumber()+".xml",
                             "xml",false,true);
                System.out.println("allow me to write you some XML :)");
                
            }else{
                tree = af.createTreeFactory().create();
            }
            
            IDataPointSetFactory dpsf = af.createDataPointSetFactory(tree);
            
          
            
            if(thisPlot.getNDataSources()==1) {
                //System.out.println("HERE!");
                Vector<DataPoint> points = thisPlot.getDataPoints();
                
                //data = dpsf.create(thisPlot.getTitle(),thisPlot.getTitle(),2);     //sort this out
                //data = dpsf.create("/data",thisPlot.getTitle(),2);
                data = dpsf.create("/data",title,2);
                
                for(int i=0;i<points.size();i++) {
                    
                    DataPoint p = points.elementAt(i);
                    data.addPoint();
                    data.point(i).coordinate(0).setValue(p.getX());
                    data.point(i).coordinate(1).setValue(p.getY()*thisPlot.getSysScale());
                    data.point(i).coordinate(1).setErrorPlus(p.getYUp()*thisPlot.getSysScale());
                    data.point(i).coordinate(1).setErrorMinus(p.getYDown()*thisPlot.getSysScale());
                    
                    
                }
                
                
            }else{
                System.out.println("SHOULDNT BE HERE");
            }
            
            if(path!=null) {
                tree.commit();
            }
            
        }catch(Exception e) {
            e.printStackTrace();
        }
        
        return data;
    }
    
    public static void writeGraphics(DataPlot plot, String path) {
        try{

        File plotDir = new File(path);
        //System.out.println("I am checking "+path);
        if (!plotDir.exists()){
        try {
            //FileUtils.forceMkdir(plotDir);
            plotDir.mkdirs();
        } catch (Exception io) {
            System.out.println("JetWebHist: failed to make "+plotDir);
            return;
        }       
        }


            IAnalysisFactory af = IAnalysisFactory.create();
            IPlotterFactory pf = af.createPlotterFactory();
            IPlotter page1 = pf.create("page1");
        page1.setParameter("plotterWidth","200");
        page1.setParameter("plotterHeight","200");

            
        IDataPointSet histMC=null;
        IDataPointSet histReal=null;
            page1.createRegions(1,1);
            
        DataPlot comparisonPlot = null;

        //page1.region(0).style().dataStyle().markerStyle().setColor("black");
        //page1.region(0).style().dataStyle().markerStyle().setShape("circle");

        page1.region(0).style().setParameter("showStatisticsBox","false");
        page1.region(0).setTitle(plot.getTitle());
        //      page1.region(0).style().setParameter("showTitle","false");


        //This styles overwrite some of the region styles
        IPlotterStyle styleReal = pf.createPlotterStyle();
        styleReal.dataStyle().markerStyle().setParameter("shape","1");
        styleReal.dataStyle().markerStyle().setParameter("color","black");
        IPlotterStyle styleMC = pf.createPlotterStyle();
        styleMC.dataStyle().markerStyle().setParameter("shape","2");
        styleMC.dataStyle().markerStyle().setParameter("color","red");

            //list available parameters

        /*      
            String[] styles = styleReal.availableParameters();
            for(int t = 0;t<styles.length;t++) {
        System.out.println(styles[t]);
            }
        */
            
            if (plot instanceof FittedPlot){
                comparisonPlot = ((FittedPlot)plot).getCompPlot();
        histMC = buildHist(plot,null,"Simulation"); //set at path if you want xml

            } else if (plot instanceof PredictedPlot){
                comparisonPlot = ((PredictedPlot)plot).getCompPlot();
        histMC = buildHist(plot,null,"Simulation"); //set at path if you want xml

        } else {
        histReal = buildHist(plot,null,"Measurement"); //set at path if you want xml
        }

        if ((comparisonPlot!=null)){
        histReal = buildHist(comparisonPlot,null,"Measurement"); //set at path if you want xml
            }
            
            if(plot.isLogarithmic()) {
                System.out.println("Using logarithmic scale");
                double ymx = plot.getYMax();
                double ymn = plot.getYMinNonZero();
                
                if(comparisonPlot !=null) {
                    ymx = Math.max(ymx,comparisonPlot.getYMax());
                    ymn = Math.max(ymn,comparisonPlot.getYMinNonZero());
                    
                }
                
                page1.region(0).style().yAxisStyle().setParameter("scale","logarithmic");
                page1.region(0).setYLimits(ymn/2.0,ymx*2.0);
            }else {
                double ymx = plot.getYMax();
                double ymn = plot.getYMinNonZero();
                
                if(comparisonPlot !=null) {
                    ymx = Math.max(ymx,comparisonPlot.getYMax());
                    ymn = Math.max(ymn,comparisonPlot.getYMinNonZero());
                    
                }
                page1.region(0).setYLimits(0,ymx*1.1);
                page1.region(0).style().xAxisStyle().setLabel(plot.getXLabel());
                page1.region(0).style().yAxisStyle().setLabel(plot.getYLabel());
                
            }

        if (histMC!=null) page1.region(0).plot(histMC,styleMC);     
        page1.region(0).plot(histReal,styleReal);
            
            page1.region(0).style().xAxisStyle().setLabel(plot.getXLabel());
            page1.region(0).style().yAxisStyle().setLabel(plot.getYLabel());
            
        Properties props = new Properties();
        props.setProperty("plotWidth",String.valueOf(300));
        props.setProperty("plotHeight",String.valueOf(300));

            //System.out.println("JetWebHist writing: "+path);
            PlotterUtilities.writeToFile(page1,path+"/plot"+plot.getNumber()
                     +"."+JetWebConfig.graphicsType,
                     JetWebConfig.graphicsType,props);
        }catch (Exception e) {
            e.printStackTrace();
        }
    return;
    }
    
    
}

Generated Wed Jan 17 09:14:27 GMT 2007