MdcMcHitCnv.cxx

Go to the documentation of this file.

#ifndef MdcMcHitCnv_CXX
#define MdcMcHitCnv_CXX 1
 
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/ObjectVector.h"
 
#include "TClonesArray.h"
 
#include "EventModel/EventModel.h"
#include "McTruth/McEvent.h"  //TDS object
#include "McTruth/MdcMcHit.h" //TDS object
#include "RootCnvSvc/Mc/McCnv.h"
#include "RootCnvSvc/Mc/MdcMcHitCnv.h"
#include "RootCnvSvc/RootAddress.h"
#include "RootEventData/TMcEvent.h"
#include "RootEventData/TMdcMc.h" // standard root object
 
// Instantiation of a static factory class used by clients to create
// instances of this service
// static CnvFactory<MdcMcHitCnv> s_factory;
// const ICnvFactory& MdcMcHitCnvFactory = s_factory;
 
MdcMcHitCnv::MdcMcHitCnv( ISvcLocator* svc ) : RootEventBaseCnv( classID(), svc ) {
  // Here we associate this converter with the /Event path on the TDS.
  MsgStream log( msgSvc(), "MdcMcHitCnv" );
  // log << MSG::DEBUG << "Constructor called for " << objType() << endmsg;
  m_rootBranchname = "m_mdcMcHitCol";
  // declareObject(EventModel::MC::MdcMcHitCol, objType(), m_rootTreename, m_rootBranchname);
  m_adresses.push_back( &m_mdcMcHitCol );
  m_mdcMcHitCol = 0;
}
 
StatusCode MdcMcHitCnv::TObjectToDataObject( DataObject*& refpObject ) {
  // creation of TDS object from root object
 
  MsgStream log( msgSvc(), "MdcMcHitCnv" );
  log << MSG::DEBUG << "MdcMcHitCnv::TObjectToDataObject" << endmsg;
  StatusCode sc = StatusCode::SUCCESS;
 
  // create the TDS location for the MdcMc Collection
  MdcMcHitCol* mdcMcTdsCol = new MdcMcHitCol;
  refpObject               = mdcMcTdsCol;
 
  // now convert
  if ( !m_mdcMcHitCol ) return sc;
  TIter   mdcMcIter( m_mdcMcHitCol );
  TMdcMc* mdcMcRoot = 0;
  while ( ( mdcMcRoot = (TMdcMc*)mdcMcIter.Next() ) )
  {
    unsigned int id            = mdcMcRoot->getId();
    unsigned int trackIndex    = mdcMcRoot->getTrackIndex();
    double       xPosition     = mdcMcRoot->getPositionX();
    double       yPosition     = mdcMcRoot->getPositionY();
    double       zPosition     = mdcMcRoot->getPositionZ();
    double       driftDistance = mdcMcRoot->getDriftDistance();
    double       depositEnergy = mdcMcRoot->getDepositEnergy();
    int          positionFlag  = mdcMcRoot->getPositionFlag();
 
    MdcMcHit* mdcMcTds                    = new MdcMcHit;
    m_common.m_rootMdcMcHitMap[mdcMcRoot] = mdcMcTds;
 
    mdcMcTds->setIdentifier( id );
    mdcMcTds->setTrackIndex( trackIndex );
    mdcMcTds->setCurrentTrackPID( mdcMcRoot->getCurrentTrackPID() );
    mdcMcTds->setIsSecondary( mdcMcRoot->getIsSecondary() );
    mdcMcTds->setPositionX( xPosition );
    mdcMcTds->setPositionY( yPosition );
    mdcMcTds->setPositionZ( zPosition );
    mdcMcTds->setMomentumX( mdcMcRoot->getMomentumX() );
    mdcMcTds->setMomentumY( mdcMcRoot->getMomentumY() );
    mdcMcTds->setMomentumZ( mdcMcRoot->getMomentumZ() );
    mdcMcTds->setDriftDistance( driftDistance );
    mdcMcTds->setDepositEnergy( depositEnergy );
    mdcMcTds->setPositionFlag( positionFlag );
    mdcMcTds->setFlightLength( mdcMcRoot->getFlightLength() );
    mdcMcTds->setCreatorProcess( std::string( mdcMcRoot->getCreatorProcess().Data() ) );
    mdcMcTds->setDigiIdx( mdcMcRoot->getDigiIdx() );
 
    mdcMcTdsCol->push_back( mdcMcTds );
  }
  // m_mdcMcHitCol->Delete();  // wensp add 2005/12/30
  delete m_mdcMcHitCol;
  m_mdcMcHitCol = 0;
  return StatusCode::SUCCESS;
}
 
