CgemMcHitCnv.cxx

Go to the documentation of this file.

#ifndef CgemMcHitCnv_CXX
#define CgemMcHitCnv_CXX  1
 
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/ObjectVector.h"
 
#include "TClonesArray.h"
#include "TArrayI.h"
 
#include "EventModel/EventModel.h"
#include "McTruth/McEvent.h"   //TDS object
#include "McTruth/CgemMcHit.h"   //TDS object
#include "RootEventData/TCgemMc.h" // standard root object
#include "RootEventData/TMcEvent.h"
#include "RootCnvSvc/Mc/McCnv.h"
#include "RootCnvSvc/Mc/CgemMcHitCnv.h"
#include "RootCnvSvc/RootAddress.h"
 
// Instantiation of a static factory class used by clients to create
// instances of this service
//static CnvFactory<CgemMcHitCnv> s_factory;
//const ICnvFactory& CgemMcHitCnvFactory = s_factory;
 
CgemMcHitCnv::CgemMcHitCnv(ISvcLocator* svc)
: RootEventBaseCnv(classID(), svc)
{
    // Here we associate this converter with the /Event path on the TDS.
    MsgStream log(msgSvc(), "CgemMcHitCnv");
    //log << MSG::DEBUG << "Constructor called for " << objType() << endreq;
    m_rootBranchname ="m_cgemMcHitCol";
    //declareObject(EventModel::MC::CgemMcHitCol, objType(), m_rootTreename, m_rootBranchname);
    m_adresses.push_back(&m_cgemMcHitCol);
    m_cgemMcHitCol=0;
}
 
// --- root to tds
StatusCode CgemMcHitCnv::TObjectToDataObject(DataObject*& refpObject) {
  // creation of TDS object from root object
 
    MsgStream log(msgSvc(), "CgemMcHitCnv");
    log << MSG::DEBUG << "CgemMcHitCnv::TObjectToDataObject" << endreq;
    StatusCode sc=StatusCode::SUCCESS;
 
    // create the TDS location for the CgemMc Collection
    CgemMcHitCol* cgemMcTdsCol = new CgemMcHitCol;
    refpObject=cgemMcTdsCol;
 
 
    // now convert
    if (!m_cgemMcHitCol) return sc;
    TIter cgemMcIter(m_cgemMcHitCol);
    TCgemMc *cgemMcRoot = 0;
    while ((cgemMcRoot = (TCgemMc*)cgemMcIter.Next())) 
    {
      unsigned int ID_track   = cgemMcRoot->GetTrackID              ();
      unsigned int ID_layer   = cgemMcRoot->GetLayerID              ();
      //Add by sunxh
      unsigned int pdg_code   = cgemMcRoot->GetPDGCode              ();
      unsigned int ID_parent  = cgemMcRoot->GetParentID             ();
      //end
      double       E_deposit  = cgemMcRoot->GetTotalEnergyDeposit   ();
      double       XYZ_pre_x  = cgemMcRoot->GetPositionXOfPrePoint  ();
      double       XYZ_pre_y  = cgemMcRoot->GetPositionYOfPrePoint  ();
      double       XYZ_pre_z  = cgemMcRoot->GetPositionZOfPrePoint  ();
      double       XYZ_post_x = cgemMcRoot->GetPositionXOfPostPoint ();
      double       XYZ_post_y = cgemMcRoot->GetPositionYOfPostPoint ();
      double       XYZ_post_z = cgemMcRoot->GetPositionZOfPostPoint ();
      double       P_pre_x    = cgemMcRoot->GetMomentumXOfPrePoint  ();
      double       P_pre_y    = cgemMcRoot->GetMomentumYOfPrePoint  ();
      double       P_pre_z    = cgemMcRoot->GetMomentumZOfPrePoint  ();
      double       P_post_x   = cgemMcRoot->GetMomentumXOfPostPoint ();
      double       P_post_y   = cgemMcRoot->GetMomentumYOfPostPoint ();
      double       P_post_z   = cgemMcRoot->GetMomentumZOfPostPoint ();
      //Add by sunxh
      TArrayI  ID_Identifier  = cgemMcRoot->GetIdentifier       ();
      //end
      TString  creatorProcess = cgemMcRoot->GetCreatorProcess();
      unsigned int isSecondary= cgemMcRoot->GetIsSecondary();
      double       trkLen_pre = cgemMcRoot->GetFlightLengthPrePoint();
      double       trkLen_post= cgemMcRoot->GetFlightLengthPostPoint();
      const vector<int>& DigiIdxVec     = cgemMcRoot->GetDigiIdxVec();
      const vector<int>& XclusterIdxVec = cgemMcRoot->GetXclusterIdxVec();
      const vector<int>& VclusterIdxVec = cgemMcRoot->GetVclusterIdxVec();
      
      //log << MSG::INFO << "My Message::CgemMcHitCnv::TObjectToDataObject::idMSB="<<idMSB<<" idLSB="<<idLSB<<",trackID="<<trackIndex<<",edep="<<depositEnergy <<" x="<<xPosition<<" y="<<yPosition<<" z="<<zPosition<< endreq;
 
      CgemMcHit *cgemMcTds = new CgemMcHit;
      m_common.m_rootCgemMcHitMap[cgemMcRoot] = cgemMcTds;
    
      cgemMcTds->SetTrackID              ( ID_track   );
      cgemMcTds->SetLayerID              ( ID_layer   );
      //Add by sunxh
      cgemMcTds->SetPDGCode              ( pdg_code   );
      cgemMcTds->SetParentID             ( ID_parent   );
      //end
      cgemMcTds->SetTotalEnergyDeposit   ( E_deposit  );
      cgemMcTds->SetPositionXOfPrePoint  ( XYZ_pre_x  );
      cgemMcTds->SetPositionYOfPrePoint  ( XYZ_pre_y  );
      cgemMcTds->SetPositionZOfPrePoint  ( XYZ_pre_z  );
      cgemMcTds->SetPositionXOfPostPoint ( XYZ_post_x );
      cgemMcTds->SetPositionYOfPostPoint ( XYZ_post_y );
      cgemMcTds->SetPositionZOfPostPoint ( XYZ_post_z );
      cgemMcTds->SetMomentumXOfPrePoint  ( P_pre_x    );
      cgemMcTds->SetMomentumYOfPrePoint  ( P_pre_y    );
      cgemMcTds->SetMomentumZOfPrePoint  ( P_pre_z    );
      cgemMcTds->SetMomentumXOfPostPoint ( P_post_x   );
      cgemMcTds->SetMomentumYOfPostPoint ( P_post_y   );
      cgemMcTds->SetMomentumZOfPostPoint ( P_post_z   );
      cgemMcTds->SetCreatorProcess       ( creatorProcess.Data());
      //Add by sunxh
      int tmp[2000];
      for(int j = 0; j < ID_Identifier.GetSize(); j++){
          tmp[j] = ID_Identifier.GetAt(j);
      }
      cgemMcTds->AddIdentifier(tmp, ID_Identifier.GetSize());
      //end
      cgemMcTds->SetIsSecondary(isSecondary);
      cgemMcTds->SetFlightLengthPrePoint(trkLen_pre);
      cgemMcTds->SetFlightLengthPostPoint(trkLen_post);
      cgemMcTds->SetDigiIdxVec(DigiIdxVec);
      cgemMcTds->SetXclusterIdxVec(XclusterIdxVec);
      cgemMcTds->SetVclusterIdxVec(VclusterIdxVec);
      cgemMcTdsCol->push_back(cgemMcTds);
    }
    //m_cgemMcHitCol->Delete();  // wensp add 2005/12/30
    delete m_cgemMcHitCol;
    m_cgemMcHitCol = 0;
    return StatusCode::SUCCESS;
}
 
