runOnlineDisplayGeomTest.C File Reference

#include "basefunctions.hxx"

Go to the source code of this file.

Functions
int	runOnlineDisplayGeomTest ()

Function Documentation

int runOnlineDisplayGeomTest

(

)

PndGeoSttPar* fSttParameters = (PndGeoSttPar*) rtdb->getContainer("PndGeoSttPar");

/// works around the "geometry not supported by map" error /// HACK

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();

/////////////////////////////////////////////////////////

Definition at line 3 of file runOnlineDisplayGeomTest.C.

References c1, ctime, Double_t, DrawIsochrones(), filedigi(), filereco(), fRun, i, iVerbose, kDCH, kDRC, kDSK, kEMC, kFTS, kGEM, kHYP, kHYPG, kLUMI, kMDT, kMVD, kRPC, kSTT, kTOF, kTPC, rootlogon(), rtdb, rtime, timer, treedigi, and TString.

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

    enum DetectorId {
        kDCH,kDRC,kDSK,kEMC,kGEM,kLUMI,kMDT,kMVD,kRPC,kSTT,kTPC,kTOF,kFTS,kHYPG,kHYP};


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

        // Reco Parameters
        Double_t stthitlifetime = 200;

    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_streamdisplay.root";

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

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

    TTree *recotree = (TTree*) filereco.Get("pndsim");
    //TClonesArray *mvdstripreco = new TClonesArray("PndSdsHit");
    //recotree->SetBranchAddress("MVDHitsStrip",&mvdstripreco);
    //TClonesArray *mvdpixelreco = new TClonesArray("PndSdsHit");
    //recotree->SetBranchAddress("MVDHitsPixel",&mvdpixelreco);


    FairRunAna *fRun = new FairRunAna();
    fRun->SetInputFile(simFileName.Data());
    fRun->AddFriend(recoFileName.Data());
    fRun->AddFriend(digiFileName.Data());
    fRun->SetOutputFile(outFileName.Data());
    fRun->Init();

    FairRuntimeDb* rtdb = fRun->GetRuntimeDb();

    // load geometry parameters
    //PndSttMapCreator* mapper = new PndSttMapCreator(fSttParameters);
    //TClonesArray* tubearray = mapper->FillTubeArray();

    PndOnlineGeometryManager *online_geometry = new PndOnlineGeometryManager(rtdb, parFileName.Data());
    //online_geometry->LoadRTDB();



    PndOnlineHitProducer *online_hit_producer = new PndOnlineHitProducer();
    online_hit_producer->LoadStream(kSTT, "STTSortedHits");
    online_hit_producer->LoadStream(kMVD, "MVDHitsStrip");
    online_hit_producer->LoadStream(kMVD, "MVDHitsPixel");


    PndOnlineManager *online = new PndOnlineManager();
    online->AddHitProducer(online_hit_producer);
    online->AddGeometryManager(online_geometry);
    online->SetActiveDetector(kSTT,stthitlifetime);
    online->SetActiveDetector(kMVD,stthitlifetime);

    TClonesArray* tubearray = online_geometry->GetDetectorGeometry(kSTT);

    cerr << " size of the STT tube array = " << tubearray->GetEntriesFast() << endl;

    // load task
    PndOnlineSttTripletFinder *triplet_finder = new PndOnlineSttTripletFinder(online, tubearray);
    //triplet_finder->Init();
    online->AddTask((FairTask*)triplet_finder);

    // initialize display
    TCanvas* c1 = new TCanvas("c1");
    c1->Range(-42,-42,42,42);
    c1->SetCanvasSize(1200, 1200);
    TText* mytext = new TText();


    // event loop!
    int current_time = 0;
    //int delta_t = 5;
    int delta_t = 20;
    int final_time = 300;
    online->Init();

    while( current_time < final_time ) {
        //online->LoadHits( PndOnlineManager::kHESRRevolution );
        online->LoadHits( delta_t );
        //online->LoadHits( current_time );
        online->Process();
        //online->Clear();
        current_time += delta_t;


        //Draw the results
        c1->Clear();
        c1->cd();
        DrawDetector();
        cout << " Drawing stuff " << endl;
        DrawIsochrones(current_time-stthitlifetime, online->GetFairHitObjectList(kSTT), tubearray);
        //DrawFairHits(drawstackstriphits);
        //DrawFairHits(drawstackpixelhits);
        //DrawTripletTracks(innertriplets, midtriplets, 10);
        //DrawTripletTracks(innertriplets, outertriplets, 45);
        //DrawTripletTracks(midtriplets, outertriplets, 25);

        
        // now print out the triplets
        //online->PrintTracks();
        TObjArray *tracks = online->GetTrackObjectList();
        for(int i=0; i<tracks->GetEntriesFast(); ++i) {
            PndOnlineTrack *trk_ptr = (PndOnlineTrack *)(tracks->At(i));
            cout << " triplet cms;  ";  trk_ptr->Vertex().Print();
        }
        

        //online->GetTrackList().clear();

        mytext->DrawText(-42, -42, TString::Format("%d ns",current_time));
        c1->Update();
        //c1->Print("hitdisplay.png");
        c1->Print("hitdisplay.gif+15");
    }

    // -----   Finish   -------------------------------------------------------
    timer.Stop();
    Double_t rtime = timer.RealTime();
    Double_t ctime = timer.CpuTime();
    cout << endl << endl;
    cout << "Macro finished succesfully." << endl;
    cout << "Real time " << rtime << " s, CPU time " << ctime << " s" << endl;
    cout << endl;
    // ------------------------------------------------------------------------


  return 0;
}