Model.hh

/* 
Beam Delivery Simulation (BDSIM) Copyright (C) Royal Holloway, 
University of London 2001 - 2023.
 
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 MODEL_H
#define MODEL_H
 
#include "Rtypes.h" // for classdef
 
#include "RebdsimTypes.hh"
 
class BDSOutputROOTEventModel;
class TTree;
 
class Model
{
public:
  Model();
  Model(bool debugIn,
        int  dataVersionIn = 0);
  virtual ~Model();
 
  std::vector<std::string> SamplerNames() const;
  std::vector<std::string> SamplerCNames() const;
  std::vector<std::string> SamplerSNames() const;
 
  std::vector<std::string> CollimatorNames() const;
 
  int DataVersion() const {return dataVersion;}
  
  inline void SetDataVersion(int dataVersionIn) {dataVersion = dataVersionIn;}
 
  void SetBranchAddress(TTree* t,
                        bool                      allBranchesOn    = true,
                        const RBDS::VectorString* branchesToTurnOn = nullptr);
 
  BDSOutputROOTEventModel* model;
 
private:
  bool debug;
  int  dataVersion;
  
  ClassDef(Model, 2);
};
 
#endif