static UInt_t color2rgb(TColor *col)
{
// returns RGB value of color
return ((UInt_t(col->GetRed()*255) << 16) +
(UInt_t(col->GetGreen()*255) << 8) +
UInt_t(col->GetBlue()*255));
}
pair <UInt_t, UInt_t> OnlineConfig::GetPageDim(UInt_t page)
{
// If defined in the config, will return those dimensions
// for the indicated page. Otherwise, will return the
// calculated dimensions required to fit all histograms.
pair <UInt_t, UInt_t> outDim;
// This is the page index in sConfFile.
UInt_t page_index = pageInfo[page].first;
UInt_t size1 = 2;
if (IsLogy(page)) size1 = 3; // last word is "logy"
// If the dimensions are defined, return them.
if(sConfFile[page_index].size()>size1-1) {
if(sConfFile[page_index].size() == size1) {
outDim = make_pair(UInt_t(atoi(sConfFile[page_index][1])),
UInt_t(atoi(sConfFile[page_index][1])));
return outDim;
} else if (sConfFile[page_index].size() == size1+1) {
outDim = make_pair(UInt_t(atoi(sConfFile[page_index][1])),
UInt_t(atoi(sConfFile[page_index][2])));
return outDim;
} else {
cout << "Warning: newpage command has too many arguments. "
<< "Will automatically determine dimensions of page."
<< endl;
}
}
// If not defined, return the "default."
UInt_t draw_count = GetDrawCount(page);
UInt_t dim = UInt_t(TMath::Nint(sqrt(double(draw_count+1))));
outDim = make_pair(dim,dim);
return outDim;
}
void usbCommTest(const Char_t* cfn="commTestConf.dat") {
R__ASSERT(bufSize>TSnIOStatusFrame::kMaxSizeOf);
// connect
const Int_t fd = openDevice();
Printf("connected");
// wait for status update
Char_t* b = buf;
Int_t nb=0;
Bool_t talking=kFALSE;
while (kTRUE) {
nb = readNextBytes(fd, b, 1u);
if (nb<0) {
if (talking==kFALSE) {
Printf("receiving data...");
}
//fprintf(stderr,"%02X ",*b);
b += -nb;
talking=kTRUE;
}
if (talking && nb>0) {
break;
}
gSystem->Sleep(1);
}
Printf("done receiving");
try {
Printf("Read status from buffer");
TSnIOStatusFrame::ReadFrom(buf, status, evt);
} catch (const std::exception& e) {
Printf("status frame read except: %s", e.what());
close(fd);
return;
}
try {
Printf("Read config from file");
readConfig(cfn);
} catch (const std::exception& e) {
Printf("read config except: %s", e.what());
close(fd);
return;
}
Char_t* cb=buf;
try {
Printf("Write config to buffer");
cb = TSnIOConfigFrame::WriteTo(buf, conf, runf);
} catch (const std::exception& e) {
Printf("write config except: %s", e.what());
close(fd);
return;
}
/*
while (1) {
Int_t a=1;
sendData(fd, &a, 4);
}
*/
Printf("Sending config (%u bytes)\n",UInt_t(cb-buf));
nb = sendData(fd, buf, UInt_t(cb-buf));
Printf("Sent %u bytes", nb);
close(fd);
}
UInt_t SetSeedOwn() {
UInt_t fSeed;
UInt_t fSeed1;
UInt_t fSeed2;
TUUID u;
UChar_t uuid[16];
u.GetUUID(uuid);
fSeed = UInt_t(uuid[3])*16777216 + UInt_t(uuid[2])*65536 + UInt_t(uuid[1])*256 + UInt_t(uuid[0]);
fSeed1 = UInt_t(uuid[7])*16777216 + UInt_t(uuid[6])*65536 + UInt_t(uuid[5])*256 + UInt_t(uuid[4]);
fSeed2 = UInt_t(uuid[11])*16777216 + UInt_t(uuid[10])*65536 + UInt_t(uuid[9])*256 + UInt_t(uuid[8]);
// use also the other bytes
UInt_t seed3 = UInt_t(uuid[15])*16777216 + UInt_t(uuid[14])*65536 + UInt_t(uuid[13])*256 + int(uuid[12]);
fSeed2 += seed3;
// TRandom r3(0);
// fSeed = static_cast<UInt_t> (4294967296.*r3.Rndm());
// fSeed1 = static_cast<UInt_t> (4294967296.*r3.Rndm());
// fSeed2 = static_cast<UInt_t> (4294967296.*r3.Rndm());
if (fSeed < 2) fSeed += 2UL;
if (fSeed1 < 8) fSeed1 += 8UL;
if (fSeed2 < 16) fSeed2 += 16UL;
UInt_t iy = fSeed ^ fSeed1 ^ fSeed2;
if(iy) return iy;
}
void processDYmm(Int_t maxEntries=-1)
{
std::cout << "DY range=" << DYtools::minMass << " .. " << DYtools::maxMass << "\n";
TVersion_t inpVersion=_verMu1;
inpVersion=_verMu76X;
inpVersion=_verMuApproved;
if (inpVersion!=_verMu1) {
if (DYtools::nMassBins!=DYtools::nMassBins43) {
std::cout << "a potential DYbinning.h problem\n";
return;
}
}
TString srcPath="/media/ssd/v20160214_1st_CovarianceMatrixInputs/";
srcPath="/mnt/sdb/andriusj/v20160214_1st_CovarianceMatrixInputs/";
TString dataFName=srcPath + "Input3/ROOTFile_ntuple_CovarianceMatrixInput.root";
if (inpVersion==_verMu76X) {
srcPath="/mnt/sdb/andriusj/v20160527_1st_CovarianceMatrixInputs_76X/";
dataFName=srcPath + "Input3/ROOTFile_Input_CovarianceMatrix.root";
}
else if (inpVersion==_verMuApproved) {
srcPath="/mnt/sdb/andriusj/v20160915_CovInput_ApprovedResults/";
dataFName=srcPath + "ROOTFile_Input_CovarianceMatrix.root";
}
DYmm13TeV_t data(dataFName);
data.DeactivateBranches();
data.ActivateBranches("Momentum_postFSR_Lead Momentum_postFSR_Sub Momentum_preFSR_Lead Momentum_preFSR_Sub Weight_Norm Weight_Gen Weight_PU");
data.ActivateBranches("Flag_EventSelection");
TH1D *h1postFsrInAccSel_M= new TH1D("h1_postFsrInAccSel_M", "postFSR selected events in acceptance;M_{gen,postFSR} [GeV];count",DYtools::nMassBins,DYtools::massBinEdges);
TH1D *h1postFsrInAccSel_MW= new TH1D("h1_postFsrInAccSel_MW", "postFSR selected events in acceptance (weighted);M_{gen,postFSR} [GeV];weighted count",DYtools::nMassBins,DYtools::massBinEdges);
TH1D *h1postFsrInAccSel_MWPU= new TH1D("h1_postFsrInAccSel_MWPU", "postFSR selected events in acceptance (weighted wPU);M_{gen,postFSR} [GeV];weighted (wPU) count",DYtools::nMassBins,DYtools::massBinEdges);
TH1D *h1postFsrInAcc_M= new TH1D("h1_postFsrInAcc_M", "postFSR in acceptace;M_{gen,postFSR} [GeV];count",DYtools::nMassBins,
DYtools::massBinEdges);
TH1D *h1postFsrInAcc_MW= new TH1D("h1_postFsrInAcc_Mweighted", "postFSR in acceptance;M_{gen,postFSR} [GeV];weighted count", DYtools::nMassBins,
DYtools::massBinEdges);
TH1D *h1postFsrInAcc_MWPU= new TH1D("h1_postFsrInAcc_MweightedPU", "postFSR in acceptance (weighted wPU);M_{gen,postFSR} [GeV];weighted (wPU) count", DYtools::nMassBins,
DYtools::massBinEdges);
TH1D *h1postFsr_M= new TH1D("h1_postFsr_M", "postFSR;M_{gen,postFSR} [GeV];count",DYtools::nMassBins,
DYtools::massBinEdges);
TH1D *h1postFsr_MW= new TH1D("h1_postFsr_Mweighted", "postFSR;M_{gen,postFSR} [GeV];weighted count", DYtools::nMassBins,
DYtools::massBinEdges);
TH1D *h1preFsr_M= new TH1D("h1_preFsr_M", "preFSR;M_{gen,preFSR} [GeV];count",DYtools::nMassBins,
DYtools::massBinEdges);
TH1D *h1preFsr_MW= new TH1D("h1_preFsr_Mweighted", "preFSR;M_{gen,preFSR} [GeV];weighted count", DYtools::nMassBins,
DYtools::massBinEdges);
prepareHisto(h1postFsrInAccSel_M);
prepareHisto(h1postFsrInAccSel_MW);
prepareHisto(h1postFsrInAccSel_MWPU);
prepareHisto(h1postFsrInAcc_M);
prepareHisto(h1postFsrInAcc_MW);
prepareHisto(h1postFsrInAcc_MWPU);
prepareHisto(h1postFsr_M);
prepareHisto(h1postFsr_MW);
prepareHisto(h1preFsr_M);
prepareHisto(h1preFsr_MW);
TH2D* h2FSRmig= new TH2D("h2FSRmig","FSR migration", DYtools::nMassBins, DYtools::massBinEdges, DYtools::nMassBins, DYtools::massBinEdges);
h2FSRmig->Sumw2();
RooUnfoldResponse fsrResp(h1postFsr_MW,h1preFsr_MW,h2FSRmig,"rooUnf_fsrResp","fsrResp;M_{#mu#mu} [GeV];FSR unfolded yield");
fsrResp.UseOverflow();
TH2D* h2effAccFSRmig= new TH2D("h2effAccFSRmig","eff x Acc x FSR migration", DYtools::nMassBins, DYtools::massBinEdges, DYtools::nMassBins, DYtools::massBinEdges);
h2effAccFSRmig->Sumw2();
RooUnfoldResponse effAccFsrResp(h1postFsrInAccSel_MW,h1preFsr_MW,h2effAccFSRmig,"rooUnf_effAccFsrResp","effAccFsrResp;M_{#mu#mu} [GeV];eff Acc FSR unfolded yield");
effAccFsrResp.UseOverflow();
UInt_t nEvents= data.GetEntries();
std::cout << "process data\n";
for (UInt_t iEntry=0; iEntry<nEvents; iEntry++) {
if (data.GetEntry(iEntry)<0) break;
if (iEntry%100000==0) std::cout << "iEntry=" << iEntry << Form("(%4.2lf%%)\n",iEntry*100/double(nEvents));
if ((maxEntries>0) && (iEntry>UInt_t(maxEntries))) {
std::cout << "debug run\n";
break;
}
double w= data.Weight_Norm * data.Weight_Gen;
if (iEntry<100) std::cout << "w=" << w << ": weight_norm=" << data.Weight_Norm << ", gen=" << data.Weight_Gen << "\n";
double mPostFsr= (*data.Momentum_postFSR_Lead + *data.Momentum_postFSR_Sub).M();
double mPreFsr= (*data.Momentum_preFSR_Lead + *data.Momentum_preFSR_Sub).M();
if (DYtools::InAcceptance_mm(data.Momentum_postFSR_Lead,data.Momentum_postFSR_Sub)) {
if (data.Flag_EventSelection) {
h1postFsrInAccSel_M->Fill( mPostFsr, 1. );
h1postFsrInAccSel_MW->Fill( mPostFsr, w );
h1postFsrInAccSel_MWPU->Fill( mPostFsr, w * data.Weight_PU );
}
h1postFsrInAcc_M->Fill( mPostFsr, 1. );
h1postFsrInAcc_MW->Fill( mPostFsr, w );
h1postFsrInAcc_MWPU->Fill( mPostFsr, w * data.Weight_PU );
}
h1postFsr_M->Fill( mPostFsr, 1. );
h1postFsr_MW->Fill( mPostFsr, w );
h1preFsr_M->Fill( mPreFsr, 1. );
h1preFsr_MW->Fill( mPreFsr, w );
h2FSRmig->Fill( mPostFsr, mPreFsr, w );
if (0) {
if ( ! DYtools::InsideMassRange(mPreFsr) )
std::cout << "preFSR mass " << mPreFsr << ", postFSR mass " << mPostFsr << "\n";
}
if ( ! DYtools::InsideMassRange(mPreFsr) &&
DYtools::InsideMassRange(mPostFsr) ) {
std::cout << "fake detected\n";
fsrResp.Fake(mPostFsr,w);
}
else if ( DYtools::InsideMassRange(mPreFsr) &&
! DYtools::InsideMassRange(mPostFsr) ) {
fsrResp.Miss(mPreFsr,w);
}
else {
fsrResp.Fill(mPostFsr,mPreFsr,w);
}
int sel=(DYtools::InAcceptance_mm(data.Momentum_postFSR_Lead,data.Momentum_postFSR_Sub) && data.Flag_EventSelection && DYtools::InsideMassRange(mPostFsr)) ? 1:0;
if (sel) h2effAccFSRmig->Fill(mPostFsr,mPreFsr,w);
if ( ! DYtools::InsideMassRange(mPreFsr) && sel ) {
effAccFsrResp.Fake(mPostFsr,w);
}
else if ( DYtools::InsideMassRange(mPreFsr) && ! sel ) {
effAccFsrResp.Miss(mPreFsr,w);
}
else {
effAccFsrResp.Fill(mPostFsr,mPreFsr,w);
}
}
RooUnfoldBayes bayesFSR( &fsrResp, h1postFsr_MW, 4 );
TH1D *h1preFsrUnf= (TH1D*) bayesFSR.Hreco()->Clone("h1preFsrUnf");
h1preFsrUnf->SetTitle("unfolded postFSR->preFSR");
h1preFsrUnf->SetDirectory(0);
histoStyle(h1preFsrUnf,kRed,24);
h1preFsrUnf->GetXaxis()->SetMoreLogLabels();
h1preFsrUnf->GetXaxis()->SetNoExponent();
TCanvas *cFSRTest=plotHisto(h1preFsrUnf,"cFSRTest",1,1,"LPE1");
plotHistoSame(h1preFsr_MW,"cFSRTest","LPE");
RooUnfoldBayes bayesEffAccFsr( &effAccFsrResp, h1postFsrInAccSel_MW, 4 );
TH1D *h1EffAccFsrUnf= (TH1D*) bayesEffAccFsr.Hreco()->Clone("h1EffAccFsrUnf");
h1EffAccFsrUnf->SetTitle("unfolded postFsrInAccSel -> preFSR");
h1EffAccFsrUnf->SetDirectory(0);
histoStyle(h1EffAccFsrUnf,kRed,24);
h1EffAccFsrUnf->GetXaxis()->SetMoreLogLabels();
h1EffAccFsrUnf->GetXaxis()->SetNoExponent();
TCanvas *cScaleToPreFsrTest= plotHisto(h1EffAccFsrUnf,"cScaleToPreFsrTest",1,1,"LPE1");
histoStyle(h1preFsr_MW,kBlue,5);
plotHistoSame(h1preFsr_MW,"cScaleToPreFsrTest","LPE");
TH1D* h1Eff=(TH1D*)h1postFsrInAccSel_MW->Clone("h1Eff");
h1Eff->SetDirectory(0);
if (!h1Eff->GetSumw2()) h1Eff->Sumw2();
h1Eff->SetTitle("Efficiency;M_{#mu#mu,GEN postFSR} [GeV];postFSR_inAcc_Sel/postFSR_inAcc");
h1Eff->Divide(h1postFsrInAccSel_MW,h1postFsrInAcc_MW,1,1,"B");
TH1D* h1EffPU=(TH1D*)h1postFsrInAccSel_MWPU->Clone("h1EffPU");
h1EffPU->SetDirectory(0);
if (!h1EffPU->GetSumw2()) h1EffPU->Sumw2();
h1EffPU->SetTitle("Efficiency (wPU);M_{#mu#mu,GEN postFSR} [GeV];postFSR_inAcc_Sel(wPU)/postFSR_inAcc(wPU)");
h1EffPU->Divide(h1postFsrInAccSel_MWPU,h1postFsrInAcc_MWPU,1,1,"B");
histoStyle(h1EffPU,kBlue,24);
TCanvas *cEffCmp=plotHisto(h1Eff,"cEff_noPU_vs_wPU",1,0,"LPE1");
plotHistoSame(h1EffPU,"cEff_noPU_vs_wPU","LPE");
TH1D* h1Acc=(TH1D*)h1postFsrInAcc_MW->Clone("h1Acc");
h1Acc->SetDirectory(0);
if (!h1Acc->GetSumw2()) h1Acc->Sumw2();
h1Acc->SetTitle("Acceptance;M_{#mu#mu,GEN postFSR} [GeV];postFSR_inAcc/postFSR");
h1Acc->Divide(h1postFsrInAcc_MW,h1postFsr_MW,1,1,"B");
TH1D* h1EffPUAcc=(TH1D*)h1postFsrInAccSel_MWPU->Clone("h1EffPUAcc");
h1EffPUAcc->SetDirectory(0);
if (!h1EffPUAcc->GetSumw2()) h1EffPUAcc->Sumw2();
h1EffPUAcc->SetTitle("Efficiency(wPU) x Acc;M_{#mu#mu,GEN postFSR} [GeV];postFSR_inAccSel(wPU)/h1postFsr_MW(noPU)");
h1EffPUAcc->Divide(h1postFsrInAccSel_MWPU,h1postFsr_MW,1,1,"B");
TH1D *h1FSRCorr_binByBin=(TH1D*)h1preFsr_MW->Clone("h1FSRCorr_binByBin");
h1FSRCorr_binByBin->SetDirectory(0);
if (!h1FSRCorr_binByBin->GetSumw2()) h1FSRCorr_binByBin->Sumw2();
h1FSRCorr_binByBin->SetTitle("FSR correction bin-by-bin;M_{#mu#mu} [GeV];preFSR/postFSR");
h1FSRCorr_binByBin->Divide(h1preFsr_MW,h1postFsr_MW,1.,1.,"B");
TString fname="dymm_test_" + versionName(inpVersion) + TString(".root");
if (maxEntries>0) fname.ReplaceAll(".root","_debug.root");
TFile fout(fname,"RECREATE");
h1Eff->Write();
h1EffPU->Write();
h1Acc->Write();
h1EffPUAcc->Write();
h1FSRCorr_binByBin->Write();
fsrResp.Write();
effAccFsrResp.Write();
h1postFsrInAccSel_M->Write();
h1postFsrInAccSel_MW->Write();
h1postFsrInAcc_M->Write();
h1postFsrInAcc_MW->Write();
h1postFsr_M->Write();
h1postFsr_MW->Write();
h1preFsr_M->Write();
h1preFsr_MW->Write();
h1preFsrUnf->Write();
h2FSRmig->Write();
h1EffAccFsrUnf->Write();
h2effAccFSRmig->Write();
cEffCmp->Write();
cFSRTest->Write();
cScaleToPreFsrTest->Write();
TObjString timeTag(DayAndTimeTag(0));
timeTag.Write("timeTag");
fout.Close();
std::cout << "file <" << fout.GetName() << "> created\n";
}
