BDSOutputROOTEventCollimator.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 BDSOUTPUTROOTEVENTCOLLIMATOR_H
#define BDSOUTPUTROOTEVENTCOLLIMATOR_H
 
#ifndef __ROOTBUILD__
#include "BDSOutputROOTEventCollimatorInfo.hh"
#include "G4Types.hh"
class BDSHitCollimator;
#endif
 
#include "TObject.h"
 
#include <set>
#include <utility>
#include <vector>
 
class BDSOutputROOTParticleData;
 
class BDSOutputROOTEventCollimator: public TObject
{
public:
  BDSOutputROOTEventCollimator();  
 
  bool               primaryInteracted;
  bool               primaryStopped;
  int                n;
  std::vector<float> energy; 
  std::vector<float> energyDeposited;
  std::vector<float> xIn;
  std::vector<float> yIn;
  std::vector<float> zIn;
  std::vector<float> xpIn;
  std::vector<float> ypIn;
  std::vector<float> zpIn;
  std::vector<float> T;
  std::vector<float> weight;
  std::vector<int>   partID;
  std::vector<int>   parentID;
  std::vector<int>   turn;
  std::vector<bool>  firstPrimaryHitThisTurn;
 
  std::vector<float> impactParameterX;
  std::vector<float> impactParameterY;
 
  std::vector<bool>  isIon;
  std::vector<int>   ionA;
  std::vector<int>   ionZ;
 
  std::set<int>      turnSet;
 
  std::vector<int>   charge;
  std::vector<float> kineticEnergy;
  std::vector<float> mass;
  std::vector<float> rigidity;
 
  double totalEnergyDeposited; 
  
  virtual ~BDSOutputROOTEventCollimator();
#ifndef __ROOTBUILD__
  void Fill(const BDSHitCollimator*                 hit,
            const BDSOutputROOTEventCollimatorInfo& info,
            const std::pair<G4double, G4double>&    differences,
            G4bool                                  storeHits);
  void FillExtras(G4bool fillIonInfo,
                  G4bool fillLinks);
 
  inline void SetPrimaryStopped(G4bool primaryStoppedIn) {primaryStopped = primaryStoppedIn;}
  
#endif
  void Fill(const BDSOutputROOTEventCollimator* other);
 
  virtual void Flush();
 
  static BDSOutputROOTParticleData* particleTable;
 
  ClassDef(BDSOutputROOTEventCollimator, 1);
};
 
 
#endif