RMGGeneratorMUSUNCosmicMuons.hh

// Copyright (C) 2024 Moritz Neuberger <https://orcid.org/0009-0001-8471-9076>
//
// This program is free software: you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License as published by the Free
// Software Foundation, either version 3 of the License, or (at your option) any
// later version.
//
// This program 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 Lesser General Public License for more
// details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
 
#ifndef _RMG_GENERATOR_MUSUN_COSMIC_MUONS_HH_
#define _RMG_GENERATOR_MUSUN_COSMIC_MUONS_HH_
 
#include <filesystem>
 
#include "CLHEP/Units/SystemOfUnits.h"
#include "G4GenericMessenger.hh"
#include "G4ParticleGun.hh"
 
#include "RMGAnalysisReader.hh"
#include "RMGVGenerator.hh"
#include "RMGVVertexGenerator.hh"
 
namespace u = CLHEP;

struct RMGGeneratorMUSUNCosmicMuons_Data {
    G4int fID;
    G4int fType;
    G4double fEkin;
    G4double fX;
    G4double fY;
    G4double fZ;
    G4double fTheta;
    G4double fPhi;
    G4double fPx;
    G4double fPy;
    G4double fPz;
};
 
 
class G4Event;
class RMGGeneratorMUSUNCosmicMuons : public RMGVGenerator {
 
  public:
 
    RMGGeneratorMUSUNCosmicMuons();
    ~RMGGeneratorMUSUNCosmicMuons() = default;
 
    RMGGeneratorMUSUNCosmicMuons(RMGGeneratorMUSUNCosmicMuons const&) = delete;
    RMGGeneratorMUSUNCosmicMuons& operator=(RMGGeneratorMUSUNCosmicMuons const&) = delete;
    RMGGeneratorMUSUNCosmicMuons(RMGGeneratorMUSUNCosmicMuons&&) = delete;
    RMGGeneratorMUSUNCosmicMuons& operator=(RMGGeneratorMUSUNCosmicMuons&&) = delete;

    void GeneratePrimaries(G4Event*) override;
    void SetParticlePosition(G4ThreeVector) override{};

    void BeginOfRunAction(const G4Run*) override;
    void EndOfRunAction(const G4Run*) override;
 
  private:
 
    void DefineCommands();
    void SetMUSUNFile(G4String pathToFile);

    bool PrepareCopy(std::string pathToFile);
 
    std::unique_ptr<G4ParticleGun> fGun = nullptr;
    std::unique_ptr<G4GenericMessenger> fMessenger = nullptr;
    G4String fPathToFile = "";
    std::filesystem::path fPathToTmpFolder;
    std::filesystem::path fPathToTmpFile;
 
    static RMGAnalysisReader* fAnalysisReader;
    static bool fHasCartesianMomentum;
 
    static RMGGeneratorMUSUNCosmicMuons_Data* fInputData;
};
 
#endif
 
// vim: tabstop=2 shiftwidth=2 expandtab