#include <TagWriterAlg.h>

Inheritance diagram for TagWriterAlg:

Public Member Functions
	TagWriterAlg (const std::string &name, ISvcLocator *pSvcLocator)

StatusCode	initialize ()

StatusCode	execute ()

StatusCode	finalize ()

StatusCode	getTagInfo ()

void	openOutputTagFile (std::string file)

Detailed Description

Definition at line 14 of file TagWriterAlg.h.

Constructor & Destructor Documentation

◆ TagWriterAlg()

TagWriterAlg::TagWriterAlg	(	const std::string &	name,
		ISvcLocator *	pSvcLocator )

Definition at line 38 of file TagWriterAlg.cxx.

                                                                          :
  Algorithm(name, pSvcLocator) {
  
  //declareProperty("InputTagFile", m_inputTagFile);
  declareProperty("OutputTagFile", m_outputTagFile);
  //declareProperty("FilterTotalCharged", m_filterTotalCharged=2);
}

Member Function Documentation

◆ execute()

StatusCode TagWriterAlg::execute ( )

Definition at line 100 of file TagWriterAlg.cxx.

                                 {
 
  MsgStream log(msgSvc(), name());
  log << MSG::INFO << "in execute()" << endreq;
 
  //Write Tag during reconstruction
  if(m_dstOutput!="")
  {
    if(m_entry==-1)
    {
      std::cout << "TagWriterAlg: write tag during reconstruction" << std::endl;
      m_dstFile = m_dstOutput;
      std::cout<< "dst file: "<< m_dstFile << std::endl;
      openOutputTagFile(m_outputTagFile[0]);
      m_oFileTree->Fill();
    }
  }
  else  //Write Tag after reconstruction
  {
    if( m_dstFile != m_rootInterface->getCurrentFileName() )
    {
      std::cout<<"TagWriterAlg, current file: "<<m_rootInterface->getCurrentFileName()<<std::endl;
      m_fileNum++;
      m_dstFile = m_rootInterface->getCurrentFileName();
 
      m_entry=-1;
      if(m_fileNum>0)
      {
        m_oFile->Write();
        delete m_oTree;
        delete m_oFileTree;
        delete m_oFile;
      }
    
      log << MSG::INFO << "open ROOT output TAG file: "<<m_outputTagFile[m_fileNum]<<endreq;
      openOutputTagFile(m_outputTagFile[m_fileNum]);
      m_oFileTree->Fill();
    }
  }
 
  m_entry++;
  getTagInfo();  
  log << MSG::INFO << "writing TAG information to ROOT TAG file"<<endreq;
  m_oTree->Fill();
 
  m_entry_in_alg++;
  return StatusCode::SUCCESS;
}

◆ finalize()

StatusCode TagWriterAlg::finalize ( )

Definition at line 184 of file TagWriterAlg.cxx.

                                  {
 
  MsgStream log(msgSvc(), name());
 
  m_oFile->Write();
 
  log << MSG::INFO << "in finalize()" << endmsg;
  return StatusCode::SUCCESS;
}

◆ getTagInfo()

StatusCode TagWriterAlg::getTagInfo ( )

Definition at line 151 of file TagWriterAlg.cxx.

{
  MsgStream log(msgSvc(), name());
  log << MSG::INFO << "reading event data from DST"<<endreq;
  
  SmartDataPtr<Event::EventHeader> eventHeader(eventSvc(),"/Event/EventHeader");
  if(!eventHeader)
  {
    log << MSG::FATAL << "Could not find EventHeader." << endreq;
    return StatusCode::FAILURE;
  }
 
  m_runNo=eventHeader->runNumber();
  m_eventId=eventHeader->eventNumber();
  log << MSG::DEBUG <<"run, evtnum = " << m_runNo << " , "<< m_eventId <<endreq;
 
  SmartDataPtr<EvtRecEvent> evtRecEvent(eventSvc(), EventModel::EvtRec::EvtRecEvent);
  if(!evtRecEvent)
  {
    log << MSG::FATAL << "Could not find EvtRecEvent." << endreq;
    return StatusCode::FAILURE;
  }
  log << MSG::DEBUG <<"ncharg, nneu, tottks = " 
      << evtRecEvent->totalCharged() << " , "
      << evtRecEvent->totalNeutral() << " , "
      << evtRecEvent->totalTracks() <<endreq;
  m_totalCharged = evtRecEvent->totalCharged();
  m_totalNeutral = evtRecEvent->totalNeutral();
  m_totalTrks = evtRecEvent->totalTracks();
  return StatusCode::SUCCESS;
}

Referenced by execute().

◆ initialize()

StatusCode TagWriterAlg::initialize ( )

Definition at line 47 of file TagWriterAlg.cxx.

                                   {
  MsgStream log(msgSvc(), name());
 
  log << MSG::INFO << "in initialize()" << endmsg;
 
  SmartIF<IJobOptionsSvc> iSvc(serviceLocator()->service("JobOptionsSvc"));
  if ( iSvc.isValid() ) {
    std::string  dll;  //sub so name
    const std::vector<const Property*>* ps = 0;
    if ( iSvc->getProperties("RootCnvSvc") != 0 ) 
      ps = iSvc->getProperties("RootCnvSvc");
    if ( iSvc->getProperties("EventCnvSvc") != 0 )
      ps = iSvc->getProperties("EventCnvSvc");
    if ( iSvc->getProperties("WriteDst") != 0 )
      ps = iSvc->getProperties("WriteDst");
 
    if( ps !=0 )
    {
      for(int i=0;i<ps->size();i++)
      {
        if( (*ps)[i]->name() == "digiRootOutputFile" )
         
        { 
          m_dstOutput = (*ps)[i]->toString();
          std::cout<<"read from JobOptionSvc, dstOutput: "<< m_dstOutput<<std::endl;
        }
      }
    }
  }
 
 
 
  m_rootInterface=RootInterface::Instance(log);
  log << MSG::INFO <<"RootInteface TotalFileNum: "<<m_rootInterface->getTotalFileNum()<<endreq;
  std::cout<<"getTotalFileNum: "<<m_rootInterface->getTotalFileNum()<<std::endl;
 
  if(m_rootInterface->getTotalFileNum()>0)
  {
    if(m_rootInterface->getTotalFileNum() != m_outputTagFile.size() )
    {
      std::cout<<"the input file num != output file number, please check your jobOptions"<<std::endl;
      exit(-1);
    }
  }
  
  m_entry = -1;
  m_entry_in_alg = -1;
  m_dstFile = "";
  m_fileNum=-1; 
  return StatusCode::SUCCESS;
}

◆ openOutputTagFile()

void TagWriterAlg::openOutputTagFile ( std::string file )

Definition at line 196 of file TagWriterAlg.cxx.

{
  m_oFile = new TFile(ofile.c_str(),"RECREATE","ROOT file for TAG");
  m_oTree = new TTree("Tag","Tags for DST file");
  m_oTree->Branch("entry",&m_entry, "entry/I");
  m_oTree->Branch("runNo",&m_runNo, "runNo/I");
  m_oTree->Branch("eventId",&m_eventId, "eventId/I");
  m_oTree->Branch("totalCharged",&m_totalCharged, "totalCharged/I");
  m_oTree->Branch("totalNeutral",&m_totalNeutral, "totalNeutral/I");
  m_oTree->Branch("totalTrks",&m_totalTrks, "totalTrks/I");
  m_oFileTree = new TTree("File", "DST file names");
  m_oFileTree->Branch("fileName", &m_dstFile);
 
}