RawDataTofMcHitCnv.hpp

Go to the documentation of this file.

#pragma once
 
#include "Identifier/TofID.h"
#include "McTruth/TofMcHit.h"
#include "RawDataCnv/EventManagement/McTruthDataStructure.h"
#include "RawDataCnv/EventManagement/TofMcHitBuilder.h"
 
#include "RawDataBaseCnv.h"
 
extern const CLID& CLID_TofMcHitCol;
 
class RawDataTofMcHitCnv : public RawDataBaseCnv {
private:
  /// TofMcHitBuilder for packing or unpacking data
  TofMcHitBuilder m_tofMcHitBuilder;
 
public:
  RawDataTofMcHitCnv( ISvcLocator* svc ) : RawDataBaseCnv( svc, classID() ) {}
 
  static const CLID& classID() { return CLID_TofMcHitCol; }
 
  StatusCode createObj( IOpaqueAddress* pAddr, DataObject*& pObj ) {
    MsgStream log( msgSvc(), "RawDataTofMcHitCnv" );
 
    // This converter will create an empty TofMcHitCol on the TDS
    TofMcHitCol* tofMcHitCol = new TofMcHitCol;
    pObj                     = tofMcHitCol;
 
    RAWEVENT* evt = m_inputSvc->currentEvent();
    if ( evt == NULL )
    {
      // log << MSG::ERROR << "RawDataCnv has no event!" << endmsg;
      return StatusCode::FAILURE;
    }
 
    assert( ( evt->getTofTruth().size() ) % 9 == 0 );
 
    TofTruth_t m_TofTruth;
    TofMcHit*  tofMcHit;
 
    std::vector<uint32_t>::const_iterator iter  = evt->getTofTruth().begin();
    std::vector<uint32_t>::const_iterator eiter = evt->getTofTruth().end();
 
    for ( int tofMcHitId = 0; iter != eiter; tofMcHitId++ )
    {
      // retrieve the TofTruth data
      m_tofMcHitBuilder.unPack( iter, eiter, m_TofTruth );
      // construct the identifier
      Identifier   ident;
      unsigned int barrel_ec = m_TofTruth.partId;
      // for scintillator
      if ( barrel_ec >= 0 && barrel_ec <= 2 )
      {
        unsigned int layer = 0;
        if ( m_TofTruth.scinNb > TofID::getPHI_BARREL_MAX() )
        {
          layer = 1;
          m_TofTruth.scinNb -= ( TofID::getPHI_BARREL_MAX() + 1 );
        }
        ident = TofID::cell_id( barrel_ec, layer, m_TofTruth.scinNb, 0 );
      }
      // for ETF(MRPC)
      else if ( barrel_ec >= 3 )
      {
        unsigned int endcap = 0;
        if ( barrel_ec == 4 ) { endcap = 1; }
        ident = TofID::cell_id( 3, endcap, m_TofTruth.scinNb, m_TofTruth.strip, 0 );
      }
      else
      { log << MSG::ERROR << "barrel_ec is not correct! barrel_ec=" << barrel_ec << endmsg; }
 
      // construct the TofMcHit
      tofMcHit = new TofMcHit(
          ident, m_TofTruth.trackIndex, m_TofTruth.x / ( m_tofMcHitBuilder.m_xCoeff * 1.0 ),
          m_TofTruth.y / ( m_tofMcHitBuilder.m_yCoeff * 1.0 ),
          m_TofTruth.z / ( m_tofMcHitBuilder.m_zCoeff * 1.0 ),
          m_TofTruth.px / ( m_tofMcHitBuilder.m_pxCoeff * 1.0 ),
          m_TofTruth.py / ( m_tofMcHitBuilder.m_pyCoeff * 1.0 ),
          m_TofTruth.pz / ( m_tofMcHitBuilder.m_pzCoeff * 1.0 ),
          m_TofTruth.trackLength / ( m_tofMcHitBuilder.m_trackLengthCoeff * 1.0 ),
          m_TofTruth.time / ( m_tofMcHitBuilder.m_timeCoeff * 1.0 ) );
      // And add the stuff to the container
      tofMcHitCol->push_back( tofMcHit );
    }
 
    return StatusCode::SUCCESS;
  }
 
  StatusCode createRep( DataObject* pObj, IOpaqueAddress*& pAddr ) {
    // convert PixelRaw in the container into ByteStream
    // MsgStream log(msgSvc(), "RawDataTofMcHitCnv");
 
    WriteRawEvent*& re = m_cnvSvc->getRawEvent();
 
    if ( re == 0 )
    {
      // log << " get RawEvent failed !" << endmsg;
      return StatusCode::FAILURE;
    }
 
    SmartDataPtr<TofMcHitCol> mcHitCol( dataProvider(), EventModel::MC::TofMcHitCol );
 
    if ( mcHitCol == 0 )
    {
      // log << "no TofMcHitCol found" << endmsg;
      return StatusCode::FAILURE;
    }
 
    StatusCode sc = m_tofMcHitBuilder.pack( mcHitCol, re );
 
    return sc;
  }
};