DBFitManager.java

Go to the documentation of this file.

package cedar.jetweb.db;

import java.util.Vector;

import cedar.jetweb.JetWebConfig;
import cedar.jetweb.JetWebException;
import cedar.jetweb.model.Model;
import cedar.jetweb.model.MCProcessType;
import cedar.jetweb.job.CutCollection;
import cedar.jetweb.model.fit.Fit;
import cedar.jetweb.model.plots.PlotSelection;
import cedar.jetweb.model.RunSeries;

import java.io.File;

import java.sql.*;

import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;

public abstract class DBFitManager {
    public static final String SUCCESS = "Operation Completed Successfully";

    public static synchronized String selectFromDB(Fit fit)
        throws JetWebException {
        String query = "";

        try {
            if (fit.getId() >= 0) {
                query = "SELECT * FROM fit WHERE fit_id=" + fit.getId();
            } else {
                query = "SELECT * FROM fit WHERE mdl_id=" + fit.getModel().getId() +
                    " AND csn_set_id=" + fit.getPlotSelection().getId();
            }
        } catch (Exception e) {
            throw new JetWebException(e,"Error getting model");
        }

        String gotit = "";

        try {
            Statement stmt = DBConfig.getConnection().createStatement();
            ResultSet rs = stmt.executeQuery(query);

            if (rs.next()) {
                gotit = fill(fit, rs);
            } 

        rs.close();
        stmt.close();
        } catch (SQLException e) {
            try {
                System.out.println("Error reading fit table " + e + "ModelId=" +
                    fit.getModel().getId() + " CSSet=" + fit.getPlotSelection().getId());
                throw new JetWebDBException(e, query, "FitId:" + fit.getId());
            } catch (JetWebException je) {
                System.out.println("Error getting model " + je + "FitId:" + fit.getId());
                throw je;
            }
        }

        return gotit;
    }

    private static String fill(Fit fit, ResultSet rs) throws JetWebException {
        String rtn = SUCCESS;

        try {
            fit.setId(rs.getInt("fit_id"));
        Model model = new Model(rs.getInt("mdl_id"));
            //fit.setModel(model);
            fit.setModelId(model.getId());

        //need to get a list of runseries that generate data for this fit
        String select  = 
        "SELECT runseries_id FROM runseries_collection, model "+
        "WHERE " + 
        "runseries_collection.runseries_collection_id = " +
        "model.runseries_collection_id AND " + 
        "model.mdl_id = " + model.getId();

        Statement stmt = DBConfig.getConnection().createStatement();
        ResultSet runseriesRs = stmt.executeQuery(select);

        HashMap<MCProcessType, CutCollection> processCuts = 
        new HashMap<MCProcessType, CutCollection>();

        while(runseriesRs.next()){
        Integer rsid = runseriesRs.getInt(1);

        if(rsid!=null){
            //RunSeries runseries = new RunSeries(rsid);
            RunSeries runseries = RunSeries.Maker(rsid);
            MCProcessType proc = runseries.getMCProcessType();
            CutCollection cuts = runseries.getCutCollection();

            CutCollection existingCuts = 
            processCuts.get(proc);

            if(existingCuts==null ||
               existingCuts.isWithin(cuts)
               ){
            processCuts.put(proc, cuts);
            }

        }

        System.out.println("processCuts = " + processCuts);

        }

        PlotSelection selection  = 
        new PlotSelection(rs.getInt("csn_set_id"));
        
        for(Map.Entry entry: processCuts.entrySet()){
        selection.restrictCutCollection
            ((MCProcessType)entry.getKey(), 
             (CutCollection)entry.getValue());
        }

        //fit.setPlotSelection(new PlotSelection(rs.getInt("csn_set_id")));
        fit.setPlotSelection(selection);
        Date fdate = rs.getTimestamp("time");
            fit.setDate(fdate);

            if (!(model.getDate() == null)) {
                if (fdate.before(model.getDate())) {
                    return "Fit is older than Model";
                }
            }

            fit.setScale(rs.getFloat("scalefactor"));

            double[] tmpChi2 = { 0., 0. };
            tmpChi2[0] = rs.getFloat("chi2");
            tmpChi2[1] = rs.getFloat("dof");
            fit.setChi2(tmpChi2);
        } catch (SQLException e) {
            System.out.println("Error reading fit table " + e + "ModelId=" +
                fit.getModel().getId() + " CSSet=" + fit.getPlotSelection().getId());
            throw new JetWebDBException(e, null,
                "Error reading fit table " + "ModelId=" + fit.getModel().getId() + " CSSet=" +
                fit.getPlotSelection().getId());
        } catch (JetWebException je) {
            System.out.println("Error getting model " + je);
        }

        return rtn;
    }

