beamtime/cosy-jan12/go4/STS/TSTSClusterizer.cxx (r4864/r3090)

Go to the documentation of this file.

#include "TSTSClusterizer.h"
#include "TMath.h"

TSTSClusterizer::TSTSClusterizer( Int_t det, Int_t side ) :
        fDetector( det ),
        fSide( side )
{
        fTimeTolerance = kDefTimeTolerance;
        fSpaceTolerance = kDefSpaceTolerance;
        fTimeWindow = kDefTimeWindow;
        fLatestDigiTime = 0;
        CheckTimeWindowGtTimeTol();
}



TSTSClusterizer::~TSTSClusterizer()
{

}



void TSTSClusterizer::CheckTimeWindowSufficient( const TSTSDigi & newDigi )
{
    if( - (newDigi.GetTime() - fLatestDigiTime ) + fTimeTolerance >= fTimeWindow ) {
        fTimeWindow = fTimeWindow * 2;
        printf("Error in TSTSClusterizer::CheckTimeWindowSufficient: Time window is too small. Increasing by factor of 2... New time window is %lld. Happened at time %lld\n", fTimeWindow, newDigi.GetTime() );
    }
    if( newDigi.GetTime() > fLatestDigiTime ) {
        fLatestDigiTime = newDigi.GetTime();
    }
}

void TSTSClusterizer::CheckTimeWindowGtTimeTol() {
    if( fTimeTolerance >= fTimeWindow ) {
        printf("Error in TSTSClusterizer: fTimeWindow is not larger than fTimeTolerance. Setting fTimeWindow = fTimeTolerance, which is %lld\n", fTimeTolerance );
        fTimeWindow = fTimeTolerance;
    }
}



Bool_t TSTSClusterizer::Match( const TSTSDigi & digi1, const TSTSDigi & digi2 ) const
{
        return ( TMath::Abs( digi1.GetTime() - digi2.GetTime() ) <= fTimeTolerance )
                        && ( TMath::Abs( digi1.GetStrip() - digi2.GetStrip() ) <= fSpaceTolerance );
}



Bool_t TSTSClusterizer::Match( const TSTSCluster & cluster, const TSTSDigi & digi ) const
{
        for( Int_t iDigi = 0; iDigi < cluster.GetNDigis(); iDigi++ ) {
                if( Match( digi, cluster.GetDigi( iDigi ) ) ) {
                        return kTRUE;
                }
        }
        return kFALSE;
}


/*
 * Expects that: cluster exists, itDigi valid
 */
void TSTSClusterizer::MoveDigiToCluster( TSTSCluster & cluster, const std::deque<TSTSDigi>::iterator & itDigi )
{
   cluster.AddDigi(*itDigi);
   fDigis.erase(itDigi);
}



void TSTSClusterizer::InsertDigiSorted( const TSTSDigi & digi )
{
    std::deque<TSTSDigi>::iterator it = fDigis.end();
    while( it != fDigis.begin() && ( it - 1 )->GetTime() > digi.GetTime() ) {
        --it;
    }
    fDigis.insert( it, digi );
}


void TSTSClusterizer::PrintBuffer() const
{
    printf( "===> Printing clusterizer buffer. Side=%c, Det=%d\n", TSTSTopology::GetSideName( fSide ), fDetector );
    for( std::deque<TSTSDigi>::const_iterator it = fDigis.begin(); it != fDigis.end(); ++it) {
        it->Print();
    }
}



void TSTSClusterizer::FeedDigi( TSTSDigi const & digi )
{
    InsertDigiSorted( digi );
    CheckTimeWindowSufficient( digi );
    // if( fTimeWindow > 10000 && fSide == 1 ) PrintBuffer();
}



Bool_t TSTSClusterizer::IsDigiBufferFilled( Bool_t noTrigger ) const
{
    if( fDigis.size() == 0 ) {
        return kFALSE;
    }
    if( noTrigger ) {
        if( fDigis.back().GetTime() - fDigis.front().GetTime() < fTimeWindow ) {
            return kFALSE;
        }
    }
    return kTRUE;
}



/*
 * When noTrigger is kFALSE, has to be called only when all Digis are fed.
 */
Bool_t TSTSClusterizer::FetchCluster( Bool_t noTrigger, TSTSCluster & cluster )
{
    cluster.Reset();
    if( ! IsDigiBufferFilled( noTrigger ) ) {
        return kFALSE;
    }

    Long64_t clustMaxTime = fDigis.front().GetTime();
    MoveDigiToCluster( cluster, fDigis.begin() );

    Bool_t matchingDigiFound;
    do {
        matchingDigiFound = kFALSE;
        std::deque<TSTSDigi>::iterator fitDigiIter = fDigis.begin();

        while( fitDigiIter != fDigis.end() && fitDigiIter->GetTime() - clustMaxTime <= fTimeTolerance ) {
            if( Match( cluster, *fitDigiIter ) ) {
                if( noTrigger ) {
                    if( (fDigis.back().GetTime() - fitDigiIter->GetTime()) < fTimeTolerance ) {
                        fTimeWindow = fTimeWindow * 2;
                        printf("Error in TClusterizer::FetchCluster: Too small time window. Increasing by factor of 2. New fTimeWindow is %lld\n", fTimeWindow );
                    }
                }
                clustMaxTime = fitDigiIter->GetTime();
                MoveDigiToCluster( cluster, fitDigiIter );
                matchingDigiFound = kTRUE;
                break;
            }
            fitDigiIter++;
        }

    } while (matchingDigiFound );

    return kTRUE;
}



void TSTSClusterizer::SetTimeTolerance( Long64_t timeTolerance )
{
    fTimeTolerance = timeTolerance;
    CheckTimeWindowGtTimeTol();
}



void TSTSClusterizer::SetTimeWindow( Long64_t timeWindow )
{
    fTimeWindow = timeWindow;
    CheckTimeWindowGtTimeTol();
}