// --- tds to root
StatusCode CgemMcHitCnv::DataObjectToTObject(DataObject* obj,RootAddress* rootaddr) {
 
  MsgStream log(msgSvc(), "CgemMcHitCnv");
  log << MSG::DEBUG << "CgemMcHitCnv::DataObjectToTObject" << endreq;
  StatusCode sc=StatusCode::SUCCESS;
 
  CgemMcHitCol * cgemMcHitCnvTds=dynamic_cast<CgemMcHitCol *> (obj);
  if (!cgemMcHitCnvTds) {
    log << MSG::ERROR << "Could not downcast to CgemMcHitCol" << endreq;
    return StatusCode::FAILURE;
  }
 
  DataObject *evt;
  m_eds->findObject(EventModel::MC::Event,evt);
  if (evt==NULL) {
    log << MSG::ERROR << "Could not get McEvent in TDS "  << endreq;
    return StatusCode::FAILURE;
  }
  McEvent * devtTds=dynamic_cast<McEvent *> (evt);
  if (!devtTds) {
    log << MSG::ERROR << "CgemMcHitCnv:Could not downcast to TDS McEvent" << endreq;
  }
  IOpaqueAddress *addr;
 
  m_cnvSvc->getMcCnv()->createRep(evt,addr); 
  TMcEvent *McEvt=m_cnvSvc->getMcCnv()->getWriteObject();
 
  const TObjArray *m_cgemMcHitCol = McEvt->getCgemMcHitCol();
  if (!m_cgemMcHitCol) return sc;
  McEvt->clearCgemMcHitCol(); //necessary in case there is I/O at the same time since array is static
  CgemMcHitCol::const_iterator cgemMcTds;
 
  for (cgemMcTds = cgemMcHitCnvTds->begin(); cgemMcTds != cgemMcHitCnvTds->end(); cgemMcTds++) 
  {
    unsigned int ID_track   = (*cgemMcTds)->GetTrackID              ();
    unsigned int ID_layer   = (*cgemMcTds)->GetLayerID              ();
    //Add by sunxh
    unsigned int pdg_code   = (*cgemMcTds)->GetPDGCode              ();
    unsigned int ID_parent  = (*cgemMcTds)->GetParentID             ();
    //end
    double       E_deposit  = (*cgemMcTds)->GetTotalEnergyDeposit   ();
    double       XYZ_pre_x  = (*cgemMcTds)->GetPositionXOfPrePoint  ();
    double       XYZ_pre_y  = (*cgemMcTds)->GetPositionYOfPrePoint  ();
    double       XYZ_pre_z  = (*cgemMcTds)->GetPositionZOfPrePoint  ();
    double       XYZ_post_x = (*cgemMcTds)->GetPositionXOfPostPoint ();
    double       XYZ_post_y = (*cgemMcTds)->GetPositionYOfPostPoint ();
    double       XYZ_post_z = (*cgemMcTds)->GetPositionZOfPostPoint ();
    double       P_pre_x    = (*cgemMcTds)->GetMomentumXOfPrePoint  ();
    double       P_pre_y    = (*cgemMcTds)->GetMomentumYOfPrePoint  ();
    double       P_pre_z    = (*cgemMcTds)->GetMomentumZOfPrePoint  ();
    double       P_post_x   = (*cgemMcTds)->GetMomentumXOfPostPoint ();
    double       P_post_y   = (*cgemMcTds)->GetMomentumYOfPostPoint ();
    double       P_post_z   = (*cgemMcTds)->GetMomentumZOfPostPoint ();
    //Add by sunxh
    TArrayI  ID_Identifier  = (*cgemMcTds)->GetIdentifier       ();
    //end
    string      creatorProcess = (*cgemMcTds)->GetCreatorProcess();
    unsigned int isSecondary= (*cgemMcTds)->GetIsSecondary();
    double       trkLen_pre = (*cgemMcTds)->GetFlightLengthPrePoint();
    double       trkLen_post= (*cgemMcTds)->GetFlightLengthPostPoint();
    const vector<int>& DigiIdxVec    = (*cgemMcTds)->GetDigiIdxVec();
    const vector<int>& XclusterIdxVec= (*cgemMcTds)->GetXclusterIdxVec();
    const vector<int>& VclusterIdxVec= (*cgemMcTds)->GetVclusterIdxVec();
 
    //log << MSG::INFO << "My Message::CgemMcHitCnv::DataObjectToTObject::idMSB="<<idMSB<<" idLSB="<<idLSB<<",trackID="<<trackIndex<<",edep="<<depositEnergy <<" x="<<xPosition<<" y="<<yPosition<<" z="<<zPosition<< endreq;
 
    TCgemMc *cgemMcRoot = new TCgemMc();
    //m_common.m_cgemMcHitMap[(*cgemMcTds)] = cgemMcRoot;
 
    cgemMcRoot->SetTrackID              ( ID_track   );
    cgemMcRoot->SetLayerID              ( ID_layer   );
    cgemMcRoot->SetPDGCode              ( pdg_code   );
    cgemMcRoot->SetParentID             ( ID_parent  );
    cgemMcRoot->SetTotalEnergyDeposit   ( E_deposit  );
    cgemMcRoot->SetPositionXOfPrePoint  ( XYZ_pre_x  );
    cgemMcRoot->SetPositionYOfPrePoint  ( XYZ_pre_y  );
    cgemMcRoot->SetPositionZOfPrePoint  ( XYZ_pre_z  );
    cgemMcRoot->SetPositionXOfPostPoint ( XYZ_post_x );
    cgemMcRoot->SetPositionYOfPostPoint ( XYZ_post_y );
    cgemMcRoot->SetPositionZOfPostPoint ( XYZ_post_z );
    cgemMcRoot->SetMomentumXOfPrePoint  ( P_pre_x    );
    cgemMcRoot->SetMomentumYOfPrePoint  ( P_pre_y    );
    cgemMcRoot->SetMomentumZOfPrePoint  ( P_pre_z    );
    cgemMcRoot->SetMomentumXOfPostPoint ( P_post_x   );
    cgemMcRoot->SetMomentumYOfPostPoint ( P_post_y   );
    cgemMcRoot->SetMomentumZOfPostPoint ( P_post_z   );
    //Add by sunxh
    int tmp[2000];
    for(int j = 0; j < ID_Identifier.GetSize(); j++){
        tmp[j] = ID_Identifier.GetAt(j);
    }
    cgemMcRoot->AddIdentifier(tmp, ID_Identifier.GetSize());
    //end
    cgemMcRoot->SetCreatorProcess(creatorProcess);
    cgemMcRoot->SetIsSecondary(isSecondary);
    cgemMcRoot->SetFlightLengthPrePoint(trkLen_pre);
    cgemMcRoot->SetFlightLengthPostPoint(trkLen_post);
    cgemMcRoot->SetDigiIdxVec(DigiIdxVec);
    cgemMcRoot->SetXclusterIdxVec(XclusterIdxVec);
    cgemMcRoot->SetVclusterIdxVec(VclusterIdxVec);
    McEvt->addCgemMc(cgemMcRoot);
  }
 
  return StatusCode::SUCCESS;
}
#endif