    public static synchronized void addFit(Fit fit) throws JetWebException {
        String insertFit = makeInsertFitString(fit);

        try {
            Statement stmt = DBConfig.getConnection().createStatement();
            stmt.executeUpdate(insertFit);
        stmt.close();
        } catch (SQLException e) {
            try {
                System.out.println("Error inserting fit results " + e + "ModelId=" +
                    fit.getModel().getId() + " CSSet=" + fit.getPlotSelection().getId());
                throw new JetWebDBException(e, insertFit,
                    "Error inserting fit results " + "ModelId=" + fit.getModel().getId() +
                    " CSSet=" + fit.getPlotSelection().getId());
            } catch (JetWebException je) {
                System.out.println("Error getting model " + je);
                throw je;
            }
        }

        try {
            // Set the fitId in the Fit object
            fit.setId(selectFitId(fit.getModel().getId(), fit.getPlotSelection().getId()));
        } catch (JetWebException je) {
            System.out.println("Error getting model " + je);
            throw je;
        }
    }

    private static int selectFitId(int mdlId, int cssId)
        throws JetWebException {
        int fitId = 0;
        String query = "SELECT fit_id FROM fit WHERE mdl_id=" + mdlId + " AND csn_set_id=" + cssId;

        try {
            Statement stmt = DBConfig.getConnection().createStatement();
            ResultSet rs = stmt.executeQuery(query);

            if (rs.next()) {
                fitId = rs.getInt("fit_id");
            }
        rs.close();
        stmt.close();
        } catch (SQLException e) {
            System.out.println("Error reading fit table " + e + "ModelId=" + mdlId + " CSSet=" +
                cssId);
            throw new JetWebDBException(e, query,
                "Error reading fit table " + "ModelId=" + mdlId + " CSSet=" + cssId);
        }

        return fitId;
    }

    public static synchronized void deleteAll(int fitId)
        throws JetWebException {
        String deleteString = "";
        String deleteString1 = "DELETE FROM fitted_prediction WHERE fit_id=" + fitId;
        String deleteString2 = "DELETE FROM fitted_point WHERE fit_id=" + fitId;
        String deleteString3 = "DELETE FROM fit WHERE fit_id=" + fitId;

    //System.out.println("Called Fit.fit()");
    //JetWebException e2 = new JetWebException("help","help");
    //e2.printStackTrace();



        try {
            Statement stmt = DBConfig.getConnection().createStatement();
            deleteString = deleteString1;
        System.out.println(deleteString);
            System.out.println(stmt.executeUpdate(deleteString));
            deleteString = deleteString2;
        System.out.println(deleteString);
            System.out.println(stmt.executeUpdate(deleteString));
            deleteString = deleteString3;
        System.out.println(deleteString);
            System.out.println(stmt.executeUpdate(deleteString));
            stmt.close();
        } catch (SQLException e) {
            System.out.println("Error deleting fit results " + e + "FitId=" + fitId);
            throw new JetWebDBException(e, deleteString, "FitId:" + fitId);
        }
    }

    private static String makeInsertFitString(Fit fit) throws JetWebException {
        String ins1 = "INSERT INTO fit (";
        String ins2 = "VALUES(";

        if (fit.getId() > 0) {
            ins1 = ins1 + "fit_id,";
            ins2 = ins2 + fit.getId() + ",";
        }

        ins1 = ins1 + "mdl_id,csn_set_id,scalefactor," + "chi2,dof)";

        fit.setDate(new Date());

        try {
            ins2 = ins2 + fit.getModel().getId() + "," + fit.getPlotSelection().getId() + "," + fit.getScale();
        } catch (JetWebException je) {
            System.out.println("Error getting model " + je);
        }

        ins2 = ins2 + "," + fit.getChi2()[0] + "," + fit.getChi2()[1] + ")";

        //System.out.println(ins1+ins2);
        return ins1 + ins2;
    }

    public static void zeroFit(int fitId) {
        String zeroIt = "update fitted_point set y=0.0,dy_up=0.0,dy_down=0.0 where fit_id=" +
            fitId;

        try {
            Statement stmt = DBConfig.getConnection().createStatement();
            stmt.executeUpdate(zeroIt);
        stmt.close();
        } catch (SQLException e) {
            System.out.println("DBFitManager: Error executing " + zeroIt);
        }

        zeroIt = "update fitted_prediction set chi2=0.0,dof=0.0 where fit_id=" + fitId;

        try {
            Statement stmt = DBConfig.getConnection().createStatement();
            stmt.executeUpdate(zeroIt);
        stmt.close();
        } catch (SQLException e) {
            System.out.println("DBFitManager: Error executing " + zeroIt);
        }
    }

    public static synchronized Vector<Fit> select(Model model)
        throws JetWebException {

    Vector<Fit> fits = new Vector<Fit>();
        String query = "";

        try {
        query = "SELECT fit_id FROM fit WHERE mdl_id=" + model.getId();
        } catch (Exception e) {
            throw new JetWebException(e,"Error getting model");
        }

        try {
            Statement stmt = DBConfig.getConnection().createStatement();
            ResultSet rs = stmt.executeQuery(query);

            while (rs.next()) {
        Fit fit = new Fit(rs.getInt("fit_id"));
        fits.add(fit);
            }
        rs.close();
        stmt.close();
        } catch (SQLException e) {
        throw new JetWebDBException(e, query, "ModelId:" + model.getId());
        }

        return fits;
    }


}

Generated Wed Jan 17 09:14:27 GMT 2007