rootana/src/framework/EventNavigator.h

Go to the documentation of this file.

#ifndef EVENTNAVIGATOR_H
#define EVENTNAVIGATOR_H


//C++/STL
#include <stdexcept>
#undef DEFER
//ROOT
class TFile;
class TTree;

//Local
#include "definitions.h"
#include "TGlobalData.h"
#include "BankIter.h"
#include "SetupRecord.h"
#include "LoopSequence.h"


class io_error : public std::runtime_error {
public:
  io_error() 
    : std::runtime_error("I/O failed")  
  {}
};


class ARGUMENTS;



class EventNavigator {
  typedef IDs::channel ChannelID;
  typedef IDs::generator GeneratorID;
  typedef IDs::source SourceID;
  typedef IDs::Generator_t GeneratorStr;
  typedef IDs::Config_t Config;


 public:
  enum OutputMode {kNew = 0, kOverwrite }; //, kAppend};

  //The initial 
  static EventNavigator& InitInstance(const ARGUMENTS& command_line);

  static EventNavigator& Instance();

  ~EventNavigator(){}
  
  const TSetupData* const GetSetupData() {return fSetupData;}

  static const SetupRecord& GetSetupRecord() {return *Instance().fSetupRecord;}

  const LoopSequence& GetLoopSequence() const;

  const char* GetOutputFileName()const;

  const LoopSequence& MakeLoopSequence(const ARGUMENTS& args);

  Bool_t ConnectInput(const char* input_name);

  
  Bool_t ConnectOutput(const char* output_name, OutputMode mode = kOverwrite);

  Bool_t ConnectOutputFile(const char* output_file_name, Bool_t append =false);

  Long64_t EntryNo() const {return fEntryNo;}

  Long64_t GetInputNEntries() const {return fRawTree->GetEntriesFast();}

  Long64_t GetStartEntry() const;
  
  Long64_t GetStopEntry() const;

  Long64_t GetLoopNEntries() const
  { return GetStopEntry() - GetStartEntry(); }
  

  inline Int_t NextEntry() {
    return LoadEntry(fEntryNo + 1);
  }

  inline Int_t GetEntry(Long64_t entry){
    return LoadEntry(entry);
  }

  
  Int_t WriteCurrentEntry();

  void SetSavePulseIslands(bool save = 1){fCopyRaw =save;};

  void Adopt(PulseIslandList& pil, const SourceID& sid);
  void Adopt(AnalysedPulseList& apl, const SourceID& sid);
  void Adopt(DetectorPulseList& dpl, const SourceID& sid);
  //void Adopt(MuonEventList& mel, const SourceID& sid);
  void Adopt(PulseIslandList& pil, const ChannelID& cid);
  

  void AdoptTemporary(PulseIslandList& pil, const SourceID& sid);
  void AdoptTemporary(AnalysedPulseList& apl, const SourceID& sid);
  void AdoptTemporary(DetectorPulseList& dpl, const SourceID& sid); 
  //void AdoptTemporary(MuonEventList& mel, const SourceID& sid);
  void AdoptTemporary(PulseIslandList& pil, const ChannelID& cid); 

#ifdef DEFER
  const PulseIslandList& FindIslandBank(const SourceID& sid) const;
  const PulseIslandList& FindAnalysedBank(const SourceID& sid) const;
  const PulseIslandList& FindDetectorBank(const SourceID& sid) const;
#endif 

  //TODO 
  // What shortcuts alowing strings/enums are sensible & useful? 
  // If any?
  /*
  const PulseIslandList& FindIslandBank(ChannelID& cid,
                                        GeneratorID& gid = "*");
  const PulseIslandList& FindIslandBank(const ChannelID& cid,
                                        const GeneratorID& gid);
  const AnalysedPulseList& FindAnalysedBank(ChannelID cid, 
                                            GeneratorStr gid = "*",
                                            Config cfg ="*") const;
  const PulseIslandList& FindIslandBank(const ChannelID& cid,
                                        const GeneratorID& gid);
  const DetectorPulseList& FindDetectorBank(ChannelID cid,
                                            GeneratorStr gid = "*",
                                            Config cfg ="*" ) const;  
  //const MuonEventList& GetMuonEventList(GeneratorID gid = "",
  //                                      int count = 0) const;
  */

  PulseIslandBankIter MatchIslandBanks(const SourceID& sid) const;
  PulseIslandBankIter MatchAnalysedBanks(const SourceID& sid) const;
  PulseIslandBankIter MatchDetectorBanks(const SourceID& sid) const;

  PulseIslandBankIter MatchIslandBanks(const ChannelID& cid,
                                       GeneratorStr gid ="") const;

  AnalysedPulseBankIter MatchAnalysedBanks(const ChannelID& cid,
                                           GeneratorStr gid ="") const;
  
  AnalysedPulseBankIter MatchDetectorBanks(const ChannelID& cid,
                                           GeneratorStr gid ="") const;

  void Close();

  TGlobalData* GetRawData()const {return fRawData;}
  TTree* GetRawTree()const {return fRawTree;}
  void DumpRawTree()const;

protected:
  TGlobalData* ConnectRawData(TFile* raw_file);

  Bool_t VerifyRawData(TTree* raw_tree);

  TSetupData* ConnectSetupData(TFile* raw_file);

  Bool_t VerifySetupData(TTree* setup_tree);


  Bool_t MirrorRawInputFormat();

  Int_t LoadEntry(Long64_t entry);


private: 
  EventNavigator();
  EventNavigator(const ARGUMENTS& command_line);
  EventNavigator(const EventNavigator& src){};
  EventNavigator& operator=(const EventNavigator& rhs){return *this;}

  static EventNavigator* fInstance; 

  Bool_t fCopyRaw;

  TFile* fRawInput;

  TFile* fOutput;

  Int_t fEntryNo;

  TTree* fRawTree;

  TTree* fSetupTree;

  TTree* fEventTree;

  TTree* fOutputTreeTPI;

  TGlobalData* fRawData;
  
  TSetupData* fSetupData;

  SetupRecord* fSetupRecord;

  LoopSequence* fLoopSequence;


};


//inline Int_t EventNavigator::NextEntry() {return LoadEntry(fEntryNo + 1);}

#endif //EVENTNAVIGATOR_H

// emacs metadata
// Local Variables:
// mode: c++
// End:

---