TRestDetectorHitsReductionProcess.cxx

//
 
#include "TRestDetectorHitsReductionProcess.h"
 
using namespace std;
 
ClassImp(TRestDetectorHitsReductionProcess);
 
TRestDetectorHitsReductionProcess::TRestDetectorHitsReductionProcess() { Initialize(); }
 
TRestDetectorHitsReductionProcess::TRestDetectorHitsReductionProcess(const char* configFilename) {
    Initialize();
    if (LoadConfigFromFile(configFilename) == -1) {
        LoadDefaultConfig();
    }
}
 
TRestDetectorHitsReductionProcess::~TRestDetectorHitsReductionProcess() {}
 
void TRestDetectorHitsReductionProcess::LoadDefaultConfig() {
    SetName("hitsReductionProcess");
    SetTitle("Default config");
 
    fStartingDistance = 0.5;
    fMinimumDistance = 3;
    fDistanceFactor = 1.5;
    fMaxNodes = 30;
}
 
void TRestDetectorHitsReductionProcess::Initialize() {
    SetSectionName(this->ClassName());
    SetLibraryVersion(LIBRARY_VERSION);
 
    fInputHitsEvent = nullptr;
    fOutputHitsEvent = nullptr;
}
 
void TRestDetectorHitsReductionProcess::LoadConfig(const string& configFilename, const string& name) {
    if (LoadConfigFromFile(configFilename, name) == -1) LoadDefaultConfig();
}
 
void TRestDetectorHitsReductionProcess::InitProcess() {}
 
TRestEvent* TRestDetectorHitsReductionProcess::ProcessEvent(TRestEvent* inputEvent) {
    fInputHitsEvent = (TRestDetectorHitsEvent*)inputEvent;
    fOutputHitsEvent = fInputHitsEvent;
 
    Int_t initialHits = fOutputHitsEvent->GetNumberOfHits();
 
    // Reducing the hits
    TRestHits* hits = fOutputHitsEvent->GetHits();
 
    Double_t distance = fStartingDistance;
    while (distance < fMinimumDistance || hits->GetNumberOfHits() > fMaxNodes) {
        Bool_t merged = true;
        while (merged) {
            merged = false;
            for (unsigned int i = 0; i < hits->GetNumberOfHits(); i++) {
                for (unsigned int j = i + 1; j < hits->GetNumberOfHits(); j++) {
                    if (hits->GetDistance2(i, j) < distance * distance) {
                        hits->MergeHits(i, j);
                        merged = true;
                    }
                }
            }
        }
        distance *= fDistanceFactor;
    }
 
    Int_t finalHits = fOutputHitsEvent->GetNumberOfHits();
 
    if (this->GetVerboseLevel() == TRestStringOutput::REST_Verbose_Level::REST_Debug) {
        cout << "TRestDetectorHitsReductionProcess : Initial number of hits : " << initialHits << endl;
        cout << "TRestDetectorHitsReductionProcess : Final number of hits : " << finalHits << endl;
    }
 
    /*
       cout << "output event" << endl;
       cout << "+++++++++++++++++" << endl;
       fOutputHitsEvent->PrintEvent();
       cout << "+++++++++++++++++" << endl;
       getchar();
       */
 
    // cout << "Number output of tracks : " <<
    // fOutputTrackEvent->GetNumberOfTracks() << endl;
 
    return fOutputHitsEvent;
}
 
void TRestDetectorHitsReductionProcess::EndProcess() {}
 
void TRestDetectorHitsReductionProcess::InitFromConfigFile() {
    fStartingDistance = GetDblParameterWithUnits("startingDistance");
    fMinimumDistance = GetDblParameterWithUnits("minimumDistance");
    fDistanceFactor = StringToDouble(GetParameter("distanceStepFactor"));
    fMaxNodes = StringToDouble(GetParameter("maxNodes"));
}