RawDataMcParticleCnv.hpp

Go to the documentation of this file.

#pragma once
 
#include "McTruth/McParticle.h"
#include "RawDataCnv/EventManagement/McParticleBuilder.h"
#include "RawDataCnv/EventManagement/McTruthDataStructure.h"
#include "RawDataCnv/EventManagement/RAWEVENT.h"
#include "RawDataCnv/EventManagement/RawEventDef.h"
 
#include "RawDataBaseCnv.h"
 
extern const CLID& CLID_McParticleCol;
 
class RawDataMcParticleCnv : public RawDataBaseCnv {
private:
  McParticleBuilder m_mcParticleBuilder;
 
public:
  RawDataMcParticleCnv( ISvcLocator* svc ) : RawDataBaseCnv( svc, classID() ) {}
 
  static const CLID& classID() { return CLID_McParticleCol; }
 
  StatusCode createObj( IOpaqueAddress* pAddr, DataObject*& pObj ) {
    // MsgStream log(msgSvc(), "RawDataMcParticleCnv");
 
    // This converter will create an empty McParticleCol on the TDS
    McParticleCol* mcParticleCol = new McParticleCol;
    pObj                         = mcParticleCol;
 
    RAWEVENT* evt = m_inputSvc->currentEvent();
    if ( evt == NULL )
    {
      // log << MSG::ERROR << "RawDataCnv has no event!" << endmsg;
      return StatusCode::FAILURE;
    }
 
    const BufferHolder& mcParBuf = evt->getMcParBuf();
    if ( mcParBuf.nBuf() == 0 )
    {
      // cerr << "no McParticle Data Buffer found!!!" << endl;
      return StatusCode::FAILURE;
    }
    uint32_t* buf = mcParBuf( 0 );
 
    uint32_t nParticle = ( buf[0] >> 16 );
    // uint32_t nVertex   = (buf[0] & 0xFFFF);
    double* vPointer = (double*)( buf + nParticle * 11 + 1 );
 
    for ( uint32_t i = 0; i < nParticle; i++ )
    {
      McParticle* mcPar = new McParticle;
      m_mcParticleBuilder.unPack( ( buf + 1 + i * 11 ), vPointer, mcPar );
      mcParticleCol->push_back( mcPar );
    }
 
    // Get primary McParticles
    SmartRefVector<McParticle> primaryParticleCol;
    McParticleCol::iterator    iter_particle = mcParticleCol->begin();
    for ( ; iter_particle != mcParticleCol->end(); iter_particle++ )
    {
      if ( ( *iter_particle )->primaryParticle() )
      {
        McParticle* mcPart = (McParticle*)( *iter_particle );
        primaryParticleCol.push_back( mcPart );
      }
    }
 
    if ( primaryParticleCol.empty() )
    {
      // log << MSG::WARNING << "createObj error: no primary particle found!" << endmsg;
    }
 
    // Add mother particle recursively
    SmartRefVector<McParticle>::iterator iter_primary = primaryParticleCol.begin();
    // std::cout << "primaryParticleCol.size() = " << primaryParticleCol.size() << std::endl;
    for ( ; iter_primary != primaryParticleCol.end(); iter_primary++ )
    {
 
      if ( !( *iter_primary )->leafParticle() ) addMother( ( *iter_primary ), mcParticleCol );
    }
 
    return StatusCode::SUCCESS;
  }
 
  StatusCode createRep( DataObject* /*pObj*/, IOpaqueAddress*& /*pAddr*/ ) {
    // convert PixelRaw in the container into ByteStream
    // MsgStream log(msgSvc(), "RawDataMcParticleCnv");
 
    WriteRawEvent*& re = m_cnvSvc->getRawEvent();
    if ( re == 0 )
    {
      // log << " get RawEvent failed !" << endmsg;
      return StatusCode::FAILURE;
    }
 
    SmartDataPtr<McParticleCol> mcParticleCol( dataProvider(), EventModel::MC::McParticleCol );
    if ( mcParticleCol == 0 )
    {
      // log << "no McParticleCol found" << endmsg;
      return StatusCode::FAILURE;
    }
 
    return m_mcParticleBuilder.pack( mcParticleCol, re );
  }
 
  void addMother( McParticle* currentParticle, McParticleCol* particleCol ) {
 
    if ( currentParticle->leafParticle() ) return;
 
    bool                    found = false;
    McParticleCol::iterator iter  = particleCol->begin();
    // std::cout << "particleCol->size() = " << particleCol->size() << std::endl;
    for ( ; iter != particleCol->end(); iter++ )
    {
      if ( currentParticle->vertexIndex1() == ( *iter )->vertexIndex0() )
      {
        found = true;
        ( *iter )->setMother( currentParticle );
        currentParticle->addDaughter( *iter );
        addMother( ( *iter ), particleCol );
      }
    }
 
    if ( !found )
      std::cout << "RawDataMcParticleCnv::addMother: inconsistency was found!" << std::endl;
  }
};