TRestDetectorHitsRotateAndTranslateProcess.cxx

 
 
#include "TRestDetectorHitsRotateAndTranslateProcess.h"
 
using namespace std;
 
#include <TRandom3.h>
 
ClassImp(TRestDetectorHitsRotateAndTranslateProcess);
 
TRestDetectorHitsRotateAndTranslateProcess::TRestDetectorHitsRotateAndTranslateProcess() { Initialize(); }
 
TRestDetectorHitsRotateAndTranslateProcess::TRestDetectorHitsRotateAndTranslateProcess(
    const char* configFilename) {
    Initialize();
 
    if (LoadConfigFromFile(configFilename)) {
        LoadDefaultConfig();
    }
 
    PrintMetadata();
}
 
TRestDetectorHitsRotateAndTranslateProcess::~TRestDetectorHitsRotateAndTranslateProcess() {
    // TRestDetectorHitsRotateAndTranslateProcess destructor
}
 
void TRestDetectorHitsRotateAndTranslateProcess::LoadDefaultConfig() {
    SetTitle("Default config");
 
    fDeltaX = 1.0;
    fDeltaY = 1.0;
    fDeltaZ = 1.0;
    fAlpha = 0.;
    fBeta = 0.;
    fGamma = 0.;
}
 
void TRestDetectorHitsRotateAndTranslateProcess::Initialize() {
    SetSectionName(this->ClassName());
    SetLibraryVersion(LIBRARY_VERSION);
 
    fDeltaX = 1.0;
    fDeltaY = 1.0;
    fDeltaZ = 1.0;
    fAlpha = 0.;
    fBeta = 0.;
    fGamma = 0.;
 
    fInputHitsEvent = nullptr;
    fOutputHitsEvent = nullptr;
}
 
void TRestDetectorHitsRotateAndTranslateProcess::LoadConfig(string configFilename) {
    if (LoadConfigFromFile(configFilename)) LoadDefaultConfig();
 
    PrintMetadata();
}
 
void TRestDetectorHitsRotateAndTranslateProcess::InitProcess() {
    // Function to be executed once at the beginning of process
    // (before starting the process of the events)
 
    // Start by calling the InitProcess function of the abstract class.
    // Comment this if you don't want it.
    // TRestEventProcess::InitProcess();
}
 
TRestEvent* TRestDetectorHitsRotateAndTranslateProcess::ProcessEvent(TRestEvent* inputEvent) {
    fInputHitsEvent = (TRestDetectorHitsEvent*)inputEvent;
 
    fOutputHitsEvent = fInputHitsEvent;
    // fInputHitsEvent->CloneTo(fOutputHitsEvent);
 
    TVector3 meanPosition = fOutputHitsEvent->GetMeanPosition();
    for (unsigned int hit = 0; hit < fOutputHitsEvent->GetNumberOfHits(); hit++) {
        fOutputHitsEvent->GetHits()->RotateIn3D(hit, fAlpha, fBeta, fGamma, meanPosition);
        fOutputHitsEvent->GetHits()->Translate(hit, fDeltaX, fDeltaY, fDeltaZ);
    }
 
    if (fOutputHitsEvent->GetNumberOfHits() == 0) return nullptr;
 
    RESTDebug << "Number of hits rotated: " << fInputHitsEvent->GetNumberOfHits() << RESTendl;
    return fOutputHitsEvent;
}
 
void TRestDetectorHitsRotateAndTranslateProcess::EndProcess() {
    // Function to be executed once at the end of the process
    // (after all events have been processed)
 
    // Start by calling the EndProcess function of the abstract class.
    // Comment this if you don't want it.
    // TRestEventProcess::EndProcess();
}
 
void TRestDetectorHitsRotateAndTranslateProcess::InitFromConfigFile() {
    fDeltaX = GetDblParameterWithUnits("deltaX");
    fDeltaY = GetDblParameterWithUnits("deltaY");
    fDeltaZ = GetDblParameterWithUnits("deltaZ");
 
    fAlpha = StringToDouble(GetParameter("alpha"));  // rotation angle around Z
    fBeta = StringToDouble(GetParameter("beta"));    // rotation angle around Y
    fGamma = StringToDouble(GetParameter("gamma"));  // rotation angle around X
 
    // Conversion to radians
    fAlpha = fAlpha * TMath::Pi() / 180.;
    fBeta = fBeta * TMath::Pi() / 180.;
    fGamma = fGamma * TMath::Pi() / 180.;
}