+TTree* TStatToolkit::WriteStatusToTree(TObject* oStatusGr)
+{
+ //
+ // Create Tree with Integers for each status variable flag (warning, outlier, physacc).
+ // (by Patrick Reichelt)
+ //
+ // input: either a TMultiGraph with status of single variable, which
+ // was computed by TStatToolkit::MakeStatusMultGr(),
+ // or a TObjArray which contains up to 10 of such variables.
+ // example: TTree* statusTree = WriteStatusToTree( TStatToolkit::MakeStatusMultGr(tree, "tpcItsMatch:run", "", sCriteria.Data(), 0) );
+ // or : TTree* statusTree = TStatToolkit::WriteStatusToTree(oaMultGr);
+ //
+ // output tree: 1=flag is true, 0=flag is false, -1=flag was not computed.
+ // To be rewritten to the pcstream
+
+ TObjArray* oaMultGr = NULL;
+ Bool_t needDeletion=kFALSE;
+ if (oStatusGr->IsA() == TObjArray::Class()) {
+ oaMultGr = (TObjArray*) oStatusGr;
+ }
+ else if (oStatusGr->IsA() == TMultiGraph::Class()) {
+ oaMultGr = new TObjArray(); needDeletion=kTRUE;
+ oaMultGr->Add((TMultiGraph*) oStatusGr);
+ }
+ else {
+ Printf("WriteStatusToTree(): Error! 'oStatusGr' must be a TMultiGraph or a TObjArray of them!");
+ return 0;
+ }
+ // variables for output tree
+ const int nvarsMax=10;
+ const int ncritMax=5;
+ Int_t currentRun;
+ Int_t treevars[nvarsMax*ncritMax];
+ TString varnames[nvarsMax*ncritMax];
+ for (int i=0; i<nvarsMax*ncritMax; i++) treevars[i]=-1;
+
+ Printf("WriteStatusToTree(): writing following variables to TTree (maybe only subset of listed criteria filled)");
+ for (Int_t vari=0; vari<nvarsMax; vari++)
+ {
+ if (vari < oaMultGr->GetEntriesFast()) {
+ varnames[vari*ncritMax+0] = Form("%s_statisticOK", ((TMultiGraph*) oaMultGr->At(vari))->GetName());
+ varnames[vari*ncritMax+1] = Form("%s_Warning", ((TMultiGraph*) oaMultGr->At(vari))->GetName());
+ varnames[vari*ncritMax+2] = Form("%s_Outlier", ((TMultiGraph*) oaMultGr->At(vari))->GetName());
+ varnames[vari*ncritMax+3] = Form("%s_PhysAcc", ((TMultiGraph*) oaMultGr->At(vari))->GetName());
+ varnames[vari*ncritMax+4] = Form("%s_Extra", ((TMultiGraph*) oaMultGr->At(vari))->GetName());
+ }
+ else {
+ varnames[vari*ncritMax+0] = Form("dummy");
+ varnames[vari*ncritMax+1] = Form("dummy");
+ varnames[vari*ncritMax+2] = Form("dummy");
+ varnames[vari*ncritMax+3] = Form("dummy");
+ varnames[vari*ncritMax+4] = Form("dummy");
+ }
+ cout << " " << varnames[vari*ncritMax+0].Data() << " " << varnames[vari*ncritMax+1].Data() << " " << varnames[vari*ncritMax+2].Data() << " " << varnames[vari*ncritMax+3].Data() << " " << varnames[vari*ncritMax+4].Data() << endl;
+ }
+
+ TTree* statusTree = new TTree("statusTree","statusTree");
+ statusTree->Branch("run", ¤tRun );
+ statusTree->Branch(varnames[ 0].Data(), &treevars[ 0]);
+ statusTree->Branch(varnames[ 1].Data(), &treevars[ 1]);
+ statusTree->Branch(varnames[ 2].Data(), &treevars[ 2]);
+ statusTree->Branch(varnames[ 3].Data(), &treevars[ 3]);
+ statusTree->Branch(varnames[ 4].Data(), &treevars[ 4]);
+ statusTree->Branch(varnames[ 5].Data(), &treevars[ 5]);
+ statusTree->Branch(varnames[ 6].Data(), &treevars[ 6]);
+ statusTree->Branch(varnames[ 7].Data(), &treevars[ 7]);
+ statusTree->Branch(varnames[ 8].Data(), &treevars[ 8]);
+ statusTree->Branch(varnames[ 9].Data(), &treevars[ 9]);
+ statusTree->Branch(varnames[10].Data(), &treevars[10]);
+ statusTree->Branch(varnames[11].Data(), &treevars[11]);
+ statusTree->Branch(varnames[12].Data(), &treevars[12]);
+ statusTree->Branch(varnames[13].Data(), &treevars[13]);
+ statusTree->Branch(varnames[14].Data(), &treevars[14]);
+ statusTree->Branch(varnames[15].Data(), &treevars[15]);
+ statusTree->Branch(varnames[16].Data(), &treevars[16]);
+ statusTree->Branch(varnames[17].Data(), &treevars[17]);
+ statusTree->Branch(varnames[18].Data(), &treevars[18]);
+ statusTree->Branch(varnames[19].Data(), &treevars[19]);
+ statusTree->Branch(varnames[20].Data(), &treevars[20]);
+ statusTree->Branch(varnames[21].Data(), &treevars[21]);
+ statusTree->Branch(varnames[22].Data(), &treevars[22]);
+ statusTree->Branch(varnames[23].Data(), &treevars[23]);
+ statusTree->Branch(varnames[24].Data(), &treevars[24]);
+ statusTree->Branch(varnames[25].Data(), &treevars[25]);
+ statusTree->Branch(varnames[26].Data(), &treevars[26]);
+ statusTree->Branch(varnames[27].Data(), &treevars[27]);
+ statusTree->Branch(varnames[28].Data(), &treevars[28]);
+ statusTree->Branch(varnames[29].Data(), &treevars[29]);
+ statusTree->Branch(varnames[30].Data(), &treevars[30]);
+ statusTree->Branch(varnames[31].Data(), &treevars[31]);
+ statusTree->Branch(varnames[32].Data(), &treevars[32]);
+ statusTree->Branch(varnames[33].Data(), &treevars[33]);
+ statusTree->Branch(varnames[34].Data(), &treevars[34]);
+ statusTree->Branch(varnames[35].Data(), &treevars[35]);
+ statusTree->Branch(varnames[36].Data(), &treevars[36]);
+ statusTree->Branch(varnames[37].Data(), &treevars[37]);
+ statusTree->Branch(varnames[38].Data(), &treevars[38]);
+ statusTree->Branch(varnames[39].Data(), &treevars[39]);
+ statusTree->Branch(varnames[40].Data(), &treevars[40]);
+ statusTree->Branch(varnames[41].Data(), &treevars[41]);
+ statusTree->Branch(varnames[42].Data(), &treevars[42]);
+ statusTree->Branch(varnames[43].Data(), &treevars[43]);
+ statusTree->Branch(varnames[44].Data(), &treevars[44]);
+ statusTree->Branch(varnames[45].Data(), &treevars[45]);
+ statusTree->Branch(varnames[46].Data(), &treevars[46]);
+ statusTree->Branch(varnames[47].Data(), &treevars[47]);
+ statusTree->Branch(varnames[48].Data(), &treevars[48]);
+ statusTree->Branch(varnames[49].Data(), &treevars[49]);
+
+ // run loop
+ Double_t graphX; // x-position of marker (0.5, 1.5, ...)
+ Double_t graphY; // if >0 -> warning/outlier/physacc! if =-0.5 -> no warning/outlier/physacc
+ TList* arrRuns = (TList*) ((TGraph*) ((TMultiGraph*) oaMultGr->At(0))->GetListOfGraphs()->At(0))->GetXaxis()->GetLabels();
+ //'TAxis->GetLabels()' returns THashList of TObjString, but using THashList gives compilation error "... incomplete type 'struct THashList' "
+ for (Int_t runi=0; runi<arrRuns->GetSize(); runi++)
+ {
+ currentRun = atoi( arrRuns->At(runi)->GetName() );
+ //Printf(" runi=%2i, name: %s \t run number: %i", runi, arrRuns->At(runi)->GetName(), currentRun);
+
+ // status variable loop
+ for (Int_t vari=0; vari<oaMultGr->GetEntriesFast(); vari++)
+ {
+ TMultiGraph* multGr = (TMultiGraph*) oaMultGr->At(vari);
+
+ // criteria loop
+ // the order is given by TStatToolkit::MakeStatusMultGr().
+ // criterion #1 is 'statisticOK' and mandatory, the rest is optional. (#0 is always True, thus skipped)
+ for (Int_t criti=1; criti<multGr->GetListOfGraphs()->GetEntries(); criti++)
+ {
+ TGraph* grCriterion = (TGraph*) multGr->GetListOfGraphs()->At(criti);
+ graphX = -1, graphY = -1;
+ grCriterion->GetPoint(runi, graphX, graphY);
+ treevars[(vari)*ncritMax+(criti-1)] = (graphY>0)?1:0;
+ }
+ }
+ statusTree->Fill();
+ }
+
+ if (needDeletion) delete oaMultGr;
+
+ return statusTree;
+}
+
+
+void TStatToolkit::MakeSummaryTree(TTree* treeIn, TTreeSRedirector *pcstream, TObjString & sumID, TCut &selection){
+ //
+ // Make a summary tree for the input tree
+ // For the moment statistic works only for the primitive branches (Float/Double/Int)
+ // Extension recursive version planned for graphs a and histograms
+ //
+ // Following statistics are exctracted:
+ // - Standard: mean, meadian, rms
+ // - LTM robust statistic: mean60, rms60, mean90, rms90
+ // Parameters:
+ // treeIn - input tree
+ // pctream - Output redirector
+ // sumID - ID as will be used in output tree
+ // selection - selection criteria define the set of entries used to evaluat statistic
+ //
+ TObjArray * brArray = treeIn->GetListOfBranches();
+ Int_t tEntries= treeIn->GetEntries();
+ Int_t nBranches=brArray->GetEntries();
+ TString treeName = treeIn->GetName();
+ treeName+="Summary";
+
+ (*pcstream)<<treeName.Data()<<"entries="<<tEntries;
+ (*pcstream)<<treeName.Data()<<"ID.="<<&sumID;
+
+ TMatrixD valBranch(nBranches,7);
+ for (Int_t iBr=0; iBr<nBranches; iBr++){
+ TString brName= brArray->At(iBr)->GetName();
+ Int_t entries=treeIn->Draw(brArray->At(iBr)->GetName(),selection);
+ if (entries==0) continue;
+ Double_t median, mean, rms, mean60,rms60, mean90, rms90;
+ mean = TMath::Mean(entries,treeIn->GetV1());
+ median= TMath::Median(entries,treeIn->GetV1());
+ rms = TMath::RMS(entries,treeIn->GetV1());
+ TStatToolkit::EvaluateUni(entries, treeIn->GetV1(), mean60,rms60,TMath::Min(TMath::Max(2., 0.60*entries),Double_t(entries)));
+ TStatToolkit::EvaluateUni(entries, treeIn->GetV1(), mean90,rms90,TMath::Min(TMath::Max(2., 0.90*entries),Double_t(entries)));
+ valBranch(iBr,0)=mean;
+ valBranch(iBr,1)=median;
+ valBranch(iBr,2)=rms;
+ valBranch(iBr,3)=mean60;
+ valBranch(iBr,4)=rms60;
+ valBranch(iBr,5)=mean90;
+ valBranch(iBr,6)=rms90;
+ (*pcstream)<<treeName.Data()<<
+ brName+"_Mean="<<valBranch(iBr,0)<<
+ brName+"_Median="<<valBranch(iBr,1)<<
+ brName+"_RMS="<<valBranch(iBr,2)<<
+ brName+"_Mean60="<<valBranch(iBr,3)<<
+ brName+"_RMS60="<<valBranch(iBr,4)<<
+ brName+"_Mean90="<<valBranch(iBr,5)<<
+ brName+"_RMS90="<<valBranch(iBr,6);
+ }
+ (*pcstream)<<treeName.Data()<<"\n";
+}
+
+
+
+TMultiGraph* TStatToolkit::MakeStatusLines(TTree * tree, const char * expr, const char * cut, const char * alias)
+{
+ //
+ // Create status lines for trending using MakeGraphSparse(), very similar to MakeStatusMultGr().
+ // (by Patrick Reichelt)
+ //
+ // format of expr : varname:xaxis (e.g. meanTPCncl:run, but 'varname' can be any string that you need for seach-and-replace)
+ // format of cut : char like in TCut
+ // format of alias: varname_OutlierMin:varname_OutlierMax:varname_WarningMin:varname_WarningMax:varname_PhysAccMin:varname_PhysAccMax:varname_RobustMean
+ //
+ TObjArray* oaVar = TString(expr).Tokenize(":");
+ if (oaVar->GetEntries()<2) {
+ printf("Expression has to be of type 'varname:xaxis':\t%s\n", expr);
+ return 0;
+ }
+ char varname[50];
+ char var_x[50];
+ snprintf(varname,50,"%s", oaVar->At(0)->GetName());
+ snprintf(var_x ,50,"%s", oaVar->At(1)->GetName());
+ //
+ TString sAlias(alias);
+ if (sAlias.IsNull()) { // alias for default usage set here:
+ sAlias = "varname_OutlierMin:varname_OutlierMax:varname_WarningMin:varname_WarningMax:varname_PhysAccMin:varname_PhysAccMax:varname_RobustMean";
+ }
+ sAlias.ReplaceAll("varname",varname);
+ TObjArray* oaAlias = TString(sAlias.Data()).Tokenize(":");
+ if (oaAlias->GetEntries()<2) {
+ printf("Alias must have 2-7 arguments:\t%s\n", alias);
+ return 0;
+ }
+ char query[200];
+ TMultiGraph* multGr = new TMultiGraph();
+ Int_t colArr[7] = {kRed, kRed, kOrange, kOrange, kGreen+1, kGreen+1, kGray+2};
+ const Int_t ngr = oaAlias->GetEntriesFast();
+ for (Int_t i=0; i<ngr; i++){
+ snprintf(query,200, "%s:%s", oaAlias->At(i)->GetName(), var_x);
+ multGr->Add( (TGraphErrors*) TStatToolkit::MakeGraphSparse(tree,query,cut,29,colArr[i],1.5) );
+ }
+ //
+ multGr->SetName(varname);
+ multGr->SetTitle(varname);
+ delete oaVar;
+ delete oaAlias;
+ return multGr;
+}
+
+
+TH1* TStatToolkit::DrawHistogram(TTree * tree, const char* drawCommand, const char* cuts, const char* histoname, const char* histotitle, Int_t nsigma, Float_t fraction )