BDSOutputROOTEventLoss.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 BDSOUTPUTROOTEVENTLOSS_H
#define BDSOUTPUTROOTEVENTLOSS_H
 
#ifndef __ROOTBUILD__
class BDSHitEnergyDeposition;
class BDSTrajectoryPoint;
class BDSTrajectoryPointHit;
#endif
 
#include "TObject.h"
 
#include <vector>
 
class BDSOutputROOTEventLoss: public TObject
{
public:
  BDSOutputROOTEventLoss();
  virtual ~BDSOutputROOTEventLoss();
  
  int                       n;       
  std::vector<float>        energy;  
  std::vector<float>        S;       
  std::vector<float>        weight;  
 
  std::vector<int>          partID;  
  std::vector<int>          trackID; 
  std::vector<int>          parentID;
  
  std::vector<int>          modelID; 
  std::vector<int>          turn;    
 
  std::vector<float>        x;
  std::vector<float>        y;
  std::vector<float>        z;
  
  std::vector<float>        X;
  std::vector<float>        Y;
  std::vector<float>        Z;
 
  std::vector<float>        T; 
 
  std::vector<float>        stepLength; 
  std::vector<float>        preStepKineticEnergy; 
 
  std::vector<int>          postStepProcessType;
  std::vector<int>          postStepProcessSubType;
  
  void Fill(const BDSOutputROOTEventLoss* other);
  virtual void Flush();
  
#ifndef __ROOTBUILD__
  BDSOutputROOTEventLoss(bool storeTurnIn,
                         bool storeLinksIn,
                         bool storeModleIDIn,
                         bool storeLocalIn,
                         bool storeGobalIn,
                         bool storeTimeIn,
                         bool storeStepLengthIn,
                         bool storePreStepKineticEnergyIn,
                         bool storePhysicsProcessesIn);
  void Fill(const BDSTrajectoryPointHit* hit);
  void Fill(const BDSHitEnergyDeposition* hit);
 
  bool storeTurn       = false; 
  bool storeLinks      = false; 
  bool storeModelID    = false; 
  bool storeLocal      = false; 
  bool storeGlobal     = false; 
  bool storeTime       = false; 
  bool storeStepLength = false; 
  bool storePreStepKineticEnergy = false; 
  bool storePhysicsProcesses = false;
#endif
 
  ClassDef(BDSOutputROOTEventLoss,5);
};
 
#endif