TRestDetectorReadoutEventViewer.cxx

 
#include "TRestDetectorReadoutEventViewer.h"
 
using namespace std;
 
Int_t planeId = 0;
 
ClassImp(TRestDetectorReadoutEventViewer)
 
    TRestDetectorReadoutEventViewer::TRestDetectorReadoutEventViewer() {
    Initialize();
}
 
TRestDetectorReadoutEventViewer::~TRestDetectorReadoutEventViewer() {}
 
void TRestDetectorReadoutEventViewer::Initialize() {
    TRestEventViewer::Initialize();
 
    fCanvasXY = new TCanvas("ReadoutMap", "ReadoutMap");
    fCanvasXZYZ = new TCanvas("XZYZ", "XZYZ");
    fCanvasXZYZ->Divide(2, 1);
 
    fHistoXZ = nullptr;
    fHistoYZ = nullptr;
 
    fSignalEvent = new TRestDetectorSignalEvent();
    SetEvent(fSignalEvent);
}
 
void TRestDetectorReadoutEventViewer::SetReadout(TRestDetectorReadout* readout) {
    // Finalize the instantiation based on argument TRestDetectorReadout
    fReadout = readout;
    cout << "WARNING : Only plane 0 is drawn. Implementation to draw several "
            "planes or to choose the plane must be implemented."
         << endl;
    fReadout->PrintMetadata();
    TRestDetectorReadoutPlane* plane = &(*fReadout)[planeId];
    fHistoXY = plane->GetReadoutHistogram();
    plane->GetBoundaries(xmin, xmax, ymin, ymax);
}
 
void TRestDetectorReadoutEventViewer::AddEvent(TRestEvent* ev) {
    // Finalize the drawing of current event, adding to the per-channel-signal
    // vs. time drawn by the generic event viewer, the three 2D histograms of
    // the XY, XZ and YZ projections.
    TRestEventViewer::AddEvent(ev);
 
    if (fPad == nullptr) return;
 
    fSignalEvent = (TRestDetectorSignalEvent*)ev;
 
    // XY histo is expected to have always same binning => reset it.
    // (X|Y)Z may change from event to event => delete (and later on re-create).
    fHistoXY->Reset(0);
    if (fHistoXZ != nullptr) {
        delete fHistoXZ;
        fHistoXZ = nullptr;
    }
    if (fHistoYZ != nullptr) {
        delete fHistoYZ;
        fHistoYZ = nullptr;
    }
 
    DrawReadoutPulses();
    fCanvasXY->cd();
    fHistoXY->Draw("COLZ");
    fCanvasXY->Update();
 
    fCanvasXZYZ->cd(1);
    fHistoXZ->Draw("CONTZ");
    fCanvasXZYZ->cd(2);
    fHistoYZ->Draw("CONTZ");
    fCanvasXZYZ->Update();
}
 
