DstReformAlg.cxx

Go to the documentation of this file.

#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/SmartIF.h"
#include <GaudiKernel/IProperty.h>
#include <TFile.h>
#include <TTree.h>
#include <iostream>
#include <map>
#include <vector>
 
#include "DstReformAlg.h"
 
#include "CLHEP/Geometry/Point3D.h"
#ifndef ENABLE_BACKWARDS_COMPATIBILITY
typedef HepGeom::Point3D<double> HepPoint3D;
#endif
 
using namespace CLHEP;
 
/////////////////////////////////////////////////////////////////////////////
DECLARE_COMPONENT( DstReformAlg )
DstReformAlg::DstReformAlg( const std::string& name, ISvcLocator* pSvcLocator )
    : Algorithm( name, pSvcLocator ) {
 
  declareProperty( "OutputDstFile", m_outputFile );
}
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
StatusCode DstReformAlg::initialize() {
  MsgStream log( msgSvc(), name() );
 
  log << MSG::INFO << "in initialize()" << endmsg;
  m_evtNum = 0;
 
  Gaudi::svcLocator()->service( "TagFilterSvc", m_tagFilterSvc ).ignore();
 
  m_dstDataType = m_tagFilterSvc->getDstDataType();
  std::cout << "init DstReformAlg, m_dstDataType: " << m_dstDataType << std::endl;
 
  return StatusCode::SUCCESS;
}
 
