TRestWimpSensitivity.h

/*************************************************************************
 * This file is part of the REST software framework.                     *
 *                                                                       *
 * Copyright (C) 2016 GIFNA/TREX (University of Zaragoza)                *
 * For more information see http://gifna.unizar.es/trex                  *
 *                                                                       *
 * REST 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, either version 3 of the License, or     *
 * (at your option) any later version.                                   *
 *                                                                       *
 * REST 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 a copy of the GNU General Public License along with   *
 * REST in $REST_PATH/LICENSE.                                           *
 * If not, see http://www.gnu.org/licenses/.                             *
 * For the list of contributors see $REST_PATH/CREDITS.                  *
 *************************************************************************/
 
#ifndef RestCore_TRestWimpSensitivity
#define RestCore_TRestWimpSensitivity
 
#include <TH1D.h>
#include <TRestMetadata.h>
#include <TRestWimpNucleus.h>
 
class TRestWimpSensitivity : public TRestMetadata {
   private:
    std::vector<TRestWimpNucleus> fNuclei;
    Double_t fWimpDensity = 0.3;
    Double_t fLabVelocity = 232;
    Double_t fEscapeVelocity = 544;
    Double_t fRmsVelocity = 220;
    Double_t fExposure = 365. * 0.32;
    Double_t fBackground = 1;
    // TODO add option to use a histogram for the spectra
    TVector2 fEnergySpectra = TVector2(0, 2);
    Double_t fEnergySpectraStep = 0.01;
    TVector2 fEnergyRange = TVector2(0.1, 1.1);
    Bool_t fUseQuenchingFactor = true;
 
    std::map<std::string, TH1D*> quenchingFactor;  
 
   public:
    TRestWimpSensitivity(const char* configFilename, const std::string& name = "");
 
    ~TRestWimpSensitivity();
 
    void Initialize() override;
    void InitFromConfigFile() override;
    void PrintMetadata() override;
 
    void ReadNuclei();
    const Double_t GetSensitivity(const double wimpMass);
    void CalculateQuenchingFactor();
    bool isEnergySpectraWideEnough();
    const std::string BuildOutputFileName(const std::string& extension = ".txt");
 
    std::map<std::string, TH1D*> GetRecoilSpectra(const double wimpMass, const double crossSection);
    std::map<std::string, TH1D*> GetFormFactor();
    inline auto GetQuenchingFactor() { return quenchingFactor; }
    inline std::vector<TRestWimpNucleus>& GetNuclei() { return fNuclei; };
    inline auto GetEnergySpectra() { return fEnergySpectra; }
    inline auto GetEnergySpectraStep() { return fEnergySpectraStep; }
    inline auto GetEnergyRange() { return fEnergyRange; }
    inline auto GetWimpDensity() { return fWimpDensity; }
    inline auto GetLabVelocity() { return fLabVelocity; }
    inline auto GetEscapeVelocity() { return fEscapeVelocity; }
    inline auto GetRmsVelocity() { return fRmsVelocity; }
    inline auto GetExposure() { return fExposure; }
    inline auto GetBackground() { return fBackground; }
    inline auto GetUseQuenchingFactor() { return fUseQuenchingFactor; }
 
    void SetNuclei(const std::vector<TRestWimpNucleus>& nuclei) { fNuclei = nuclei; }
    void SetEnergySpectra(const TVector2& energySpectra) { fEnergySpectra = energySpectra; }
    void SetEnergySpectraStep(const Double_t energySpectraStep) { fEnergySpectraStep = energySpectraStep; }
    void SetEnergyRange(const TVector2& energyRange) { fEnergyRange = energyRange; }
    void SetWimpDensity(const Double_t wimpDensity) { fWimpDensity = wimpDensity; }
    void SetLabVelocity(const Double_t labVelocity) { fLabVelocity = labVelocity; }
    void SetEscapeVelocity(const Double_t escapeVelocity) { fEscapeVelocity = escapeVelocity; }
    void SetRmsVelocity(const Double_t rmsVelocity) { fRmsVelocity = rmsVelocity; }
    void SetExposure(const Double_t exposure) { fExposure = exposure; }
    void SetBackground(const Double_t background) { fBackground = background; }
    void SetUseQuenchingFactor(const Bool_t useQuenchingFactor) { fUseQuenchingFactor = useQuenchingFactor; }
 
    ClassDefOverride(TRestWimpSensitivity, 1);
};
 
#endif