TRestCombinedMask.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 "TRestCombinedMask.h"
 
#include "TRandom3.h"
 
ClassImp(TRestCombinedMask);
 
TRestCombinedMask::TRestCombinedMask() : TRestPatternMask() { Initialize(); }
 
TRestCombinedMask::TRestCombinedMask(const char* cfgFileName, std::string name)
    : TRestPatternMask(cfgFileName) {
    Initialize();
 
    LoadConfigFromFile(fConfigFileName, name);
 
    if (GetVerboseLevel() >= TRestStringOutput::REST_Verbose_Level::REST_Info) PrintMetadata();
}
 
TRestCombinedMask::~TRestCombinedMask() {}
 
void TRestCombinedMask::Initialize() {
    SetSectionName(this->ClassName());
    SetType("Combined");
 
    for (const auto mask : fMasks) {
        mask->Initialize();
        if (mask->GetMaskRadius() > fMaskRadius) fMaskRadius = mask->GetMaskRadius();
    }
}
 
Int_t TRestCombinedMask::GetRegion(Double_t& x, Double_t& y) {
    Int_t region = 0;
    Double_t xIn = x;
    Double_t yIn = y;
    for (const auto mask : fMasks) {
        Int_t id = mask->GetRegion(x, y);
 
        // The (x,y) coordinates are transformed for each mask.
        // We need to recover the original point to be evaluated for each mask.
        x = xIn;
        y = yIn;
 
        RESTDebug << "TRestCombinedMask::GetRegion. Mask type: " << mask->GetType() << " region : " << id
                  << RESTendl;
        if (id == 0) return 0;
 
        region = id + region * mask->GetMaxRegions();
    }
 
    return region;
}
 
void TRestCombinedMask::InitFromConfigFile() {
    TRestMetadata::InitFromConfigFile();
 
    int cont = 0;
    TRestPatternMask* msk = (TRestPatternMask*)this->InstantiateChildMetadata(cont, "Mask");
    while (msk != nullptr) {
        AddMask(msk);
        cont++;
        msk = (TRestPatternMask*)this->InstantiateChildMetadata(cont, "Mask");
    }
    Initialize();
}
 
void TRestCombinedMask::PrintMetadata() {
    TRestPatternMask::PrintMetadata();
 
    PrintMask();
 
    Int_t cont = 1;
    for (const auto mask : fMasks) {
        RESTMetadata << " == MASK " << cont << " == " << RESTendl;
        cont++;
        mask->PrintMask();
        RESTMetadata << "++++" << RESTendl;
    }
}
 
void TRestCombinedMask::PrintMask() {
    PrintMaskMembers();
    RESTMetadata << "++++" << RESTendl;
}
 
void TRestCombinedMask::PrintMaskMembers() {
    RESTMetadata << " - Number of masks : " << fMasks.size() << RESTendl;
}