StatusCode MdcMcHitCnv::DataObjectToTObject( DataObject* obj, RootAddress* rootaddr ) {
 
  MsgStream log( msgSvc(), "MdcMcHitCnv" );
  log << MSG::DEBUG << "MdcMcHitCnv::DataObjectToTObject" << endmsg;
  StatusCode sc = StatusCode::SUCCESS;
 
  MdcMcHitCol* mdcMcHitCnvTds = dynamic_cast<MdcMcHitCol*>( obj );
  if ( !mdcMcHitCnvTds )
  {
    log << MSG::ERROR << "Could not downcast to MdcMcHitCol" << endmsg;
    return StatusCode::FAILURE;
  }
 
  DataObject* evt;
  m_eds->findObject( EventModel::MC::Event, evt );
  if ( evt == NULL )
  {
    log << MSG::ERROR << "Could not get McEvent in TDS " << endmsg;
    return StatusCode::FAILURE;
  }
  McEvent* devtTds = dynamic_cast<McEvent*>( evt );
  if ( !devtTds )
  { log << MSG::ERROR << "MdcMcHitCnv:Could not downcast to TDS McEvent" << endmsg; }
  IOpaqueAddress* addr;
 
  m_cnvSvc->getMcCnv()->createRep( evt, addr );
  TMcEvent* McEvt = m_cnvSvc->getMcCnv()->getWriteObject();
 
  const TObjArray* m_mdcMcHitCol = McEvt->getMdcMcHitCol();
  if ( !m_mdcMcHitCol ) return sc;
  McEvt->clearMdcMcHitCol(); // necessary in case there is I/O at the same time since array is
                             // static
  MdcMcHitCol::const_iterator mdcMcTds;
 
  for ( mdcMcTds = mdcMcHitCnvTds->begin(); mdcMcTds != mdcMcHitCnvTds->end(); mdcMcTds++ )
  {
    UInt_t   id            = ( ( *mdcMcTds )->identify() ).get_value();
    UInt_t   trackIndex    = ( *mdcMcTds )->getTrackIndex();
    Double_t xPosition     = ( *mdcMcTds )->getPositionX();
    Double_t yPosition     = ( *mdcMcTds )->getPositionY();
    Double_t zPosition     = ( *mdcMcTds )->getPositionZ();
    Double_t driftDistance = ( *mdcMcTds )->getDriftDistance();
    Double_t depositEnergy = ( *mdcMcTds )->getDepositEnergy();
    Int_t    positionFlag  = ( *mdcMcTds )->getPositionFlag();
 
    TMdcMc* mdcMcRoot = new TMdcMc();
    // m_common.m_mdcMcHitMap[(*mdcMcTds)] = mdcMcRoot;
 
    mdcMcRoot->setId( id );
    mdcMcRoot->setTrackIndex( trackIndex );
    mdcMcRoot->setCurrentTrackPID( ( *mdcMcTds )->getCurrentTrackPID() );
    mdcMcRoot->setIsSecondary( ( *mdcMcTds )->getIsSecondary() );
    mdcMcRoot->setPositionX( xPosition );
    mdcMcRoot->setPositionY( yPosition );
    mdcMcRoot->setPositionZ( zPosition );
    mdcMcRoot->setMomentumX( ( *mdcMcTds )->getMomentumX() );
    mdcMcRoot->setMomentumY( ( *mdcMcTds )->getMomentumY() );
    mdcMcRoot->setMomentumZ( ( *mdcMcTds )->getMomentumZ() );
    mdcMcRoot->setDriftDistance( driftDistance );
    mdcMcRoot->setDepositEnergy( depositEnergy );
    mdcMcRoot->setPositionFlag( positionFlag );
    mdcMcRoot->setFlightLength( ( *mdcMcTds )->getFlightLength() );
    mdcMcRoot->setCreatorProcess( ( *mdcMcTds )->getCreatorProcess() );
    mdcMcRoot->setDigiIdx( ( *mdcMcTds )->getDigiIdx() );
 
    McEvt->addMdcMc( mdcMcRoot );
  }
 
  return StatusCode::SUCCESS;
}
#endif