void writeEStructExample( const char* filelist,
const char* outputDir,
const char* scriptDir,
int maxNumEvents = 0 ) {
gROOT->LoadMacro("load2ptLibs.C");
load2ptLibs();
char cutFile[1024];
sprintf(cutFile,"%s/CutsFile.txt",scriptDir);
gROOT->LoadMacro("getOutFileName.C"); // for laying out file names
const char* scratchDir = "CuCu200data";
// define centrality (just making stuff up below!!!)
StEStructCentrality* cent=StEStructCentrality::Instance();
const double mbBins[]={2,30,100,180,240,320,400,999999};
int mbNBins=1+1+1+1+1+1+1+1;
cent->setCentralities(mbBins,mbNBins);
int numberOfAnalyses=mbNBins-1;
// create the low-level reader (here for MuDst)
// reader = reader interface + pointer to Data Maker + cut classes
//***** depends on PDSF vs RCF .... below is RCF version of scheduler
// char fileListFile[1024];
// sprintf(fileListFile,"%s/%s",scriptDir,filelist);
StMuDstMaker* mk = new StMuDstMaker(0,0,"",filelist,".",5000);
// create the generic EStructAnalysisMaker
StEStructAnalysisMaker *estructMaker = new StEStructAnalysisMaker("EStruct2pt");
// Set up the reader with event/track cuts
StEStructEventCuts* ecuts = new StEStructEventCuts(cutFile);
StEStructTrackCuts* tcuts = new StEStructTrackCuts(cutFile);
StEStructMuDstReader* reader = new StEStructMuDstReader(mk,ecuts,tcuts,false);
estructMaker->SetEventReader(reader);
// create the QAHist object (must come after centrality and cutbin objects)
int EventType = 0; // real data
StEStructQAHists* qaHists = new StEStructQAHists(EventType);
qaHists->initHistograms();
estructMaker->SetQAHists(qaHists);
StEStructEventWriter** analysis = new StEStructEventWriter*[mbNBins];
StEStructEventMaker** writeMakers = new StEStructEventMaker*[mbNBins];
int analysisMode = 0; // 2pt correlations mode selection
for(int i=0;i<numberOfAnalyses;i++){
int ic=i;
if(numberOfAnalyses==1)ic=-1;
TString writerName("EStructWriterMaker"); writerName+=i;
writeMakers[i]=new StEStructEventMaker(writerName.Data());
analysis[i]=new StEStructEventWriter(writeMakers[i]);
analysis[i]->setOutputFileName(getOutFileName(outputDir,scratchDir,"event",ic));
/*
analysis[i]->setQAHists(qaHists); // only 1 QA Object in this case
analysis[i]->setZBuffLimits(ecuts); // common to all
*/
analysis[i]->setAnalysisIndex(i);
}
estructMaker->SetAnalyses(analysis,numberOfAnalyses);
// --- now do the work ---
estructMaker->Init();
estructMaker->startTimer();
int counter=0, istat=0, i=0;
while (istat!=2) {
istat=estructMaker->Make(); // now includes filling qa histograms
i++; counter++;
if (counter==200) {
cout<<"doing event ="<<i<<endl;
counter=0;
estructMaker->writeDiagnostics(0);
}
if ( maxNumEvents!=0 && i>=maxNumEvents ) {
istat=2;
}
}
estructMaker->stopTimer();
//--- now write out stats and cuts ---
ofstream ofs(getOutFileName(outputDir,scratchDir,"stats"));
estructMaker->logAllStats(ofs);
ecuts->printCuts(ofs);
tcuts->printCuts(ofs);
ofs<<endl;
ofs.close();
// --> root cut file
TFile* tf=new TFile(getOutFileName(outputDir,scratchDir,"cuts"),"RECREATE");
ecuts->writeCutHists(tf);
tcuts->writeCutHists(tf);
tf->Close();
// --- write out the data
estructMaker->Finish();
}
void combineHistograms3(const char *dirName, const char **inNames, const char *outName, int nCent) {
// -- Calculate \Delta\rho/\sqrt{\rho_{ref}} for all quantities, i.e. Phi-Phi, Eta-Eta, DPhi-DEta etc.
//
// root.exe -q -b combineHistograms3.C("dir","inFileList","outFileName",numFiles)
//
// inFileList contains centrality information, so I can have a list
// like {"Data1", "Data2",...} or {"Sum0_1", "Sum2_3",...}
// Before getting here we use hadd to add histograms from different jobs
// creating files we call Data{n}.root (where n is a centrality bin)
// then we may combine centralities be merging the Data{n} files tagging
// histograms with z-vertex index and creating files Sum{n1}_{n2}.root where
// n1 and n2 are the limits of the centrality bin range.
gROOT->LoadMacro("load2ptLibs.C");
load2ptLibs();
gSystem->Load("StEStructPoolSupport.so");
TFile *tf;
StEStructSupport *ehelp;
TH2D **ptsedp;
TH2D **ptsedpC;
TH2D **ptdedp;
TH2D **ptdedpC;
TH2D **sedp;
TH2D **sedpC;
TH2D **dedp;
TH2D **dedpC;
TH2D **ptetaeta;
TH2D **ptetaetaC;
TH2D **etaeta;
TH2D **etaetaC;
TH2D **ptetaetaSS;
TH2D **ptetaetaSSC;
TH2D **etaetaSS;
TH2D **etaetaSSC;
TH2D **ptetaetaAS;
TH2D **ptetaetaASC;
TH2D **etaetaAS;
TH2D **etaetaASC;
TH2D **ptphiphi;
TH2D **ptphiphiC;
TH2D **phiphi;
TH2D **phiphiC;
const char* binName[]={"all","awayside","nearside","soft","softAS","softNS","neck","neckAS","neckNS","hard","hardAS","hardNS","softHard","softHardAS","softHardNS"};
const char* chargeName[] = {"_LS_", "_US_", "_CD_", "_CI_"};
const char* chargeType[] = {"_PP_", "_PM_", "_MP_", "_MM_"};
char outFileName[1024];
sprintf(outFileName,"%s/%s.root",dirName,outName);
TFile *out = new TFile(outFileName,"RECREATE");
char inFileName[1024];
// For axial space near/away side is not useful.
int axialBins[] = {0, 3, 6, 9, 12};
for (int ic=0;ic<nCent;ic++) {
printf("Centrality %2i: d2N/dEdP pHat_{A+} pHat_{A-} pHat_{B+} pHat_{B-}\n",ic);
for (int ibin=0;ibin<5;ibin++) {
sprintf(inFileName,"%s/%s%s.root",dirName,inNames[ic],binName[axialBins[ibin]]);
tf = new TFile(inFileName);
tf->cd();
ehelp = new StEStructSupport(tf,0);
ehelp->msilent = true;
ehelp->mapplyDEtaFix = false;
ehelp->mPairNormalization = true;
ehelp->mIdenticalPair = true;
ehelp->setBGMode(0);
int subtract = 1;
ptsedpC = ehelp->buildPtCommon("SEtaDPhi",2,subtract);
ptdedpC = ehelp->buildPtCommon("DEtaDPhi",2,subtract);
ptetaetaC = ehelp->buildPtCommon("EtaEta",2,subtract);
ptetaetaSSC = ehelp->buildPtCommon("EtaEtaSS",2,subtract);
ptetaetaASC = ehelp->buildPtCommon("EtaEtaAS",2,subtract);
ptphiphiC = ehelp->buildPtCommon("PhiPhi",2,subtract);
ptsedp = ehelp->buildPtChargeTypes("SEtaDPhi",2,subtract);
ptdedp = ehelp->buildPtChargeTypes("DEtaDPhi",2,subtract);
ptetaeta = ehelp->buildPtChargeTypes("EtaEta",2,subtract);
ptetaetaSS = ehelp->buildPtChargeTypes("EtaEtaSS",2,subtract);
ptetaetaAS = ehelp->buildPtChargeTypes("EtaEtaAS",2,subtract);
ptphiphi = ehelp->buildPtChargeTypes("PhiPhi",2,subtract);
sedpC = ehelp->buildCommon("SEtaDPhi",2);
dedpC = ehelp->buildCommon("DEtaDPhi",2);
etaetaC = ehelp->buildCommon("EtaEta",2);
etaetaSSC = ehelp->buildCommon("EtaEtaSS",2);
etaetaASC = ehelp->buildCommon("EtaEtaAS",2);
phiphiC = ehelp->buildCommon("PhiPhi",2);
sedp = ehelp->buildChargeTypes("SEtaDPhi",2);
dedp = ehelp->buildChargeTypes("DEtaDPhi",2);
etaeta = ehelp->buildChargeTypes("EtaEta",2);
etaetaSS = ehelp->buildChargeTypes("EtaEtaSS",2);
etaetaAS = ehelp->buildChargeTypes("EtaEtaAS",2);
phiphi = ehelp->buildChargeTypes("PhiPhi",2);
out->cd();
for (int icharge=0;icharge<4;icharge++) {
TString name(binName[axialBins[ibin]]);
name += "_NDEtaDPhi"; name += chargeName[icharge]; name += ic;
if (dedp) {
dedp[icharge]->SetName(name.Data());
dedp[icharge]->SetTitle(name.Data());
dedp[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_NSEtaDPhi"; name += chargeName[icharge]; name += ic;
if (sedp) {
sedp[icharge]->SetName(name.Data());
sedp[icharge]->SetTitle(name.Data());
sedp[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_PtDEtaDPhi"; name += chargeName[icharge]; name += ic;
if (ptdedp) {
ptdedp[icharge]->SetName(name.Data());
ptdedp[icharge]->SetTitle(name.Data());
ptdedp[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_PtSEtaDPhi"; name += chargeName[icharge]; name += ic;
if (ptsedp) {
ptsedp[icharge]->SetName(name.Data());
ptsedp[icharge]->SetTitle(name.Data());
ptsedp[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_NDEtaDPhi"; name += chargeType[icharge]; name += ic;
if (dedpC) {
dedpC[icharge]->SetName(name.Data());
dedpC[icharge]->SetTitle(name.Data());
dedpC[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_NSEtaDPhi"; name += chargeType[icharge]; name += ic;
if (sedpC) {
sedpC[icharge]->SetName(name.Data());
sedpC[icharge]->SetTitle(name.Data());
sedpC[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_PtDEtaDPhi"; name += chargeType[icharge]; name += ic;
if (ptdedpC) {
ptdedpC[icharge]->SetName(name.Data());
ptdedpC[icharge]->SetTitle(name.Data());
ptdedpC[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_PtSEtaDPhi"; name += chargeType[icharge]; name += ic;
if (ptsedpC) {
ptsedpC[icharge]->SetName(name.Data());
ptsedpC[icharge]->SetTitle(name.Data());
ptsedpC[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_NEtaEta"; name += chargeName[icharge]; name += ic;
if (etaeta) {
etaeta[icharge]->SetName(name.Data());
etaeta[icharge]->SetTitle(name.Data());
etaeta[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_PtEtaEta"; name += chargeName[icharge]; name += ic;
if (ptetaeta) {
ptetaeta[icharge]->SetName(name.Data());
ptetaeta[icharge]->SetTitle(name.Data());
ptetaeta[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_NEtaEtaSS"; name += chargeName[icharge]; name += ic;
if (etaetaSS) {
etaetaSS[icharge]->SetName(name.Data());
etaetaSS[icharge]->SetTitle(name.Data());
etaetaSS[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_PtEtaEtaSS"; name += chargeName[icharge]; name += ic;
if (ptetaetaSS) {
ptetaetaSS[icharge]->SetName(name.Data());
ptetaetaSS[icharge]->SetTitle(name.Data());
ptetaetaSS[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_NEtaEtaAS"; name += chargeName[icharge]; name += ic;
if (etaetaAS) {
etaetaAS[icharge]->SetName(name.Data());
etaetaAS[icharge]->SetTitle(name.Data());
etaetaAS[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_PtEtaEtaAS"; name += chargeName[icharge]; name += ic;
if (ptetaetaAS) {
ptetaetaAS[icharge]->SetName(name.Data());
ptetaetaAS[icharge]->SetTitle(name.Data());
ptetaetaAS[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_NEtaEta"; name += chargeType[icharge]; name += ic;
if (etaetaC) {
etaetaC[icharge]->SetName(name.Data());
etaetaC[icharge]->SetTitle(name.Data());
etaetaC[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_PtEtaEta"; name += chargeType[icharge]; name += ic;
if (ptetaetaC) {
ptetaetaC[icharge]->SetName(name.Data());
ptetaetaC[icharge]->SetTitle(name.Data());
ptetaetaC[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_NEtaEtaSS"; name += chargeType[icharge]; name += ic;
if (etaetaSSC) {
etaetaSSC[icharge]->SetName(name.Data());
etaetaSSC[icharge]->SetTitle(name.Data());
etaetaSSC[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_PtEtaEtaSS"; name += chargeType[icharge]; name += ic;
if (ptetaetaSSC) {
ptetaetaSSC[icharge]->SetName(name.Data());
ptetaetaSSC[icharge]->SetTitle(name.Data());
ptetaetaSSC[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_NEtaEtaAS"; name += chargeType[icharge]; name += ic;
if (etaetaASC) {
etaetaASC[icharge]->SetName(name.Data());
etaetaASC[icharge]->SetTitle(name.Data());
etaetaASC[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_PtEtaEtaAS"; name += chargeType[icharge]; name += ic;
if (ptetaetaASC) {
ptetaetaASC[icharge]->SetName(name.Data());
ptetaetaASC[icharge]->SetTitle(name.Data());
ptetaetaASC[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_NPhiPhi"; name += chargeName[icharge]; name += ic;
if (phiphi) {
phiphi[icharge]->SetName(name.Data());
phiphi[icharge]->SetTitle(name.Data());
phiphi[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_PtPhiPhi"; name += chargeName[icharge]; name += ic;
if (ptphiphi) {
ptphiphi[icharge]->SetName(name.Data());
ptphiphi[icharge]->SetTitle(name.Data());
ptphiphi[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_NPhiPhi"; name += chargeType[icharge]; name += ic;
if (phiphiC) {
phiphiC[icharge]->SetName(name.Data());
phiphiC[icharge]->SetTitle(name.Data());
phiphiC[icharge]->Write();
}
TString name(binName[axialBins[ibin]]);
name += "_PtPhiPhi"; name += chargeType[icharge]; name += ic;
if (ptphiphiC) {
ptphiphiC[icharge]->SetName(name.Data());
ptphiphiC[icharge]->SetTitle(name.Data());
ptphiphiC[icharge]->Write();
}
}
// Calculate and print scale factors
double scale = ehelp->getCIdNdEtadPhi();
double *ptHat = ehelp->getptHat();
printf(" bin %8s %7.3f %7.3f %7.3f %7.3f %7.3f\n",binName[axialBins[ibin]],scale,ptHat[0],ptHat[1],ptHat[2],ptHat[3]);
delete [] ptHat;
delete tf;
delete ehelp;
}
}
TH2D **ytyt = 0;
TH2D **ytytC = 0;
TH2D **sytdyt = 0;
TH2D **sytdytC = 0;
gROOT->LoadMacro("minimizeNegative.C");
double *sFactor[2][3], *eSFactor[2][3];
for (int it=0;it<2;it++) {
for (int ibin=0;ibin<3;ibin++) {
sFactor[it][ibin] = new double[nCent];
eSFactor[it][ibin] = new double[nCent];
}
}
float sf[2];
bool scaleYt = false;
double argList[10];
double start = 0.95;
double step = 0.001;
double bmin = 0.9;
double bmax = 1.1;
int errFlag = 0;
minData.mCorrType = 1;
minData.mLambda = 10;
// For ytyt space soft/neck/hard is not useful.
int ytytBins[] = {0, 1, 2};
for (int ic=0;ic<nCent;ic++) {
for (int ibin=0;ibin<3;ibin++) {
sprintf(inFileName,"%s/%s%s.root",dirName,inNames[ic],binName[ytytBins[ibin]]);
tf = new TFile(inFileName);
tf->cd();
ehelp = new StEStructSupport(tf,0);
ehelp->msilent = true;
ehelp->mapplyDEtaFix = false;
ehelp->mPairNormalization = false;
ehelp->mIdenticalPair = true;
ehelp->mYtYtNormalization = true;
// Make sure LS histogram for zBin 0 is in file.
if (0 == ehelp->histogramExists("YtYt", 0)) {
continue;
}
// A lot of stuff so we can find a scaling factor for \rho_{ref}
// such that \Delta\rho is almost always positive.
minuit = new TMinuit(1);
if (!scaleYt) {
ehelp->setBGMode(0);
ehelp->mYtYtNormalization = true;
sf[0] = 1;
sf[1] = 1;
cout << ">>>>>Not fitting scale factors for centrality " << ic << " yt bin " << ibin << ", Using YtYtNormalization = " << ehelp->mYtYtNormalization << endl;
ytyt = ehelp->buildChargeTypes("YtYt",5,sf);
ytytC = ehelp->buildCommon("YtYt",5,sf);
sytdyt = ehelp->buildChargeTypes("SYtDYt",5,sf);
sytdytC = ehelp->buildCommon("SYtDYt",5,sf);
} else {
ehelp->setBGMode(1);
minData.mSupport = ehelp;
minData.mChargeType = 0;
argList[0] = -1;
minuit->mnexcm("SET PRINT",argList,1,errFlag);
minuit->SetFCN(minimizeNegative);
minuit->mnparm( 0, "rho_ref scale factor", start, step, bmin, bmax, errFlag );
minuit->SetErrorDef(1);
minuit->SetPrintLevel(-1);
argList[0] = 1;
minuit->mnexcm("SET STR",argList,1,errFlag);
argList[0] = 500;
cout << ">>>>>Starting scale factor 0 fit for centrality " << ic << " yt bin " << ibin << endl;
minuit->mnexcm("MIGRAD",argList,1,errFlag);
double cVal, eVal;
minuit->GetParameter(0,cVal,eVal);
sFactor[0][ibin][ic] = cVal;
eSFactor[0][ibin][ic] = eVal;
if (4 == errFlag) {
minuit->mnexcm("MINOS",argList,1,errFlag);
minuit->GetParameter(0,cVal,eVal);
sFactor[0][ibin][ic] = cVal;
eSFactor[0][ibin][ic] = eVal;
if (4 == errFlag) {
cout << "++++Out of chesse error; Fit failed. Using scale factor = 1" << endl;
sFactor[0][ibin][ic] = 1;
eSFactor[0][ibin][ic] = 100;
}
}
delete minuit;
// Seems like I should be able to reset the TMinuit object to do a
// different fit. Didn't work on my first attempts, so I just create
// a new one.
minData.mChargeType = 1;
minuit = new TMinuit(1);
argList[0] = -1;
minuit->mnexcm("SET PRINT",argList,1,errFlag);
minuit->SetFCN(minimizeNegative);
minuit->mnparm( 0, "rho_ref scale factor", start, step, bmin, bmax, errFlag );
minuit->SetErrorDef(1);
minuit->SetPrintLevel(-1);
argList[0] = 1;
minuit->mnexcm("SET STR",argList,1,errFlag);
argList[0] = 500;
cout << ">>>>>Starting scale factor 1 fit for centrality " << ic << " yt bin " << ibin << endl;
minuit->mnexcm("MIGRAD",argList,1,errFlag);
minuit->GetParameter(0,cVal,eVal);
sFactor[1][ibin][ic] = cVal;
eSFactor[1][ibin][ic] = eVal;
if (4 == errFlag) {
minuit->mnexcm("MINOS",argList,1,errFlag);
minuit->GetParameter(0,cVal,eVal);
sFactor[1][ibin][ic] = cVal;
eSFactor[1][ibin][ic] = eVal;
if (4 == errFlag) {
cout << "++++Out of chesse error; Fit failed. Using scale factor = 1" << endl;
sFactor[1][ibin][ic] = 1;
eSFactor[1][ibin][ic] = 100;
}
}
sf[0] = sFactor[0][ibin][ic];
sf[1] = sFactor[1][ibin][ic];
ytyt = ehelp->buildChargeTypes("YtYt",5,sf);
ytytC = ehelp->buildCommon("YtYt",5,sf);
sytdyt = ehelp->buildChargeTypes("SYtDYt",5,sf);
sytdytC = ehelp->buildCommon("SYtDYt",5,sf);
}
delete minuit;
out->cd();
for (int icharge=0;icharge<4;icharge++) {
if (ytyt) {
TString name(binName[ytytBins[ibin]]);
name += "_YtYt"; name += chargeName[icharge]; name += ic;
ytyt[icharge]->SetName(name.Data());
ytyt[icharge]->SetTitle(name.Data());
ytyt[icharge]->Write();
}
if (ytytC) {
TString name(binName[ytytBins[ibin]]);
name += "_YtYt"; name += chargeType[icharge]; name += ic;
ytytC[icharge]->SetName(name.Data());
ytytC[icharge]->SetTitle(name.Data());
ytytC[icharge]->Write();
}
if (sytdyt) {
TString name(binName[ytytBins[ibin]]);
name += "_SYtDYt"; name += chargeName[icharge]; name += ic;
sytdyt[icharge]->SetName(name.Data());
sytdyt[icharge]->SetTitle(name.Data());
sytdyt[icharge]->Write();
}
if (sytdytC) {
TString name(binName[ytytBins[ibin]]);
name += "_SYtDYt"; name += chargeType[icharge]; name += ic;
sytdytC[icharge]->SetName(name.Data());
sytdytC[icharge]->SetTitle(name.Data());
sytdytC[icharge]->Write();
}
}
delete tf;
delete ehelp;
}
}
if (scaleYt) {
cout << "<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<" << endl;
cout << " Here are Yt scale factors for rho_{ref}" << endl;
printf(" Centrality all LS SS LS AS LS all US SS US AS US \n");
for (int ic=0;ic<nCent;ic++) {
printf("%6i %9.3f(%3.0f) %7.3f(%3.0f) %7.3f(%3.0f) %7.3f(%3.0f) %7.3f(%3.0f) %7.3f(%3.0f)\n",ic,
sFactor[0][0][ic],1000*eSFactor[0][0][ic],sFactor[0][1][ic],1000*eSFactor[0][1][ic],sFactor[0][2][ic],1000*eSFactor[0][2][ic],
sFactor[1][0][ic],1000*eSFactor[1][0][ic],sFactor[1][1][ic],1000*eSFactor[1][1][ic],sFactor[1][2][ic],1000*eSFactor[1][2][ic]);
}
for (int it=0;it<2;it++) {
for (int ibin=0;ibin<3;ibin++) {
delete sFactor[it][ibin];
delete eSFactor[it][ibin];
}
}
}
out->Close();
// delete out;
}
/************************************************************************
* $Id: doEStructEmpty.C,v 1.2 2007/05/27 22:46:27 msd Exp $
*
* Author: Jeff Porter
*
* example code for reading in MuDst files contained int "fileListFile"
* and running the 2pt correlations analysis, producing hist files in,
*
* outputDir/jobName/data/XX/
*
* where XX = event level selection imposed by the cut files.
* This example (from PP) contains 3 selections on event mult
*
*************************************************************************/
void doEStructEmpty(const char* fileListFile, const char* outputDir, const char* cutFile, const char* jobName, int cutBinMode=0, int maxNumEvents=0, int analysisMode=0){
gROOT->LoadMacro("load2ptLibs.C");
load2ptLibs();
gROOT->LoadMacro("getOutFileName.C"); // for laying out file names
// define centrality
StEStructCentrality* cent=StEStructCentrality::Instance();
//int numCent = 2;
//const double temp[2]={0,2000}; //=1 centrality
int numCent = 2;
const double temp[2]={2,2000}; //=1 centrality
//int numCent = 5;
//const double temp[5]={2,250,500,750,2000}; //=1 centrality
cent->setCentralities(temp,numCent);
int numberOfAnalyses=numCent-1;
// choose the mode for the binning
StEStructCutBin* cb=StEStructCutBin::Instance();
cb->setMode(cutBinMode);
int mbNCutBins = cb->getNumBins();
// create the low-level reader (here for MuDst)
// reader = reader interface + pointer to Data Maker + cut classes
StMuDstMaker* mk = new StMuDstMaker(0,0,"",fileListFile,".",5000);
// create the generic EStructAnalysisMaker
StEStructAnalysisMaker *estructMaker = new StEStructAnalysisMaker("EStruct2pt");
// Set up the reader with event/track cuts
StEStructEventCuts* ecuts = new StEStructEventCuts(cutFile);
StEStructTrackCuts* tcuts = new StEStructTrackCuts(cutFile);
StEStructMuDstReader* reader = new StEStructMuDstReader(mk,ecuts,tcuts);
estructMaker->SetEventReader(reader);
// create the QAHist object (must come after centrality and cutbin objects)
int EventType = 0; // real data
//StEStructQAHists* qaHists = new StEStructQAHists(EventType);
//qaHists->initHistograms();
//estructMaker->SetQAHists(qaHists);
//StEStruct2ptCorrelations** analysis = new StEStruct2ptCorrelations*[numberOfAnalyses];
StEStructEmptyAnalysis** analysis = new StEStructEmptyAnalysis*[numberOfAnalyses];
//StEStructPairCuts* pcuts = new StEStructPairCuts(cutFile);
for(int i=0;i<numberOfAnalyses;i++){
int ic=i;
if(numberOfAnalyses==1)ic=-1;
//analysis[i]=new StEStruct2ptCorrelations(pcuts,analysisMode);
analysis[i]=new StEStructEmptyAnalysis();
analysis[i]->setOutputFileName(getOutFileName(outputDir,jobName,"data",ic));
//analysis[i]->setQAHists(qaHists); // only 1 QA Object in this case
//analysis[i]->setZBuffLimits(ecuts); // common to all
//analysis[i]->setAnalysisIndex(i);
}
estructMaker->SetAnalyses(analysis,numberOfAnalyses);
// --- now do the work ---
estructMaker->Init();
estructMaker->startTimer();
int counter=0, istat=0, i=0;
while (istat!=2) {
istat=estructMaker->Make(); // now includes filling qa histograms
i++; counter++;
if (counter==200) {
cout<<"doing event ="<<i<<endl;
counter=0;
//estructMaker->writeDiagnostics(0);
}
if ( maxNumEvents!=0 && i>=maxNumEvents ) {
istat=2;
}
}
estructMaker->stopTimer();
//--- now write out stats and cuts ---
ofstream ofs(getOutFileName(outputDir,jobName,"stats"));
estructMaker->logAllStats(ofs);
ecuts->printCuts(ofs);
tcuts->printCuts(ofs);
//pcuts->printCuts(ofs);
ofs<<endl;
ofs.close();
// --> root cut file
TFile* tf=new TFile(getOutFileName(outputDir,jobName,"cuts"),"RECREATE");
ecuts->writeCutHists(tf);
tcuts->writeCutHists(tf);
tf->Close();
// --> root qa histogram file
//estructMaker->writeQAHists(getOutFileName(outputDir,jobName,"QA"));
// --- write out the data
estructMaker->Finish();
}
/************************************************************************
* $Id: doEStructPythia.C,v 1.4 2006/04/04 22:15:57 porter Exp $
*
* Author: Jeff Porter
*
* example code for reading in MuDst files contained int "fileListFile"
* and running the 2pt correlations analysis, producing hist files in,
*
* outputDir/jobName/data/XX/
*
* where XX = event level selection imposed by the cut files.
* This example (from PP) contains 3 selections on event mult
*
*************************************************************************/
void doEStructPythia( const char* filelist,
const char* outputDir,
const char* scriptDir,
int maxNumEvents = 0 ) {
gROOT->LoadMacro("load2ptLibs.C");
load2ptLibs();
gROOT->LoadMacro("loadPythiaLibs.C");
loadPythiaLibs();
gROOT->LoadMacro("getPythia.C");
char* jobid=gSystem->Getenv("JOBID");
const char* rframe="CMS";
const char* cproj ="p";
const char* ctarg ="p";
float rts = 200.0;
int pythiaTune = 0; // 1=TuneA, 2=TuneB, all else = minbias standard
TPythia6* pythia=getPythia(rframe,cproj,ctarg,rts,pythiaTune,jobid);
char cutFile[1024];
sprintf(cutFile,"%s/CutsFile.txt",scriptDir);
gROOT->LoadMacro("getOutFileName.C"); // for laying out file names
const char* scratchDir = "PPPythia";
// define centrality
StEStructCentrality* cent=StEStructCentrality::Instance();
const double temp[2]={0,2000};
cent->setCentralities(temp,2);
int mbNBins=cent->numCentralities()-1;
// choose the mode for the binning
int cutBinMode = 1;
StEStructCutBin* cb=StEStructCutBin::Instance();
cb->setMode(cutBinMode);
int mbNCutBins = cb->getNumBins();
// create the generic EStructAnalysisMaker
StEStructAnalysisMaker *estructMaker = new StEStructAnalysisMaker("EStruct2Pythia");
// create the QAHist object (must come after centrality and cutbin objects)
int EventType = 2; // pythia from generator
StEStructQAHists* qaHists = new StEStructQAHists(EventType);
qaHists->initHistograms();
estructMaker->SetQAHists(qaHists);
// Set up the reader with event/track cuts
StEStructEventCuts* ecuts = new StEStructEventCuts(cutFile);
StEStructTrackCuts* tcuts = new StEStructTrackCuts(cutFile);
StEStructPythia* reader = new StEStructPythia(pythia,ecuts,tcuts,false,0,maxNumEvents);
estructMaker->SetEventReader(reader);
StEStruct2ptCorrelations** analysis = new StEStructEmptyAnalysis*[mbNBins];
// StEStructPairCuts* pcuts = new StEStructPairCuts(cutFile);
int analysisMode = 0; // 2pt correlations mode selection
for(int i=0;i<mbNBins;i++){
int ic=i;
if(mbNBins==1)ic=-1;
analysis[i]=new StEStructEmptyAnalysis;
analysis[i]->setOutputFileName(getOutFileName(outputDir,scratchDir,"data",ic));
analysis[i]->setQAHists(qaHists); // only 1 QA Object in this case
analysis[i]->setZBuffLimits(ecuts); // common to all
}
estructMaker->SetAnalyses(analysis,mbNBins);
// --- now do the work ---
estructMaker->Init();
estructMaker->startTimer();
int counter=0, istat=0, i=0;
while (istat!=2) {
istat=estructMaker->Make(); // now includes filling qa histograms
i++; counter++;
if (counter==100) {
cout<<"doing event ="<<i<<endl;
counter=0;
}
if ( maxNumEvents!=0 && i>=maxNumEvents ) {
istat=2;
}
}
estructMaker->stopTimer();
//--- now write out stats and cuts ---
ofstream ofs(getOutFileName(outputDir,scratchDir,"stats"));
estructMaker->logAllStats(ofs);
ecuts->printCuts(ofs);
tcuts->printCuts(ofs);
pcuts->printCuts(ofs);
ofs<<endl;
ofs.close();
// --> root cut file
TFile* tf=new TFile(getOutFileName(outputDir,scratchDir,"cuts"),"RECREATE");
ecuts->writeCutHists(tf);
tcuts->writeCutHists(tf);
tf->Close();
// --> root qa histogram file
estructMaker->writeQAHists(getOutFileName(outputDir,scratchDir,"QA"));
// --- write out the data
estructMaker->Finish();
}
