SegmentRetriever.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TrkJiveXML/SegmentRetriever.h"
 
#include "TrkSegment/Segment.h"
#include "TrkSegment/TrackSegment.h"
#include "TrkSegment/SegmentCollection.h"
#include "TrkRIO_OnTrack/RIO_OnTrack.h"
 
#include "JiveXML/DataType.h"
 
namespace JiveXML {
 
  SegmentRetriever::SegmentRetriever(const std::string& type,const std::string& name,const IInterface* parent):
    AthAlgTool(type,name,parent),
    m_typeName("Segment"){
 
      //Only declare the interface
      declareInterface<IDataRetriever>(this);
  }
  
  StatusCode SegmentRetriever::retrieve(ToolHandle<IFormatTool> &FormatTool) {
    
    //be verbose
    if (msgLvl(MSG::DEBUG)) msg(MSG::DEBUG) << "Retrieving " << dataTypeName() <<endmsg; 
 
    //Get an iterator over all segement collections
    SG::ConstIterator<Trk::SegmentCollection> CollectionItr, CollectionsEnd;
    if ((evtStore()->retrieve(CollectionItr, CollectionsEnd)).isFailure()){
      if (msgLvl(MSG::ERROR)) msg(MSG::ERROR) << "Unable to retrieve iterator for Segment collection" << endmsg;
      return StatusCode::RECOVERABLE;
    }
 
    //Loop over segment collections
    for (; CollectionItr!=CollectionsEnd; ++CollectionItr) {
 
      //Get number of segments in this collection
      Trk::SegmentCollection::size_type NSegs=(*CollectionItr).size();
 
      //Define the data vectors we want to fill and create space
      DataVect x; x.reserve(NSegs);
      DataVect y; y.reserve(NSegs);
      DataVect z; z.reserve(NSegs);
      DataVect phi; phi.reserve(NSegs);
      DataVect theta; theta.reserve(NSegs);
      DataVect numHits; numHits.reserve(NSegs);
      DataVect hits;
      
      //Loop over the segments
      Trk::SegmentCollection::const_iterator SegmentItr;
      for (SegmentItr=(*CollectionItr).begin(); SegmentItr!=(*CollectionItr).end(); ++SegmentItr) {
        
        //Retrive primite variabels
        x.emplace_back((*SegmentItr)->globalPosition().x()/10.);
        y.emplace_back((*SegmentItr)->globalPosition().y()/10.);
        z.emplace_back((*SegmentItr)->globalPosition().z()/10.);
        phi.emplace_back((*SegmentItr)->localParameters()[Trk::phi]);
        theta.emplace_back((*SegmentItr)->localParameters()[Trk::theta]);
 
        //Count number of valid (non-null) RIO_OnTracks
        int NRoTs = 0;
        //Reserve space for expected number
        hits.reserve(hits.size()+(*SegmentItr)->containedMeasurements().size());
 
        //Loop over segment measurments
        std::vector< const Trk::MeasurementBase * >::const_iterator measItr, measEnd;
        measItr=(*SegmentItr)->containedMeasurements().begin();
        measEnd=(*SegmentItr)->containedMeasurements().end();
 
        //Now loop over measurements
        for (; measItr!=measEnd; ++measItr) {
          
          //dynamic_cast to RIO_OnTrack - will return NULL if not a RIO_OnTrack object
          const Trk::RIO_OnTrack *RoT = dynamic_cast<const Trk::RIO_OnTrack*>(*measItr);  
 
          //Ignore failed dynamic_casts
          if (!RoT) continue ;
 
          //Add the hit
          hits.emplace_back(RoT->identify().get_compact() );
          //count as valid
          NRoTs++;
        }
 
        //Store number of hits for this segement
        numHits.emplace_back(NRoTs);
      }
 
      //Add data to our map
      DataMap DataMap;
      DataMap["x"] = x;
      DataMap["y"] = y;
      DataMap["z"] = z;
      DataMap["phi"] = phi;
      DataMap["theta"] = theta;
      DataMap["numHits"] = numHits;
 
      //Hits are stored as multiple with average size given in XML header
      if (NSegs > 0) {
        std::string multiple = "hits multiple=\"" + DataType( hits.size()*1./numHits.size()).toString() + "\"";
        DataMap[multiple] = hits;
      }
    
      //forward data to formating tool
      if ( FormatTool->AddToEvent(dataTypeName(), CollectionItr.key(), &DataMap).isFailure())
            return StatusCode::RECOVERABLE;
      
      //Be verbose
      if (msgLvl(MSG::DEBUG)) {
        msg(MSG::DEBUG) << dataTypeName() << " collection " << CollectionItr.key();
        msg(MSG::DEBUG) << " retrieved with " << NSegs << " entries"<< endmsg;
      }
 
    }//Loop over segment collections
    
    //All collections retrieved - done
    return StatusCode::SUCCESS;
 
  } // retrive
} //namespace