TRestRingsMask.cxx

/*************************************************************************
 * This file is part of the REST software framework.                     *
 *                                                                       *
 * Copyright (C) 2016 GIFNA/TREX (University of Zaragoza)                *
 * For more information see https://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 https://www.gnu.org/licenses/.                            *
 * For the list of contributors see $REST_PATH/CREDITS.                  *
 *************************************************************************/
 
 
#include "TRestRingsMask.h"
 
#include "TRandom3.h"
 
ClassImp(TRestRingsMask);
 
TRestRingsMask::TRestRingsMask() : TRestPatternMask() { Initialize(); }
 
TRestRingsMask::TRestRingsMask(const char* cfgFileName, std::string name) : TRestPatternMask(cfgFileName) {
    Initialize();
 
    LoadConfigFromFile(fConfigFileName, name);
 
    if (GetVerboseLevel() >= TRestStringOutput::REST_Verbose_Level::REST_Info) PrintMetadata();
}
 
TRestRingsMask::~TRestRingsMask() {}
 
void TRestRingsMask::Initialize() {
    SetSectionName(this->ClassName());
    SetType("Rings");
 
    if (fNRings > 0) GenerateRings();
}
 
Int_t TRestRingsMask::GetRegion(Double_t& x, Double_t& y) {
    if (TRestPatternMask::GetRegion(x, y)) return 0;
 
    Double_t r = TMath::Sqrt(x * x + y * y);
    int cont = 0;
    for (const auto& ringRadius : fRingsRadii) {
        if (r < ringRadius.second && r >= ringRadius.first) return cont % fMaxRegions + 1;
        cont++;
    }
    return 0;
}
 
void TRestRingsMask::GenerateRings() {
    fRingsRadii.clear();
    for (int n = 0; n < fNRings; n++) {
        Double_t inR = fInitialRadius + n * fRingsGap;
        Double_t outR = fRingsThickness + fInitialRadius + n * fRingsGap;
 
        std::pair<Double_t, Double_t> p(inR, outR);
        fRingsRadii.push_back(p);
    }
}
 
void TRestRingsMask::PrintMetadata() {
    TRestPatternMask::PrintMetadata();
 
    PrintMaskMembers();
    RESTMetadata << "----" << RESTendl;
}
 
void TRestRingsMask::PrintMask() {
    PrintCommonPatternMembers();
    RESTMetadata << "----" << RESTendl;
    PrintMaskMembers();
}
 
void TRestRingsMask::PrintMaskMembers() {
    if (fRingsGap > 0) RESTMetadata << " - Rings gap : " << fRingsGap << " mm" << RESTendl;
    if (fRingsThickness > 0) RESTMetadata << " - Rings thickness : " << fRingsThickness << " mm" << RESTendl;
    if (fNRings > 0) RESTMetadata << " - Number of rings : " << fNRings << RESTendl;
    if (fInitialRadius > 0) RESTMetadata << " - Initial radius : " << fInitialRadius << " mm" << RESTendl;
    if (fRingsRadii.size() > 0) {
        if (fNRings > 0) RESTMetadata << "-----" << RESTendl;
        RESTMetadata << "Inner radii : (";
        for (unsigned int n = 0; n < fRingsRadii.size(); n++) {
            if (n > 0) RESTMetadata << ", ";
            RESTMetadata << fRingsRadii[n].first;
        }
        RESTMetadata << ")" << RESTendl;
    }
    if (fRingsRadii.size() > 0) {
        RESTMetadata << "Outter radii : (";
        for (unsigned int n = 0; n < fRingsRadii.size(); n++) {
            if (n > 0) RESTMetadata << ", ";
            RESTMetadata << fRingsRadii[n].second;
        }
        RESTMetadata << ")" << RESTendl;
    }
}
 
void TRestRingsMask::PrintRings() {
    for (unsigned int n = 0; n < fRingsRadii.size(); n++)
        std::cout << n << " - " << fRingsRadii[n].first << " - " << fRingsRadii[n].second << std::endl;
}