trackplotter.cxx File Reference

#include "trackplotter.h"
#include "TROOT.h"
#include "TApplication.h"
#include "TGTableLayout.h"
#include "TGFrame.h"
#include "TList.h"
#include "TGWindow.h"
#include "TRootEmbeddedCanvas.h"
#include "TEllipse.h"
#include "TBox.h"
#include "TArrow.h"
#include "TPolyLine.h"
#include "TGButton.h"
#include "TCanvas.h"
#include "TRandom.h"
#include "TGDoubleSlider.h"
#include "TGaxis.h"
#include "TColor.h"
#include "Riostream.h"
#include "TEnv.h"
#include "PlotValNtuple.h"

Go to the source code of this file.

Functions
int	main (int argc, char **argv)

Function Documentation

main()

int main	(	int	argc,
		char **	argv )

Definition at line 569 of file trackplotter.cxx.

                                {
//    TApplication theApp("App", &argc, argv);
 
    if (gROOT->IsBatch()) {
        fprintf(stderr, "%s: cannot run in batch mode\n", argv[0]);
        return 1;
    }
    // -----------------------------------
    // Read the current TrackPlotter setup
 
    // get the TEnv object
    TEnv* trackPlotterSetup = new TEnv("trackplotter.conf");
    // get the ntuple filename
    //char *filename = new char[80];
    const char* filename = trackPlotterSetup->GetValue("TrackingNtupleFileName","TrkValidation.root");
    // track ntuple directory and tree name
    //char *trackNtupleDir = new char[80];
    const char* trackNtupleDir = trackPlotterSetup->GetValue("TrackingNtupleDirectory","Validation");
    //char *trackNtupleTree = new char[80];
    const char* trackNtupleTree = trackPlotterSetup->GetValue("TrackingNtupleTree","Tracks");
 
    // use DAF data?
    bool doDAFdata = false;
    doDAFdata = trackPlotterSetup->GetValue("PlotDAFdata",0);
 
    // use RIO data?
    bool doRIOdata = true;
    doRIOdata = trackPlotterSetup->GetValue("PlotPrepRawData", 1);
    //std::cout << "use RIO: -" << trackPlotterSetup->GetValue("PlotPrepRawData", "True") <<"-" << std::endl;
    // rio ntuple directory and tree name
    //char *rioNtupleDir = new char[80];
    const char* rioNtupleDir = trackPlotterSetup->GetValue("PrepRawDataNtupleDirectory","Validation");
    //char *trackNtupleTree = new char[80];
    const char* rioNtupleTree = trackPlotterSetup->GetValue("PrepRawDataNtupleTree","TRT_RIOs");
 
//     trackPlotterSetup->SetValue("TrackingNtupleFileName", filename);
//     trackPlotterSetup->SetValue("TrackingNtupleDirectory", trackNtupleDir);
//     trackPlotterSetup->SetValue("TrackingNtupleTree", trackNtupleTree);
//     trackPlotterSetup->SetValue("PlotDAFdata", doDAFdata);
//     trackPlotterSetup->SetValue("PlotPrepRawData", doRIOdata);
//     trackPlotterSetup->SetValue("PrepRawDataNtupleDirectory", rioNtupleDir);
//     trackPlotterSetup->SetValue("PrepRawDataNtupleTree", rioNtupleTree);
//     trackPlotterSetup->Save();
    trackPlotterSetup->Print();
 
    TColor *color_301 = new TColor(301,0.7,0.0,0.0,"");
    TColor *color_302 = new TColor(302,0.0,0.7,0.0,"");
    TColor *color_303 = new TColor(303,0.0,0.0,0.7,"");
    TColor *color_306 = new TColor(306,0.7,0.0,0.7,"");
    TColor *color_310 = new TColor(310,0.8,0.8,0.8,"");
    TColor *color_311 = new TColor(311,0.55,0.0,0.0,"");
 
//    int trackNumber = 0;
    if ( argc >= 2 ){ // argc should be 2 for correct execution
        //delete[] filename;
        filename = argv[1];
    }else {
        // We print argv[0] assuming it is the program name
        std::cout<<"usage: "<< argv[0] <<" <filename>"<< std::endl;
        //filename = "TrkValidation.root";        
//        trackNumber = atoi(argv[2]);
    }
    //if ( argc >= 3 ){ 
    //    doDAFdata = bool(argv[2]);
    //}
    std::cout<<"Using file "<<filename << (doRIOdata ? " including PrepRawData" : "") << (doDAFdata ? " with DAF information" : "") << std::endl;
 
    TApplication theApp("App", &argc, argv); 
    TFile *f = new TFile(filename);
    if (!f) {
        std::cout << "ERROR: could not find " << filename << std::endl;
        return 1;
    }
    
    char rootFileName[100];
    sprintf(rootFileName,"%s:/%s",filename, trackNtupleDir);
    //std::cout<<"cd:  "<<rootFileName << std::endl;
    f->cd(rootFileName);
    
    TTree *tree = (TTree*)gDirectory->Get(trackNtupleTree);
    if (!tree) {
        std::cout << "ERROR: could not get the tree: " << filename << ":/" << trackNtupleDir << "/" << trackNtupleTree << std::endl;
            return 1;    
    }
    //delete filename;
    // get the ntuple with the track data
    TrackDataNtupleTree* trackNtuple = new TrackDataNtupleTree(tree, doDAFdata);
 
    //std::cout<<"got TrackDataNtupleTree"<< std::endl;
    // get the RIOs
    
    sprintf(rootFileName,"%s:/%s",filename, rioNtupleDir);
    //f->cd("TrkValidation.root:/FitterValidation");
    //std::cout<<"cd:  "<<rootFileName << std::endl;
    f->cd(rootFileName);
    
    TTree *RIOtree = 0;
    TrkValidationNtupleRIOs* rioNtuple = 0;
    if (doRIOdata) {
        RIOtree = (TTree*)gDirectory->Get(rioNtupleTree);
        if (!RIOtree) {
            std::cout << "ERROR: could not get the tree: " << filename << ":/" << rioNtupleDir << "/" << rioNtupleTree << std::endl;
                return 1;    
        }
        // get the ntuple with the rio data
        rioNtuple = new TrkValidationNtupleRIOs(RIOtree);
    }
    //delete filename;
 
    PlotValNtuple theNtuplePlotter( *trackNtuple, rioNtuple);
 
    Viewer viewer(gClient->GetRoot(), &theNtuplePlotter);
 
    theApp.Run();
 
    delete color_301; delete color_302;
    delete color_303; delete color_306;
    delete color_310; delete color_311;
    delete trackNtuple;
    delete rioNtuple;
    return 0;
}