hps-java/src/main/java/org/lcsim/hps/evio
diff -u -r1.7 -r1.8
--- TestRunTriggeredReconToLcio.java 21 Sep 2012 23:22:18 -0000 1.7
+++ TestRunTriggeredReconToLcio.java 24 Sep 2012 23:51:46 -0000 1.8
@@ -10,6 +10,8 @@
import java.util.Queue;
import org.lcsim.event.EventHeader;
import org.lcsim.event.MCParticle;
+import org.lcsim.event.SimCalorimeterHit;
+import org.lcsim.event.SimTrackerHit;
import org.lcsim.hps.conditions.QuietBaseLCSimEvent;
import org.lcsim.hps.monitoring.HPSCalibrationDriver;
import org.lcsim.hps.recon.ecal.HPSTriggerDriver;
@@ -38,6 +40,10 @@
Queue<EventHeader> events = null;
private int ecalMode = EventConstants.ECAL_PULSE_INTEGRAL_MODE;
List<MCParticle> mcParticles = null;
+ List<SimTrackerHit> trackerHits = null;
+ List<SimCalorimeterHit> ecalHits = null;
+ static final String ecalCollectionName = "EcalHits";
+ static final String trackerCollectionName = "TrackerHits";
public TestRunTriggeredReconToLcio() {
}
@@ -100,14 +106,19 @@
protected void process(EventHeader event) {
if (event.hasCollection(MCParticle.class, MCEvent.MC_PARTICLES)) {
mcParticles = event.getMCParticles();
+ ecalHits = event.getSimCalorimeterHits(ecalCollectionName);
+ trackerHits = event.getSimTrackerHits(trackerCollectionName);
}
+
if (HPSTriggerDriver.triggerBit()) {
EventHeader lcsimEvent = new QuietBaseLCSimEvent(HPSCalibrationDriver.runNumber(), event.getEventNumber(), event.getDetectorName());
events.add(lcsimEvent);
System.out.println("Creating LCIO event " + eventNum);
lcsimEvent.put(MCEvent.MC_PARTICLES, mcParticles);
- System.out.println("Adding " + mcParticles.size() + " MCParticles");
+ lcsimEvent.put(ecalCollectionName, ecalHits);
+ lcsimEvent.put(trackerCollectionName, trackerHits);
+ System.out.println("Adding " + mcParticles.size() + " MCParticles, " + ecalHits.size() + " SimCalorimeterHits, " + trackerHits.size() + " SimTrackerHits");
++eventNum;
}