/home/bes3soft/bes3soft/Boss/7.0.2/dist/7.0.2/Event/RootIO/RootIO-00-01-31/RootIO/digiRootWriterAlg.h

Go to the documentation of this file.

#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/AlgFactory.h"
#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/SmartDataPtr.h"
#include "GaudiKernel/Algorithm.h"

#include "EventModel/Event.h"
#include "EventModel/EventModel.h"
#include "RawEvent/DigiEvent.h"
#include "MDCRawEvent/MdcDigi.h"

#include "RootCnvSvc/Util.h"

#include "TROOT.h"
#include "TFile.h"
#include "TTree.h"
#include "TDirectory.h"

#include "DigiRootData/DigiEvent.h"

#include "RootCnvSvc/commonData.h"

#include "RootIO/IRootIoSvc.h"

class digiRootWriterAlg : public Algorithm
{       
public:
    
    digiRootWriterAlg(const std::string& name, ISvcLocator* pSvcLocator);
    
    StatusCode initialize();
   
    StatusCode execute();
    
    StatusCode finalize();

private:

    StatusCode writeDigiEvent();


    StatusCode writeMdcDigi();

    void writeEvent();
    void close();
   
    TFile *m_digiFile;
    TTree *m_digiTree;
    DigiEvent *m_digiEvt;
    std::string m_fileName;
    std::string m_treeName;
    int m_splitMode;
    int m_bufSize;
    int m_compressionLevel;
    int m_autoSaveEvents;
    
    commonData m_common;
    IRootIoSvc* m_rootIoSvc;

};

static const AlgFactory<digiRootWriterAlg>  Factory;
const IAlgFactory& digiRootWriterAlgFactory = Factory;

digiRootWriterAlg::digiRootWriterAlg(const std::string& name, 
                                 ISvcLocator* pSvcLocator) : 
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);

}

StatusCode digiRootWriterAlg::initialize()
{
    // 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::execute()
{
    // 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::writeDigiEvent() {
    // 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;
}

StatusCode digiRootWriterAlg::writeMdcDigi() {
    // 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;
}

void digiRootWriterAlg::writeEvent() 
{
    // 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;
}

void digiRootWriterAlg::close() 
{
    // 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::finalize()
{
    close();
    
    StatusCode sc = StatusCode::SUCCESS;
    return sc;
}

---