TRestAxionEvent.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 "TRestAxionEvent.h"
 
#include "TRestTools.h"
 
using namespace std;
using namespace TMath;
 
ClassImp(TRestAxionEvent);
 
TRestAxionEvent::TRestAxionEvent() {
    Initialize();
    fPad = NULL;
}
 
TRestAxionEvent::~TRestAxionEvent() {}
 
void TRestAxionEvent::Initialize() { TRestEvent::Initialize(); }
 
TPad* TRestAxionEvent::DrawEvent(TString option) {
    vector<string> optList = TRestTools::GetOptions((string)option);
 
    for (unsigned int n = 0; n < optList.size(); n++) {
        if (optList[n] == "print") this->PrintEvent();
    }
 
    optList.erase(std::remove(optList.begin(), optList.end(), "print"), optList.end());
 
    if (optList.size() == 0) optList.push_back("TODO");
 
    if (fPad != NULL) {
        delete fPad;
        fPad = NULL;
    }
 
    fPad = new TPad(this->GetName(), " ", 0, 0, 1, 1);
    fPad->Divide(1, 1);
    fPad->Draw();
 
    // TOBE implemented if necessary
 
    return fPad;
}
 
void TRestAxionEvent::RotateXZ(const TVector3& center, Double_t theta, Double_t phi) {
    TVector3 ref = fPosition - center;
 
    ref.RotateX(theta);
    ref.RotateZ(phi);
 
    fPosition = ref + center;
 
    fDirection.RotateX(theta);
    fDirection.RotateZ(phi);
}
 
void TRestAxionEvent::RotateZX(const TVector3& center, Double_t phi, Double_t theta) {
    TVector3 ref = fPosition - center;
 
    ref.RotateZ(phi);
    ref.RotateX(theta);
 
    fPosition = ref + center;
 
    fDirection.RotateZ(phi);
    fDirection.RotateX(theta);
}
 
void TRestAxionEvent::RotateXY(const TVector3& center, Double_t pitch, Double_t yaw) {
    TVector3 ref = fPosition - center;
 
    ref.RotateX(pitch);
    ref.RotateY(yaw);
 
    fPosition = ref + center;
 
    fDirection.RotateX(pitch);
    fDirection.RotateY(yaw);
}
 
void TRestAxionEvent::RotateYX(const TVector3& center, Double_t yaw, Double_t pitch) {
    TVector3 ref = fPosition - center;
 
    ref.RotateY(yaw);
    ref.RotateX(pitch);
 
    fPosition = ref + center;
 
    fDirection.RotateY(yaw);
    fDirection.RotateX(pitch);
}
 
void TRestAxionEvent::Translate(const TVector3& delta) { fPosition += delta; }
 
void TRestAxionEvent::PrintEvent() {
    TRestEvent::PrintEvent();
 
    cout << "Mass : " << GetMass() * units("eV") << " eV" << endl;
    cout << "Energy : " << GetEnergy() << " keV" << endl;
    cout << "Position : ( " << fPosition.X() << ", " << fPosition.Y() << ", " << fPosition.Z() << " ) mm"
         << endl;
    cout << "Direction : ( " << fDirection.X() << ", " << fDirection.Y() << ", " << fDirection.Z() << " )"
         << endl;
    cout << endl;
}