BDSPSPopulationScaled.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 BDSPSPOPULATIONSCALED_H
#define BDSPSPOPULATIONSCALED_H
#include "G4String.hh"
#include "G4THitsMap.hh"
#include "G4TrackLogger.hh"
#include "G4Types.hh"
#include "G4VPrimitiveScorer.hh"
 
#include <map>
#include <vector>
 
class G4PhysicsVector;
 
class BDSPSPopulationScaled: public G4VPrimitiveScorer
{
public:
  explicit BDSPSPopulationScaled(const G4String& name, G4int depth=0);
  
  BDSPSPopulationScaled(const G4String& scorerName,
                        const G4String& pathname,
                        G4int           depth=0);
 
  virtual ~BDSPSPopulationScaled() override;
 
public:
  void Initialize(G4HCofThisEvent* HCE) override;
  void EndOfEvent(G4HCofThisEvent* HCE) override;
  void clear() override;
 
  G4bool ProcessHits(G4Step* aStep, G4TouchableHistory*) override;
 
  virtual G4double GetConversionFactor(G4int    particleID,
                                       G4double kineticEnergy,
                                       G4double angle) const;
 
  static std::vector<G4String> LoadDirectoryContents(const G4String& dirname);
  G4int NearestNeighbourAngleIndex(const std::vector<G4double>& vec, G4double value) const;
  G4int NearestNeighbourIonPID(const std::vector<G4int>& vec, G4int value) const;
  G4int GetZFromParticleID(G4int particleID) const;
  void PrintAll() override;
  
private:
  G4int HCID;
  G4THitsMap<G4double>* EvtMap;
  
  std::map<G4int, G4TrackLogger>  fCellTrackLogger;
  std::map< G4int, std::map<G4int, G4PhysicsVector*> > conversionFactors;
  std::map< G4int, std::vector<G4int> > ionParticleIDs;
  std::vector<G4double> angles;
};
 
#endif