DataOut.h

Go to the documentation of this file.

#ifndef DataOut_H
#define DataOut_H

#include <string>
#include <iostream>
#include <map>

#include "WCSimRootEvent.hh"
#include "WCSimRootGeom.hh"
#include "WCSimRootOptions.hh"

#include "TFile.h"
#include "TTree.h"

#include "Tool.h"
#include "TimeDelta.h"
#include "Stopwatch.h"

class DataOut: public Tool {


 public:

  DataOut();
  bool Initialise(std::string configfile,DataModel &data);
  bool Execute();
  bool Finalise();


 private:
  void ExecuteSubDet(WCSimRootEvent * wcsim_event, std::vector<SubSample> & samples, WCSimRootEvent * original_wcsim_event = 0);
  void CreateSubEvents(WCSimRootEvent * wcsim_event);
  //Get the WCSim "date", used later to give the hits the correct absolute time.
  // Also add the trigger offset from the config file
  TimeDelta GetOffset(WCSimRootEvent * original_wcsim_event = 0);
  void FillHits(WCSimRootEvent * wcsim_event, std::vector<SubSample> & samples);
  void AddTruthInfo(WCSimRootEvent * wcsim_event, WCSimRootEvent * original_wcsim_event, const TimeDelta & time_shift);
  //Set some trigger header infromation that requires all the hits to be 
  // present to calculate e.g. sumq
  void FinaliseSubEvents(WCSimRootEvent * wcsim_event);

  std::string m_output_filename;
  TFile * m_output_file;
  TTree * m_event_tree;
  WCSimRootEvent * m_id_wcsimevent_triggered;
  WCSimRootEvent * m_od_wcsimevent_triggered;

  TriggerInfo * m_all_triggers;
  TimeDelta m_trigger_offset;

  int m_event_num;

  bool m_save_only_failed_hits;
  bool m_save_multiple_hits_per_trigger;

  util::Stopwatch * m_stopwatch;
  std::string m_stopwatch_file;

  int m_verbose;

  std::stringstream m_ss;

  void StreamToLog(int level) {
    Log(m_ss.str(), level, m_verbose);
    m_ss.str("");
  }

  enum LogLevel {FATAL=-1, ERROR=0, WARN=1, INFO=2, DEBUG1=3, DEBUG2=4, DEBUG3=5};
};


#endif