StatusCode DstReformAlg::execute() {
  UInt_t tagData0 = m_tagFilterSvc->getTagData0();
  UInt_t tagData1 = m_tagFilterSvc->getTagData1();
  UInt_t tagData2 = m_tagFilterSvc->getTagData2();
  UInt_t tagData3 = m_tagFilterSvc->getTagData3();
  UInt_t tagData4 = m_tagFilterSvc->getTagData4();
 
  // tagData0 and tagData1 is for DST reform, tagData0 is first priority
 
  // for DTagSetAlg and LTagSetAlg, if tagData0=3, then reform these events with
  // tagData1(nGoodCharged).
  int dstDataType = m_tagFilterSvc->getDstDataType();
  if ( ( dstDataType == 2 || dstDataType == 4 ) && ( tagData0 == 3 ) )
  { cmap[tagData0 + tagData1].push_back( m_evtNum ); }
  else cmap[tagData0].push_back( m_evtNum );
 
  // add tag information to each event
  if ( m_dstDataType == 1 )
  {
    evtmap[m_evtNum].push_back( tagData0 );
    evtmap[m_evtNum].push_back( tagData1 );
    evtmap[m_evtNum].push_back( tagData2 );
    evtmap[m_evtNum].push_back( tagData3 );
    evtmap[m_evtNum].push_back( tagData4 );
  }
  else if ( ( m_dstDataType == 2 ) || ( m_dstDataType == 3 ) || ( m_dstDataType == 4 ) )
  {
    UInt_t tagData5 = m_tagFilterSvc->getTagData5();
    UInt_t tagData6 = m_tagFilterSvc->getTagData6();
    UInt_t tagData7 = m_tagFilterSvc->getTagData7();
    UInt_t tagData8 = m_tagFilterSvc->getTagData8();
    UInt_t tagData9 = m_tagFilterSvc->getTagData9();
    evtmap[m_evtNum].push_back( tagData1 );
    evtmap[m_evtNum].push_back( tagData2 );
    evtmap[m_evtNum].push_back( tagData3 );
    evtmap[m_evtNum].push_back( tagData4 );
    evtmap[m_evtNum].push_back( tagData5 );
    evtmap[m_evtNum].push_back( tagData6 );
    evtmap[m_evtNum].push_back( tagData7 );
    evtmap[m_evtNum].push_back( tagData8 );
    evtmap[m_evtNum].push_back( tagData9 );
  }
 
  m_evtNum++;
 
  return StatusCode::SUCCESS;
}
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
StatusCode DstReformAlg::finalize() {
  MsgStream log( msgSvc(), name() );
 
  std::string fn;
 
  auto iProp = service<IProperty>( "EventCnvSvc" );
  if ( iProp )
  {
    iProp->getProperty( "digiRootInputFile", fn ).ignore();
    std::string::size_type p1 = fn.find( '\'' ) + 1;
    std::string::size_type p2 = fn.find( '\'', p1 );
    fn                        = fn.substr( p1, p2 - p1 );
    std::cout << "input dst file name: " << fn << std::endl;
  }
 
  TFile* f1 = TFile::Open( fn.c_str(), "READ" );
  TFile* f2 = TFile::Open( m_outputFile.c_str(), "RECREATE" );
  TTree* t1 = (TTree*)f1->Get( "Event" );
 
  log << MSG::INFO << "in finalize()" << endmsg;
  TTree* jt1  = (TTree*)( f1->Get( "JobInfoTree" ) );
  TTree* jt2  = jt1->CloneTree();
  TTree* t0   = new TTree( "Metadata", "" );
  Int_t  mode = -1, begin = 0, end = 0;
  t0->Branch( "mode", &mode, "mode/I" );
  t0->Branch( "begin", &begin, "begin/I" );
  t0->Branch( "end", &end, "end/I" );
 
  for ( std::map<int, std::vector<long int>>::iterator it = cmap.begin(); it != cmap.end();
        ++it )
  {
    mode  = it->first;
    begin = end;
    end += ( it->second ).size();
    std::cout << "mode: " << mode << " begin: " << begin << " end: " << end << std::endl;
    t0->Fill();
  }
 
  TTree* t2      = t1->CloneTree( 0 );
  Int_t  nEvents = t1->GetEntries();
  std::cout << "Total nEvents: " << nEvents << std::endl;
  Int_t  origEntry = 0, currEntry = 0;
  UInt_t tagData0, tagData1, tagData2, tagData3, tagData4;
  UInt_t tagData5, tagData6, tagData7, tagData8;
  TTree* t3 = new TTree( "Entries", "" );
  t3->Branch( "origEntry", &origEntry, "origEntry/I" );
  t3->Branch( "currEntry", &currEntry, "currEntry/I" );
  t3->Branch( "tagData0", &tagData0, "tagData0/i" );
  t3->Branch( "tagData1", &tagData1, "tagData1/i" );
  t3->Branch( "tagData2", &tagData2, "tagData2/i" );
  t3->Branch( "tagData3", &tagData3, "tagData3/i" );
  t3->Branch( "tagData4", &tagData4, "tagData4/i" );
  if ( ( m_dstDataType == 2 ) || ( m_dstDataType == 3 ) || ( m_dstDataType == 4 ) )
  {
    t3->Branch( "tagData5", &tagData5, "tagData5/i" );
    t3->Branch( "tagData6", &tagData6, "tagData6/i" );
    t3->Branch( "tagData7", &tagData7, "tagData7/i" );
    t3->Branch( "tagData8", &tagData8, "tagData8/i" );
  }
 
  for ( std::map<int, std::vector<long int>>::iterator it = cmap.begin(); it != cmap.end();
        ++it )
  {
    int nNt = it->second.size();
    std::cout << "Writing " << nNt << "\t events with " << it->first << " TagData0..."
              << std::endl;
    for ( int i = 0; i < nNt; ++i )
    {
      origEntry = ( it->second )[i];
      t1->GetEntry( origEntry );
      tagData0 = ( evtmap[origEntry] )[0];
      tagData1 = ( evtmap[origEntry] )[1];
      tagData2 = ( evtmap[origEntry] )[2];
      tagData3 = ( evtmap[origEntry] )[3];
      tagData4 = ( evtmap[origEntry] )[4];
      if ( ( m_dstDataType == 2 ) || ( m_dstDataType == 3 ) || ( m_dstDataType == 4 ) )
      {
        tagData5 = ( evtmap[origEntry] )[5];
        tagData6 = ( evtmap[origEntry] )[6];
        tagData7 = ( evtmap[origEntry] )[7];
        tagData8 = ( evtmap[origEntry] )[8];
      }
      t2->Fill();
      t3->Fill();
      currEntry++;
    }
    ( it->second ).clear();
  }
 
  f2->Write();
  f2->Close();
  return StatusCode::SUCCESS;
}