EventAnalysis.hh

/* 
Beam Delivery Simulation (BDSIM) Copyright (C) Royal Holloway, 
University of London 2001 - 2022.
 
This file is part of BDSIM.
 
BDSIM is free software: you can redistribute it and/or modify 
it under the terms of the GNU General Public License as published 
by the Free Software Foundation version 3 of the License.
 
BDSIM is distributed in the hope that it will be useful, but 
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
 
You should have received a copy of the GNU General Public License
along with BDSIM.  If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef EVENTANALYSIS_H
#define EVENTANALYSIS_H
 
#include <vector>
 
#include "Analysis.hh"
 
#include "Rtypes.h" // for classdef
 
class Event;
class HistogramDefSet;
class PerEntryHistogramSet;
class SamplerAnalysis;
class TChain;
class TFile;
 
class EventAnalysis: public Analysis
{
public:
  EventAnalysis();
  
  EventAnalysis(Event*   eventIn,
                TChain*  chain,
                bool     perEntryAnalysis    = true,
                bool     processSamplersIn   = true,
                bool     debugIn             = false,
                double   printModuloFraction = 0.01,
                bool     emittanceOnTheFlyIn = false,
                long int eventStartIn        = 0,
                long int eventEndIn          = -1,
                const std::string& primaryParticleName = "");
 
  virtual ~EventAnalysis() noexcept;
 
  virtual void Execute();
 
  virtual void Process();
 
  virtual void SimpleHistograms();
 
  virtual void Terminate();
 
  virtual void Write(TFile* outputFileName);
 
protected:
  Event* event; 
  std::vector<SamplerAnalysis*> samplerAnalyses; 
  std::vector<std::vector<std::vector<double> > > opticalFunctions; 
 
  void PreparePerEntryHistogramSets();
  void AccumulatePerEntryHistogramSets(long int entryNumber);
  void TerminatePerEntryHistogramSets();
  
  void CheckSpectraBranches();
 
  void FillHistogram(HistogramDefSet* definition);
 
private:
  void SetPrintModuloFraction(double fraction);
 
  void Initialise();
 
  void ProcessSamplers(bool firstTime = false);
 
  int  printModulo;       
  bool processSamplers;   
  bool emittanceOnTheFly; 
  long int eventStart;    
  long int eventEnd;      
 
  std::vector<PerEntryHistogramSet*> perEntryHistogramSets;
 
  std::map<HistogramDefSet*, std::vector<TH1*> > simpleSetHistogramOutputs;
  
  ClassDef(EventAnalysis,1);
};
 
#endif