Author: [log in to unmask]
Date: Wed Aug 26 23:40:18 2015
New Revision: 3416
Log:
Set the TrackType of the track based on the tracking strategy that was used to find it.
Modified:
java/trunk/tracking/src/main/java/org/hps/recon/tracking/TrackDataDriver.java
Modified: java/trunk/tracking/src/main/java/org/hps/recon/tracking/TrackDataDriver.java
=============================================================================
--- java/trunk/tracking/src/main/java/org/hps/recon/tracking/TrackDataDriver.java (original)
+++ java/trunk/tracking/src/main/java/org/hps/recon/tracking/TrackDataDriver.java Wed Aug 26 23:40:18 2015
@@ -17,6 +17,7 @@
import org.lcsim.fit.helicaltrack.HelicalTrackCross;
import org.lcsim.fit.helicaltrack.HelicalTrackHit;
import org.lcsim.fit.helicaltrack.HelicalTrackStrip;
+import org.lcsim.recon.tracking.seedtracker.SeedTrack;
import org.lcsim.util.Driver;
/**
@@ -108,6 +109,23 @@
// Loop over all the tracks in the event
for (Track track : tracks) {
+ //
+ // Set the TrackType of the track based on the tracking
+ // strategy that was used to find it.
+ //
+
+ // Get the name of the strategy used to find this track
+ SeedTrack seedTrack = (SeedTrack) track;
+ String strategyName = seedTrack.getStrategy().getName();
+
+ // If a TrackType is associated with this strategy, set it.
+ // Otherwise, just move on and stick with the default value
+ // of zero.
+ TrackType trackType = TrackType.getType(strategyName);
+ if (trackType != null) {
+ seedTrack.setTrackType(trackType.getType());
+ }
+
totalT0 = 0;
totalHits = 0;
t0Residuals.clear();
|