lcsim/test/org/lcsim/contrib/tracking

diff -u -r1.5 -r1.6
--- TrackerHitCheaterTest.java	7 Aug 2007 17:36:38 -0000	1.5
+++ TrackerHitCheaterTest.java	9 Aug 2007 20:11:02 -0000	1.6
@@ -54,15 +54,17 @@

         LCSimLoop loop = new LCSimLoop();
         loop.setLCIORecordSource(file);
         loop.add( new TestDriver() );

-        loop.loop(10);

+        loop.loop(1);

         loop.dispose();
     }
     
     class TestDriver extends Driver

-    {

+    {    	
+    	private int nevents=0;

         protected void process(EventHeader event)

-        {
-            

+        {        	
+        	int nhits=0;
+        	        	        	

             // Get the SimTrackerHits
             List<SimTrackerHit> simulated_hits = event.get(SimTrackerHit.class, "SiTrackerBarrel_RO");            
             System.out.println("Number of input SimTrackerHits: "+simulated_hits.size());

@@ -70,16 +72,32 @@

             List<TrackerHit> tracker_hits = _hit_cheater.makeTrackerHits(simulated_hits);
             System.out.println("Number of output TrackerHits: "+tracker_hits.size());
             

+            if (nevents == 0)
+            {
+            	assertEquals("Incorrect number of TrackerHits.",tracker_hits.size(),8);
+            }
+            

             for (TrackerHit hit : tracker_hits)
             {

-                System.out.println("Location of output TrackerHit: ["+hit.getPosition()[0]+

+            	assertTrue("Missing MCParticles.",((BaseTrackerHitMC)hit).mcParticles().size() > 0);
+            	assertTrue("Missing SimTrackerHits.",((BaseTrackerHitMC)hit).getSimHits().size() > 0);
+            	
+            	if (nhits<2)
+            	{
+            		System.out.println("Location of output TrackerHit: ["+hit.getPosition()[0]+

                                                                    ","+hit.getPosition()[1]+
                                                                    ","+hit.getPosition()[2]+"]");

-                System.out.println("Number of MCParticles contributing: "+((BaseTrackerHitMC)hit).mcParticles().size());
-                System.out.println("Number of SimTrackerHits contributing: "+((BaseTrackerHitMC)hit).getSimHits().size());

+                	System.out.println("Number of MCParticles contributing: "+((BaseTrackerHitMC)hit).mcParticles().size());
+                	System.out.println("Number of SimTrackerHits contributing: "+((BaseTrackerHitMC)hit).getSimHits().size());                	
+            	}
+            	else if (nhits == 2) {
+            		System.out.println("[SNIP]");
+            	}
+                ++nhits;

             }        
 //            event.put("CheatedTrackerHits",trackerhits,BaseTrackerHitMC.class,0);        
             

+            ++nevents;            

         }
     }
 }