TRestAxionSolarFlux.cxx

/******************** REST disclaimer ***********************************
 * 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.                  *
 *************************************************************************/
 
 
#include "TRestAxionSolarFlux.h"
using namespace std;
 
ClassImp(TRestAxionSolarFlux);
 
TRestAxionSolarFlux::TRestAxionSolarFlux() : TRestMetadata() {}
 
TRestAxionSolarFlux::TRestAxionSolarFlux(const char* cfgFileName, string name) : TRestMetadata(cfgFileName) {
    RESTDebug << "Entering TRestAxionSolarFlux constructor( cfgFileName, name )" << RESTendl;
    RESTDebug << "File: " << cfgFileName << " Name: " << name << RESTendl;
}
 
TRestAxionSolarFlux::~TRestAxionSolarFlux() {}
 
void TRestAxionSolarFlux::Initialize() {
    SetLibraryVersion(LIBRARY_VERSION);
 
    if (!fTablesLoaded) LoadTables();
 
    if (!fRandom) {
        delete fRandom;
        fRandom = nullptr;
    }
 
    if (fRandom != nullptr) {
        delete fRandom;
        fRandom = nullptr;
    }
 
    fRandom = new TRandom3(fSeed);
    fSeed = fRandom->TRandom::GetSeed();
}
 
TH1F* TRestAxionSolarFlux::GetFluxHistogram(string fname, Double_t binSize) {
    string fullPathName = SearchFile(fname);
 
    std::vector<std::vector<Double_t>> fluxData;
    TRestTools::ReadASCIITable(fullPathName, fluxData, 3);
 
    TH2F* originalHist =
        new TH2F(Form("FluxTable_%s", GetName()), "", 100, 0., 1., (Int_t)(20. / binSize), 0., 20.);
 
    for (const auto& data : fluxData) {
        Double_t r = 0.005 + data[0];
        Double_t en = data[1] - 0.005;
        Double_t flux = data[2] * binSize;  // flux in cm-2 s-1 bin-1
 
        originalHist->Fill(r, en, flux);
    }
 
    return (TH1F*)originalHist->ProjectionY();
}
 
TCanvas* TRestAxionSolarFlux::DrawFluxFile(string fname, Double_t binSize) {
    if (fCanvas != nullptr) {
        delete fCanvas;
        fCanvas = nullptr;
    }
    fCanvas = new TCanvas("canv", "This is the canvas title", 1400, 1200);
    fCanvas->Draw();
 
    TPad* pad1 = new TPad("pad1", "This is pad1", 0.01, 0.02, 0.99, 0.97);
    pad1->Draw();
 
    fCanvas->cd();
    pad1->cd();
 
    GetFluxHistogram(fname, binSize)->Draw("hist");
 
    return fCanvas;
}
 
Double_t TRestAxionSolarFlux::GetFluxAtEnergy(Double_t energy, Double_t m) {
    TH1F* h = GetEnergySpectrum(m);
    return h->GetBinContent(h->FindBin(energy));
}
 
TCanvas* TRestAxionSolarFlux::DrawSolarFlux() {
    if (fCanvas != nullptr) {
        delete fCanvas;
        fCanvas = nullptr;
    }
    fCanvas = new TCanvas("canv", "This is the canvas title", 1200, 500);
    fCanvas->Draw();
 
    TPad* pad1 = new TPad("pad1", "This is pad1", 0.01, 0.02, 0.99, 0.97);
    pad1->Draw();
 
    pad1->cd();
    pad1->SetLogy();
    pad1->SetRightMargin(0.09);
    pad1->SetLeftMargin(0.15);
    pad1->SetBottomMargin(0.15);
 
    TH1F* ht = GetEnergySpectrum();
    ht->SetLineColor(kBlack);
    ht->SetFillStyle(4050);
    ht->SetFillColor(kBlue - 10);
 
    ht->Draw("hist");
 
    return fCanvas;
}
 
void TRestAxionSolarFlux::PrintMetadata() {
    TRestMetadata::PrintMetadata();
 
    RESTMetadata << " - Coupling type : " << fCouplingType << RESTendl;
    RESTMetadata << " - Coupling strength : " << fCouplingStrength << RESTendl;
    RESTMetadata << "--------" << RESTendl;
    RESTMetadata << " - Random seed : " << fSeed << RESTendl;
    RESTMetadata << "--------" << RESTendl;
}