tools/auxi.C

Go to the documentation of this file.

// **** some auxilliary functions in auxtut.C ****
// - FairRunAna* initrun(TString prefix, TString outfile, int min=-1, int max=-1) --> Init FairRunAna
// - plotmyhistos()               --> Plots all histograms in current TDirectory on a autosized canvas
// - writemyhistos()              --> Writes all histos in current TFile
// - fillM(RhoCandList l, TH1* h) --> Fill mass histogram h with masses of candidates in l
// - RemoveGeoManager()           --> Temporary fix for error on macro exit
// **** some auxilliary functions in auxtut.C ****

#include "TGeoManager.h"

void CloseGeoManager()
{
  if (gROOT->GetVersionInt() >= 60602) {
    gGeoManager->GetListOfVolumes()->Delete();
    gGeoManager->GetListOfShapes()->Delete();
    delete gGeoManager;
  }
}

bool checkfile(TString fn)
{
        bool fileok=true;
        TFile fff(fn);
        if (fff.IsZombie()) fileok=false;
        TTree *t=(TTree*)fff.Get("pndsim");
        if (t==0x0) fileok=false;

        if (!fileok) cout <<"Skipping broken file '"<<fn<<"'"<<endl;
        return fileok;
}

FairRunAna* initrun(TString prefix, TString outfile, int min=-1, int max=-1)
{
        // *** the files coming from the simulation
        TString inPidFile  = prefix+"_pid.root";    // this file contains the PndPidCandidates and McTruth
        TString inParFile  = prefix+"_par.root";

        // *** PID table with selection thresholds; can be modified by the user
        TString pidParFile = TString(gSystem->Getenv("VMCWORKDIR"))+"/macro/params/all.par";

        // *** initialization
        FairLogger::GetLogger()->SetLogToFile(kFALSE);
        FairRunAna* fRun = new FairRunAna();
        FairRuntimeDb* rtdb = fRun->GetRuntimeDb();
        fRun->SetSource(new FairFileSource(inPidFile));

        // *** setup parameter database
        FairParRootFileIo* parIO = new FairParRootFileIo();
        parIO->open(inParFile);
        FairParAsciiFileIo* parIOPid = new FairParAsciiFileIo();
        parIOPid->open(pidParFile.Data(),"in");

        rtdb->setFirstInput(parIO);
        rtdb->setSecondInput(parIOPid);
        rtdb->setOutput(parIO);

        fRun->SetOutputFile(outfile);

        return fRun;
}

void plotmyhistos(std::vector<TH1*> h, int maxy=700, double asp = 1.1)
{
        int N = h.size();
        if (N<=0) return;

        int nx=sqrt(N);
        int ny=nx;
        while(nx*ny<N) nx++;

        int dim = maxy/ny;
        if (dim>400) dim=400;

        TCanvas *c=new TCanvas("c","c",10,10,dim*nx*asp, dim*ny);

        c->Divide(nx,ny, 0.0005,0.0005);

        for (int i=0;i<N;++i)
        {
                c->cd(i+1);
                h[i]->Draw();
        }
}

void plotmyhistos(int maxy=700, double asp = 1.1)
{
    std::vector<TH1*> h;

    //somehow either one or the other is filled, depending on
    // whether a TFile was opened, or we write to one
    int Nkey = gDirectory->GetListOfKeys()->GetSize();
    int Nlst = gDirectory->GetList()->GetSize();

    if( Nkey==0 && Nlst==0) return;

    if (Nkey>0)
    {
        TIter next(gDirectory->GetListOfKeys());
        TKey *key;
        while ((key = (TKey*)next()))
        {
            TClass *cl = gROOT->GetClass(key->GetClassName());
            if (!cl->InheritsFrom("TH1")) continue;
            TH1 *hist = (TH1*)key->ReadObj();
            h.push_back(hist);
        }
    }
    else
    {
        TList *hl=gDirectory->GetList();
        for (int i=0;i<hl->GetSize();++i)
        {
            TString cn = hl->At(i)->ClassName();
            if (cn.BeginsWith("TH1") || cn.BeginsWith("TH2")) h.push_back((TH1*)hl->At(i));
        }
    }

    plotmyhistos(h, maxy, asp);
}

int writemyhistos(int maxy=800, double asp = 1.1)
{
        TList *hl=gDirectory->GetList();
        std::vector<TH1*> h;
        for (int i=0;i<hl->GetSize();++i)
        {
                TString cn = hl->At(i)->ClassName();
                if (cn.BeginsWith("TH1") || cn.BeginsWith("TH2")) h.push_back((TH1*)hl->At(i));
        }

        int N = h.size();
        if (N==0) return 0;

        for (int i=0;i<N;++i) h[i]->Write();

        return N;
}

int fillM(RhoCandList &l, TH1* h)
{
        for (int i=0;i<l.GetLength();++i) h->Fill(l[i]->M());

        return l.GetLength();
}

int fillP(RhoCandList &l, TH1* h)
{
        for (int i=0;i<l.GetLength();++i) h->Fill(l[i]->P());

        return l.GetLength();
}