CscPrepDataRetriever.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#include "CscPrepDataRetriever.h"
 
#include "MuonFullIDHelper.h"
#include "MuonReadoutGeometry/CscReadoutElement.h"
 
namespace JiveXML {
 
  //--------------------------------------------------------------------------
 
   CscPrepDataRetriever::CscPrepDataRetriever(const std::string& type, const std::string& name, const IInterface* parent):
     AthAlgTool(type,name,parent)
   {
 
     declareInterface<IDataRetriever>(this);
 
   }
 
  //--------------------------------------------------------------------------
 
  StatusCode CscPrepDataRetriever::initialize(){
    
    ATH_MSG_DEBUG("Initializing retriever for " << dataTypeName()); 
    
    ATH_CHECK(m_sgKey.initialize());
    ATH_CHECK( m_idHelperSvc.retrieve() );
    return StatusCode::SUCCESS;
  }        
                
  //--------------------------------------------------------------------------
 
  StatusCode CscPrepDataRetriever::retrieve(ToolHandle<IFormatTool> &FormatTool) {
 
    //be verbose
    ATH_MSG_VERBOSE("Retrieving " << dataTypeName()); 
 
    SG::ReadHandle<Muon::CscPrepDataContainer> cscContainer(m_sgKey);
 
 
    int ndata = 0;
    Muon::CscPrepDataContainer::const_iterator containerIt;
    for (containerIt=cscContainer->begin(); containerIt!=cscContainer->end(); ++containerIt) {
               ndata += (*containerIt)->size();
    }
 
    //Make the vectors to contain the information and reserve space accordingly
    DataVect x; x.reserve(ndata);
    DataVect y; y.reserve(ndata);
    DataVect z; z.reserve(ndata);
    DataVect lengthVec; lengthVec.reserve(ndata);
    DataVect chargeVec; chargeVec.reserve(ndata);
    DataVect idVec; idVec.reserve(ndata);
    DataVect identifierVec; identifierVec.reserve(ndata);
    DataVect barcode; barcode.reserve(ndata);
 
    for (containerIt=cscContainer->begin(); containerIt!=cscContainer->end(); ++containerIt) {
      const Muon::CscPrepDataCollection *cscCollection = *containerIt;
 
      Muon::CscPrepDataCollection::const_iterator collectionIt;
      for (collectionIt=cscCollection->begin(); collectionIt!=cscCollection->end(); ++collectionIt) {
 
        const Muon::CscPrepData *data = *collectionIt;
        const MuonGM::CscReadoutElement *element = data->detectorElement();
        Identifier id = data->identify();
          
        if (!element) {
           ATH_MSG_WARNING("No MuonGM::CscReadoutElement for hit " << id);
          continue;
        }
 
    Amg::Vector3D globalPos = element->stripPos(id);
        double length = element->stripLength(id);
        int charge = data->charge();
 
        x.push_back(DataType(globalPos.x()/CLHEP::cm));
        y.push_back(DataType(globalPos.y()/CLHEP::cm));
        z.push_back(DataType(globalPos.z()/CLHEP::cm));
        lengthVec.push_back(DataType(length/CLHEP::cm));
        chargeVec.push_back(DataType(charge));
        identifierVec.push_back(DataType(MuonFullIDHelper::getFullID(id, m_idHelperSvc->cscIdHelper())));
        idVec.push_back(DataType( id.get_compact() ));
        barcode.push_back(DataType(0));
      }
    }
 
    DataMap myDataMap;
    myDataMap["x"] = x;
    myDataMap["y"] = y;
    myDataMap["z"] = z;
    myDataMap["length"] = lengthVec;
    myDataMap["charge"] = chargeVec;
    myDataMap["identifier"] = identifierVec;
    myDataMap["id"] = idVec;
    myDataMap["barcode"] = barcode;
 
    //Be verbose
    ATH_MSG_DEBUG(dataTypeName() << ": "<< x.size());
 
    //forward data to formating tool
    //return FormatTool->AddToEvent(dataTypeName(), m_sgKey, &myDataMap);
    std::string emptyStr="";
    return FormatTool->AddToEvent(dataTypeName(), m_sgKey.key(), &myDataMap);  
  }
 
  //--------------------------------------------------------------------------
}