ShapesDoPlot.java

Go to the documentation of this file.

package cedar.hepdata.webapp.pages;

import cedar.hepdata.model.*;
import cedar.hepdata.xml.*;
import cedar.hepdata.db.*;

import org.apache.tapestry5.annotations.*;
import org.apache.tapestry5.ioc.annotations.*;
import org.apache.tapestry5.services.*;

import org.hibernate.*;
import org.hibernate.criterion.*;
import java.util.*;
//import java.lang.Math;
import java.lang.*;

import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;

import org.apache.tapestry5.ComponentResources;
import org.apache.tapestry5.Link;
import org.apache.tapestry5.MarkupWriter;
import org.apache.tapestry5.StreamResponse;
import org.apache.tapestry5.annotations.*;
import org.apache.tapestry5.ioc.annotations.*;
import org.apache.tapestry5.annotations.*;
import org.apache.tapestry5.ioc.annotations.*;
import org.apache.tapestry5.ioc.services.TypeCoercer;
import org.apache.tapestry5.dom.Element;
import org.apache.tapestry5.services.Response;
import org.jfree.chart.ChartUtilities;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.title.TextTitle;
import org.jfree.data.xy.*;

import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.NumberAxis;
import org.jfree.chart.axis.ValueAxis;
import org.jfree.chart.axis.LogarithmicAxis;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.renderer.xy.XYErrorRenderer;
import org.jfree.chart.renderer.xy.XYItemRenderer;
import org.jfree.data.xy.XYIntervalSeries;
import org.jfree.data.xy.XYIntervalSeriesCollection;
import org.jfree.data.Range;

public class ShapesDoPlot{

    @Inject
    private org.hibernate.Session _session;
    @Inject 
    private ComponentResources _resources;

    @Persist
    private int _nplots;
    @Persist
    private List <Integer> _plots = new ArrayList();
    @Persist
    private Map <Integer,String> _expnames = new HashMap();
    @Persist
    private Map <Integer,Integer> _irns = new HashMap();
    @Persist
    private Map <Integer,Integer> _dsids = new HashMap();
    @Persist
    private Map <Integer,Integer> _yaids = new HashMap();
    @Persist
    private Map <Integer,Boolean> _ckbxs = new HashMap();
    @Persist
    private Map <Integer,String> _ecms = new HashMap();
    @Persist
    private Map <Integer,String> _vars = new HashMap();
    
    @Persist
    private String _ylinlog;
   
    public void setPlots(List plots){ _plots=plots; _nplots=_plots.size()-1; }
    public void setExpnames(Map expnames) { _expnames = expnames; }
    public void setIrns(Map irns) { _irns = irns; }
    public void setDsids(Map dsids) { _dsids = dsids; }
    public void setYaids(Map yaids) { _yaids = yaids; }
    public void setCkbxs(Map ckbxs) { _ckbxs = ckbxs; }
    public void setEcms(Map ecms) { _ecms = ecms; }
    public void setVars(Map vars) { _vars = vars; }
    public void setYlinlog(String ylinlog){ _ylinlog = ylinlog; } 
   
    private int _irn;
    private int _dsid;
    private int _yaid;   
     
    private YAxis getYAxis(){return this.getYAxis(0);}
    private YAxis getYAxis(int nplot){   
        Query q =null;
        q = _session.createQuery("select y from Paper p, Dataset d, YAxis y where p._spiresId=:irn and d._localId=:dsid and d._paper=p and y._localId=:yaid and y._dataset=d");
        q.setLong("irn",_irns.get(nplot));
        q.setLong("dsid",_dsids.get(nplot));
        q.setLong("yaid",_yaids.get(nplot));
        return (YAxis) q.list().get(0);
    
    }  
    
    static private int numberRun = 0;
    static private HashMap seriesMap = new HashMap();
    static private String titlestring = "Title:";

    public Link getChart1(){    
        if (numberRun > 10000) {
            numberRun=0;
        }
        numberRun += _nplots+1;
        int _nn = -1;
        for (int n = 0; n<=_nplots; n++){
           if( _ckbxs.get(n) ){         
             _nn += 1;
             XYIntervalSeries series = getDataSeries(n,getYAxis(n));
             seriesMap.put(numberRun+_nn,series);
           }  
        }
        return _resources.createActionLink("chart", false, new Object[]{"400","400",numberRun,_nn+1});
    }

