CompositeParticleRetriever.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#include "AnalysisJiveXML/CompositeParticleRetriever.h"
 
#include "CompositeParticleEvent/CompositeParticle.h"
#include "CompositeParticleEvent/CompositeParticleContainer.h"
 
// for associations:
//#include "Particle/TrackParticleContainer.h"
//#include "CaloEvent/CaloClusterContainer.h"
 
#include "CLHEP/Units/SystemOfUnits.h"
 
namespace JiveXML {
 
  CompositeParticleRetriever::CompositeParticleRetriever(const std::string& type,const std::string& name,const IInterface* parent):
    AthAlgTool(type,name,parent),
    m_typeName("CompositeParticle")
  {
 
    //Only declare the interface
    declareInterface<IDataRetriever>(this);
 
    declareProperty("StoreGateKey", m_sgKey = "AllObjects", 
        "Collection to be first in output, shown in Atlantis without switching");
  }
   
  StatusCode CompositeParticleRetriever::retrieve(ToolHandle<IFormatTool> &FormatTool) {
    
    if (msgLvl(MSG::DEBUG)) msg(MSG::DEBUG)  << "in retrieveAll()" << endmsg;
    
    SG::ConstIterator<CompositeParticleContainer> iterator, end;
    const CompositeParticleContainer* compPart;
    
    //obtain the default collection first
    if (msgLvl(MSG::DEBUG)) msg(MSG::DEBUG)  << "Trying to retrieve " << dataTypeName() << " (" << m_sgKey << ")" << endmsg;
    StatusCode sc = evtStore()->retrieve(compPart, m_sgKey);
    if (sc.isFailure() ) {
      if (msgLvl(MSG::WARNING)) msg(MSG::WARNING) << "Collection " << m_sgKey << " not found in SG " << endmsg; 
    }else{
      DataMap data = getData(compPart);
      if ( FormatTool->AddToEvent(dataTypeName(), m_sgKey, &data).isFailure()){
    if (msgLvl(MSG::WARNING)) msg(MSG::WARNING) << "Collection " << m_sgKey << " not found in SG " << endmsg;
      }else{
         if (msgLvl(MSG::DEBUG)) msg(MSG::DEBUG)  << dataTypeName() << " (" << m_sgKey << ") CompositeParticle retrieved" << endmsg;
      }
    }
 
    //obtain all other collections from StoreGate
    if (( evtStore()->retrieve(iterator, end)).isFailure()){
       if (msgLvl(MSG::WARNING)) msg(MSG::WARNING)  << "Unable to retrieve iterator for Jet collection" << endmsg;
//        return StatusCode::WARNING;
    }
      
    for (; iterator!=end; ++iterator) {
       if (iterator.key()!=m_sgKey) {
          if (msgLvl(MSG::DEBUG)) msg(MSG::DEBUG)  << "Trying to retrieve all " << dataTypeName() << " (" << iterator.key() << ")" << endmsg;
            DataMap data = getData(&(*iterator));
            if ( FormatTool->AddToEvent(dataTypeName(), iterator.key(), &data).isFailure()){
           if (msgLvl(MSG::WARNING)) msg(MSG::WARNING) << "Collection " << iterator.key() << " not found in SG " << endmsg;
        }else{
          if (msgLvl(MSG::DEBUG)) msg(MSG::DEBUG) << dataTypeName() << " (" << iterator.key() << ") CompositeParticle retrieved" << endmsg;
            }
      }
    }     
    //All collections retrieved okay
    return StatusCode::SUCCESS;
  }
 
 
  const DataMap CompositeParticleRetriever::getData(const CompositeParticleContainer* cpcont) {
    
    if (msgLvl(MSG::DEBUG)) msg(MSG::DEBUG) << "retrieve()" << endmsg;
 
    DataMap DataMap;
 
    DataVect phi; phi.reserve(cpcont->size());
    DataVect eta; eta.reserve(cpcont->size());
    DataVect et; et.reserve(cpcont->size());
    DataVect mass; mass.reserve(cpcont->size());
    DataVect energy; energy.reserve(cpcont->size());
    DataVect px; px.reserve(cpcont->size());
    DataVect py; py.reserve(cpcont->size());
    DataVect pz; pz.reserve(cpcont->size());
    DataVect pdgId; pdgId.reserve(cpcont->size());
    DataVect typeEV; typeEV.reserve(cpcont->size());
    DataVect charge; charge.reserve(cpcont->size());
    DataVect dataType; dataType.reserve(cpcont->size());
    DataVect label; label.reserve(cpcont->size());
 
    //planned, not working yet:
//    DataVect childID; childID.reserve(cpcont->size());
//    DataVect motherID; motherID.reserve(cpcont->size());
 
    // for associations. Just placeholder here, for full implementation
    // see MuonRetriever.
//    DataVect clusterKeyVec; clusterKeyVec.reserve(cpcont->size());
//    DataVect clusterIndexVec; clusterIndexVec.reserve(cpcont->size());
//    DataVect clusterKeyVec; clusterKeyVec.reserve(cpcont->size());
//    DataVect clusterIndexVec; clusterIndexVec.reserve(cpcont->size());
 
    CompositeParticleContainer::const_iterator compPartItr  = cpcont->begin();
    CompositeParticleContainer::const_iterator compPartItrE = cpcont->end();
 
    std::string typeLabel = "n_a"; // same as in TruthParticleRetriever
    int pdgId2 = 0;
    for (; compPartItr != compPartItrE; ++compPartItr) {
      phi.push_back(DataType((*compPartItr)->phi()));
      eta.push_back(DataType((*compPartItr)->eta()));
      et.push_back(DataType((*compPartItr)->et()/CLHEP::GeV));
      mass.push_back(DataType((*compPartItr)->m()/CLHEP::GeV));
      energy.push_back( DataType((*compPartItr)->e()/CLHEP::GeV ) );
      px.push_back( DataType((*compPartItr)->px()/CLHEP::GeV ) );
      py.push_back( DataType((*compPartItr)->py()/CLHEP::GeV ) );
      pz.push_back( DataType((*compPartItr)->pz()/CLHEP::GeV ) );
 
      charge.push_back( DataType( (*compPartItr)->charge() ));
      dataType.push_back( DataType( (*compPartItr)->dataType() ));
      typeLabel = "n_a";
      pdgId2 = (*compPartItr)->pdgId();
      pdgId.push_back( DataType( pdgId2 ));
      if( abs(pdgId2) == 11) typeLabel = "EV_Electron";
      if( abs(pdgId2) == 12) typeLabel = "EV_NeutrinoElectron";
      if( abs(pdgId2) == 13) typeLabel = "EV_Muon";
      if( abs(pdgId2) == 14) typeLabel = "EV_NeutrinoMuon";
      if( abs(pdgId2) == 15) typeLabel = "EV_Tau";
      if( abs(pdgId2) == 16) typeLabel = "EV_NeutrinoTau";
      if( abs(pdgId2) == 6) typeLabel = "EV_Top";  
      if( abs(pdgId2) == 5) typeLabel = "EV_Bottom";
      if( abs(pdgId2) == 22) typeLabel = "EV_Photon";
      if( abs(pdgId2) == 23) typeLabel = "EV_Z0";
      if( pdgId2 == 24) typeLabel = "EV_Wplus";
      if( pdgId2 == -24) typeLabel = "EV_Wminus";
      typeEV.push_back( DataType( typeLabel ));
      label.push_back( DataType( "none" ) );
 
//    childID.push_back( DataType( "none" ) ); // placeholders
//    motherID.push_back( DataType( "none" ) );
    }
    // four-vectors
    DataMap["phi"] = phi;
    DataMap["eta"] = eta;
    DataMap["et"] = et;
    DataMap["energy"] = energy;
    DataMap["mass"] = mass;
    DataMap["px"] = px;
    DataMap["py"] = py;
    DataMap["pz"] = pz;
    DataMap["pdgId"] = pdgId;
    DataMap["typeEV"] = typeEV;
    DataMap["charge"] = charge;
    DataMap["dataType"] = dataType;
    DataMap["label"] = label;
    
//    DataMap["childID"] = childID;
//    DataMap["motherID"] = motherID;
 
    if (msgLvl(MSG::DEBUG)) {
      msg(MSG::DEBUG) << dataTypeName() << " retrieved with " << phi.size() << " entries"<< endmsg;
    }
 
    //All collections retrieved okay
    return DataMap;
 
  } // retrieve
 
  //--------------------------------------------------------------------------
  
} // JiveXML namespace