digiRootWriterAlg Class Reference

Writes Digi TDS data to a persistent ROOT file. on the digiRootWriterAlg of GLAST. More...

#include <digiRootWriterAlg.h>

List of all members.

Public Member Functions
	digiRootWriterAlg (const std::string &name, ISvcLocator *pSvcLocator)
StatusCode	initialize ()
	Handles setup by opening ROOT file in write mode and creating a new TTree.
StatusCode	execute ()
	Orchastrates reading from TDS and writing to ROOT for each event.
StatusCode	finalize ()
	Closes the ROOT file and cleans up.
Private Member Functions
StatusCode	writeDigiEvent ()
	Retrieves event Id and run Id from TDS and fills the McEvent ROOT object.
StatusCode	writeMdcDigi ()
void	writeEvent ()
	Calls TTree::Fill for each event and clears m_digiEvt.
void	close ()
	Performs the final write to the ROOT file and closes.
Private Attributes
TFile *	m_digiFile
	ROOT file pointer.
TTree *	m_digiTree
	ROOT tree pointer.
DigiEvent *	m_digiEvt
	Top-level Monte Carlo ROOT object.
std::string	m_fileName
	name of the output ROOT file
std::string	m_treeName
	name of the TTree in the ROOT file
int	m_splitMode
	ROOT split mode.
int	m_bufSize
	Buffer Size for the ROOT file.
int	m_compressionLevel
	Compression level for the ROOT file.
int	m_autoSaveEvents
	auto save every events
commonData	m_common
IRootIoSvc *	m_rootIoSvc

---

Detailed Description

Writes Digi TDS data to a persistent ROOT file. on the digiRootWriterAlg of GLAST.

Definition at line 31 of file digiRootWriterAlg.h.

---

Constructor & Destructor Documentation

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

Definition at line 88 of file digiRootWriterAlg.h.

References m_autoSaveEvents, m_bufSize, m_compressionLevel, m_fileName, m_splitMode, and m_treeName.

                                                           : 
Algorithm(name, pSvcLocator)
{
    // Input parameters available to be set via the jobOptions file
    declareProperty("digiRootFile",m_fileName="digi.root");
    declareProperty("splitMode", m_splitMode=1);
    declareProperty("bufferSize", m_bufSize=64000);
    declareProperty("compressionLevel", m_compressionLevel=1);
    //declareProperty("treeName", m_treeName="Digi");
    declareProperty("treeName", m_treeName="Rec"); //wensp modified on 20050515 for test
    declareProperty("autoSave", m_autoSaveEvents=1000);

}

---

Member Function Documentation

void digiRootWriterAlg::close

(

)

[private]

Performs the final write to the ROOT file and closes.

Definition at line 247 of file digiRootWriterAlg.h.

References m_digiTree.

Referenced by finalize().

{
    // Purpose and Method:  Writes the ROOT file at the end of the run.
    //    The TObject::kWriteDelete parameter is used in the Write method
    //    replacing TObject::kOverwrite - supposed to be safer
    //    since ROOT will periodically write to the ROOT file when the bufSize
    //    is filled.  Writing would create 2 copies of the same tree to be
    //    stored in the ROOT file, if we did not specify kOverwrite.

    TDirectory *saveDir = gDirectory;
    TFile *f = m_digiTree->GetCurrentFile();
    //m_digiFile->cd();
    f->cd();
    m_digiTree->BuildIndex("m_runId", "m_eventId");
    f->Write(0, TObject::kWriteDelete);
    f->Close();
    saveDir->cd();
    return;
}

StatusCode digiRootWriterAlg::execute

(

)

Orchastrates reading from TDS and writing to ROOT for each event.

Definition at line 143 of file digiRootWriterAlg.h.

References calibUtil::ERROR, m_digiEvt, m_digiFile, m_fileName, msgSvc(), writeDigiEvent(), writeEvent(), and writeMdcDigi().

{
    // Purpose and Method:  Called once per event.  This method calls
    //   the appropriate methods to read data from the TDS and write data
    //   to the ROOT file.

    MsgStream log(msgSvc(), name());

    StatusCode sc = StatusCode::SUCCESS;
    
    if (!m_digiFile->IsOpen()) {
        log << MSG::ERROR << "ROOT file " << m_fileName 
            << " could not be opened for writing." << endreq;
        return StatusCode::FAILURE;
    }
    
    m_digiEvt->Clear();

    sc = writeDigiEvent();
    if (sc.isFailure()) {
        log << MSG::ERROR << "Failed to write DigiEvent" << endreq;
        return sc;
    }


    sc = writeMdcDigi();
    if (sc.isFailure()) {
        log << MSG::ERROR << "Failed to write Tkr Digi Collection" << endreq;
        return sc;
    }
  
    writeEvent();
    return sc;
}

