TgcPrepDataRetriever.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TgcPrepDataRetriever.h"
 
#include "MuonFullIDHelper.h"
#include "MuonReadoutGeometry/TgcReadoutElement.h"
 
namespace JiveXML {
 
  //--------------------------------------------------------------------------
 
  TgcPrepDataRetriever::TgcPrepDataRetriever(const std::string& type,const std::string& name, const IInterface* parent):
    AthAlgTool(type, name, parent)
  {
 
    declareInterface<IDataRetriever>(this);    
  }
 
  //--------------------------------------------------------------------------
 
  StatusCode TgcPrepDataRetriever::initialize(){
    
    ATH_MSG_DEBUG("Initializing retriever for " << dataTypeName()); 
    ATH_CHECK(m_sgKey.initialize());
    ATH_CHECK( m_idHelperSvc.retrieve() );
 
    return StatusCode::SUCCESS;
  }
 
  //--------------------------------------------------------------------------
 
  StatusCode TgcPrepDataRetriever::retrieve(ToolHandle<IFormatTool> &FormatTool) {
 
    //be verbose
    ATH_MSG_DEBUG("Retrieving " << dataTypeName()); 
 
    SG::ReadHandle<Muon::TgcPrepDataContainer> tgcContainer(m_sgKey);
 
    int ndata = 0;
    Muon::TgcPrepDataContainer::const_iterator containerIt;
    for (containerIt=tgcContainer->begin(); containerIt!=tgcContainer->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 swidthVec; swidthVec.reserve(ndata);
    DataVect lwidthVec; lwidthVec.reserve(ndata);
    DataVect idVec; idVec.reserve(ndata);
    DataVect identifierVec; identifierVec.reserve(ndata);
    DataVect barcode; barcode.reserve(ndata);
 
    for (containerIt=tgcContainer->begin(); containerIt!=tgcContainer->end(); ++containerIt) {
      const Muon::TgcPrepDataCollection *tgcCollection = *containerIt;
 
      Muon::TgcPrepDataCollection::const_iterator collectionIt;
      for (collectionIt=tgcCollection->begin(); collectionIt!=tgcCollection->end(); ++collectionIt) {
 
        const Muon::TgcPrepData *data = *collectionIt;
        const MuonGM::TgcReadoutElement *element = data->detectorElement();
        Identifier id = data->identify();
 
        if (!element) {
          if (msgLvl(MSG::WARNING)) ATH_MSG_WARNING("No MuonGM::TgcReadoutElement for hit " << id);
          continue;
        }
 
    Amg::Vector3D globalPos; 
        double shortWidth, longWidth, length;
        if (m_idHelperSvc->tgcIdHelper().isStrip(id)) {
          int gasGap = m_idHelperSvc->tgcIdHelper().gasGap(id);
          int strip = m_idHelperSvc->tgcIdHelper().channel(id);
          shortWidth = element->stripShortWidth(gasGap, strip);
          longWidth = element->stripLongWidth(gasGap, strip);
          length = element->stripLength();
          globalPos = element->stripPos(gasGap, strip);
        } else {
          int gasGap = m_idHelperSvc->tgcIdHelper().gasGap(id);
          int gang = m_idHelperSvc->tgcIdHelper().channel(id);
          shortWidth = element->gangShortWidth(gasGap, gang);
          longWidth = element->gangLongWidth(gasGap, gang);
          length = element->gangRadialLength(gasGap, gang);
          globalPos = element->wireGangPos(gasGap, gang);
        }
 
        x.push_back(DataType(globalPos.x()/CLHEP::cm));
        y.push_back(DataType(globalPos.y()/CLHEP::cm));
        z.push_back(DataType(globalPos.z()/CLHEP::cm));
        swidthVec.push_back(DataType(shortWidth/CLHEP::cm));
        lwidthVec.push_back(DataType(longWidth/CLHEP::cm));
        lengthVec.push_back(DataType(length/CLHEP::cm));
        identifierVec.push_back(DataType(MuonFullIDHelper::getFullID(id, m_idHelperSvc->tgcIdHelper())));
        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["swidth"] = swidthVec;
    myDataMap["lwidth"] = lwidthVec;
    myDataMap["identifier"] = identifierVec;
    myDataMap["id"] = idVec;
    myDataMap["barcode"] = barcode;
 
    //Be verbose
    ATH_MSG_DEBUG(dataTypeName() << ": "<< x.size());
 
    //return FormatTool->AddToEvent(dataTypeName(), m_sgKey, &myDataMap);
    std::string emptyStr="";
    return FormatTool->AddToEvent(dataTypeName(), m_sgKey.key(), &myDataMap);  
  }
}