TRestAxionTransmissionProcess.cxx

/*************************************************************************
 * 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 "TRestAxionTransmissionProcess.h"
using namespace std;
 
ClassImp(TRestAxionTransmissionProcess);
 
TRestAxionTransmissionProcess::TRestAxionTransmissionProcess() { Initialize(); }
 
TRestAxionTransmissionProcess::TRestAxionTransmissionProcess(char* cfgFileName) {
    Initialize();
 
    LoadConfig(cfgFileName);
}
 
TRestAxionTransmissionProcess::~TRestAxionTransmissionProcess() { delete fAxionEvent; }
 
void TRestAxionTransmissionProcess::LoadDefaultConfig() {
    SetName(this->ClassName());
    SetTitle("Default config");
}
 
void TRestAxionTransmissionProcess::LoadConfig(std::string cfgFilename, std::string name) {
    if (LoadConfigFromFile(cfgFilename, name)) LoadDefaultConfig();
}
 
void TRestAxionTransmissionProcess::Initialize() {
    SetSectionName(this->ClassName());
    SetLibraryVersion(LIBRARY_VERSION);
 
    fAxionEvent = new TRestAxionEvent();
}
 
void TRestAxionTransmissionProcess::InitProcess() {
    RESTDebug << "Entering ... TRestAxionGeneratorProcess::InitProcess" << RESTendl;
 
    RegisterAllObservables();
 
    fXrayWindows.clear();
    for (const auto& wName : fWindowNames) {
        TRestAxionXrayWindow* w = (TRestAxionXrayWindow*)GetMetadata(wName);
 
        if (w == nullptr) {
            RESTError << "TRestAxionTransmissionProcess. Window definition with name : " << wName
                      << " not found!" << RESTendl;
        } else {
            fXrayWindows.push_back(w);
        }
    }
}
 
TRestEvent* TRestAxionTransmissionProcess::ProcessEvent(TRestEvent* evInput) {
    fAxionEvent = (TRestAxionEvent*)evInput;
 
    TVector3 inPos = fAxionEvent->GetPosition();
    TVector3 inDir = fAxionEvent->GetDirection();
 
    TVector3 newPos = REST_Physics::MoveToPlane(inPos, inDir, TVector3(0, 0, 1), TVector3(0, 0, 0));
 
    Double_t transmission = 1;
    Double_t x = newPos.X();
    Double_t y = newPos.Y();
    Double_t z = newPos.Z();
    Double_t en = fAxionEvent->GetEnergy();
 
    RESTDebug << "Particle position to evaluate window transmission. " << RESTendl;
    RESTDebug << "X : " << x << " Y: " << y << " Z: " << z << RESTendl;
 
    for (const auto& window : fXrayWindows) {
        transmission *= window->GetTransmission(en, x, y);
    }
    RESTDebug << "Transmission: " << transmission << RESTendl;
 
    SetObservableValue("transmission", transmission);
 
    if (GetVerboseLevel() >= TRestStringOutput::REST_Verbose_Level::REST_Debug) {
        fAxionEvent->PrintEvent();
 
        if (GetVerboseLevel() >= TRestStringOutput::REST_Verbose_Level::REST_Extreme) GetChar();
    }
 
    return fAxionEvent;
}
 
void TRestAxionTransmissionProcess::InitFromConfigFile() {
    TRestEventProcess::InitFromConfigFile();
 
    // This is the additional code required by the process to read window names
    TiXmlElement* windowDefinition = GetElement("window");
    while (windowDefinition) {
        fWindowNames.push_back(GetFieldValue("name", windowDefinition));
 
        windowDefinition = GetNextElement(windowDefinition);
    }
}