EventWriterTool.h

Go to the documentation of this file.

#pragma once
 
#include <Gaudi/Property.h>
#include <GaudiKernel/AlgTool.h>
#include <GaudiKernel/DataStoreItem.h>
#include <GaudiKernel/IDataProviderSvc.h>
#include <TFile.h>
#include <TTree.h>
#include <string>
#include <vector>
 
#include "EventWriter/IEventWriterTool.h"
#include "RootCnvSvc/IRootCnvSvc.h"
#include "RootCnvSvc/commonData.h"
#include "RootEventData/TJobInfo.h"
 
class EventWriterTool : public extends<AlgTool, IEventWriterTool> {
public:
  using extends::extends;
 
  StatusCode write() override;
 
  /// Overriding initialize / finalize
  StatusCode initialize() override;
  StatusCode finalize() override;
 
private:
  void           addItem( std::vector<DataStoreItem*>& items, const std::string& descriptor );
  DataStoreItem* findItem( const std::string& path );
  std::string    getDecayOptions();
 
private:
  int         m_st{ 0 };
  commonData  m_commonData;
 
  std::vector<DataStoreItem*> m_itemLists{};
 
  TFile*    m_outputFile{ nullptr };
  TTree*    m_outputTree{ nullptr };
  TTree*    m_jobInfoTree{ nullptr };
  TJobInfo* m_jobInfo{ nullptr };
 
  /// Services
  IRootCnvSvc*      m_cnvSvc{ nullptr };
  IDataProviderSvc* m_pDataProvider{ nullptr };
 
  /// Properties
  IntegerProperty     m_mode{ this, "RunMode", 2 };
  StringProperty      m_dofileName{ this, "digiRootOutputFile", "event.rec" };
  StringArrayProperty m_itemNames{ this,
                                   "ItemList",
                                   {
                                       "/Event",
                                       "/Event/EventHeader",
                                       "/Event/Navigator",
                                       "/Event/Digi",
                                       "/Event/Digi/MdcDigiCol",
                                       "/Event/Digi/TofDigiCol",
                                       "/Event/Digi/EmcDigiCol",
                                       "/Event/Digi/MucDigiCol",
                                       "/Event/Hlt",
                                       "/Event/Hlt/HltRawCol",
                                       "/Event/Hlt/HltInf",
                                       "/Event/Recon",
                                       "/Event/Recon/RecMdcTrackCol",
                                       "/Event/Recon/RecMdcDedxCol",
                                       "/Event/Recon/RecMdcKalTrackCol",
                                       "/Event/Recon/RecEsTimeCol",
                                       "/Event/Recon/RecExtTrackCol",
                                       "/Event/Recon/RecTofTrackCol",
                                       "/Event/Recon/RecEmcHitCol",
                                       "/Event/Recon/RecEmcClusterCol",
                                       "/Event/Recon/RecEmcShowerCol",
                                       "/Event/Recon/RecMucTrackCol",
                                       "/Event/Recon/EvtRecEvent",
                                       "/Event/Recon/EvtRecTrackCol",
                                   } };
};