PndFtsHoughSpacePeak.h

Go to the documentation of this file.

#ifndef PndFtsHoughSpacePeak_H
#define PndFtsHoughSpacePeak_H

#include "PndFtsHoughTrackerTask.h"

// Root Class Headers ----------------
#include "PndTrackCand.h"
#include "Rtypes.h"                     // for Double_t, Int_t, etc
#include "FairLogger.h" // for FairLogger, MESSAGE_ORIGIN
#include <set>





class PndFtsHoughSpacePeak : public TObject {
public:
        static const Int_t noVal = -1;

        // Constructors / Destructors ---------
        PndFtsHoughSpacePeak( Int_t height = noVal, Int_t firstBin = noVal, Int_t firstHitIdx = noVal );
        ~PndFtsHoughSpacePeak();

        inline void resetBins();
        inline void replaceBins( Int_t height, Int_t firstBin, Int_t firstHitIdx );
        inline void addBin(Int_t binNumber, Int_t hitIdx);

        void setFinished(Bool_t newVal){ fFinished = newVal; };

        Bool_t isFinished() const{ return fFinished; };
        Int_t getHeight() const{ return fHeight; };

        const std::set<Int_t>& getBins() const { return fBins; };
        const std::set<Int_t>& getHitIds() const { return fHitIds; };

        inline Bool_t binsOverlapWith(const PndFtsHoughSpacePeak& toCheck);
        inline void mergeWith(const PndFtsHoughSpacePeak& toAdd);


private:
        std::set< Int_t > fBins; // globalbins belonging to the peak
        std::set< Int_t > fHitIds; // hit indices belonging to the peak
        Int_t fHeight; // height of peak
        Bool_t fFinished;

        ClassDef(PndFtsHoughSpacePeak, 1);

};

void PndFtsHoughSpacePeak::addBin(Int_t binNumber, Int_t hitIdx){
        fBins.insert(binNumber);
        fHitIds.insert(hitIdx);
}

void PndFtsHoughSpacePeak::replaceBins( Int_t height, Int_t firstBin, Int_t firstHitIdx ){
        resetBins();
        addBin(firstBin, firstHitIdx);
        fHeight = height;
}


void PndFtsHoughSpacePeak::resetBins(){
        fBins.clear();
        fHitIds.clear();
        fFinished = kFALSE;
        fHeight = noVal;
}

Bool_t PndFtsHoughSpacePeak::binsOverlapWith(const PndFtsHoughSpacePeak& toCheck){
        // return kTRUE iif toAdd has bins in common with this
        const std::set< Int_t > binsToSearch = toCheck.getBins();
        // try to find elements from toAdd in this
        for (std::set< Int_t >::iterator it = binsToSearch.begin(); it != binsToSearch.end(); ++it){
                std::set< Int_t >::iterator itFind = fBins.find( *it );
                if ( itFind != fBins.end() ) return kTRUE; // element from binsToSearch was found in this
        }
        return kFALSE; // none of the elements from binsToSearch was found in this
}

void PndFtsHoughSpacePeak::mergeWith(const PndFtsHoughSpacePeak& toAdd) {
        // merge bins
        const std::set< Int_t > binsToMerge = toAdd.getBins();
        fBins.insert( binsToMerge.begin(), binsToMerge.end() );
        // merge hit indices
        const std::set< Int_t > hitsToMerge = toAdd.getHitIds();
        fHitIds.insert(  hitsToMerge.begin(), hitsToMerge.end()  );
}

#endif