printsttcoords.C

Go to the documentation of this file.

#include "TStopwatch.h"
#include "TTree.h"
#include "TFile.h"
#include "TClonesArray.h"
#include "basefunctions.hxx"
//#include "FairRunAna.h"
//#include "FairParRootFileIo.h"
//#include "FairRuntimeDb.h"
//#include "PndSttTrackFinderReal.h"
//#include "PndMCTrack.h"
//#include "PndSttHit.h"
//#include "PndTrack.h"

// xy-projection of time-based MVD and STT Hits
// m.mertens@fz-juelich.de

int printsttcoords() {
        gROOT->Reset();
        // Verbosity level (0=quiet, 1=event level, 2=track level, 3=debug)
        Int_t iVerbose = 0;

        TStopwatch timer;
        timer.Start();
        gROOT->LoadMacro("$VMCWORKDIR/gconfig/rootlogon.C");
        rootlogon();
        //gSystem->Load("libSttMvdTracking");

        TString simFileName = "Sim_Dpm_500.root";
        TString parFileName = "Sim_Dpm_500_params.root";
        TString digiFileName = "Sim_Dpm_500_digi.root";
        //TString recoFileName = "Sim_Dpm_500_reco.root";
        TString outFileName = "Sim_Dpm_500_printsttcoords.root";

        TFile filedigi(digiFileName.Data());
        //TFile filereco(recoFileName.Data());

        TTree *treedigi = (TTree*) filedigi.Get("pndsim");
        TClonesArray *sttsortedhits = new TClonesArray("PndSttHit");
        treedigi->SetBranchAddress("STTSortedHits",&sttsortedhits);

        FairRunAna *fRun= new FairRunAna();
        fRun->SetInputFile(simFileName.Data());
        //fRun->AddFriend(recoFile.Data());
        fRun->AddFriend(digiFileName.Data());
        //fRun->AddFriend(trackF.Data());
        fRun->SetOutputFile(outFileName.Data());
        //fRun->RunWithTimeStamps();
        FairRuntimeDb* rtdb = fRun->GetRuntimeDb();
        FairParRootFileIo* parInput1 = new FairParRootFileIo();
        parInput1->open(parFileName.Data());
        rtdb->setFirstInput(parInput1);

        //works around the "geometry not supported by map" error
        PndSttTrackFinderReal* sttTrackFinder = new PndSttTrackFinderReal(0);
        PndSttFindTracks* sttFindTracks = new PndSttFindTracks("Track Finder", "FairTask", sttTrackFinder, iVerbose);
        sttFindTracks->AddHitCollectionName("STTHit", "STTPoint");
        //sttFindTracks->SetPersistence(kFALSE);
        fRun->AddTask(sttFindTracks);
        fRun->Init();
        PndGeoSttPar* fSttParameters = (PndGeoSttPar*) rtdb->getContainer("PndGeoSttPar");
        PndSttMapCreator* mapper = new PndSttMapCreator(fSttParameters);
        TClonesArray* tubearray = mapper->FillTubeArray();

        PrintSttStats(tubearray);

        /*
        for (Int_t entrynumber = 0; entrynumber < treedigi->GetEntriesFast(); entrynumber++) {
                treedigi->GetEntry(entrynumber);
                for (Int_t clonesnumber = 0; clonesnumber < sttsortedhits->GetEntriesFast(); clonesnumber++) {
                        PndSttHit* myHit = (PndSttHit*) sttsortedhits->At(clonesnumber);
                        TVector3 hitPos;
                        myHit->Position(hitPos);
                        Int_t tubeID = myHit->GetTubeID();
                        PndSttTube* myTube = (PndSttTube*) tubearray->At(tubeID);
                        TVector3 tubePos = myTube->GetPosition();
                        cout << entrynumber << " " << clonesnumber << " " << myHit->GetTimeStamp() << " " << tubeID << " " << hitPos.X() << " " << hitPos.Y() << " - " << tubePos.X() << " " << tubePos.Y() << endl;
                }
        }
        */
  return 0;
}