McParticleBuilder.cxx

Go to the documentation of this file.

#include "RawDataCnv/EventManagement/McParticleBuilder.h"
#include <fstream>
#include <iostream>
using namespace std;
 
McParticleBuilder::McParticleBuilder() : Builder() {
  // initialize(Builder::m_confFile);
}
 
void McParticleBuilder::unPack( uint32_t* buf, double* vPointer, McParticle* mcPar ) {
  uint32_t trkIndex = ( buf[0] >> 16 );
  uint32_t vIndex0  = ( buf[0] >> 8 ) & 0xFF;
  uint32_t vIndex1  = buf[0] & 0xFF;
 
  double*          ptmp      = (double*)( buf + 3 );
  double*          v0Pointer = vPointer + 4 * vIndex0;
  HepLorentzVector initMomentum( ptmp[0], ptmp[1], ptmp[2], ptmp[3] );
  HepLorentzVector initPosition( v0Pointer[0], v0Pointer[1], v0Pointer[2], v0Pointer[3] );
 
  mcPar->initialize( buf[1], buf[2], initMomentum, initPosition, "" );
  mcPar->setTrackIndex( trkIndex );
  mcPar->setVertexIndex0( vIndex0 );
 
  if ( vIndex1 == 0xFF ) { mcPar->setVertexIndex1( -99 ); }
  else
  {
    mcPar->setVertexIndex1( vIndex1 );
 
    double*          v1Pointer = vPointer + 4 * vIndex1;
    HepLorentzVector finalPosition( v1Pointer[0], v1Pointer[1], v1Pointer[2], v1Pointer[3] );
    mcPar->finalize( finalPosition );
  }
 
  return;
}
 
StatusCode McParticleBuilder::pack( McParticleCol* mcParticleCol, WriteRawEvent*& re ) {
  if ( mcParticleCol == 0 )
  {
    cerr << "McParticleBuilder::pack can't get mcParticleCol" << endl;
    return StatusCode::FAILURE;
  }
 
  uint32_t     nParticle = mcParticleCol->size();
  uint32_t     nVertex   = 0;
  vector<bool> vFlag( nParticle + 1, false );
  double*      vPointer = (double*)( m_buf + nParticle * 11 + 1 );
 
  uint32_t index = 0;
  int      tIndex, vIndex0, vIndex1;
  int      parId;
  uint32_t statusFlag;
 
  m_buf[index++] = ( nParticle << 16 );
 
  McParticleCol::const_iterator itPar = mcParticleCol->begin();
  for ( ; itPar != mcParticleCol->end(); itPar++ )
  {
    tIndex  = ( *itPar )->trackIndex();
    vIndex0 = ( *itPar )->vertexIndex0();
    vIndex1 = ( *itPar )->vertexIndex1();
    if ( vIndex1 == -99 ) vIndex1 = 0xFF;
    parId                         = ( *itPar )->particleProperty();
    statusFlag                    = ( *itPar )->statusFlags();
    const HepLorentzVector& p4vec = ( *itPar )->initialFourMomentum();
    m_buf[index++]                = ( ( tIndex << 16 ) | ( vIndex0 << 8 ) | ( vIndex1 ) );
    m_buf[index++]                = (uint32_t)parId;
    m_buf[index++]                = statusFlag;
    double* ptmp                  = (double*)( m_buf + index );
    ptmp[0]                       = p4vec.x();
    ptmp[1]                       = p4vec.y();
    ptmp[2]                       = p4vec.z();
    ptmp[3]                       = p4vec.t();
    index += 8;
 
    if ( !vFlag[vIndex0] )
    {
      const HepLorentzVector& initPos = ( *itPar )->initialPosition();
      vPointer[vIndex0 * 4 + 0]       = initPos.px();
      vPointer[vIndex0 * 4 + 1]       = initPos.py();
      vPointer[vIndex0 * 4 + 2]       = initPos.pz();
      vPointer[vIndex0 * 4 + 3]       = initPos.e();
      vFlag[vIndex0]                  = true;
      nVertex++;
    }
 
    if ( ( vIndex1 != 0xFF ) && ( !vFlag[vIndex1] ) )
    {
      const HepLorentzVector& finalPos = ( *itPar )->finalPosition();
      vPointer[vIndex1 * 4 + 0]        = finalPos.px();
      vPointer[vIndex1 * 4 + 1]        = finalPos.py();
      vPointer[vIndex1 * 4 + 2]        = finalPos.pz();
      vPointer[vIndex1 * 4 + 3]        = finalPos.e();
      vFlag[vIndex1]                   = true;
      nVertex++;
    }
  }
 
  m_buf[0] |= nVertex;
 
  append2event( re, 0xf10000, ( nParticle * 11 + nVertex * 8 + 1 ) );
 
  return StatusCode::SUCCESS;
}
 
// initialize re2te tables
 
StatusCode McParticleBuilder::initialize( string& initFile ) {
  ifstream f;
 
  // read init file
  f.open( initFile.c_str() );
 
  if ( f.bad() )
  {
    cerr << "Error: could not open file " << initFile << endl;
    return StatusCode::FAILURE;
  }
 
  if ( !Builder::find( f, "##McParticleConf", initFile ) )
  {
    cerr << "Error: could not find '##McParticleConf' in file " << initFile << endl;
    return StatusCode::FAILURE;
  }
 
  f.close();
 
  return StatusCode::SUCCESS;
}
 
uint32_t McParticleBuilder::getTEID( uint32_t /*reid*/ ) { return 0; }
 
uint32_t McParticleBuilder::getREID( uint32_t /*teid*/ ) { return 0; }