slicPandora/src

PfoProcessor.cpp 1.2 -> 1.3

diff -u -r1.2 -r1.3
--- PfoProcessor.cpp	8 Mar 2010 23:00:53 -0000	1.2
+++ PfoProcessor.cpp	9 Mar 2010 20:13:42 -0000	1.3
@@ -1,11 +1,17 @@

-// $Id: PfoProcessor.cpp,v 1.2 2010/03/08 23:00:53 jeremy Exp $

+// $Id: PfoProcessor.cpp,v 1.3 2010/03/09 20:13:42 jeremy Exp $

 #include "PfoProcessor.h"
 
 // lcio

-#include "EVENT/LCIO.h"
-#include "IMPL/LCFlagImpl.h"

+
+#include "EVENT/LCCollection.h"
+#include "EVENT/SimCalorimeterHit.h"
+#include "EVENT/CalorimeterHit.h"

 #include "IMPL/LCCollectionVec.h"

+#include "IMPL/ReconstructedParticleImpl.h"

 #include "IMPL/ClusterImpl.h"

+#include "IMPL/LCFlagImpl.h" 
+#include "EVENT/LCIO.h"
+#include "EVENT/Track.h"

 
 // slicPandora
 #include "PfoConstructionAlgorithm.h"

@@ -14,6 +20,11 @@

 using IMPL::LCCollectionVec;
 using IMPL::LCFlagImpl;
 using IMPL::ClusterImpl;

+using IMPL::ReconstructedParticleImpl;
+using EVENT::CalorimeterHit;
+using EVENT::Track;
+using std::cout;
+using std::endl;

 
 PfoProcessor::PfoProcessor()
     : EventProcessor("PfoProcessor")

@@ -22,22 +33,84 @@

 PfoProcessor::~PfoProcessor()
 {}

-void PfoProcessor::processEvent(EVENT::LCEvent*)

