rtang/XIAEventBuilder
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

GAGGPIDCutCreator.C can edit existing cuts

Browse Source
This commit is contained in:
nrb@Debain10 2022-03-18 21:14:43 -04:00
parent 9700accb0f
commit 46e5700a34
1 changed files with 74 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

87
GAGGPIDCutCreator.C
View File

@ -12,20 +12,24 @@
void GAGGPIDCutCreator(TString dataList,
TString saveFileName = "GAGGPIDCuts.root",
int detID = -100, // -100 = new, -X = append from X, X = only X
int peakRange=1200,
int tailRange=400,
bool isLogz = false,
TString gate = "",
TString treeName = "tree"
float frac = 0.3
){
printf("================ Graphic Cut Creator for GAGG ============== \n");
TChain * chain = new TChain(treeName);
TChain * chain = new TChain("tree");
chain->Add(dataList);
chain->GetListOfFiles()->Print();
Long64_t numEntries = chain->GetEntries();
printf(" total number of entries : %llu, using the first %.0f%% data.\n", numEntries, frac*100.);
TString varX, varY, tag;
gStyle->SetOptStat("neiou");
@ -36,17 +40,54 @@ void GAGGPIDCutCreator(TString dataList,
cCutCreator->Update();
if( isLogz ) cCutCreator->cd()->SetLogz();
TCutG * cut = NULL;
TObjArray * cutList = new TObjArray();
TString expression;
TString expression, gate;
TH2F * h[NGAGG];
int count = 0;
for (Int_t i = 0; i < NGAGG; i++) {
TFile * cutFile;
TCutG * cut = NULL;
TObjArray * cutList;
/// load the cutFile and load the cutList
int startID = 0;
int stopID = NGAGG;
if( detID != -100 ) {
cutFile = new TFile(saveFileName, "UPDATE");
bool listExist = cutFile->GetListOfKeys()->Contains("cutList");
if( !listExist ) {
cutList = new TObjArray();
}else{
cutList = (TObjArray*) cutFile->FindObjectAny("cutList");
int numCut = cutList->GetLast()+1;
printf("----- found %d cuts \n", numCut);
for( int k = 0; k < numCut; k++){
if( cutList->At(k) != NULL ){
printf("found a cut at %2d \n", k);
}else{
printf(" No cut at %2d \n", k);
}
}
}
if( detID < 0 ) startID = abs(detID);
if( detID >= 0 ){
startID = detID;
stopID = detID+1;
}
}else{
cutFile = new TFile(saveFileName, "recreate");
cutList = new TObjArray();
}
printf("=================== plotting histogram, start at GAGG-%0d \n", startID);
int count = 0;
for (Int_t i = startID; i < stopID; i++) {
varX.Form("gaggT"); varY.Form("gaggP");
h[i] = new TH2F(Form("h%d", i), Form("GAGG-%d",i), 500, 0, tailRange, 500, 0, peakRange);
@ -54,7 +95,10 @@ void GAGGPIDCutCreator(TString dataList,
expression.Form("%s:%s>>h%d", varY.Data(), varX.Data(),i);
gate.Form("gaggID==%d", i);
chain->Draw(expression, gate, "col");
chain->Draw(expression, gate, "col", numEntries * frac);
cut = (TCutG*) cutList->At(i);
cut->Draw("same");
if( h[i]->Integral() < 1000 ) {
h[i]->SetMarkerStyle(20);
@ -62,7 +106,9 @@ void GAGGPIDCutCreator(TString dataList,
h[i]->Draw("");
}
printf("======== make a graphic cut on the plot (double click to stop), %d-th cut: ", i );
printf("======== make a graphic cut on the plot (double click to stop), %2d-th cut: ", i );
if( detID != -100 )
cCutCreator->Modified(); cCutCreator->Update();
@ -71,7 +117,7 @@ void GAGGPIDCutCreator(TString dataList,
cut = (TCutG*) gROOT->FindObject("CUTG");
if( cut == NULL ){
printf(" stopped by user. no file saved or changed. \n");
printf(" stopped by user.\n");
break;
}
@ -81,17 +127,26 @@ void GAGGPIDCutCreator(TString dataList,
cut->SetVarY(varY.Data());
cut->SetTitle(tag);
cut->SetLineColor(i+1);
if( detID != -100 ){
cutList->AddAt(cut, abs(detID));
}else{
cutList->Add(cut);
}
printf(" cut-%d \n", i);
count ++;
}
TFile * cutFile = new TFile(saveFileName, "recreate");
cutList->Write("cutList", TObject::kSingleKey);
if( detID == -100) {
printf("====> saved %d cuts into %s\n", count, saveFileName.Data());
}else{
if( detID < 0 ) {
printf("====> ReDo %d cuts into %s, from GAGG-%2d\n", count, saveFileName.Data(), startID);
}else{
printf("====> ReDo %d cuts into %s, GAGG-%2d\n", count, saveFileName.Data(), startID);
}
}
gROOT->ProcessLine(".q");
}