RMGGermaniumOutputScheme.hh

// Copyright (C) 2022 Luigi Pertoldi <https://orcid.org/0000-0002-0467-2571>
//
// 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_GERMANIUM_OUTPUT_SCHEME_HH_
#define _RMG_GERMANIUM_OUTPUT_SCHEME_HH_
 
#include <memory>
#include <optional>
#include <set>
 
#include "G4AnalysisManager.hh"
#include "G4GenericMessenger.hh"
 
#include "RMGDetectorHit.hh"
#include "RMGGermaniumDetector.hh"
#include "RMGOutputTools.hh"
#include "RMGVOutputScheme.hh"
 
class G4Event;
class RMGGermaniumOutputScheme : public RMGVOutputScheme {
 
  public:
 
    RMGGermaniumOutputScheme();

    void AssignOutputNames(G4AnalysisManager* ana_man) override;

    void StoreEvent(const G4Event* event) override;

    bool ShouldDiscardEvent(const G4Event* event) override;

    std::optional<bool> StackingActionNewStage(int) override;

    std::optional<G4ClassificationOfNewTrack> StackingActionClassify(const G4Track*, int) override;

    void SetEdepCutLow(double threshold) { fEdepCutLow = threshold; }

    void SetEdepCutHigh(double threshold) { fEdepCutHigh = threshold; }

    void AddEdepCutDetector(int det_uid) { fEdepCutDetectors.insert(det_uid); }

    void SetPositionMode(RMGOutputTools::PositionMode mode) { fPositionMode = mode; }

    void SetClusterDistance(double threshold) { fPreClusterPars.cluster_distance = threshold; }

    void SetClusterDistanceSurface(double threshold) {
      fPreClusterPars.cluster_distance_surface = threshold;
    }

    void SetSurfaceThickness(double thickness) { fPreClusterPars.surface_thickness = thickness; }

    void SetClusterTimeThreshold(double threshold) {
      fPreClusterPars.cluster_time_threshold = threshold;
    }

    void SetElectronTrackEnergyThreshold(double threshold) {
      fPreClusterPars.track_energy_threshold = threshold;
    }
 
    void EndOfRunAction(const G4Run*) override;
 
  protected:
 
    [[nodiscard]] std::string GetNtupleNameFlat() const override { return "germanium"; }
 
  private:
 
    RMGDetectorHitsCollection* GetHitColl(const G4Event*);
    void SetPositionModeString(std::string mode);
 
    std::vector<std::unique_ptr<G4GenericMessenger>> fMessengers;
    void DefineCommands();
 
    double fEdepCutLow = -1;
    double fEdepCutHigh = -1;
    std::set<int> fEdepCutDetectors;
 
    bool fDiscardPhotonsIfNoGermaniumEdep = false;
    bool fDiscardZeroEnergyHits = true;
 
    bool fStoreSinglePrecisionEnergy = false;
    bool fStoreSinglePrecisionPosition = false;
 
    bool fStoreTrackID = false;
    bool fPreClusterHits = true;
    bool fStoreVelocity = false;

    RMGOutputTools::ClusterPars fPreClusterPars{};
 
    // mode of position to store
    RMGOutputTools::PositionMode fPositionMode = RMGOutputTools::PositionMode::kAverage;
 
    std::map<std::string, G4ThreeVector> fDetectorOrigins;
};
 
#endif
 
// vim: tabstop=2 shiftwidth=2 expandtab