void TRestDetectorReadoutEventViewer::DrawReadoutPulses() {
    int readoutChannel, daqChannel;
    double charge;
 
    Int_t modId;
    TRestDetectorReadoutModule* module = nullptr;
    TRestDetectorReadoutChannel* channel;
 
    int maxIndex;
 
    zmin = 1E9;
    zmax = 1E-9;
 
    double z;
    for (int i = 0; i < fSignalEvent->GetNumberOfSignals(); i++)
        for (int j = 0; j < fSignalEvent->GetSignal(i)->GetNumberOfPoints(); j++) {
            z = fSignalEvent->GetSignal(i)->GetTime(j);
            if (z < zmin) zmin = z;
            if (z > zmax) zmax = z;
        }
 
    zmax++;
    zmin--;
 
    fHistoXZ = new TH2D("XZ", "XZ", 100, xmin, xmax, 100, zmin, zmax);
    fHistoYZ = new TH2D("YZ", "YZ", 100, ymin, ymax, 100, zmin, zmax);
 
    for (int i = 0; i < fSignalEvent->GetNumberOfSignals(); i++) {
        daqChannel = fSignalEvent->GetSignal(i)->GetSignalID();
 
        TRestDetectorReadoutPlane* plane = &(*fReadout)[planeId];
        for (size_t m = 0; m < plane->GetNumberOfModules(); m++) {
            module = &(*plane)[m];
 
            if (module->IsDaqIDInside(daqChannel)) break;
        }
        modId = module->GetModuleID();
 
        readoutChannel = module->DaqToReadoutChannel(daqChannel);
        cout << "daqChannel " << daqChannel << " readoutChannel " << readoutChannel << endl;
 
        if ((channel = GetChannel(readoutChannel)) == nullptr) {
            continue;
        }
 
        int nPixels = channel->GetNumberOfPixels();
 
        Double_t xRead = plane->GetX(modId, readoutChannel);
        Double_t yRead = plane->GetY(modId, readoutChannel);
 
        // Pixel readout
        Int_t xStrip = 0;
        Int_t yStrip = 0;
        if (TMath::IsNaN(yRead)) xStrip = 1;
        if (TMath::IsNaN(xRead)) yStrip = 1;
        if (xStrip == 0 && yStrip == 0) {
            for (int j = 0; j < fSignalEvent->GetSignal(i)->GetNumberOfPoints(); j++)
                fHistoXZ->Fill(xRead, fSignalEvent->GetSignal(i)->GetTime(j),
                               fSignalEvent->GetSignal(i)->GetData(j));
            for (int j = 0; j < fSignalEvent->GetSignal(i)->GetNumberOfPoints(); j++)
                fHistoYZ->Fill(yRead, fSignalEvent->GetSignal(i)->GetTime(j),
                               fSignalEvent->GetSignal(i)->GetData(j));
        }
        // Strips readout
        else {
            if (yStrip == 0) {
                for (int j = 0; j < fSignalEvent->GetSignal(i)->GetNumberOfPoints(); j++)
                    fHistoXZ->Fill(xRead, fSignalEvent->GetSignal(i)->GetTime(j),
                                   fSignalEvent->GetSignal(i)->GetData(j));
            } else if (xStrip == 0) {
                for (int j = 0; j < fSignalEvent->GetSignal(i)->GetNumberOfPoints(); j++) {
                    fHistoYZ->Fill(yRead, fSignalEvent->GetSignal(i)->GetTime(j),
                                   fSignalEvent->GetSignal(i)->GetData(j));
                }
            }
        }
 
        // Only the maximum is plotted for the XY readout
        maxIndex = fSignalEvent->GetSignal(i)->GetMaxIndex();
        charge = fSignalEvent->GetSignal(i)->GetData(maxIndex);
 
        for (int px = 0; px < nPixels; px++) {
            // cout<< "X : " << module->GetPixelCenter( readoutChannel, px ).X()<<" Y
            // : "<<module->GetPixelCenter( readoutChannel, px ).Y()<<" Ch:
            // "<<charge<<endl;
            fHistoXY->Fill(module->GetPixelCenter(readoutChannel, px).X(),
                           module->GetPixelCenter(readoutChannel, px).Y(), charge);
        }
    }
}
 
TRestDetectorReadoutChannel* TRestDetectorReadoutEventViewer::GetChannel(int readoutChannel) {
    TRestDetectorReadoutPlane* plane = &(*fReadout)[0];
    for (size_t n = 0; n < plane->GetNumberOfModules(); n++) {
        if ((*plane)[n].GetChannel(readoutChannel) == nullptr) continue;
        return (*plane)[n].GetChannel(readoutChannel);
    }
 
    cout << "Readout channel " << readoutChannel << " not found" << endl;
    return nullptr;
}
 
TRestDetectorReadoutModule* TRestDetectorReadoutEventViewer::GetModule(int readoutChannel) {
    TRestDetectorReadoutPlane* plane = &(*fReadout)[0];
    for (int n = 0; n < fReadout->GetNumberOfModules(); n++) {
        if ((*plane)[n].GetChannel(readoutChannel) == nullptr) continue;
        return &(*plane)[n];
    }
 
    cout << "Readout channel " << readoutChannel << " not found" << endl;
    return nullptr;
}