4 added files
lcsim/src/org/lcsim/contrib/Partridge/TestSeedTracker
diff -N MyStrategy.java
--- /dev/null 1 Jan 1970 00:00:00 -0000
+++ MyStrategy.java 13 Apr 2008 07:08:19 -0000 1.1
@@ -0,0 +1,55 @@
+/*
+ * MyStrategy.java
+ *
+ * Created on April 12, 2008, 8:42 PM
+ *
+ */
+
+package org.lcsim.contrib.Partridge.TestSeedTracker;
+
+import java.util.ArrayList;
+import java.util.List;
+import org.lcsim.contrib.seedtracker.SeedLayer;
+import org.lcsim.contrib.seedtracker.SeedLayer.SeedType;
+import org.lcsim.contrib.seedtracker.SeedStrategy;
+import org.lcsim.geometry.subdetector.BarrelEndcapFlag;
+
+public class MyStrategy {
+ private List<SeedStrategy> _strategylist;
+
+ public MyStrategy() {
+
+ // Create a list of strategies
+ _strategylist = new ArrayList<SeedStrategy>();
+
+ // Create an outside-in strategy for the barrel
+ List<SeedLayer> lyrlist = new ArrayList<SeedLayer>();
+ lyrlist.add(new SeedLayer("VertexBarrel",0,BarrelEndcapFlag.BARREL,SeedType.Extend));
+ lyrlist.add(new SeedLayer("VertexBarrel",1,BarrelEndcapFlag.BARREL,SeedType.Extend));
+ lyrlist.add(new SeedLayer("VertexBarrel",2,BarrelEndcapFlag.BARREL,SeedType.Extend));
+ lyrlist.add(new SeedLayer("VertexBarrel",3,BarrelEndcapFlag.BARREL,SeedType.Extend));
+ lyrlist.add(new SeedLayer("VertexBarrel",4,BarrelEndcapFlag.BARREL,SeedType.Extend));
+ lyrlist.add(new SeedLayer("TrackerBarrel",0,BarrelEndcapFlag.BARREL,SeedType.Extend));
+ lyrlist.add(new SeedLayer("TrackerBarrel",1,BarrelEndcapFlag.BARREL,SeedType.Confirm));
+ lyrlist.add(new SeedLayer("TrackerBarrel",2,BarrelEndcapFlag.BARREL,SeedType.Seed));
+ lyrlist.add(new SeedLayer("TrackerBarrel",3,BarrelEndcapFlag.BARREL,SeedType.Seed));
+ lyrlist.add(new SeedLayer("TrackerBarrel",4,BarrelEndcapFlag.BARREL,SeedType.Seed));
+ SeedStrategy outsideinbarrel = new SeedStrategy("OutsideInBarrel",lyrlist);
+
+ // Set parameters for this strategy
+ outsideinbarrel.putMinPT(1.0); // Set minimum pT at 1 GeV
+ outsideinbarrel.putMaxDCA(1.0); // Set maximum d0 at 1 mm
+ outsideinbarrel.putMaxZ0(1.0); // Set maximum z0 at 1 mm
+ outsideinbarrel.putMinConfirm(1); // Require at least 1 confirm hit
+ outsideinbarrel.putMinHits(7); // Require at least 7 total hits
+ outsideinbarrel.putMaxChisq(50.); // Set maximum chi^2 at 50
+ outsideinbarrel.putBadHitChisq(15.); // Set chi^2 change that gives preference to excluding the hit
+
+ // Add the strategy to our strategy list
+ _strategylist.add(outsideinbarrel);
+ }
+
+ public List<SeedStrategy> getStrategies() {
+ return _strategylist;
+ }
+}
lcsim/src/org/lcsim/contrib/Partridge/TestSeedTracker
diff -N MyTrackerDriver.java
--- /dev/null 1 Jan 1970 00:00:00 -0000
+++ MyTrackerDriver.java 13 Apr 2008 07:08:19 -0000 1.1
@@ -0,0 +1,53 @@
+/*
+ * MyTrackerDriver.java
+ *
+ * Created on March 29, 2006, 4:58 PM
+ *
+ */
+
+package org.lcsim.contrib.Partridge.TestSeedTracker;
+
+import org.lcsim.contrib.seedtracker.SeedTracker;
+import org.lcsim.event.EventHeader;
+import org.lcsim.fit.helicaltrack.HelicalTrackHitDriver;
+import org.lcsim.fit.helicaltrack.HelicalTrackHitDriver.HitType;
+import org.lcsim.util.Driver;
+
+/**
+ * Driver for testing the SeedTracker track finding algorithm. This driver causes
+ * the SmearMCHits and SeedTracker drivers to be invoked for each event.
+ * @author Richard Partridge
+ * @version 1.0
+ */
+public class MyTrackerDriver extends Driver
+{
+ /**
+ * Creates a new instance of MyTrackerDriver
+ */
+ public MyTrackerDriver()
+ {
+ // Add the hit digitization driver (this example uses the virtual segment digitization code)
+ add(new VSExampleDriver());
+
+ // Add a driver to create HelicalTrackHits from the digitized hits
+ HelicalTrackHitDriver hitdriver = new HelicalTrackHitDriver();
+ hitdriver.addCollection("StandardTrackerHits",HitType.VirtualSegmentation);
+ add(hitdriver);
+
+ // Create a list of strategies to be used in track finding
+ MyStrategy strategylist = new MyStrategy();
+
+ // Add a driver to do the track finding
+ add(new SeedTracker(strategylist.getStrategies()));
+ }
+
+ /**
+ * Process the current event
+ * @param event EventHeader for this event
+ */
+ public void process(EventHeader event)
+ {
+ super.process(event);
+ return;
+ }
+}
lcsim/src/org/lcsim/contrib/Partridge/TestSeedTracker
diff -N VSExampleDriver.java
--- /dev/null 1 Jan 1970 00:00:00 -0000
+++ VSExampleDriver.java 13 Apr 2008 07:08:19 -0000 1.1
@@ -0,0 +1,117 @@
+package org.lcsim.contrib.Partridge.TestSeedTracker;
+
+import java.util.*;
+
+import hep.aida.*;
+import org.lcsim.event.EventHeader;
+import org.lcsim.event.SimTrackerHit;
+import org.lcsim.recon.cat.util.NoSuchParameterException;
+import org.lcsim.units.clhep.SystemOfUnits;
+import org.lcsim.util.Driver;
+import org.lcsim.util.aida.AIDA;
+import hep.physics.vec.BasicHep3Vector;
+import hep.physics.vec.Hep3Vector;
+import hep.physics.vec.VecOp;
+
+import org.lcsim.contrib.onoprien.tracking.clustering.ClusteringDriver;
+import org.lcsim.contrib.onoprien.tracking.clustering.clusterers.NearestNeighborClusterer;
+import org.lcsim.contrib.onoprien.tracking.digitization.SimToDigiDriver;
+import org.lcsim.contrib.onoprien.tracking.digitization.SimToDigiConverter;
+import org.lcsim.contrib.onoprien.tracking.digitization.algorithms.ConverterSimple;
+import org.lcsim.contrib.onoprien.tracking.geom.SegmentationManager;
+import org.lcsim.contrib.onoprien.tracking.geom.Segmenter;
+import org.lcsim.contrib.onoprien.tracking.geom.Sensor;
+import org.lcsim.contrib.onoprien.tracking.geom.SensorType;
+import org.lcsim.contrib.onoprien.tracking.hit.DigiTrackerHit;
+import org.lcsim.contrib.onoprien.tracking.hit.TrackerCluster;
+import org.lcsim.contrib.onoprien.tracking.hit.TrackerHit;
+import org.lcsim.contrib.onoprien.tracking.hitmaking.HitMakingDriver;
+import org.lcsim.contrib.onoprien.tracking.hitmaking.TrackerHitConverter;
+import org.lcsim.contrib.onoprien.tracking.hitmaking.TrackerHitMaker;
+import org.lcsim.contrib.onoprien.tracking.hitmaking.hitmakers.TrackerHitMakerBasic;
+import org.lcsim.contrib.onoprien.tracking.mctruth.MCTruth;
+import org.lcsim.contrib.onoprien.tracking.mctruth.MCTruthDriver;
+import org.lcsim.contrib.onoprien.tracking.mctruth.SimGroup;
+
+/**
+ * An Example of how to define virtual segmentation of the tracker,
+ * and use {@link TrackerHitConverter} to create
+ * "standard" <tt>org.lcsim.event.TrackerHits</tt>.
+ * Parameters are chosen to define reasonable segmentation for SiD01.
+ *
+ *
+ *
+ * @author D. Onoprienko
+ * @version $Id: VSExampleDriver.java,v 1.1 2008/04/13 07:08:19 partridge Exp $
+ */
+public class VSExampleDriver extends Driver {
+
+// -- Constructors : ----------------------------------------------------------
+
+ public VSExampleDriver() {
+
+ add(new MCTruthDriver());
+
+ // Segmentation description :
+
+ Segmenter segmenter = new VSExampleSegmenter();
+ SegmentationManager segMan = new SegmentationManager(segmenter);
+ SegmentationManager.setDefaultInstance(segMan);
+ add(segMan);
+
+ // Digitization :
+
+ SimToDigiConverter converter = new ConverterSimple();
+ SimToDigiDriver conversionDriver = new SimToDigiDriver(converter);
+ conversionDriver.set("ADD_INPUT_LIST_NAME", "TkrBarrHits");
+ conversionDriver.set("ADD_INPUT_LIST_NAME", "VtxBarrHits");
+ conversionDriver.set("OUTPUT_MAP_NAME", "DigiTrackerHits");
+ add(conversionDriver);
+
+ // Clustering :
+
+ ClusteringDriver clusteringDriver = new ClusteringDriver(new NearestNeighborClusterer());
+ clusteringDriver.set("INPUT_MAP_NAME","DigiTrackerHits");
+ clusteringDriver.set("OUTPUT_MAP_NAME","TrackerClusters");
+ add(clusteringDriver);
+
+ // Conversion to "standard" org.lcsim.event.TrackerHit : ------------------
+
+ TrackerHitConverter hitConverter = new TrackerHitConverter();
+ hitConverter.set("INPUT_CLUSTER_MAP_NAME", "TrackerClusters");
+ hitConverter.set("OUTPUT_HIT_LIST_NAME", "StandardTrackerHits");
+ add(hitConverter);
+ }
+
+
+// -- Event processing : ------------------------------------------------------
+
+ public void process(EventHeader event) {
+
+ System.out.println(" ");
+ System.out.println("Event "+event.getEventNumber());
+
+ super.process(event);
+
+ MCTruth mcTruth = (MCTruth) event.get("MCTruth");
+
+ System.out.println(" ");
+
+ int n = 0;
+ HashMap<Sensor, List<DigiTrackerHit>> digiMap =
+ (HashMap<Sensor, List<DigiTrackerHit>>) event.get("DigiTrackerHits");
+ for (List<DigiTrackerHit> digiList : digiMap.values()) n += digiList.size();
+ System.out.println("Created " + n + " DigiTrackerHits on " + digiMap.keySet().size() + " sensors");
+
+ n = 0;
+ HashMap<Sensor, List<TrackerCluster>> clusterMap =
+ (HashMap<Sensor, List<TrackerCluster>>) event.get("TrackerClusters");
+ for (List<TrackerCluster> clusterList : clusterMap.values()) n += clusterList.size();
+ System.out.println("Created " + n + " TrackerClusters on " + clusterMap.keySet().size() + " sensors");
+
+ n = 0;
+ List<org.lcsim.event.TrackerHit> oldHitList = (List<org.lcsim.event.TrackerHit>) event.get("StandardTrackerHits");
+ System.out.println("Created " + oldHitList.size() + " org.lcsim.event.TrackerHits");
+ }
+
+}
lcsim/src/org/lcsim/contrib/Partridge/TestSeedTracker
diff -N VSExampleSegmenter.java
--- /dev/null 1 Jan 1970 00:00:00 -0000
+++ VSExampleSegmenter.java 13 Apr 2008 07:08:19 -0000 1.1
@@ -0,0 +1,64 @@
+package org.lcsim.contrib.Partridge.TestSeedTracker;
+
+import org.lcsim.units.clhep.SystemOfUnits;
+
+import org.lcsim.contrib.onoprien.tracking.geom.segmenters.CylindricalBarrelSegmenter;
+import org.lcsim.contrib.onoprien.tracking.geom.segmenters.DiskTrackerToWedgesSegmenter;
+import org.lcsim.contrib.onoprien.tracking.geom.segmenters.DiskTrackerToRingsSegmenter;
+import org.lcsim.contrib.onoprien.tracking.geom.segmenters.SubdetectorBasedSegmenter;
+
+/**
+ *
+ *
+ *
+ *
+ * @author D. Onoprienko
+ * @version $Id: VSExampleSegmenter.java,v 1.1 2008/04/13 07:08:19 partridge Exp $
+ */
+public class VSExampleSegmenter extends SubdetectorBasedSegmenter {
+
+// -- Constructors : ----------------------------------------------------------
+
+ public VSExampleSegmenter() {
+
+ // 25 um pixels in VTX Barrel
+
+ CylindricalBarrelSegmenter vtxBarrelSegmenter = new CylindricalBarrelSegmenter("VertexBarrel");
+ vtxBarrelSegmenter.setStripLength(25.*SystemOfUnits.micrometer);
+ vtxBarrelSegmenter.setStripWidth(25.*SystemOfUnits.micrometer);
+ setSegmenter("VertexBarrel", vtxBarrelSegmenter);
+
+ // 10 cm x 25 um strips in outer Tracker Barrel
+
+ CylindricalBarrelSegmenter trackerBarrelSegmenter = new CylindricalBarrelSegmenter("TrackerBarrel");
+ trackerBarrelSegmenter.setStripLength(10.*SystemOfUnits.cm);
+ trackerBarrelSegmenter.setStripWidth(25.*SystemOfUnits.micrometer);
+ setSegmenter("TrackerBarrel", trackerBarrelSegmenter);
+
+ // 25 um pixels in VTX Endcap
+
+// DiskTrackerToRingsSegmenter vtxEndcapSegmenter = new DiskTrackerToRingsSegmenter("VertexEndcap");
+// vtxEndcapSegmenter.setStripLength(25.*SystemOfUnits.micrometer);
+// vtxEndcapSegmenter.setStripWidth(25.*SystemOfUnits.micrometer);
+// setSegmenter("VertexEndcap", vtxEndcapSegmenter);
+
+ // 15 degrees stereo wedges in Forward Tracker
+
+// DiskTrackerToWedgesSegmenter trackerForwardSegmenter = new DiskTrackerToWedgesSegmenter("TrackerForward");
+// trackerForwardSegmenter.setStripLength(10.*SystemOfUnits.cm);
+// trackerForwardSegmenter.setStripWidth(25.*SystemOfUnits.micrometer);
+// trackerForwardSegmenter.setNumberOfPhiSlices(24);
+// setSegmenter("TrackerForward", trackerForwardSegmenter);
+
+ // 15 degrees stereo wedges in outer Tracker Endcap
+
+// DiskTrackerToWedgesSegmenter trackerEndcapSegmenter = new DiskTrackerToWedgesSegmenter("TrackerEndcap");
+// trackerEndcapSegmenter.setNumberOfRadialSlices(new int[]{3,5,8,10, 10});
+// trackerEndcapSegmenter.setStripWidth(25.*SystemOfUnits.micrometer);
+// trackerEndcapSegmenter.setNumberOfPhiSlices(24);
+// setSegmenter("TrackerEndcap", trackerEndcapSegmenter);
+ }
+
+// -- Private parts : ---------------------------------------------------------
+
+}
CVSspam 0.2.8