    public XYIntervalSeries getDataSeries(int n, YAxis yaxis){
        XAxis _xaxis =  yaxis.getDataset().getXAxis(1);
        for (String comment : yaxis.getDataset().getComments()){
            titlestring = comment;
        }
//        XYIntervalSeries series = new  XYIntervalSeries(yaxis.getDataset().getPaper().getSpiresId());
        XYIntervalSeries series = new  XYIntervalSeries(
        _expnames.get(n)
        .concat("/")
        .concat(_irns.get(n).toString())
        .concat("/")
        .concat(_dsids.get(n).toString())
        .concat("/")
        .concat(_yaids.get(n).toString())
        .concat("/")
        .concat(_ecms.get(n))
        );
        for (int ip=1; ip<=yaxis.getPoints().size(); ip++){
            double y = yaxis.getPoint(ip).getValue();
            double eplus = 0.0;
            double eminus = 0.0;
            for (Uncertainty e : yaxis.getPoint(ip).getErrors()){
               eplus = eplus + e.getPlus()*e.getPlus(); 
               eminus = eminus - e.getMinus()*e.getMinus(); 
            }
            eplus = Math.sqrt(eplus);
            eminus = Math.sqrt(eminus);
            double yhigh = y + eplus;
            double ylow = y - eplus;
            double x;
            double xhigh;
            double xlow;
            if(_xaxis.getBin(ip).getLowValue() != null  && _xaxis.getBin(ip).getHighValue() != null){
                x = (_xaxis.getBin(ip).getLowValue()+_xaxis.getBin(ip).getHighValue())/2;
                xlow =  _xaxis.getBin(ip).getLowValue();
                xhigh =  _xaxis.getBin(ip).getHighValue();
            }
            else{
                x = _xaxis.getBin(ip).getFocus();
                xhigh = _xaxis.getBin(ip).getFocus();
                xlow = _xaxis.getBin(ip).getFocus();
            }
            series.add(x,xlow,xhigh,y,ylow,yhigh);
        }
            return series;
    }
    public StreamResponse onChart(final int width, final int height, final int numberRun, int number){
 
        final XYIntervalSeriesCollection collection = new XYIntervalSeriesCollection();
        for (int n=0; n<number; n++){
            XYIntervalSeries series = (XYIntervalSeries) seriesMap.get(numberRun+n);
            seriesMap.remove(numberRun+n);
            collection.addSeries(series);
        }
        IntervalXYDataset data1 = collection;
 
        final XYItemRenderer renderer1 = new XYErrorRenderer();
        final Range rangex = new Range(0.0,1.0);
        System.out.println(rangex.toString());
        final NumberAxis rangeAxisX = new NumberAxis(getYAxis(0).getDataset().getXAxis(1).getHeader());
        rangeAxisX.setRange(rangex);
//        rangeAxisX.setAutoRangeIncludesZero(false);
        NumberAxis rangeAxisY = null;
        String _label = getYAxis(0).getHeader();
        System.out.println("log lin is " + _ylinlog);
        if(_ylinlog.equals("YLIN")) { 
           rangeAxisY = new NumberAxis(_label);
        } 
        else{
            rangeAxisY = new LogarithmicAxis(_label);
            ((LogarithmicAxis)rangeAxisY).setAllowNegativesFlag(true);
            ((LogarithmicAxis)rangeAxisY).setAutoRangeNextLogFlag(true);
        }
        final XYPlot subplot1 = new XYPlot(data1, rangeAxisX, rangeAxisY, renderer1);
        final TextTitle title = new TextTitle(titlestring);
        final JFreeChart chart = new JFreeChart(subplot1);
        chart.addSubtitle(title);
        return new StreamResponse(){
            public String getContentType(){
                return "image/png";
            }
            public InputStream getStream() throws IOException {
                BufferedImage image  = chart.createBufferedImage(width, height);
                ByteArrayOutputStream byteArray = new ByteArrayOutputStream() ;
                ChartUtilities.writeBufferedImageAsPNG(byteArray, image) ;
                return new ByteArrayInputStream(byteArray.toByteArray());
            }
            public void prepareResponse(Response response){}
        };
    }
    
}

---