GPowerSpectrumAtmoFlux.h

Go to the documentation of this file.

//____________________________________________________________________________
//____________________________________________________________________________

#pragma once

#include <TLorentzVector.h>

#include "GENIE/Tools/Flux/GAtmoFlux.h"
#include "GENIE/Framework/ParticleData/PDGCodeList.h"
#include "TH2D.h"

namespace genie {
namespace flux {

class GPowerSpectrumAtmoFlux: public GFluxI {
public:
        GPowerSpectrumAtmoFlux();
        ~GPowerSpectrumAtmoFlux();

        const PDGCodeList &FluxParticles(void) override;  
        double MaxEnergy(void) override; 
        bool GenerateNext(void) override; 
        int PdgCode(void) override; 
        double Weight(void) override; 
        const TLorentzVector& Momentum(void) override; 
        const TLorentzVector& Position(void) override; 
        bool End(void) override; 
        long int Index(void) override; 
        void Clear(Option_t *opt) override; 
        void GenerateWeighted(bool gen_weighted) override; 

        double MinEnergy(void);
        void SetSpectralIndex(double index);
        void SetUserCoordSystem(TRotation &rotation);
        void SetRadii(double Rlongitudinal, double Rtransverse);
        void SetFlavors(std::vector<int> flavors);
        void SetMinEnergy(double Emin);
        void SetMaxEnergy(double Emax);
        long int NFluxNeutrinos(void) const;
        void ResetNFluxNeutrinos(void);
        void AddFluxFile(int neutrino_pdg, string filename);
        bool LoadFluxData(void);

        double GetFlux(int flavour, double energy, double costh, double phi);
        double ComputeWeight(int flavour, double energy, double costh, double phi);
        void InitializeWeight();


private:
        PDGCodeList *fPdgCList; 
        double fMaxEvCut; 
        double fMinEvCut; 
        int fgPdgC; 
        TLorentzVector fgP4; 
        TLorentzVector fgX4; 
        double fWeight; 
        double fSpectralIndex; 
        double fRl; 
        double fRt; 
        double fGlobalGenWeight; 
        double fAgen; 
        TRotation fRotTHz2User; 
        long int fNNeutrinos; 
        vector<int> fFluxFlavour; 
    vector<string> fFluxFile; 
        map<int, TH3D*> fRawFluxHistoMap; 
        map<int, TH2D*> fRawFluxHistoMap2D; 

        bool FillFluxHisto(int nu_pdg, string filename);
        void AddAllFluxes(void);
        TH3D* CreateNormalisedFluxHisto( TH3D* hist);  // normalise flux files
        void ResetSelection(void);
        void Initialize(void);
        void CleanUp(void);
};


} // flux namespace
} // genie namespace