StatusCode digiRootWriterAlg::finalize

(

)

Closes the ROOT file and cleans up.

Definition at line 267 of file digiRootWriterAlg.h.

References close().

{
    close();
    
    StatusCode sc = StatusCode::SUCCESS;
    return sc;
}

StatusCode digiRootWriterAlg::initialize

(

)

Handles setup by opening ROOT file in write mode and creating a new TTree.

Definition at line 103 of file digiRootWriterAlg.h.

References calibUtil::ERROR, Bes_Common::INFO, m_bufSize, m_common, m_compressionLevel, m_digiEvt, m_digiFile, m_digiTree, m_fileName, m_rootIoSvc, m_splitMode, m_treeName, and msgSvc().

{
    // Purpose and Method:  Called once before the run begins.  This method
    //    opens a new ROOT file and prepares for writing.

    StatusCode sc = StatusCode::SUCCESS;
    MsgStream log(msgSvc(), name());
    
    // Use the Job options service to set the Algorithm's parameters
    // This will retrieve parameters set in the job options file
    setProperties();

    if ( service("RootIoSvc", m_rootIoSvc, true).isFailure() ){
        log << MSG::INFO << "Couldn't find the RootIoSvc!" << endreq;
        log << MSG::INFO << "No Auto Saving" << endreq;
        m_rootIoSvc = 0;
    } 

    facilities::Util::expandEnvVar(&m_fileName);

    // Save the current directory for the ntuple writer service
    TDirectory *saveDir = gDirectory;   
    // Create the new ROOT file
    m_digiFile = new TFile(m_fileName.c_str(), "RECREATE");
    if (!m_digiFile->IsOpen()) {
        log << MSG::ERROR << "ROOT file " << m_fileName 
            << " could not be opened for writing." << endreq;
        return StatusCode::FAILURE;
    }
    m_digiFile->cd();
    m_digiFile->SetCompressionLevel(m_compressionLevel);
    m_digiTree = new TTree(m_treeName.c_str(), "Bes Digitization Data");
    m_digiEvt = new DigiEvent();
    m_common.m_digiEvt = m_digiEvt;
    m_digiTree->Branch("DigiEvent","DigiEvent", &m_digiEvt, m_bufSize, m_splitMode);
    saveDir->cd();
    return sc;
    
}

StatusCode digiRootWriterAlg::writeDigiEvent

(

)

[private]

Retrieves event Id and run Id from TDS and fills the McEvent ROOT object.

Definition at line 179 of file digiRootWriterAlg.h.

References DigiEvent::initialize(), m_digiEvt, and msgSvc().

Referenced by execute().

                                             {
    // Purpose and Method:  Retrieve the Event object from the TDS and set the
    //    event and run numbers in the DigiEvent ROOT object

    MsgStream log(msgSvc(), name());
    StatusCode sc = StatusCode::SUCCESS;

    // Retrieve the Event data for this event
 SmartDataPtr<Event::EventHeader> eventHeader(eventSvc(),"/Event");
    if (!eventHeader) return sc;

    Short_t   runId = eventHeader->runNumber();
    Short_t  evtId  = eventHeader->eventNumber();
    Bool_t fromMc = true;


    m_digiEvt->initialize(evtId, runId, fromMc);
   // m_digiEvt->Print();
    
    return sc;
}

void digiRootWriterAlg::writeEvent

(

)

[private]

Calls TTree::Fill for each event and clears m_digiEvt.

Definition at line 229 of file digiRootWriterAlg.h.

References IRootIoSvc::getAutoSaveInterval(), m_digiTree, and m_rootIoSvc.

Referenced by execute().

{
    // Purpose and Method:  Stores the DigiEvent data for this event in the ROOT
    //    tree.  The m_digiEvt object is cleared for the next event.
    static int eventCounter = 0;
    TDirectory *saveDir = gDirectory;
    m_digiTree->GetCurrentFile()->cd();
    //m_digiFile->cd();
    m_digiTree->Fill();
    //m_digiEvt->Clear();
    saveDir->cd();
    ++eventCounter;
    if (m_rootIoSvc)
        if (eventCounter % m_rootIoSvc->getAutoSaveInterval() == 0) m_digiTree->AutoSave();

    return;
}

