CbmL1StsTrackFinder.cxx

Go to the documentation of this file.

/*
 *====================================================================
 *
 *  CBM Level 1 Reconstruction
 *
 *  Authors: I.Kisel,  S.Gorbunov
 *
 *  e-mail : ikisel@kip.uni-heidelberg.de
 *
 *====================================================================
 *
 *  CbmL1StsTrackFinder source file
 *
 *====================================================================
 */
#include "CbmL1StsTrackFinder.h"
 
#include "CbmKFMath.h"
#include "CbmMvdHit.h"
#include "CbmStsHit.h"
#include "CbmStsTrack.h"
#include "FairHit.h"
#include "FairMCPoint.h"
#include "FairRootManager.h"
#include "L1Algo/L1Algo.h"
#include "TClonesArray.h"
 
#include <iostream>
#include <vector>
 
using std::cout;
using std::endl;
using std::vector;
 
// -----   Default constructor   -------------------------------------------
CbmL1StsTrackFinder::CbmL1StsTrackFinder()
{
    fName = "STS Track Finder L1";
}
// -------------------------------------------------------------------------
 
// -----   Destructor   ----------------------------------------------------
CbmL1StsTrackFinder::~CbmL1StsTrackFinder() {}
// -------------------------------------------------------------------------
 
// -----   Public method Init   --------------------------------------------
void CbmL1StsTrackFinder::Init() {}
// -------------------------------------------------------------------------
 
// -----   Public method DoFind   ------------------------------------------
Int_t CbmL1StsTrackFinder::DoFind()
{
    /*
      if ( !mHitArray || !hHitArray || !sHitArray ) {
      cout << "-E- CbmL1StsTrackFinder::DoFind: "
      << "Hit arrays missing! " << mHitArray << " " <<hHitArray<<" "<< sHitArray << endl;
      return -1;
      }
    */
    if (!fTracks) {
        cout << "-E- CbmL1StsTrackFinder::DoFind: "
             << "Track array missing! " << endl;
        return -1;
    }
 
    int ntracks = fTracks->GetEntries();
 
    CbmL1* L1 = CbmL1::Instance();
    if (!L1)
        return 0;
 
    L1->Reconstruct();
 
    for (vector<CbmL1Track>::iterator it = L1->vRTracks.begin(); it != L1->vRTracks.end(); ++it) {
        CbmL1Track& T = *it;
        new ((*fTracks)[ntracks]) CbmStsTrack();
        CbmStsTrack* t = L1_DYNAMIC_CAST<CbmStsTrack*>(fTracks->At(ntracks++));
        t->SetFlag(0);
        FairTrackParam *fpar = t->GetParamFirst(), *lpar = t->GetParamLast();
        CbmKFMath::CopyTC2TrackParam(fpar, T.T, T.C);
        CbmKFMath::CopyTC2TrackParam(lpar, T.TLast, T.CLast);
        t->SetChi2(T.chi2);
        t->SetNDF(T.NDF);
        t->SetPidHypo(T.T[4] >= 0 ? 211 : -211);
 
        for (vector<int>::iterator ih = it->StsHits.begin(); ih != it->StsHits.end(); ++ih) {
            CbmL1HitStore& h = L1->vHitStore[*ih];
            double zref = L1->algo->vStations[h.iStation].z[0];
            if (h.ExtIndex < 0) {
                CbmMvdHit tmp;
                tmp.SetZ(zref);
                t->AddMvdHit(-h.ExtIndex - 1, &tmp);
            } else {
                CbmStsHit tmp;
                tmp.SetZ(zref);
                t->AddStsHit(h.ExtIndex, &tmp);
            }
        }
    }
 
    return ntracks;
}
// -------------------------------------------------------------------------