Model.cc

/* 
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/>.
*/
#include "Model.hh"
#include "RebdsimTypes.hh"
 
#include "BDSOutputROOTEventModel.hh"
 
#include "TTree.h"
 
#include <iostream>
#include <string>
#include <vector>
 
ClassImp(Model)
 
Model::Model():
  Model(false)
{;}
 
Model::Model(bool debugIn,
             int  dataVersionIn):
  model(nullptr),
  debug(debugIn),
  dataVersion(dataVersionIn)
{
  model = new BDSOutputROOTEventModel();
}
 
Model::~Model()
{
  delete model;
}
 
void Model::SetBranchAddress(TTree* t,
                             bool                      allBranchesOn,
                             const RBDS::VectorString* branchesToTurnOn)
{
  // turn off all branches by default.
  t->SetBranchStatus("*", false);
 
  if (allBranchesOn)
    {t->SetBranchStatus("*", true);}
  else if (branchesToTurnOn)
    {
      for (const auto& name : *branchesToTurnOn)
        {
          std::string nameStar = name + ".*"; // necessary because of the splitting
          if (debug)
            {std::cout << "Turning on branch \"" << nameStar << "\"" << std::endl;}
          t->SetBranchStatus(nameStar.c_str(), true);
        }
    }
  
  t->SetBranchAddress("Model.", &model);
}
 
std::vector<std::string> Model::SamplerNames() const
{
  if (model)
    {return model->samplerNamesUnique;}
  else
    {return std::vector<std::string>();}
}
 
std::vector<std::string> Model::SamplerCNames() const
{
  return model ? model->samplerCNamesUnique : std::vector<std::string>();
}
 
std::vector<std::string> Model::SamplerSNames() const
{
  return model ? model->samplerSNamesUnique : std::vector<std::string>();
}
 
std::vector<std::string> Model::CollimatorNames() const
{
  if (dataVersion > 3)
    {return model ? model->collimatorBranchNamesUnique : std::vector<std::string>();}
  else
    {return std::vector<std::string>();}
}