StatusCode digiRootWriterAlg::writeMdcDigi

(

)

[private]

Retrieves TKR digitization data from the TDS and fills the TkrDigi ROOT collection

Definition at line 201 of file digiRootWriterAlg.h.

References TRawData::initialize(), m_common, m_digiEvt, commonData::m_mdcDigiMap, EventModel::Digi::MdcDigiCol, msgSvc(), and TMdcDigi::setOverflow().

Referenced by execute().

                                           {
    // Purpose and Method:  Retrieve the TkrDigi collection from the TDS and set the
    //    TkrDigi ROOT collection

    MsgStream log(msgSvc(), name());
    StatusCode sc = StatusCode::SUCCESS;
    
    SmartDataPtr<MdcDigiCol> mdcDigiColTds(eventSvc(), EventModel::Digi::MdcDigiCol);
    if (!mdcDigiColTds) return sc;
    MdcDigiCol::const_iterator mdcDigiTds;

    for (mdcDigiTds = mdcDigiColTds->begin(); mdcDigiTds != mdcDigiColTds->end(); mdcDigiTds++) {
          UInt_t overflow  = (*mdcDigiTds)->getOverflow();
          UInt_t time      = (*mdcDigiTds)->getTimeChannel();
          UInt_t charge    = (*mdcDigiTds)->getChargeChannel();
          UInt_t id        = (*mdcDigiTds)->getIntId();
          TMdcDigi *mdcDigiRoot = new TMdcDigi();
          m_common.m_mdcDigiMap[(*mdcDigiTds)] = mdcDigiRoot;

         mdcDigiRoot->initialize(id, time ,charge);
         mdcDigiRoot->setOverflow(overflow);
         m_digiEvt->addMdcDigi(mdcDigiRoot);
//       mdcDigiRoot->Print();
    }

    return sc;
}

---

Member Data Documentation

int digiRootWriterAlg::m_autoSaveEvents [private]

auto save every events

Definition at line 78 of file digiRootWriterAlg.h.

Referenced by digiRootWriterAlg().

int digiRootWriterAlg::m_bufSize [private]

Buffer Size for the ROOT file.

Definition at line 74 of file digiRootWriterAlg.h.

Referenced by digiRootWriterAlg(), and initialize().

commonData digiRootWriterAlg::m_common [private]

Definition at line 80 of file digiRootWriterAlg.h.

Referenced by initialize(), and writeMdcDigi().

int digiRootWriterAlg::m_compressionLevel [private]

Compression level for the ROOT file.

Definition at line 76 of file digiRootWriterAlg.h.

Referenced by digiRootWriterAlg(), and initialize().

DigiEvent* digiRootWriterAlg::m_digiEvt [private]

Top-level Monte Carlo ROOT object.

Definition at line 66 of file digiRootWriterAlg.h.

Referenced by execute(), initialize(), writeDigiEvent(), and writeMdcDigi().

TFile* digiRootWriterAlg::m_digiFile [private]

ROOT file pointer.

Definition at line 62 of file digiRootWriterAlg.h.

Referenced by execute(), and initialize().

TTree* digiRootWriterAlg::m_digiTree [private]

ROOT tree pointer.

Definition at line 64 of file digiRootWriterAlg.h.

Referenced by close(), initialize(), and writeEvent().

std::string digiRootWriterAlg::m_fileName [private]

name of the output ROOT file

Definition at line 68 of file digiRootWriterAlg.h.

Referenced by digiRootWriterAlg(), execute(), and initialize().

IRootIoSvc* digiRootWriterAlg::m_rootIoSvc [private]

Definition at line 81 of file digiRootWriterAlg.h.

Referenced by initialize(), and writeEvent().

int digiRootWriterAlg::m_splitMode [private]

ROOT split mode.

Definition at line 72 of file digiRootWriterAlg.h.

Referenced by digiRootWriterAlg(), and initialize().

std::string digiRootWriterAlg::m_treeName [private]

name of the TTree in the ROOT file

Definition at line 70 of file digiRootWriterAlg.h.

Referenced by digiRootWriterAlg(), and initialize().

---