source: Idarraga/allpix/include/AllPixRun.hh @ 226

/**
 *  Author John Idarraga <idarraga@cern.ch>
 */
#ifndef AllPixRun_h
#define AllPixRun_h 1

#include "G4Run.hh"

#include "AllPixTrackerHit.hh"
#include "AllPixRunAction.hh"
#include "AllPixMedipix2Digit.hh"
#include "AllPixDigitInterface.hh"

#include <TROOT.h>
#include <TFile.h>
#include <TBranch.h>
#include <TTree.h>
#include <TChain.h>
#include <TString.h>

#include <vector>
#include <string>
#include <map>

#include <iostream>
#include <fstream>

using namespace std;

class FramesHandler;
class WriteToNtuple;
class SimpleHits;

#ifdef _EUTELESCOPE
#define __magic_trigger_cntr_ack 4 // 4 scintillators with 4 primary particle hits
#endif

class AllPixRun : public G4Run {

public:

  AllPixRun(TString, TString);
  virtual ~AllPixRun();

  virtual void RecordEvent(const G4Event*);
  void RecordHits(const G4Event*);
  void RecordDigits(const G4Event*);
  void SetLCIOBridgeFileDsc(ofstream * f, ofstream * fdut)
  { m_lciobridge_f = f; m_lciobridge_dut_f = fdut;};

  // filling frames
  void FillFramesNtuple(const G4Run *);

private:

  AllPixTrackerHitsCollection * m_hitsCollection;

  // Hits
  SimpleHits ** m_storableHits;

  // map index in frames handler to det Id
  map<int, int> m_detIdToIndex;

  // Frames ntuple  --> not storing whole Digits
  //  building frames from digits
  FramesHandler ** m_frames;
  TString m_datasetDigits;
  TString m_datasetHits;
  TString m_tempdir;

  // Information about ntuples

  // Same as number of digitizers
  G4int m_nOfDetectors;

  // Number of sensitive detectors.  Can be more than the number of
  // digitizers if the user decides to have other volumes as SDs
  G4int m_nOfSD;

  // coming from AllPixRunAction
  ofstream * m_lciobridge_f;
  ofstream * m_lciobridge_dut_f;

};

#endif