TRestDetectorReadoutMapping.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 "TRestDetectorReadoutMapping.h"
 
using namespace std;
 
ClassImp(TRestDetectorReadoutMapping);
TRestDetectorReadoutMapping::TRestDetectorReadoutMapping() {}
 
TRestDetectorReadoutMapping::~TRestDetectorReadoutMapping() {}
 
Double_t TRestDetectorReadoutMapping::GetX(Int_t nodeX) { return (fNetSizeX / fNodesX) * nodeX; }
 
Double_t TRestDetectorReadoutMapping::GetY(Int_t nodeY) { return (fNetSizeY / fNodesY) * nodeY; }
 
Int_t TRestDetectorReadoutMapping::GetNodeX(Double_t x) {
    Int_t nX = (Int_t)((x / fNetSizeX) * fNodesX);
    if (nX >= fNodesX) return fNodesX - 1;
    return nX;
}
 
Int_t TRestDetectorReadoutMapping::GetNodeY(Double_t y) {
    Int_t nY = (Int_t)((y / fNetSizeY) * fNodesY);
    if (nY >= fNodesY) return fNodesY - 1;
    return nY;
}
 
Int_t TRestDetectorReadoutMapping::GetChannel(Double_t x, Double_t y) {
    return fChannel[GetNodeX(x)][GetNodeY(y)];
}
 
Int_t TRestDetectorReadoutMapping::GetPixel(Double_t x, Double_t y) {
    return fPixel[GetNodeX(x)][GetNodeY(y)];
}
 
Int_t TRestDetectorReadoutMapping::GetNumberOfNodesNotSet() {
    Int_t counter = 0;
    for (int i = 0; i < fNodesX; i++)
        for (int j = 0; j < fNodesY; j++) {
            if (!isNodeSet(i, j)) counter++;
        }
    return counter;
}
 
Int_t TRestDetectorReadoutMapping::GetNodeX_ForChannelAndPixel(Int_t ch, Int_t px) {
    for (int i = 0; i < fNodesX; i++)
        for (int j = 0; j < fNodesY; j++) {
            if (fChannel[i][j] == ch && fPixel[i][j] == px) return i;
        }
    return -1;
}
 
Int_t TRestDetectorReadoutMapping::GetNodeY_ForChannelAndPixel(Int_t ch, Int_t px) {
    for (int i = 0; i < fNodesX; i++)
        for (int j = 0; j < fNodesY; j++) {
            if (fChannel[i][j] == ch && fPixel[i][j] == px) return j;
        }
    return -1;
}
 
void TRestDetectorReadoutMapping::Initialize(Int_t nX, Int_t nY, Double_t sX, Double_t sY) {
    fNodesX = nX;
    fNodesY = nY;
    fNetSizeX = sX;
    fNetSizeY = sY;
    fChannel.ResizeTo(fNodesX, fNodesY);
    fPixel.ResizeTo(fNodesX, fNodesY);
 
    for (int i = 0; i < fNodesX; i++)
        for (int j = 0; j < fNodesY; j++) {
            fChannel[i][j] = -1;
            fPixel[i][j] = -1;
        }
}
 
void TRestDetectorReadoutMapping::SetNode(Int_t i, Int_t j, Int_t ch, Int_t pix) {
    fChannel[i][j] = ch;
    fPixel[i][j] = pix;
}
 
Bool_t TRestDetectorReadoutMapping::isNodeSet(Int_t i, Int_t j) {
    if (fChannel[i][j] == -1 || fPixel[i][j] == -1) return false;
    return true;
}
 
Bool_t TRestDetectorReadoutMapping::AllNodesSet() {
    for (int i = 0; i < fNodesX; i++)
        for (int j = 0; j < fNodesY; j++) {
            if (!isNodeSet(i, j)) {
                cout << "Node : " << i << " , " << j << " is NOT set. Ch : " << fChannel[i][j]
                     << " Pix : " << fPixel[i][j] << endl;
                return false;
            }
        }
    return true;
}