#include <RootFileWriter.h>

Public Types
typedef const std::string	WriterArgType

Public Member Functions
	RootFileWriter (WriterArgType &fname)

virtual	~RootFileWriter ()

int	writeEvent (void *pevt, int size)

int	close ()

int	stat ()

Detailed Description

Definition at line 11 of file RootFileWriter.h.

Member Typedef Documentation

◆ WriterArgType

typedef const std::string RootFileWriter::WriterArgType

Definition at line 15 of file RootFileWriter.h.

Constructor & Destructor Documentation

◆ RootFileWriter()

RootFileWriter::RootFileWriter ( WriterArgType & fname )

Definition at line 10 of file RootFileWriter.cxx.

   : m_fname( fname),
     m_firstRound( true ),
     m_stat(0)
{
   m_file = new TFile(fname.c_str(), "RECREATE");
   m_tree = new TTree("Event","Event");
   m_jtree = new TTree("JobInfoTree", "Job info");
   m_evt  = new TBossFullEvent;
   m_jobInfo = new TJobInfo;
}

Referenced by RootFileWriter().

◆ ~RootFileWriter()

RootFileWriter::~RootFileWriter ( )

virtual

Definition at line 22 of file RootFileWriter.cxx.

{
   if ( m_file->IsOpen() ) {
      this->close();
   }
 
   delete m_file;
   //delete m_tree;
   //delete m_jtree;
   delete m_evt;
   delete m_jobInfo;
}

Member Function Documentation

◆ close()

int RootFileWriter::close ( )

Definition at line 92 of file RootFileWriter.cxx.

{
   m_jobInfo->setBossVer( getenv("BES_RELEASE") );
   m_jobInfo->addJobOptions( this->getJobOptions() );
   m_jobInfo->setDecayOptions( this->getDecayOptions() );
   m_jtree->Fill();
 
   m_file->Write();
   m_file->Close();
 
   m_stat = 1;
 
   return 0;
}

Referenced by main(), and ~RootFileWriter().

◆ stat()

int RootFileWriter::stat ( )

Definition at line 107 of file RootFileWriter.cxx.

{
   return m_stat;
}

◆ writeEvent()

int RootFileWriter::writeEvent	(	void *	pevt,
		int	size )

Definition at line 35 of file RootFileWriter.cxx.

{
   if ( size == 4 ) {
      m_stat = *((int*)pevt);
      if ( m_stat == DistBossCode::StatusFinalize ) {
     throw ReachEndOfFileList();
      }
      else {
     throw RawExMessage("[RootFileWriter] Get an invalid DistBossCode!");
      }
   }
 
   TBufferFile buffer(TBuffer::kRead, size, ((char*)pevt+4), kFALSE);
   TBossFullEvent * evt = (TBossFullEvent*)buffer.ReadObject(TBossFullEvent::Class());
 
   m_evt->copy(evt);
 
   /*
   static int ii = 0;
   std::cout << "Total: " << ++ii << "   Run: " << m_evt->m_EvtHeader->getRunId()
             << "  Evt: " << m_evt->m_EvtHeader->getEventId() << std::endl;
         */
 
   if ( m_firstRound ) {
      if (m_evt->m_EvtHeader) m_tree->Branch("TEvtHeader","TEvtHeader",&m_evt->m_EvtHeader,3200000,1);
      if (m_evt->m_digiEvt) m_tree->Branch("TDigiEvent","TDigiEvent",&m_evt->m_digiEvt,3200000,1);
      if (m_evt->m_dstEvt) m_tree->Branch("TDstEvent", "TDstEvent", &m_evt->m_dstEvt, 3200000, 1);
      if (m_evt->m_mcEvt) m_tree->Branch("TMcEvent","TMcEvent",&m_evt->m_mcEvt,3200000,1);
      if (m_evt->m_trigEvt) m_tree->Branch("TTrigEvent","TTrigEvent",&m_evt->m_trigEvt,3200000,1);
      if (m_evt->m_hltEvt) m_tree->Branch("THltEvent","THltEvent",&m_evt->m_hltEvt,3200000,1);
      if (m_evt->m_rectrackEvt) m_tree->Branch("TRecEvent", "TRecTrackEvent", &m_evt->m_rectrackEvt, 3200000, 1);
      if (m_evt->m_evtRecObject) m_tree->Branch("TEvtRecObject", "TEvtRecObject", &m_evt->m_evtRecObject, 3200000, 1);
 
      m_jtree->Branch("JobInfo",&m_jobInfo);
 
      m_firstRound = false;
   }
 
   if ( m_file->IsZombie() || (!m_file->IsOpen()) ) {
      std::cout<<"ROOT File: " << m_fname << " bad status in RootFileWriter" << std::endl;
      exit(1);
   }
 
   int nb = m_tree->Fill();
   if ( nb == -1 ) {
      std::cerr << "RootFileWriter: error in fill tree " << m_tree->GetName() << std::endl;
      exit(1);
   }
 
   delete evt;
 
   m_evt->reset();
   m_file = m_tree->GetCurrentFile();
 
   return nb;
}

Referenced by main().

The documentation for this class was generated from the following files:

Public Types

Public Member Functions