Author: [log in to unmask]
Date: Wed Jun 10 16:08:41 2015
New Revision: 3125

Log:
Add ECal MC test case (not working yet). HPSJAVA-545

Added:
    java/trunk/integration-tests/src/test/java/org/hps/test/
    java/trunk/integration-tests/src/test/java/org/hps/test/it/
    java/trunk/integration-tests/src/test/java/org/hps/test/it/EcalSimReconTest.java

Added: java/trunk/integration-tests/src/test/java/org/hps/test/it/EcalSimReconTest.java
 =============================================================================
--- java/trunk/integration-tests/src/test/java/org/hps/test/it/EcalSimReconTest.java	(added)
+++ java/trunk/integration-tests/src/test/java/org/hps/test/it/EcalSimReconTest.java	Wed Jun 10 16:08:41 2015
@@ -0,0 +1,156 @@
+package org.hps.test.it;
+
+import hep.aida.IHistogram1D;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.List;
+
+import junit.framework.TestCase;
+
+import org.hps.job.JobManager;
+import org.hps.recon.ecal.cluster.ClusterUtilities;
+import org.hps.test.util.TestFileUrl;
+import org.hps.test.util.TestOutputFile;
+import org.lcsim.event.Cluster;
+import org.lcsim.event.EventHeader;
+import org.lcsim.util.Driver;
+import org.lcsim.util.aida.AIDA;
+import org.lcsim.util.loop.LCSimLoop;
+
+/**
+ * Run readout simulation and full 2015 Engineering Run reconstruction on ECal MC input data and then check histograms of cluster data.
+ * 
+ * @author Jeremy McCormick, SLAC
+ */
+public class EcalSimReconTest extends TestCase {
+        
+    /**
+     * Test input file name on lcsim.org site.
+     */
+    private static final String FILE_NAME = "egsv3-triv2-g4v1_s2d6_HPS-EngRun2015-Nominal-v1_101.slcio";
+    
+    /**
+     * Steering resource file for readout simulation.
+     */    
+    private static final String READOUT_STEERING = "/org/hps/steering/readout/EngineeringRun2015TrigPairs1.lcsim";
+    
+    /**
+     * Steering resource file for running reconstruction.
+     */
+    private static final String RECON_STEERING = "/org/hps/steering/recon/EngineeringRun2015FullReconMC.lcsim";
+    
+    /**
+     * Run number for conditions system.
+     */
+    private static final Integer RUN = 0;
+    
+    /**
+     * Run the test.
+     * 
+     * @throws Exception if there is an uncaught exception thrown
+     */
+    public void testEcalSimRecon() throws Exception {
+        
+        // Get the input events file.
+        File inputFile = TestFileUrl.getInputFile(EcalSimReconTest.class, FILE_NAME);
+                
+        // Run readout simulation on MC input.
+        File readoutFile = new TestOutputFile(EcalSimReconTest.class, "readout.slcio");
+        JobManager job = new JobManager();
+        job.addVariableDefinition("detector", "HPS-EngRun2015-Nominal-v1");
+        job.addVariableDefinition("outputFile", readoutFile.getPath().replace(".slcio", ""));
+        job.addVariableDefinition("run", RUN.toString());
+        job.addInputFile(inputFile);
+        job.setup(READOUT_STEERING);
+        job.configure();
+        job.run();
+        System.out.println("readout ran on " + job.getLCSimLoop().getTotalCountableConsumed() + " events");
+        
+        // Run the recon on the readout output.
+        File reconFile = new TestOutputFile(EcalSimReconTest.class, "recon.slcio");
+        job = new JobManager();
+        job.addInputFile(readoutFile);
+        job.addVariableDefinition("detector", "HPS-EngRun2015-Nominal-v1");
+        job.addVariableDefinition("outputFile", reconFile.getPath().replace(".slcio", ""));
+        job.addVariableDefinition("run", RUN.toString());
+        job.setup(RECON_STEERING);
+        job.run();
+        System.out.println("recon ran on " + job.getLCSimLoop().getTotalCountableConsumed() + " events");
+        
+        // Check the recon output.
+        LCSimLoop loop = new LCSimLoop();
+        loop.setLCIORecordSource(reconFile);
+        loop.add(new EcalSimReconCheckDriver());
+        loop.loop(-1);
+        loop.dispose();
+        System.out.println("EcalSimReconCheckDriver ran on " + loop.getTotalCountableConsumed() + " events");
+    }
+    
+    /**
+     * Driver for checking test output.
+     * 
+     * @author Jeremy McCormick, SLAC
+     */
+    static class EcalSimReconCheckDriver extends Driver {
+        
+        /**
+         * Setup AIDA histogramming.
+         */
+        private final static AIDA aida = AIDA.defaultInstance();
+        
+        /**
+         * Count of clusters per event.
+         */
+        private final IHistogram1D clusCountH1D = aida.histogram1D("Cluster Count", 20, -0.5, 19.5);
+        
+        /**
+         * Individual cluster energies.
+         */
+        private final IHistogram1D clusEnergyH1D = aida.histogram1D("Cluster Energy", 100, -0.5, 9.5);
+        
+        /**
+         * Highest cluster energy in event.
+         */
+        private final IHistogram1D clusHighEnergyH1D = aida.histogram1D("Cluster Highest Energy", 100, -0.5, 9.5);
+        
+        /**
+         * First hit time in highest energy cluster.
+         */
+        private final IHistogram1D clusTimeH1D = aida.histogram1D("Cluster Time", 500, -0.5, 499.5);
+                
+        /**
+         * Process events and fill histograms from cluster collection.
+         */
+        public void process(EventHeader event) {
+            List<Cluster> clusters = event.get(Cluster.class, "EcalClustersCorr");
+            clusCountH1D.fill(clusters.size());
+            for (Cluster cluster : clusters) {
+                clusEnergyH1D.fill(cluster.getEnergy());
+            }
+            
+            Cluster highClus = ClusterUtilities.findHighestEnergyCluster(clusters);
+            clusHighEnergyH1D.fill(highClus.getEnergy());
+            clusTimeH1D.fill(highClus.getCalorimeterHits().get(0).getTime());
+        }
+        
+        /**
+         * Save histograms and perform checks on histogram statistics.
+         */
+        public void endOfData() {
+            
+            System.out.println("clusCountH1D mean: " + clusCountH1D.mean()); 
+            System.out.println("clusEnergyH1D mean: " + clusEnergyH1D.mean());
+            System.out.println("clusHighEnergyH1D mean: " + clusHighEnergyH1D.mean());
+            System.out.println("clusTimeH1D mean: " + clusTimeH1D.mean());
+            
+            // TODO: Add test assertions here.
+            
+            try {
+                aida.saveAs(new TestOutputFile(EcalSimReconTest.class, "plots.aida"));
+            } catch (IOException e) {
+                throw new RuntimeException(e);
+            }
+        }
+    }    
+}