BDSSDCollimator.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 BDSSDCOLLIMATOR_H
#define BDSSDCOLLIMATOR_H
 
#include "BDSHitCollimator.hh"
#include "BDSSensitiveDetector.hh"
 
#include "globals.hh" // geant4 types / globals
#include "G4String.hh"
 
#include <map>
#include <vector>
 
class BDSAuxiliaryNavigator;
class BDSHitEnergyDeposition;
 
class G4Step;
class G4HCofThisEvent;
class G4TouchableHistory;
class G4VHit;
 
class BDSSDCollimator: public BDSSensitiveDetector
{ 
public:
  explicit BDSSDCollimator(G4String name);
 
  BDSSDCollimator& operator=(const BDSSDCollimator&) = delete;
  BDSSDCollimator(BDSSDCollimator&) = delete;
 
  virtual ~BDSSDCollimator();
 
  virtual void Initialize(G4HCofThisEvent* HCE) override;
 
  virtual G4bool ProcessHits(G4Step* aStep,
                             G4TouchableHistory* rOHist) override;
 
  virtual G4bool ProcessHitsOrdered(G4Step* step,
                                    G4TouchableHistory*  rOHist,
                                    const std::vector<G4VHit*>& hits) override;
 
  virtual G4VHit* last() const override;
 
private:
  inline bool IsIon(const int& pdgID) const {return pdgID > 1000000000;}
  
  BDSHitsCollectionCollimator* collimatorCollection;
 
  G4String itsCollectionName;
 
  G4int itsHCID;
 
  BDSAuxiliaryNavigator* auxNavigator;
 
  std::map<BDSBeamline*, std::map<G4int, G4int> > mapping;
};
 
#endif