StripClusterTruthDecorator.cxx

Go to the documentation of this file.

#include "StripClusterTruthDecorator.h"
 
//output
#include "xAODTracking/TrackMeasurementValidation.h"
#include "xAODTracking/TrackMeasurementValidationContainer.h"
#include "xAODTracking/TrackMeasurementValidationAuxContainer.h"
 
#define AUXDATA(OBJ, TYP, NAME) \
  static const SG::AuxElement::Accessor<TYP> acc_##NAME (#NAME);  acc_##NAME(*(OBJ))
 
 
 
namespace ActsTrk {
 
  StripClusterTruthDecorator::StripClusterTruthDecorator(const std::string& name,
                             ISvcLocator *pSvcLocator) :
    AthReentrantAlgorithm(name, pSvcLocator)
  {}
 
  
  StatusCode StripClusterTruthDecorator::initialize() {
    
    // Read keys
    ATH_CHECK(m_clustercontainer_key.initialize());
    ATH_CHECK(m_associationMap_key.initialize());
        
    // Write keys
    ATH_CHECK(m_write_xaod_key.initialize());
    
    return StatusCode::SUCCESS;
  }
  
  StatusCode StripClusterTruthDecorator::execute(const EventContext& ctx)  const {
    
    SG::ReadHandle<xAOD::StripClusterContainer> StripClusterContainer(m_clustercontainer_key,ctx);
 
    if (!StripClusterContainer.isValid()) {
      ATH_MSG_ERROR("Failed Getting the strip cluster container "<<m_clustercontainer_key);
      return StatusCode::FAILURE;
    }
      
 
    const MeasurementToTruthParticleAssociation* measToTruth(nullptr);
    if (m_useTruthInfo) {
      SG::ReadHandle<MeasurementToTruthParticleAssociation> measToTruthHandle(m_associationMap_key,ctx);
      
      if (measToTruthHandle.isValid()) {
    measToTruth = &*measToTruthHandle;
      } else {
    ATH_MSG_ERROR("Measurement to Truth association map "<<m_associationMap_key.key()<<" not found!");
      }
    }
 
    // Setup outputs
    // Create the xAOD container and its auxiliary store:
    SG::WriteHandle<xAOD::TrackMeasurementValidationContainer> xaod(m_write_xaod_key,ctx);
    ATH_CHECK(xaod.record(std::make_unique<xAOD::TrackMeasurementValidationContainer>(),
              std::make_unique<xAOD::TrackMeasurementValidationAuxContainer>()));
    
    
    // loop over collection and convert to xAOD::TrackMeasurementValidation
    for( const auto* prd : *StripClusterContainer ){
      
      //This needs to be taken care of
      Identifier clusterId = Identifier((int)prd->identifier());
      if ( !clusterId.is_valid() ) {
    ATH_MSG_WARNING("Strip cluster identifier is not valid");
      }
 
      xAOD::TrackMeasurementValidation* xprd = new xAOD::TrackMeasurementValidation();
      
      //unsigned int cluster_idx = xaod->size();
      xaod->push_back(xprd);
      //Set Identifier
      xprd->setIdentifier( clusterId.get_compact() );
      
      //Set Global Position
      auto gpos = prd->globalPosition();
      xprd->setGlobalPosition(gpos.x(),gpos.y(),gpos.z());
      
      auto locpos = prd->localPosition<1>();
      // Set local error matrix
      xprd->setLocalPosition(locpos[0],  locpos[1]); 
      
      
      auto localCov = prd->localCovariance<1>();
      
      if(localCov.size() == 1){
    xprd->setLocalPositionError( localCov(0,0), 0., 0. ); 
      } else if(localCov.size() == 4){
    xprd->setLocalPositionError( localCov(0,0), localCov(1,1), localCov(0,1) );     
      } else {
    xprd->setLocalPositionError(0.,0.,0.);
      }
      
      // Get a list of all true particle contributing to the cluster
 
      if (measToTruth) {
 
    if (prd->index() >= (*measToTruth).size()) {
      ATH_MSG_ERROR("PRD index "<< prd->index()<<" not present in the measurement to truth vector with size "<<(*measToTruth).size());
      continue;
    }
    
    auto tps = (*measToTruth)[prd->index()];
    
    //TODO:: reserve / resize?
    std::vector<unsigned int> tp_indices;
    std::vector<unsigned int> tp_barcodes;
    for (auto tp : tps) {
      tp_indices.push_back(tp->index());
      tp_barcodes.push_back(tp->barcode());
    }
    
    //TODO move vectors
    AUXDATA(xprd,std::vector<unsigned int>, truth_index) = tp_indices;
    AUXDATA(xprd,std::vector<unsigned int>, truth_barcode) = tp_barcodes;
    
      }// association map exists
  
    }// close loop on clusters
    
    ATH_MSG_DEBUG( " recorded StripPrepData objects: size " << xaod->size() );
    
    return StatusCode::SUCCESS;
  }
 
 
  StatusCode StripClusterTruthDecorator::finalize(){
    
    return StatusCode::SUCCESS;
  }
  
} // ActsTrk