lcsim/test/org/lcsim/recon/tracking/seedtracker
diff -N SeedStrategyXMLTest.java
--- /dev/null 1 Jan 1970 00:00:00 -0000
+++ SeedStrategyXMLTest.java 27 Aug 2008 18:09:04 -0000 1.1
@@ -0,0 +1,31 @@
+/*
+ * To change this template, choose Tools | Templates
+ * and open the template in the editor.
+ */
+
+package org.lcsim.recon.tracking.seedtracker;
+
+import java.util.List;
+import junit.framework.TestCase;
+import org.lcsim.recon.tracking.seedtracker.digiexample.BetterPlanarStrategy;
+import org.lcsim.util.test.TestUtil.TestOutputFile;
+
+/**
+ *
+ * @author cozzy
+ */
+public class SeedStrategyXMLTest extends TestCase{
+
+ public SeedStrategyXMLTest(String testName) {
+ super(testName);
+ }
+
+ public void testXMLReadingAndWriting() throws Exception {
+ List<SeedStrategy> strategies = new BetterPlanarStrategy().getStrategies();
+ TestOutputFile file = new TestOutputFile("SeedStrategyXMLTest.xml");
+ StrategyXMLUtils.writeStrategyListToFile(strategies, file);
+ List<SeedStrategy> readStrategies = StrategyXMLUtils.getStrategyListFromFile(file);
+ assertTrue(strategies.equals(readStrategies));
+ }
+
+}
lcsim/test/org/lcsim/recon/tracking/seedtracker
diff -N SeedTrackerDigiTest.java
--- /dev/null 1 Jan 1970 00:00:00 -0000
+++ SeedTrackerDigiTest.java 27 Aug 2008 18:09:04 -0000 1.1
@@ -0,0 +1,83 @@
+/*
+ * To change this template, choose Tools | Templates
+ * and open the template in the editor.
+ */
+
+package org.lcsim.recon.tracking.seedtracker;
+
+import java.io.File;
+import java.net.URL;
+import java.util.ArrayList;
+import java.util.List;
+import junit.framework.TestSuite;
+import junit.framework.Test;
+import junit.framework.TestCase;
+import org.lcsim.contrib.SiStripSim.TrackerHitDriver_User;
+import org.lcsim.recon.tracking.seedtracker.digiexample.BetterPlanarStrategy;
+import org.lcsim.contrib.tracking.TrackerHitCheater;
+import org.lcsim.event.EventHeader;
+import org.lcsim.event.SimTrackerHit;
+import org.lcsim.event.TrackerHit;
+import org.lcsim.fit.helicaltrack.HelicalTrackHitDriver;
+import org.lcsim.fit.helicaltrack.HelicalTrackHitDriver.HitType;
+import org.lcsim.util.Driver;
+import org.lcsim.util.cache.FileCache;
+import org.lcsim.util.loop.LCSimLoop;
+
+/**
+ *
+ * @author cozzy
+ */
+public class SeedTrackerDigiTest extends TestCase{
+
+ public SeedTrackerDigiTest(String testName) {
+ super(testName);
+ }
+
+ public static Test suite() {
+ return new TestSuite(SeedTrackerDigiTest.class);
+ }
+
+ public void setUp() {
+ System.gc();
+ }
+
+ public void testSeedTrackerDigi() throws Exception {
+ URL url = new URL("http://www.lcsim.org/test/lcio/muons_Theta4-176_1-10GeV_SLIC-v2r4p6_geant4-v9r1p2_LCPhys_sid01_planar_tracker-0-10.slcio");
+ FileCache cache = new FileCache();
+ File file = cache.getCachedFile(url);
+ LCSimLoop loop = new LCSimLoop();
+ loop.setLCIORecordSource(file);
+ loop.add( new TestDriver());
+ loop.loop(5, null);
+ loop.dispose();
+ }
+
+ class TestDriver extends Driver{
+
+ public TestDriver() {
+ TrackerHitDriver_User thd = new TrackerHitDriver_User();
+ add(thd);
+ HelicalTrackHitDriver hitdriver = new HelicalTrackHitDriver();
+ hitdriver.addCollection(((TrackerHitDriver_User)thd).getStripHits1DName(),HitType.Digitized);
+ hitdriver.addCollection("CheatedVertexBarrelHits",HitType.Base);
+ add(hitdriver);
+ add( new SeedTracker(new BetterPlanarStrategy().getStrategies()) );
+
+ }
+
+ public void process(EventHeader event) {
+ List<SimTrackerHit> simhits = new ArrayList<SimTrackerHit>();
+ simhits.addAll(event.getSimTrackerHits("SiVertexBarrel_RO"));
+ simhits.addAll(event.getSimTrackerHits("SiVertexEndcap_RO"));
+ TrackerHitCheater cheater = new TrackerHitCheater();
+ List<TrackerHit> smearedhits = cheater.makeTrackerHits(simhits);
+ event.put("CheatedVertexBarrelHits", smearedhits, TrackerHit.class, 0);
+ super.process(event);
+ assertFalse("Track not found in event "+event.getEventNumber(),event.getTracks().isEmpty());
+ return;
+ }
+
+ }
+
+}
lcsim/test/org/lcsim/recon/tracking/seedtracker
diff -N SeedTrackerVSTest.java
--- /dev/null 1 Jan 1970 00:00:00 -0000
+++ SeedTrackerVSTest.java 27 Aug 2008 18:09:04 -0000 1.1
@@ -0,0 +1,69 @@
+/*
+ * To change this template, choose Tools | Templates
+ * and open the template in the editor.
+ */
+
+package org.lcsim.recon.tracking.seedtracker;
+
+import java.io.File;
+import java.net.URL;
+import junit.framework.TestSuite;
+import junit.framework.Test;
+import junit.framework.TestCase;
+import org.lcsim.recon.tracking.seedtracker.example.MyStrategy;
+import org.lcsim.recon.tracking.seedtracker.example.VSExampleDriver;
+import org.lcsim.event.EventHeader;
+import org.lcsim.fit.helicaltrack.HelicalTrackHitDriver;
+import org.lcsim.fit.helicaltrack.HelicalTrackHitDriver.HitType;
+import org.lcsim.util.Driver;
+import org.lcsim.util.cache.FileCache;
+import org.lcsim.util.loop.LCSimLoop;
+
+/**
+ *
+ * @author cozzy
+ */
+public class SeedTrackerVSTest extends TestCase{
+
+ public SeedTrackerVSTest(String testName) {
+ super(testName);
+ }
+
+ public static Test suite() {
+ return new TestSuite(SeedTrackerVSTest.class);
+ }
+
+ public void setUp() {
+ System.gc();
+ }
+
+ public void testSeedTrackerVS() throws Exception {
+ URL url = new URL("http://www.lcsim.org/test/lcio/muons10_Theta4-176_1-10GeV_SLIC_v2r2p1_sid01.slcio");
+ FileCache cache = new FileCache();
+ File file = cache.getCachedFile(url);
+ LCSimLoop loop = new LCSimLoop();
+ loop.setLCIORecordSource(file);
+ loop.add( new TestDriver());
+ loop.loop(5, null);
+ loop.dispose();
+ }
+
+ class TestDriver extends Driver{
+
+ public TestDriver() {
+ add(new VSExampleDriver());
+ HelicalTrackHitDriver hitdriver = new HelicalTrackHitDriver();
+ hitdriver.addCollection("NewTrackerHits",HitType.VirtualSegmentation);
+ add(hitdriver);
+ add( new SeedTracker(new MyStrategy().getStrategies()) );
+ }
+
+ public void process(EventHeader event) {
+ super.process(event);
+ assertFalse(event.getTracks().isEmpty());
+ return;
+ }
+
+ }
+
+}