+void PfoProcessor::processEvent(EVENT::LCEvent* event)

 {
     std::cout << "PfoProcessor::processEvent" << std::endl;

-
-    // Setup cluster collection.
-    /*

+        // first a container for the clusters...

     LCCollectionVec* clusterVec = new LCCollectionVec(EVENT::LCIO::CLUSTER);

+    // if we want to point back to the hits we need to set the flag

     LCFlagImpl clusterFlag(0);
     clusterFlag.setBit(EVENT::LCIO::CLBIT_HITS);
     clusterVec->setFlag(clusterFlag.getFlag());

+    // get the particle flow objects

     pandora::ParticleFlowObjectList particleFlowObjectList;
     PANDORA_THROW_RESULT_IF_AND_IF(STATUS_CODE_SUCCESS, STATUS_CODE_NOT_INITIALIZED, !=, PandoraApi::GetParticleFlowObjects(getJobManager()->getPandora(),

-                                                                                                                            particleFlowObjectList));

+            particleFlowObjectList));
+
+    std::cout << "Pandora found " << particleFlowObjectList.size() << " PFOs" << std::endl;
+

     LCCollectionVec* pReconstructedParticleCollection = new LCCollectionVec(EVENT::LCIO::RECONSTRUCTEDPARTICLE);

-    */

+    // get particle flow objects and create "reconstructed particles"
+    for (pandora::ParticleFlowObjectList::iterator itPFO = particleFlowObjectList.begin(), itPFOEnd = particleFlowObjectList.end();
+            itPFO != itPFOEnd; ++itPFO)
+    {
+        ReconstructedParticleImpl *pReconstructedParticle = new ReconstructedParticleImpl();
+
+        pandora::ClusterAddressList clusterAddressList = (*itPFO)->GetClusterAddressList();
+        cout << "Found " << clusterAddressList.size() << " clusters." << endl;
+        pandora::TrackAddressList trackAddressList = (*itPFO)->GetTrackAddressList();
+
+        // make LCIO clusters
+        for (pandora::ClusterAddressList::iterator itCluster = clusterAddressList.begin(), itClusterEnd = clusterAddressList.end();
+                itCluster != itClusterEnd; ++itCluster)
+        {
+            ClusterImpl *pCluster = new ClusterImpl();
+            
+            double clusterEnergy = 0.;
+            float* clusterPosition;
+            double maxE = 0;
+            for (pandora::CaloHitAddressList::iterator itHit = (*itCluster).begin(), itHitEnd = (*itCluster).end(); itHit != itHitEnd; ++itHit)
+            {
+              
+                clusterEnergy += ((CalorimeterHit*) (*itHit))->getEnergy();
+                if(((CalorimeterHit*) (*itHit))->getEnergy()>maxE)
+                {
+                    maxE = ((CalorimeterHit*) (*itHit))->getEnergy();
+                    //avert your gaze...
+                    clusterPosition = const_cast<float*>(((CalorimeterHit*) (*itHit))->getPosition());
+                }
+                pCluster->addHit((CalorimeterHit*) (*itHit), 1.0); // transform from Uid (=void*) to a CalorimeterHit*
+            }
+            pCluster->setEnergy(clusterEnergy);
+            pCluster->setPosition(clusterPosition);
+            cout << "Cluster contains " << pCluster->getCalorimeterHits().size() << " hits" << endl;
+            // add this cluster to the collection
+            clusterVec->addElement(pCluster);
+            pReconstructedParticle->addCluster(pCluster);
+        }
+
+        // add tracks
+        for (pandora::TrackAddressList::iterator itTrack = trackAddressList.begin(), itTrackEnd = trackAddressList.end(); itTrack != itTrackEnd;
+                ++itTrack)
+        {
+            pReconstructedParticle->addTrack((Track*) (*itTrack));
+        }
+
+        float momentum[3] = {(*itPFO)->GetMomentum().GetX(), (*itPFO)->GetMomentum().GetY(), (*itPFO)->GetMomentum().GetZ()};
+        pReconstructedParticle->setMomentum(momentum);
+        pReconstructedParticle->setEnergy((*itPFO)->GetEnergy());
+        pReconstructedParticle->setMass((*itPFO)->GetMass());
+        pReconstructedParticle->setCharge((*itPFO)->GetChargeSign());
+        pReconstructedParticle->setType((*itPFO)->GetParticleId());
+
+        pReconstructedParticleCollection->addElement(pReconstructedParticle);
+    }
+    // add the list of clusters to the event
+    event->addCollection(clusterVec, "ReconClusters");
+    //cout << "end of loop over pandora outpout" << endl;
+    event->addCollection(pReconstructedParticleCollection, "PandoraPFOCollection");
+    //cout << "added pfo objects to event" << endl;
+

+// $Id: EventMarkerProcessor.h,v 1.1 2010/03/09 20:13:42 jeremy Exp $ + +#ifndef EVENTMARKERPROCESSOR_H +#define EVENTMARKERPROCESSOR_H 1 + +#include "EventProcessor.h" + +class EventMarkerProcessor : public EventProcessor +{ +public: + EventMarkerProcessor() : EventProcessor("EventMarkerProcessor") {;} + virtual ~EventMarkerProcessor() {;} + +public: + + void processEvent(EVENT::LCEvent* event) + { + std::cout << std::endl; + std::cout << ">>>>>> EVENT #" << event->getEventNumber() << std::endl; + std::cout << std::endl; + } + +}; + +#endif

+// $Id: ResetPandoraProcessor.h,v 1.1 2010/03/09 20:13:42 jeremy Exp $ + +#ifndef RESETPANDORAPROCESSOR_H +#define RESETPANDORAPROCESSOR_H 1 + +// slicPandora +#include "EventProcessor.h" + +// pandora +#include "Api/PandoraApiImpl.h" + +class ResetPandoraProcessor : public EventProcessor +{ +public: + ResetPandoraProcessor() + : EventProcessor("ResetPandoraProcessor") + {;} + virtual ~ResetPandoraProcessor() {;} + +public: + void processEvent(EVENT::LCEvent*) + { + getJobManager()->getPandora().GetPandoraApiImpl()->ResetForNextEvent(); + } +}; + + +#endif

+ * implemented by sub-classes. Processors that are calling Pandora algorithms + * may ignore the LCEvent entirely and retrieve objects directly from Pandora + * via the JobManager's reference to it.

Commit in `slicPandora` on MAIN
`Makefile`	+24	-29	1.9 -> 1.10
`src/CalorimeterHitMaker.cpp`	+1	-2	1.2 -> 1.3
`/CalorimeterHitProcessor.cpp`	+2	-2	1.2 -> 1.3
`/PfoProcessor.cpp`	+83	-10	1.2 -> 1.3
`include/EventMarkerProcessor.h`	+25		added 1.1
`/ResetPandoraProcessor.h`	+28		added 1.1
`/CalorimeterHitMaker.h`	+1		1.1 -> 1.2
`/DetectorGeometry.h`	+2		1.6 -> 1.7
`/EventProcessor.h`	+10	-3	1.2 -> 1.3
`/IDDecoder.h`	+1		1.2 -> 1.3
`/JobConfig.h`	+1		1.2 -> 1.3
`/JobManager.h`	+1		1.4 -> 1.5
`/PfoConstructionAlgorithm.h`	+4		1.1 -> 1.2
`/PfoProcessor.h`	+1		1.1 -> 1.2
	+184	-46