Add Gaudi Argument for inclusion of Flow Element algorithm

Reconstruction/PFlow/PFlowUtils/share/PFlowCalibHitDecorator_algorithms.py

@@ -14,4 +14,9 @@ from PFlowUtils.PFlowUtilsConf import PFlowCalibPFODecoratorAlgorithm
 PFlowCalibPFODecoratorAlgorithm = PFlowCalibPFODecoratorAlgorithm()
 PFlowCalibPFOTruthAttributerTool = CalibClusterTruthAttributerTool("PFlowCalibPFOTruthAttributerTool")
 PFlowCalibPFODecoratorAlgorithm.TruthAttributerTool = PFlowCalibPFOTruthAttributerTool
+#allow flow element decoration if the eflowRec JO permits it
+from eflowRec.eflowRecFlags import jobproperties as jobprops
+if(jobprops.eflowRecFlags.useFlowElements==True):
+    PFlowCalibPFODecoratorAlgorithm.useFlowElements=True
+
 topSequence+=PFlowCalibPFODecoratorAlgorithm

Reconstruction/PFlow/PFlowUtils/src/PFlowCalibPFODecoratorAlgorithm.cxx

@@ -13,6 +13,7 @@ StatusCode PFlowCalibPFODecoratorAlgorithm::initialize(){
   ATH_CHECK(m_mapTruthBarcodeToTruthParticleReadHandleKey.initialize());
 
   ATH_CHECK(m_pfoWriteDecorHandleKeyNLeadingTruthParticles.initialize());
+  ATH_CHECK(m_feWriteDecorHandleKeyNLeadingTruthParticles.initialize());
 
   ATH_CHECK(m_truthAttributerTool.retrieve());
 
@@ -84,8 +85,8 @@ StatusCode PFlowCalibPFODecoratorAlgorithm::execute(const EventContext& ctx) con
 				      pfoWriteDecorHandleNLeadingTruthParticles,
 				      mapIdentifierToCalibHitsReadHandle,
 				      mapTruthBarcodeToTruthParticleReadHandle)); // end of check block 
-  bool doFlowElements=true;
-  if(doFlowElements){
+
+  if(m_useFlowElements){
     SG::WriteDecorHandle<xAOD::FlowElementContainer,std::vector<std::pair<unsigned int, double> > > feWriteDecorHandleNLeadingTruthParticles(m_feWriteDecorHandleKeyNLeadingTruthParticles,ctx);
     
     ATH_CHECK(this->LinkCalibHitPFO(

Reconstruction/PFlow/PFlowUtils/src/PFlowCalibPFODecoratorAlgorithm.h

@@ -64,6 +64,9 @@ private:
   /** Allow user to set the number of truth particles per clusterCaloCluster or PFO, in descending pt order, for which to store calibration hit enery */
   Gaudi::Property<unsigned int> m_numTruthParticles{this,"NumTruthParticles",3,"Set number of truth particles per CaloCluster/PFO for which we store calibration hit energy"};
 
+
+  Gaudi::Property<bool> m_useFlowElements{this,"useFlowElements",false,"Set decoration of flow element container as well as PFO"};
+
   // functions to do the links between either PFO or FlowElements
   StatusCode LinkCalibHitPFO(
 			     SG::WriteDecorHandle<xAOD::FlowElementContainer, std::vector< std::pair<unsigned int, double> > >& pfoWriteDecorHandle,