/**
* The Tree begins with a single short: 0x4D4D
*
* Nodes are laid out in the following format:
* Node Header (short) 2 bytes
* Non-Null Node: 0x0BAB
* Null Node: 0x1ABAB
* Node Data (unsigned char) 1 byte
* Node Frequency (int) 4 bytes
*
* The tree ends with a single short: 0x0FFF
*/
void HuffmanProcessor::WriteTree(Node* root)
{
short non_null_header = 0x0BAB;
if(root->leftChild == NULL && root->rightChild == NULL)
{
// Node header
output.write((char*)&non_null_header, sizeof(non_null_header));
// Data
output.write((char*)&root->data, sizeof(root->data));
output.write((char*)&root->frequency, sizeof(root->frequency));
// Sentinel children
short null_header = 0x1ABAB;
output.write((char*)&null_header, sizeof(short));
output.write((char*)&null_header, sizeof(short));
return;
}
// Node header and data
output.write((char*)&non_null_header, sizeof(non_null_header));
output.write((char*)&root->data, sizeof(root->data));
output.write((char*)&root->frequency, sizeof(root->frequency));
WriteTree(root->leftChild);
WriteTree(root->rightChild);
}
/*
* Play the "animal" game, in which the program attempts to guess an animal
* that the user is thinking of by asking yes or no questions. Eventually,
* the program either will guess the user's animal or run out of questions
* to ask. In the latter case, the program will ask the user to provide a
* yes-or-no question that would distinguish between the user's animal and
* the program's best guess.
* The data structure of questions and guesses is essentially a binary tree,
* with each internal node having a "yes" branch and a "no" branch. Leaves
* of the tree represent animals to be guessed by the program. If the program
* fails to guess the user's animal, it replaces one of the leaves of the tree
* by a node containing the new question, whose children are the program's
* best guess and the animal provided by the user.
* The structure of the program is simple. It initializes the question/guess
* data structure, then plays games as long as the user is interested. In each
* game, the program starts at the top of the tree (the root) and progresses
* toward the bottom (the leaves) depending on the user's responses. Once it
* reaches a leaf, it either has won or lost, and handles the situation as
* described above.
*/
int main (int argc, char *argv[])
{
char *treefile = NULL;
TreeType tree;
PositionType pos;
char *newQuestion, *newAnswer;
if (argc > 1) {
treefile = argv[1];
}
tree = InitTree (treefile);
printf("%s", "Think of an animal. I will try to guess what it is.\n"
"Please answer my questions with yes or no.\n");
while (TRUE) {
pos = Top (tree);
while (!IsLeaf (tree, pos)) {
pos = Answer(Question(tree,pos))?
YesNode(tree,pos): NoNode(tree,pos);
}
if (Answer (Guess (tree, pos))) {
printf ("I got it right!\n");
} else {
GetNewInfo (tree, pos, &newAnswer, &newQuestion);
ReplaceNode (tree, pos, newAnswer, newQuestion);
}
if (!Answer ("Want to play again? ")) {
WriteTree(tree, treefile);
exit (0);
}
}
}
static geBoolean DirTree_WriteToFile1(const DirTree *Tree, geVFile *File, long *Size)
{
DirTree_Header Header;
long StartPosition;
long EndPosition;
if (geVFile_Tell(File, &StartPosition) == GE_FALSE)
return GE_FALSE;
Header.Signature = DIRTREE_FILE_SIGNATURE;
if (geVFile_Seek(File, sizeof(Header), GE_VFILE_SEEKCUR) == GE_FALSE)
return GE_FALSE;
if (WriteTree(Tree, File) == GE_FALSE)
return GE_FALSE;
geVFile_Tell(File, &EndPosition);
Header.Size = EndPosition - StartPosition;
geVFile_Seek(File, StartPosition, GE_VFILE_SEEKSET);
if (geVFile_Write(File, &Header, sizeof(Header)) == GE_FALSE)
return GE_FALSE;
// Make sure that we end up at the end of the directory.
geVFile_Seek(File, EndPosition, GE_VFILE_SEEKSET);
*Size = Header.Size;
return GE_TRUE;
}
void CITPView::Write ()
{
CStringA str;
CResHierarchyRef::Entry *pEntry = m_sRes .GetEntry (0);
UINT32 *plMap = m_sRes .GetMultiMapTable (pEntry);
for (int j = 0; j < pEntry ->ulCount; j++)
{
int nElement = plMap [j];
const CResHierarchyRef::Element *pElement =
m_sRes .GetElement (nElement);
if (pElement ->ulType == CNwnHierarchy::ElementType_LIST)
{
UINT32 *plList = m_sRes .GetListTable (pElement);
long lCount = *plList++;
for (j = 0; j < lCount; j++)
WriteTree (plList [j], 0);
}
}
}
int WriteTree(struct TreeInfo *info, struct Node *n, FILE *sfile, int newlines)
{
if(newlines && save_linelen > 0)
saveputc(sfile, '\n')
SetRootProps(info, n);
saveputc(sfile, '(')
if(!WriteNode(info, n, sfile))
return(FALSE);
n = n->child;
while(n)
{
if(n->sibling)
{
while(n) /* write child + variations */
{
if(!WriteTree(info, n, sfile, 1))
return(FALSE);
n = n->sibling;
}
}
else
{
if(!WriteNode(info, n, sfile)) /* write child */
return(FALSE);
n = n->child;
}
}
saveputc(sfile, ')')
if(newlines != 1)
saveputc(sfile, '\n')
return(TRUE);
}
void SaveSGF(struct SGFInfo *sgf)
{
FILE *sfile;
struct Node *n;
struct TreeInfo *info;
char *c, name[500];
int nl = 0, i = 1;
sgfc = sgf; /* set curent SGFInfo context */
if(strlen(sgf->name) > 480)
PrintError(FE_DEST_NAME_TOO_LONG);
if(option_split_file)
sprintf(name, "%s_%03d.sgf", sgf->name, i);
else
strcpy(name, sgf->name);
sfile = fopen(name, "wb");
if(!sfile)
PrintError(FE_DEST_FILE_OPEN, name);
if(option_keep_head)
{
*sgf->start = '\n';
for(c = sgf->buffer; c <= sgf->start; c++)
if(fputc((*c), sfile) == EOF)
{
fclose(sfile);
PrintError(FE_DEST_FILE_WRITE, name);
}
}
save_linelen = 0;
save_chars_in_node = 0;
save_eol_in_node = 0;
n = sgf->root;
info = sgf->tree;
while(n)
{
if(!WriteTree(info, n, sfile, nl))
{
fclose(sfile);
PrintError(FE_DEST_FILE_WRITE, name);
}
nl = 2;
n = n->sibling;
info = info->next;
if(option_split_file && n)
{
fclose(sfile);
i++;
sprintf(name, "%s_%03d.sgf", sgf->name, i);
if(!(sfile = fopen(name, "wb")))
PrintError(FE_DEST_FILE_OPEN, name);
}
}
fclose(sfile);
}
void CITPView::WriteTree (int nEntry, int nIndent)
{
char szName [64];
char szBlueprint [64];
char szIndent [65];
int nList = -1;
//
// Generate the indent string
//
if (nIndent > _countof (szIndent) - 1)
nIndent = _countof (szIndent) - 1;
memset (szIndent, ' ', nIndent);
szIndent [nIndent] = 0;
//
// Init the string
//
szName [0] = szBlueprint [0] = 0;
//
// Get the entry in question
//
CResHierarchyRef::Entry *pEntry = m_sRes .GetEntry (nEntry);
//
// Collect information about what we need to add
//
UINT32 *plMap = m_sRes .GetMultiMapTable (pEntry);
for (int j = 0; j < pEntry ->ulCount; j++)
{
int nElement = plMap [j];
const CResHierarchyRef::Element *pElement =
m_sRes .GetElement (nElement);
//
// If this is a list, get the element
//
if (pElement ->ulType == CNwnHierarchy::ElementType_LIST)
{
nList = nElement;
}
//
// Otherwise, inspect the variable name to get
// the other information
//
else
{
//
// Get the name
//
CResHierarchyRef::VarName *pVarName =
m_sRes .GetVarName (pElement ->ulVarName);
//
// If this is the strref, get the name
//
if (stricmp (pVarName ->szName, "STRREF") == 0)
{
g_sDialogTlkFile .GetString (pElement ->u32Data,
szName, _countof (szName));
}
//
// If this is a name, get the name
//
else if (stricmp (pVarName ->szName, "NAME") == 0)
{
m_sRes .GetElementLongString (pElement,
szName, _countof (szName));
}
//
// If this is a resref, get the blueprint
//
else if (stricmp (pVarName ->szName, "RESREF") == 0)
{
m_sRes .GetElementShortString (pElement,
szBlueprint, _countof (szBlueprint));
}
}
}
//
// Generate the name
//
CString str (szIndent);
str += szName;
if (szBlueprint [0] != 0)
{
str += _T (" [");
str += szBlueprint;
str += _T ("]");
}
m_strText += str + "\r\n";
//
// If we have a list, add this list
//
if (nList != -1)
{
const CResHierarchyRef::Element *pElement =
m_sRes .GetElement (nList);
UINT32 *plList = m_sRes .GetListTable (pElement);
long lCount = *plList++;
for (j = 0; j < lCount; j++)
WriteTree (plList [j], nIndent + 2);
}
}
static geBoolean WriteTree(const DirTree *Tree, geVFile *File)
{
int Length;
int Terminator;
assert(Tree);
assert(Tree->Name);
Terminator = DIRTREE_LIST_NOTTERMINATED;
if (geVFile_Write(File, &Terminator, sizeof(Terminator)) == GE_FALSE)
return GE_FALSE;
// Write out the name
Length = strlen(Tree->Name) + 1;
if (geVFile_Write(File, &Length, sizeof(Length)) == GE_FALSE)
return GE_FALSE;
if (Length > 0)
{
if (geVFile_Write(File, Tree->Name, Length) == GE_FALSE)
return GE_FALSE;
}
// Write out the attribute information
if (geVFile_Write(File, &Tree->Time, sizeof(Tree->Time)) == GE_FALSE)
return GE_FALSE;
if (geVFile_Write(File, &Tree->AttributeFlags, sizeof(Tree->AttributeFlags)) == GE_FALSE)
return GE_FALSE;
if (geVFile_Write(File, &Tree->Size, sizeof(Tree->Size)) == GE_FALSE)
return GE_FALSE;
if (geVFile_Write(File, &Tree->Offset, sizeof(Tree->Offset)) == GE_FALSE)
return GE_FALSE;
if (geVFile_Write(File, &Tree->Hints.HintDataLength, sizeof(Tree->Hints.HintDataLength)) == GE_FALSE)
return GE_FALSE;
if (Tree->Hints.HintDataLength != 0)
//bug fix. someone got copy happy and forgot to remove the & from Tree->Hints.HintData
if (geVFile_Write(File, Tree->Hints.HintData, Tree->Hints.HintDataLength) == GE_FALSE)
return GE_FALSE;
// Write out the Children
if (Tree->Children)
{
WriteTree(Tree->Children, File);
}
else
{
Terminator = DIRTREE_LIST_TERMINATED;
if (geVFile_Write(File, &Terminator, sizeof(Terminator)) == GE_FALSE)
return GE_FALSE;
}
// Write out the Siblings
if (Tree->Siblings)
{
WriteTree(Tree->Siblings, File);
}
else
{
Terminator = DIRTREE_LIST_TERMINATED;
if (geVFile_Write(File, &Terminator, sizeof(Terminator)) == GE_FALSE)
return GE_FALSE;
}
return GE_TRUE;
}
int main(int argc, char * argv[]) {
// first argument - config file
// second argument - filelist
// using namespace std;
//const int CutNumb = 8;
//string CutList[CutNumb]={"No cut","Trigger","1l","lept-Veto","b-Veto","MET $>$ 50","MET $>$ 100","dPhi $>$ 1"};
// **** configuration
Config cfg(argv[1]);
string Channel="mutau";
// kinematic cuts on electrons
bool fillplots= false;
bool Wtemplate= true;
const bool isData = cfg.get<bool>("IsData");
const bool applyPUreweighting = cfg.get<bool>("ApplyPUreweighting");
const bool applyPUreweighting_vertices = cfg.get<bool>("ApplyPUreweighting_vertices");
const bool applyPUreweighting_official = cfg.get<bool>("ApplyPUreweighting_official");
const bool applyLeptonSF = cfg.get<bool>("ApplyLeptonSF");
const bool InvertTauIso = cfg.get<bool>("InvertTauIso");
const bool InvertLeptonIso = cfg.get<bool>("InvertLeptonIso");
const bool InvertMET = cfg.get<bool>("InvertMET");
const double ptElectronLowCut = cfg.get<double>("ptElectronLowCut");
const double ptElectronHighCut = cfg.get<double>("ptElectronHighCut");
const double etaElectronCut = cfg.get<double>("etaElectronCut");
const double dxyElectronCut = cfg.get<double>("dxyElectronCut");
const double dzElectronCut = cfg.get<double>("dzElectronCut");
const double isoElectronLowCut = cfg.get<double>("isoElectronLowCut");
const double isoElectronHighCut = cfg.get<double>("isoElectronHighCut");
const bool applyElectronId = cfg.get<bool>("ApplyElectronId");
// vertex cuts
const double ndofVertexCut = cfg.get<double>("NdofVertexCut");
const double zVertexCut = cfg.get<double>("ZVertexCut");
const double dVertexCut = cfg.get<double>("DVertexCut");
// kinematic cuts on muons
const double ptMuonLowCut = cfg.get<double>("ptMuonLowCut");
const double ptMuonHighCut = cfg.get<double>("ptMuonHighCut");
const double etaMuonCut = cfg.get<double>("etaMuonCut");
const double dxyMuonCut = cfg.get<double>("dxyMuonCut");
const double dzMuonCut = cfg.get<double>("dzMuonCut");
const double isoMuonLowCut = cfg.get<double>("isoMuonLowCut");
const double isoMuonHighCut = cfg.get<double>("isoMuonHighCut");
const double isoMuonHighCutQCD = cfg.get<double>("isoMuonHighCutQCD");
const bool applyMuonId = cfg.get<bool>("ApplyMuonId");
const double ptTauLowCut = cfg.get<double>("ptTauLowCut");
const double etaTauCut = cfg.get<double>("etaTauCut");
const string dataBaseDir = cfg.get<string>("DataBaseDir");
string TrigLeg ;
if (!isData) TrigLeg = cfg.get<string>("Mu17LegMC");
if (isData) TrigLeg = cfg.get<string>("Mu18LegData");
const string Mu17Tau20MuLegA = cfg.get<string>("Mu17Tau20MuLegA");
const string Mu17Tau20MuLegB = cfg.get<string>("Mu17Tau20MuLegB");
const string Mu17Tau20TauLegA = cfg.get<string>("Mu17Tau20TauLegA");
const string Mu17Tau20TauLegB = cfg.get<string>("Mu17Tau20TauLegB");
const string SingleMuonTriggerFile = cfg.get<string>("Muon17TriggerEff");
const float singleMuonTriggerPtCut = cfg.get<float>("SingleMuonTriggerPtCut");
const float singleMuonTriggerEtaCut = cfg.get<float>("SingleMuonTriggerEtaCut");
const string Region = cfg.get<string>("Region");
const string Sign = cfg.get<string>("Sign");
const double leadchargedhadrcand_dz = cfg.get<double>("leadchargedhadrcand_dz");
const double leadchargedhadrcand_dxy = cfg.get<double>("leadchargedhadrcand_dxy");
// kinematic cuts on Jets
const double etaJetCut = cfg.get<double>("etaJetCut");
const double ptJetCut = cfg.get<double>("ptJetCut");
// topological cuts
const double dRleptonsCutmutau = cfg.get<double>("dRleptonsCutmutau");
const double dZetaCut = cfg.get<double>("dZetaCut");
const double deltaRTrigMatch = cfg.get<double>("DRTrigMatch");
const bool oppositeSign = cfg.get<bool>("oppositeSign");
const bool isIsoR03 = cfg.get<bool>("IsIsoR03");
// tau
const double taupt = cfg.get<double>("taupt");
const double taueta = cfg.get<double>("taueta");
const double decayModeFinding = cfg.get<double>("decayModeFinding");
const double decayModeFindingNewDMs = cfg.get<double>("decayModeFindingNewDMs");
const double againstElectronVLooseMVA5 = cfg.get<double>("againstElectronVLooseMVA5");
const double againstMuonTight3 = cfg.get<double>("againstMuonTight3");
const double vertexz = cfg.get<double>("vertexz");
const double byCombinedIsolationDeltaBetaCorrRaw3Hits = cfg.get<double>("byCombinedIsolationDeltaBetaCorrRaw3Hits");
const unsigned int RunRangeMin = cfg.get<unsigned int>("RunRangeMin");
const unsigned int RunRangeMax = cfg.get<unsigned int>("RunRangeMax");
// vertex distributions filenames and histname
const string vertDataFileName = cfg.get<string>("VertexDataFileName");
const string vertMcFileName = cfg.get<string>("VertexMcFileName");
const string vertHistName = cfg.get<string>("VertexHistName");
// lepton scale factors
const string muonSfDataBarrel = cfg.get<string>("MuonSfDataBarrel");
const string muonSfDataEndcap = cfg.get<string>("MuonSfDataEndcap");
const string muonSfMcBarrel = cfg.get<string>("MuonSfMcBarrel");
const string muonSfMcEndcap = cfg.get<string>("MuonSfMcEndcap");
const string jsonFile = cfg.get<string>("jsonFile");
string cmsswBase = (getenv ("CMSSW_BASE"));
string fullPathToJsonFile = cmsswBase + "/src/DesyTauAnalyses/NTupleMaker/test/json/" + jsonFile;
const string MuonIdIsoFile = cfg.get<string>("MuonIdIsoEff");
const string TauFakeRateFile = cfg.get<string>("TauFakeRateEff");
// Run-lumi selector
std::vector<Period> periods;
if (isData) { // read the good runs
std::fstream inputFileStream(fullPathToJsonFile.c_str(), std::ios::in);
if (inputFileStream.fail() ) {
std::cout << "Error: cannot find json file " << fullPathToJsonFile << std::endl;
std::cout << "please check" << std::endl;
std::cout << "quitting program" << std::endl;
exit(-1);
}
for(std::string s; std::getline(inputFileStream, s); ) {
//std::fstream inputFileStream("temp", std::ios::in);
periods.push_back(Period());
std::stringstream ss(s);
ss >> periods.back();
}
}
TString MainTrigger(TrigLeg);
TString Muon17Tau20MuLegA (Mu17Tau20MuLegA );
TString Muon17Tau20MuLegB (Mu17Tau20MuLegB );
TString Muon17Tau20TauLegA (Mu17Tau20TauLegA );
TString Muon17Tau20TauLegB (Mu17Tau20TauLegB );
const double Lumi = cfg.get<double>("Lumi");
const double bTag = cfg.get<double>("bTag");
const double metcut = cfg.get<double>("metcut");
CutList.clear();
CutList.push_back("No cut");
CutList.push_back("No cut after PU");
CutList.push_back("$\\mu$");
CutList.push_back("$\\tau_h$");
CutList.push_back("Trigger");
CutList.push_back("2nd $\\ell$-Veto");
CutList.push_back("3rd $\\ell$-Veto");
CutList.push_back("Lepton SF");
CutList.push_back("TauFakeRate");
CutList.push_back("topPtRwgt");
CutList.push_back("${M}_T>60");
CutList.push_back("$ E_T^{\\rm miss}>$ 100");
CutList.push_back("Jets $<$3");
CutList.push_back("b-Veto");
CutList.push_back("$40<\\rm{Inv}_M<80");
CutList.push_back("$1.5<\\Delta R<4$");
int CutNumb = int(CutList.size());
xs=1;fact=1;fact2=1;
unsigned int RunMin = 9999999;
unsigned int RunMax = 0;
ifstream ifs("xsecs");
string line;
while(std::getline(ifs, line)) // read one line from ifs
{
fact=fact2=1;
istringstream iss(line); // access line as a stream
// we only need the first two columns
string dt,st1,st2;st1="stau2_1";st2="stau5_2";
iss >> dt >> xs >> fact >> fact2;
//ifs >> dt >> xs; // no need to read further
//cout<< " "<<dt<<" "<<endl;
//cout<< "For sample ========================"<<dt<<" xsecs is "<<xs<<" XSec "<<XSec<<" "<<fact<<" "<<fact2<<endl;
//if (dt==argv[2]) {
//if (std::string::npos != dt.find(argv[2])) {
if ( dt == argv[2]) {
XSec= xs*fact*fact2;
cout<<" Found the correct cross section "<<xs<<" for Dataset "<<dt<<" XSec "<<XSec<<endl;
}
/*
if ( argv[2] == st1) {ChiMass=100;mIntermediate=200;}
else if (argv[2] == st2) {ChiMass=200;mIntermediate=500;}
*/
if (isData) XSec=1.;
ChiMass=0.0;
}
if (XSec<0&& !isData) {cout<<" Something probably wrong with the xsecs...please check - the input was "<<argv[2]<<endl;return 0;}
std::vector<unsigned int> allRuns; allRuns.clear();
cout<<" ChiMass is "<<ChiMass<<" "<<mIntermediate<<endl;
bool doThirdLeptVeto=true;
bool doMuVeto=true;
//CutList[CutNumb]=CutListt[CutNumb];
char ff[100];
sprintf(ff,"%s/%s",argv[3],argv[2]);
if (applyPUreweighting_vertices and applyPUreweighting_official)
{std::cout<<"ERROR: Choose only ONE PU reweighting method (vertices or official, not both!) " <<std::endl; exit(-1);}
// reweighting with vertices
// reading vertex weights
TFile * fileDataNVert = new TFile(TString(cmsswBase)+"/src/"+dataBaseDir+"/"+vertDataFileName);
TFile * fileMcNVert = new TFile(TString(cmsswBase)+"/src/"+dataBaseDir+"/"+vertMcFileName);
TH1D * vertexDataH = (TH1D*)fileDataNVert->Get(TString(vertHistName));
TH1D * vertexMcH = (TH1D*)fileMcNVert->Get(TString(vertHistName));
float normVertexData = vertexDataH->GetSumOfWeights();
float normVertexMc = vertexMcH->GetSumOfWeights();
vertexDataH->Scale(1/normVertexData);
vertexMcH->Scale(1/normVertexMc);
PileUp * PUofficial = new PileUp();
TFile * filePUdistribution_data = new TFile(TString(cmsswBase)+"/src/DesyTauAnalyses/NTupleMaker/data/PileUpDistrib/Data_Pileup_2015D_Nov17.root","read");
TFile * filePUdistribution_MC = new TFile (TString(cmsswBase)+"/src/DesyTauAnalyses/NTupleMaker/data/PileUpDistrib/MC_Spring15_PU25_Startup.root", "read");
TH1D * PU_data = (TH1D *)filePUdistribution_data->Get("pileup");
TH1D * PU_mc = (TH1D *)filePUdistribution_MC->Get("pileup");
PUofficial->set_h_data(PU_data);
PUofficial->set_h_MC(PU_mc);
TFile *f10= new TFile(TString(cmsswBase)+"/src/DesyTauAnalyses/NTupleMaker/data/"+muonSfDataBarrel); // mu SF barrel data
TFile *f11 = new TFile(TString(cmsswBase)+"/src/DesyTauAnalyses/NTupleMaker/data/"+muonSfDataEndcap); // mu SF endcap data
TFile *f12= new TFile(TString(cmsswBase)+"/src/DesyTauAnalyses/NTupleMaker/data/"+muonSfMcBarrel); // mu SF barrel MC
TFile *f13 = new TFile(TString(cmsswBase)+"/src/DesyTauAnalyses/NTupleMaker/data/"+muonSfMcEndcap); // mu SF endcap MC
TGraphAsymmErrors *hEffBarrelData = (TGraphAsymmErrors*)f10->Get("ZMassBarrel");
TGraphAsymmErrors *hEffEndcapData = (TGraphAsymmErrors*)f11->Get("ZMassEndcap");
TGraphAsymmErrors *hEffBarrelMC = (TGraphAsymmErrors*)f12->Get("ZMassBarrel");
TGraphAsymmErrors *hEffEndcapMC = (TGraphAsymmErrors*)f13->Get("ZMassEndcap");
double * dataEffBarrel = new double[10];
double * dataEffEndcap = new double[10];
double * mcEffBarrel = new double[10];
double * mcEffEndcap = new double[10];
dataEffBarrel = hEffBarrelData->GetY();
dataEffEndcap = hEffEndcapData->GetY();
mcEffBarrel = hEffBarrelMC->GetY();
mcEffEndcap = hEffEndcapMC->GetY();
// Lepton Scale Factors
TH1D * MuSF_IdIso_Mu1H = new TH1D("MuIdIsoSF_Mu1H", "MuIdIsoSF_Mu1", 100, 0.5,1.5);
ScaleFactor * SF_muonIdIso;
if (applyLeptonSF) {
SF_muonIdIso = new ScaleFactor();
SF_muonIdIso->init_ScaleFactor(TString(cmsswBase)+"/src/"+TString(MuonIdIsoFile));
}
ScaleFactor * SF_muonTrigger = new ScaleFactor();
SF_muonTrigger->init_ScaleFactor(TString(cmsswBase)+"/src/"+TString(SingleMuonTriggerFile));
////////
cout<<" Will try to initialize the TFR now.... "<<endl;
ScaleFactor * SF_TFR;
bool applyTFR = true;
if (applyTFR) {
SF_TFR = new ScaleFactor();
SF_TFR->init_ScaleFactorb(TString(cmsswBase)+"/src/"+TString(TauFakeRateFile),applyTFR);
}
double Weight=0;
int nTotalFiles = 0;
int iCut=0;
double CFCounter[CutNumb];
double statUnc[CutNumb];
int iCFCounter[CutNumb];
for (int i=0;i < CutNumb; i++){
CFCounter[i] = 0;
iCFCounter[i] = 0;
statUnc[i] =0;
}
// file name and tree name
std::string rootFileName(argv[2]);
//std::ifstream fileList(argv[2]);
std::ifstream fileList(ff);
//std::ifstream fileList0(argv[2]);
std::ifstream fileList0(ff);
std::string ntupleName("makeroottree/AC1B");
std::string initNtupleName("initroottree/AC1B");
TString era=argv[3];
TString invMuStr,invTauStr,invMETStr;
if(InvertLeptonIso) invMuStr = "_InvMuIso_";
if(InvertTauIso) invTauStr = "_InvTauIso_";
if(InvertMET) invMETStr = "_InvMET_";
TString TStrName(rootFileName+invMuStr+invTauStr+invMETStr+"_"+Region+"_"+Sign);
std::cout <<" The filename will be "<<TStrName <<std::endl;
// output fileName with histograms
TFile * file;
if (isData) file = new TFile(era+"/"+TStrName+TString("_DataDriven.root"),"update");
if (!isData) file = new TFile(era+"/"+TStrName+TString(".root"),"update");
file->mkdir(Channel.c_str());
file->cd(Channel.c_str());
int nFiles = 0;
int nEvents = 0;
int selEvents = 0;
int selEventsAllMuons = 0;
int selEventsIdMuons = 0;
int selEventsIsoMuons = 0;
bool lumi=false;
bool isLowIsoMu=false;
bool isHighIsoMu = false;
bool isLowIsoTau=false;
bool isHighIsoTau = false;
std::string dummy;
// count number of files --->
while (fileList0 >> dummy) nTotalFiles++;
SetupHists(CutNumb);
if (argv[4] != NULL && atoi(argv[4])< nTotalFiles) nTotalFiles=atoi(argv[4]);
//if (nTotalFiles>50) nTotalFiles=50;
//nTotalFiles = 10;
for (int iF=0; iF<nTotalFiles; ++iF) {
std::string filen;
fileList >> filen;
std::cout << "file " << iF+1 << " out of " << nTotalFiles << " filename : " << filen << std::endl;
TFile * file_ = TFile::Open(TString(filen));
TH1D * histoInputEvents = NULL;
histoInputEvents = (TH1D*)file_->Get("makeroottree/nEvents");
if (histoInputEvents==NULL) continue;
int NE = int(histoInputEvents->GetEntries());
for (int iE=0;iE<NE;++iE)
inputEventsH->Fill(0.);
std::cout << " number of input events = " << NE << std::endl;
TTree * _inittree = NULL;
_inittree = (TTree*)file_->Get(TString(initNtupleName));
if (_inittree==NULL) continue;
Float_t genweight;
if (!isData)
_inittree->SetBranchAddress("genweight",&genweight);
Long64_t numberOfEntriesInitTree = _inittree->GetEntries();
std::cout << " number of entries in Init Tree = " << numberOfEntriesInitTree << std::endl;
for (Long64_t iEntry=0; iEntry<numberOfEntriesInitTree; iEntry++) {
_inittree->GetEntry(iEntry);
if (isData)
histWeightsH->Fill(0.,1.);
else
histWeightsH->Fill(0.,genweight);
}
TTree * _tree = NULL;
_tree = (TTree*)file_->Get(TString(ntupleName));
if (_tree==NULL) continue;
Long64_t numberOfEntries = _tree->GetEntries();
std::cout << " number of entries in Tree = " << numberOfEntries << std::endl;
AC1B analysisTree(_tree);
// if (std::string::npos != rootFileName.find("TTJetsLO") || std::string::npos != rootFileName.find("TTPow"))
//numberOfEntries = 1000;
// numberOfEntries = 1000;
for (Long64_t iEntry=0; iEntry<numberOfEntries; ++iEntry) {
Float_t weight = 1;
Float_t puweight = 1;
//float topptweight = 1;
analysisTree.GetEntry(iEntry);
nEvents++;
iCut = 0;
//std::cout << " number of entries in Tree = " << numberOfEntries <<" starting weight "<<weight<< std::endl;
if (nEvents%50000==0)
cout << " processed " << nEvents << " events" << endl;
if (fabs(analysisTree.primvertex_z)>zVertexCut) continue;
if (analysisTree.primvertex_ndof<ndofVertexCut) continue;
double dVertex = (analysisTree.primvertex_x*analysisTree.primvertex_x+
analysisTree.primvertex_y*analysisTree.primvertex_y);
if (dVertex>dVertexCut) continue;
if (analysisTree.primvertex_count<2) continue;
//isData= false;
bool lumi=false;
isLowIsoMu=false;
isHighIsoMu = false;
isLowIsoTau=false;
isHighIsoTau = false;
Float_t genweights;
float topPt = 0;
float antitopPt = 0;
bool isZTT = false;
if(!isData)
{
/* TTree *genweightsTree = (TTree*)file_->Get("initroottree/AC1B");
genweightsTree->SetBranchAddress("genweight",&genweights);
Long64_t numberOfEntriesInit = genweightsTree->GetEntries();
for (Long64_t iEntryInit=0; iEntryInit<numberOfEntriesInit; ++iEntryInit) {
genweightsTree->GetEntry(iEntryInit);
histWeightsH->Fill(0.,genweights);
}
*/ /*
for (unsigned int igent=0; igent < analysisTree.gentau_count; ++igent) {
if (analysisTree.gentau_isPrompt[igent]) isZTT = true;
}
*/
for (unsigned int igen=0; igen<analysisTree.genparticles_count; ++igen) {
// cout<< " info = " << int(analysisTree.genparticles_count) <<" "<<int(analysisTree.genparticles_pdgid[igen])<<endl;
if (analysisTree.genparticles_pdgid[igen]==6)
topPt = TMath::Sqrt(analysisTree.genparticles_px[igen]*analysisTree.genparticles_px[igen]+
analysisTree.genparticles_py[igen]*analysisTree.genparticles_py[igen]);
if (analysisTree.genparticles_pdgid[igen]==-6)
antitopPt = TMath::Sqrt(analysisTree.genparticles_px[igen]*analysisTree.genparticles_px[igen]+
analysisTree.genparticles_py[igen]*analysisTree.genparticles_py[igen]);
}
weight *= analysisTree.genweight;
lumi=true;
//cout<<" weight from init "<<genweights<< " "<<analysisTree.genweight<<" "<<weight<<endl;
/*
if (applyPUreweighting) {
int binNvert = vertexDataH->FindBin(analysisTree.primvertex_count);
float_t dataNvert = vertexDataH->GetBinContent(binNvert);
float_t mcNvert = vertexMcH->GetBinContent(binNvert);
if (mcNvert < 1e-10){mcNvert=1e-10;}
float_t vertWeight = dataNvert/mcNvert;
weight *= vertWeight;
// cout << "NVert = " << analysisTree.primvertex_count << " weight = " << vertWeight << endl;
}
*/
}
if (isData) {
XSec = 1.;
histRuns->Fill(analysisTree.event_run);
///////////////according to dimuons
int n=analysisTree.event_run;
int lum = analysisTree.event_luminosityblock;
std::string num = std::to_string(n);
std::string lnum = std::to_string(lum);
for(const auto& a : periods)
{
if ( num.c_str() == a.name ) {
//std::cout<< " Eureka "<<num<<" "<<a.name<<" ";
// std::cout <<"min "<< last->lower << "- max last " << last->bigger << std::endl;
for(auto b = a.ranges.begin(); b != std::prev(a.ranges.end()); ++b) {
// cout<<b->lower<<" "<<b->bigger<<endl;
if (lum >= b->lower && lum <= b->bigger ) lumi = true;
}
auto last = std::prev(a.ranges.end());
// std::cout <<"min "<< last->lower << "- max last " << last->bigger << std::endl;
if ( (lum >=last->lower && lum <= last->bigger )) lumi=true;
}
}
if (!lumi) continue;
//if (lumi ) cout<<" ============= Found good run"<<" "<<n<<" "<<lum<<endl;
}
if (analysisTree.event_run<RunMin)
RunMin = analysisTree.event_run;
if (analysisTree.event_run>RunMax)
RunMax = analysisTree.event_run;
//std::cout << " Run : " << analysisTree.event_run << std::endl;
bool isNewRun = true;
if (allRuns.size()>0) {
for (unsigned int iR=0; iR<allRuns.size(); ++iR) {
if (analysisTree.event_run==allRuns.at(iR)) {
isNewRun = false;
break;
}
}
}
if (isNewRun)
allRuns.push_back(analysisTree.event_run);
if (!lumi) continue;
JetsMV.clear();
ElMV.clear();
TauMV.clear();
MuMV.clear();
LeptMV.clear();
mu_index=-1;
tau_index=-1;
el_index=-1;
double MET = sqrt ( analysisTree.pfmet_ex*analysisTree.pfmet_ex + analysisTree.pfmet_ey*analysisTree.pfmet_ey);
METV.SetPx(analysisTree.pfmet_ex);
METV.SetPy(analysisTree.pfmet_ey);
METV.SetPz(analysisTree.pfmet_ez);
METV.SetPhi(analysisTree.pfmet_phi);
if(fillplots)
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
for (unsigned int ijj = 0; ijj<analysisTree.pfjet_count; ++ijj) {
JetsV.SetPxPyPzE(analysisTree.pfjet_px[ijj], analysisTree.pfjet_py[ijj], analysisTree.pfjet_pz[ijj], analysisTree.pfjet_e[ijj]);
JetsMV.push_back(JetsV);
}
for (unsigned int imm = 0; imm<analysisTree.muon_count; ++imm) {
MuV.SetPtEtaPhiM(analysisTree.muon_pt[imm], analysisTree.muon_eta[imm], analysisTree.muon_phi[imm], muonMass);
MuMV.push_back(MuV);
// mu_index=0;
}
for (unsigned int ie = 0; ie<analysisTree.electron_count; ++ie) {
ElV.SetPtEtaPhiM(analysisTree.electron_pt[ie], analysisTree.electron_eta[ie], analysisTree.electron_phi[ie], electronMass);
ElMV.push_back(ElV);
// el_index=0;
}
for (unsigned int itt = 0; itt<analysisTree.tau_count; ++itt) {
TauV.SetPtEtaPhiM(analysisTree.tau_pt[itt], analysisTree.tau_eta[itt], analysisTree.tau_phi[itt], tauMass);
TauMV.push_back(TauV);
// tau_index=0;
}
if (!isData )
{
if (applyPUreweighting) {
puweight = float(PUofficial->get_PUweight(double(analysisTree.numtruepileupinteractions)));
weight *=puweight;
}
}
// vector <string> ss; ss.push_back(.c_str());
if(fillplots)
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
//selecTable.Fill(1,0, weight );
bool trigAccept = false;
unsigned int nMainTrigger = 0;
bool isMainTrigger = false;
unsigned int nfilters = analysisTree.run_hltfilters->size();
// std::cout << "nfiltres = " << nfilters << std::endl;
for (unsigned int i=0; i<nfilters; ++i) {
// std::cout << "HLT Filter : " << i << " = " << analysisTree.run_hltfilters->at(i) << std::endl;
TString HLTFilter(analysisTree.run_hltfilters->at(i));
if (HLTFilter==MainTrigger) {
nMainTrigger = i;
isMainTrigger = true;
}
}
if (!isMainTrigger) {
std::cout << "HLT filter for Mu20 " << MainTrigger << " not found" << std::endl;
return(-1);
}
/////now clear the Mu.El.Jets again to fill them again after cleaning
MuMV.clear();
ElMV.clear();
TauMV.clear();
LeptMV.clear();
double isoMuMin = 9999;
bool mu_iso=false;
vector<int> muons; muons.clear();
for (unsigned int im = 0; im<analysisTree.muon_count; ++im) {
if (analysisTree.muon_pt[im]<ptMuonLowCut) continue;
if (fabs(analysisTree.muon_eta[im])>etaMuonCut) continue;
if (fabs(analysisTree.muon_dxy[im])>dxyMuonCut) continue;
if (fabs(analysisTree.muon_dz[im])>dzMuonCut) continue;
double absIso= analysisTree.muon_r03_sumChargedHadronPt[im]
+ max(analysisTree.muon_r03_sumNeutralHadronEt[im] + analysisTree.muon_r03_sumPhotonEt[im]
- 0.5 * analysisTree.muon_r03_sumPUPt[im],0.0);
double relIso = absIso/analysisTree.muon_pt[im];
if (relIso<isoMuonLowCut) continue;
if (applyMuonId && !analysisTree.muon_isMedium[im]) continue;
//cout<<" after muIso index "<<int(mu_index)<<" pT "<<analysisTree.muon_pt[im]<<" relIso "<<relIso<<" isoMuMin "<<isoMuMin<<" muon_count "<<analysisTree.muon_count<<" im "<<im<<" event "<<iEntry<<endl;
if (double(relIso)<double(isoMuMin)) {
isoMuMin = relIso;
mu_index = int(im);
mu_iso=true;
//cout<<" after muIso index "<<int(mu_index)<<" pT "<<analysisTree.muon_pt[im]<<" relIso "<<relIso<<" isoMuMin "<<isoMuMin<<" muon_count "<<analysisTree.muon_count<<" im "<<im<<" event "<<iEntry<<endl;
muons.push_back(im);
MuV.SetPtEtaPhiM(analysisTree.muon_pt[mu_index], analysisTree.muon_eta[mu_index], analysisTree.muon_phi[mu_index], muonMass);
MuMV.push_back(MuV);
LeptMV.push_back(MuV);
}
//cout<<" Indexes here "<<im<<" "<<mu_index<<endl;
if (relIso == isoMuMin && im != mu_index) {
//cout<<" found a pair for muons " <<relIso <<" mu_index "<<mu_index<<" pT "<<analysisTree.muon_pt[int(mu_index)]<<" new index "<<im<<" pT "<<analysisTree.muon_pt[int(im)]<<" event "<<iEntry<<endl;
analysisTree.muon_pt[im] > analysisTree.muon_pt[mu_index] ? mu_index = int(im) : mu_index = mu_index;
}
}
if (muons.size()==0 || !mu_iso ) continue;
double absIso= analysisTree.muon_r03_sumChargedHadronPt[mu_index]
+ max(analysisTree.muon_r03_sumNeutralHadronEt[mu_index] + analysisTree.muon_r03_sumPhotonEt[mu_index]
- 0.5 * analysisTree.muon_r03_sumPUPt[mu_index],0.0);
double relIso = absIso/analysisTree.muon_pt[mu_index];
if (relIso>isoMuonHighCut && !InvertLeptonIso) continue;
if (relIso>isoMuonHighCutQCD ) { isHighIsoMu=true ;isLowIsoMu=false;}
else { isHighIsoMu = false;isLowIsoMu=true;}
sort(LeptMV.begin(), LeptMV.end(),ComparePt);
if (LeptMV.size() == 0 ) continue;
if (InvertLeptonIso && !isHighIsoMu) continue;
if (!InvertLeptonIso && isHighIsoMu) continue;
if (InvertLeptonIso && isLowIsoMu) continue;
//cout<<" Iso check "<<relIso<<" InvertLeptonIso "<<InvertLeptonIso<<" isHighIsoMu "<<isHighIsoMu<<" isLowIsoMu "<<isLowIsoMu<<" cutQCD "<<isoMuonHighCutQCD<<endl;
if(fillplots)
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
double isoTauMin = 999;
bool tau_iso = false;
vector<int> tau; tau.clear();
for (unsigned int it = 0; it<analysisTree.tau_count; ++it) {
if (analysisTree.tau_pt[it] < ptTauLowCut || fabs(analysisTree.tau_eta[it])> etaTauCut) continue;
if (analysisTree.tau_decayModeFindingNewDMs[it]<decayModeFindingNewDMs) continue;
if ( fabs(analysisTree.tau_leadchargedhadrcand_dz[it])> leadchargedhadrcand_dz) continue;
if (analysisTree.tau_againstElectronVLooseMVA5[it]<againstElectronVLooseMVA5) continue;
if (analysisTree.tau_againstMuonTight3[it]<againstMuonTight3) continue;
//cout<<" "<<analysisTree.tau_byMediumCombinedIsolationDeltaBetaCorr3Hits[it]<<endl;
if (!InvertTauIso && analysisTree.tau_byCombinedIsolationDeltaBetaCorrRaw3Hits[it] > byCombinedIsolationDeltaBetaCorrRaw3Hits ) continue;
//if (!InvertTauIso && analysisTree.tau_byMediumCombinedIsolationDeltaBetaCorr3Hits[it] < 0.5 ) continue;
double tauIso = analysisTree.tau_byCombinedIsolationDeltaBetaCorrRaw3Hits[it];
if (tauIso<isoTauMin ) {
// cout<<" there was a chenge "<<tauIso<<" "<<isoTauMin<<" it "<<it<<" tau_index "<<tau_index<<" "<<analysisTree.tau_count<<endl;
isoTauMin = tauIso;
tau_iso=true;
tau_index = (int)it;
tau.push_back(tau_index);
TauV.SetPtEtaPhiM(analysisTree.tau_pt[tau_index], analysisTree.tau_eta[tau_index], analysisTree.tau_phi[tau_index], tauMass);
TauMV.push_back(TauV);
}
continue;
if (tauIso==isoTauMin && it != tau_index) {
//analysisTree.tau_pt[it] > analysisTree.tau_pt[tau_index] ? tau_index = it : tau_index = tau_index;
if (analysisTree.tau_pt[it] > analysisTree.tau_pt[tau_index] ) tau_index = (int)it ;
//cout<<" found a pair " <<tauIso <<" "<<tau_index<<" "<<it<<endl;
}
}
if (tau.size()==0 || !tau_iso ) continue;
// cout<< " Lets check "<<mu_index <<" "<<tau_index <<" "<<endl;
//cout<<" "<<endl;
////////////////////change to new tau inverted definition
double tauIsoI = analysisTree.tau_byMediumCombinedIsolationDeltaBetaCorr3Hits[tau_index];
if (tauIsoI > 0.5 && InvertTauIso) {isHighIsoTau =true;}
//else {isHighIsoTau =false ; isLowIsoTau=true;}
//if (isHighIsoTau && tauIso > 2*byCombinedIsolationDeltaBetaCorrRaw3Hits ) continue;
if (InvertTauIso && !isHighIsoTau) continue;
if (!InvertTauIso && isHighIsoTau) continue;
//if (InvertTauIso && isLowIsoTau) continue;
/*
continue;
double isoTauMin = 999;
bool tau_iso = false;
vector<int> tau; tau.clear();
for (unsigned int it = 0; it<analysisTree.tau_count; ++it) {
if (analysisTree.tau_pt[it] < ptTauLowCut || fabs(analysisTree.tau_eta[it])> etaTauCut) continue;
if (analysisTree.tau_decayModeFindingNewDMs[it]<decayModeFindingNewDMs) continue;
if ( fabs(analysisTree.tau_leadchargedhadrcand_dz[it])> leadchargedhadrcand_dz) continue;
if (analysisTree.tau_againstElectronVLooseMVA5[it]<againstElectronVLooseMVA5) continue;
if (analysisTree.tau_againstMuonTight3[it]<againstMuonTight3) continue;
//if (!InvertTauIso && analysisTree.tau_byCombinedIsolationDeltaBetaCorrRaw3Hits[it] > byCombinedIsolationDeltaBetaCorrRaw3Hits ) continue;
cout<<" "<<analysisTree.tau_byMediumCombinedIsolationDeltaBetaCorr3Hits[it]<<endl;
//aif (!InvertTauIso && analysisTree.tau_byMediumCombinedIsolationDeltaBetaCorr3Hits[it] < 0.5 ) continue;
double tauIso = analysisTree.tau_byCombinedIsolationDeltaBetaCorrRaw3Hits[it];
if (tauIso<isoTauMin ) {
// cout<<" there was a chenge "<<tauIso<<" "<<isoTauMin<<" it "<<it<<" tau_index "<<tau_index<<" "<<analysisTree.tau_count<<endl;
isoTauMin = tauIso;
tau_iso=true;
tau_index = int(it);
tau.push_back(tau_index);
TauV.SetPtEtaPhiM(analysisTree.tau_pt[tau_index], analysisTree.tau_eta[tau_index], analysisTree.tau_phi[tau_index], tauMass);
TauMV.push_back(TauV);
}
if (tauIso==isoTauMin && it != tau_index) {
analysisTree.tau_pt[it] > analysisTree.tau_pt[tau_index] ? tau_index = int(it) : tau_index = tau_index;
//cout<<" found a pair " <<tauIso <<" "<<tau_index<<" "<<it<<endl;
}
}
if (tau.size()==0 || !tau_iso ) continue;
double tauIsoI = analysisTree.tau_byMediumCombinedIsolationDeltaBetaCorr3Hits[tau_index];
if (tauIsoI > 0.5 && InvertTauIso) {isHighIsoTau =true;}
//else {isHighIsoTau =false ; isLowIsoTau=true;}
//if (isHighIsoTau && tauIso > 2*byCombinedIsolationDeltaBetaCorrRaw3Hits ) continue;
if (InvertTauIso && !isHighIsoTau) continue;
if (!InvertTauIso && isHighIsoTau) continue;
//if (InvertTauIso && isLowIsoTau) continue;
*/
double q = analysisTree.tau_charge[tau_index] * analysisTree.muon_charge[mu_index];
if (q>0 && Sign=="OS" ) continue;
if (q<0 && Sign=="SS" ) continue;
bool regionB = (q<0 && isLowIsoMu);
bool regionA = (q>0 && isLowIsoMu);
bool regionC = (q<0 && isHighIsoMu);
bool regionD = (q>0 && isHighIsoMu);
if(fillplots)
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
//cout<<" HOW MANY MUONS DO I HAVE ?? "<<muons.size()<<endl;
bool isdRLeptonMatched = false;
for (unsigned int iT=0; iT<analysisTree.trigobject_count; ++iT) {
if (analysisTree.trigobject_filters[iT][nMainTrigger]) { // Mu17 Leg
double dRtrig = deltaR(analysisTree.muon_eta[mu_index],analysisTree.muon_phi[mu_index],
analysisTree.trigobject_eta[iT],analysisTree.trigobject_phi[iT]);
if (!isData && analysisTree.trigobject_filters[iT][nMainTrigger] && analysisTree.trigobject_pt[iT]>singleMuonTriggerPtCut && dRtrig<deltaRTrigMatch)
isdRLeptonMatched = true;
if (isData && dRtrig<deltaRTrigMatch) isdRLeptonMatched=true;
}
}
if (!isdRLeptonMatched) continue;
double dR = deltaR(analysisTree.tau_eta[tau_index],analysisTree.tau_phi[tau_index],
analysisTree.muon_eta[mu_index],analysisTree.muon_phi[mu_index]);
if (dR<dRleptonsCutmutau) continue;
double ptMu1 = (double)analysisTree.muon_pt[mu_index];
double etaMu1 = (double)analysisTree.muon_eta[mu_index];
float trigweight=1.;
float Mu17EffData = (float)SF_muonTrigger->get_EfficiencyData(double(ptMu1),double(etaMu1));
float Mu17EffMC = (float)SF_muonTrigger->get_EfficiencyMC(double(ptMu1),double(etaMu1));
if (!isData) {
if (Mu17EffMC>1e-6)
trigweight = Mu17EffData / Mu17EffMC;
weight *= trigweight;
// cout<<" Trigger weight "<<trigweight<<endl;
}
if(fillplots)
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
//Set this flag if there is an opposite-charge muon pair in the event with muons separated by DR>0.15 and both passing the loose selection:
bool MuVeto=false;
if (doMuVeto){
if (muons.size()>1){
for (unsigned int imv = 0; imv<analysisTree.muon_count; ++imv) {
if ( imv != mu_index ){
double absIso= analysisTree.muon_r03_sumChargedHadronPt[imv]
+ max(analysisTree.muon_r03_sumNeutralHadronEt[imv] + analysisTree.muon_r03_sumPhotonEt[imv]
- 0.5 * analysisTree.muon_r03_sumPUPt[imv],0.0);
double relIso = absIso/analysisTree.muon_pt[imv];
double dRr = deltaR(analysisTree.muon_eta[mu_index],analysisTree.muon_phi[mu_index],
analysisTree.muon_eta[imv],analysisTree.muon_phi[imv]);
bool OSCharge = false;
if ( imv != mu_index && analysisTree.muon_charge[imv] != analysisTree.muon_charge[mu_index] ) OSCharge=true;
//if ( analysisTree.muon_charge[imv] != analysisTree.muon_charge[mu_index] && analysisTree.muon_isGlobal[imv] && analysisTree.muon_isTracker[imv] && analysisTree.muon_isPF[imv]
if ( analysisTree.muon_charge[imv] != analysisTree.muon_charge[mu_index] && analysisTree.muon_isGlobal[imv] && analysisTree.muon_isTracker[imv] && analysisTree.muon_isPF[imv]
&& analysisTree.muon_pt[imv]> 15 && fabs(analysisTree.muon_eta[imv])< 2.4 && fabs(analysisTree.muon_dxy[imv])<0.045
&& fabs(analysisTree.muon_dz[imv] < 0.2 && relIso< 0.3 && analysisTree.muon_isMedium[imv]) && dRr > 0.15 && OSCharge) //removed from last recipe
MuVeto=true;
}
}
}
}
if (MuVeto) continue;
if(fillplots)
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
bool ThirdLeptVeto=false;
if (doThirdLeptVeto){
if (analysisTree.electron_count>0) {
for (unsigned int iev = 0; iev<analysisTree.electron_count; ++iev) {
/*
double neutralIsoV = analysisTree.electron_r03_sumNeutralHadronEt[iev] + analysisTree.electron_r03_sumNeutralHadronEt[iev] + analysisTree.electron_r03_sumPhotonEt[iev] - 4*TMath::Pi()*(0.3*0.3)*analysisTree.rho;
double IsoWithEA = analysisTree.electron_r03_sumChargedHadronPt[iev] + TMath::Max(double(0), neutralIsoV);
*/
double IsoWithEA = analysisTree.electron_r03_sumChargedHadronPt[iev]
+ max(analysisTree.electron_r03_sumNeutralHadronEt[iev] + analysisTree.electron_r03_sumPhotonEt[iev]
- 0.5 * analysisTree.electron_r03_sumPUPt[iev], 0.0) ;
double relIsoV = IsoWithEA/analysisTree.electron_pt[iev];
bool electronMvaId = electronMvaIdWP90(analysisTree.electron_pt[iev], analysisTree.electron_superclusterEta[iev], analysisTree.electron_mva_id_nontrigPhys14[iev]);
if ( iev != el_index && analysisTree.electron_pt[iev] > 10 && fabs(analysisTree.electron_eta[iev]) < 2.5 && fabs(analysisTree.electron_dxy[iev])<0.045
&& fabs(analysisTree.electron_dz[iev]) < 0.2 && relIsoV< 0.3 && electronMvaId && analysisTree.electron_pass_conversion[iev]
&& analysisTree.electron_nmissinginnerhits[iev] <=1) ThirdLeptVeto=true;
}
}
if (analysisTree.muon_count>0){
for (unsigned int imvv = 0; imvv<analysisTree.muon_count; ++imvv) {
// if ( imvv != mu_index && analysisTree.muon_charge[imvv] != analysisTree.muon_charge[mu_index] ){
double absIso= analysisTree.muon_r03_sumChargedHadronPt[imvv]
+ max(analysisTree.muon_r03_sumNeutralHadronEt[imvv] + analysisTree.muon_r03_sumPhotonEt[imvv]
- 0.5 * analysisTree.muon_r03_sumPUPt[imvv],0.0);
double relIso = absIso/analysisTree.muon_pt[imvv];
if ( imvv != mu_index && analysisTree.muon_isMedium[imvv] && analysisTree.muon_pt[imvv]> 10 && fabs(analysisTree.muon_eta[imvv])< 2.4 && fabs(analysisTree.muon_dxy[imvv])<0.045
&& fabs(analysisTree.muon_dz[imvv] < 0.2 && relIso< 0.3 && analysisTree.muon_isMedium[imvv]) ) ThirdLeptVeto=true;
}
}
}
if (ThirdLeptVeto) continue;
if(fillplots)
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
if (!isData && applyLeptonSF) {
//leptonSFweight = SF_yourScaleFactor->get_ScaleFactor(pt, eta)
double ptMu1 = (double)analysisTree.muon_pt[mu_index];
double etaMu1 = (double)analysisTree.muon_eta[mu_index];
double IdIsoSF_mu1 = SF_muonIdIso->get_ScaleFactor(ptMu1, etaMu1);
MuSF_IdIso_Mu1H->Fill(IdIsoSF_mu1);
weight = weight*IdIsoSF_mu1;
}
if(fillplots)
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
TLorentzVector muVc ; muVc.SetPtEtaPhiM(analysisTree.muon_pt[mu_index], analysisTree.muon_eta[mu_index], analysisTree.muon_phi[mu_index], muonMass);
TLorentzVector tauVc; tauVc.SetPtEtaPhiM(analysisTree.tau_pt[tau_index], analysisTree.tau_eta[tau_index], analysisTree.tau_phi[tau_index], tauMass);
double MTv = mT(muVc,METV);
if (!isData && applyTFR) {
//leptonSFweight = SF_yourScaleFactor->get_ScaleFactor(pt, eta)
double ptTau1 = (double)analysisTree.tau_pt[tau_index];
double etaTau1 = (double)analysisTree.tau_eta[tau_index];
double TFRSF_mu1 = SF_TFR->get_ScaleFactor(ptTau1, etaTau1);
MuSF_IdIso_Mu1H->Fill(TFRSF_mu1);
weight = weight*TFRSF_mu1;
//cout<<" "<<TFRSF_mu1<<" for eta "<<etaTau1<< " pT "<< ptTau1<<endl;
}
if(fillplots)
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
if (!isData && ( string::npos != filen.find("TTJets") || string::npos != filen.find("TTPowHeg")) )
//if (!isData )
{
if (topPt>0.&&antitopPt>0.) {
float topptweight = topPtWeight(topPt,antitopPt);
// cout<<" "<<topPt<<" "<<antitopPt<<endl;
weight *= topptweight;
}
}
if(fillplots)
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
if (MTv<60 ) continue;
if(fillplots)
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
// for (unsigned int j=0;j<LeptMV.size();++j) cout<<" j "<<j<<" "<<LeptMV.at(j).Pt()<<endl;
// cout<<""<<endl;
////////jets cleaning
vector<int> jets; jets.clear();
TLorentzVector leptonsV, muonJ, jetsLV;
// continue;
//JetsV.SetPxPyPzE(analysisTree.pfjet_px[ij], analysisTree.pfjet_py[ij], analysisTree.pfjet_pz[ij], analysisTree.pfjet_e[ij]);
//double ETmiss = TMath::Sqrt(analysisTree.pfmet_ex*analysisTree.pfmet_ex + analysisTree.pfmet_ey*analysisTree.pfmet_ey);
double ETmiss = METV.Pt();//TMath::Sqrt(analysisTree.pfmet_ex*analysisTree.pfmet_ex + analysisTree.pfmet_ey*analysisTree.pfmet_ey);
if (InvertMET && ETmiss > 100. ) continue;
if (!InvertMET && ETmiss < 100. ) continue; //that is the nominal selection ie MET > 100
if(fillplots)
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
double ptScalarSum = -1;
bool btagged= false;
JetsMV.clear();
float jetEtaCut = 2.4;
float DRmax = 0.5;
int countjets = 0;
for (unsigned int jet=0; jet<analysisTree.pfjet_count; ++jet) {
float absJetEta = fabs(analysisTree.pfjet_eta[jet]);
if (absJetEta > etaJetCut) continue;
if (fabs(analysisTree.pfjet_pt[jet])<ptJetCut) continue;
//double Dr= deltaR(LeptMV.at(il).Eta(), LeptMV.at(il).Phi(),
bool isPFJetId = false ;
isPFJetId =looseJetiD(analysisTree,jet);
if (!isPFJetId) continue;
//for (unsigned int lep=0;LeptMV.size();lep++){
//double Dr=(LeptMV.at(lep).Eta(),LeptMV.at(lep).Phi(),
double Dr=deltaR(analysisTree.muon_eta[mu_index],analysisTree.muon_phi[mu_index],
analysisTree.pfjet_eta[jet],analysisTree.pfjet_phi[jet]);
if ( Dr < DRmax) continue;
double Drr=deltaR(analysisTree.tau_eta[tau_index],analysisTree.tau_phi[tau_index],
analysisTree.pfjet_eta[jet],analysisTree.pfjet_phi[jet]);
if ( Drr < DRmax) continue;
if (analysisTree.pfjet_btag[jet][0] > bTag) btagged = true;
JetsV.SetPxPyPzE(analysisTree.pfjet_px[jet], analysisTree.pfjet_py[jet], analysisTree.pfjet_pz[jet], analysisTree.pfjet_e[jet]);
JetsMV.push_back(JetsV);
countjets++;
}
if (countjets >2 ) continue;
if(fillplots)
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
if (btagged ) continue;
if(fillplots)
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
// pt Scalar
//cout<<" "<<mu_index<<" "<<tau_index<<" "<<MuMV.at(mu_index).M()<<" "<<TauMV.at(tau_index).M()<<endl;
TLorentzVector diL = muVc + tauVc;
if ( diL.M() < 100 ) continue;
if(fillplots)
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
/* if (ETmiss < 100) continue;
if (ETmiss < 120) continue;
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
// topological cut
//if (DZeta<dZetaCut) continue;
*/
//double dRr = deltaR(diL.Eta(), diL.Phi(), METV.Eta(), METV.Phi());
double dRr = deltaR(muVc.Eta(), muVc.Phi(), tauVc.Eta(), tauVc.Phi());
if (dRr>3 ) continue;
if(fillplots)
FillMainHists(iCut, weight, ElMV, MuMV, TauMV,JetsMV,METV, ChiMass,mIntermediate,analysisTree, Channel, mu_index,el_index,tau_index);
CFCounter[iCut]+= weight;
iCFCounter[iCut]++;
iCut++;
FillTree();
selEvents++;
} // end of file processing (loop over events in one file)
nFiles++;
delete _tree;
file_->Close();
delete file_;
}
cout<<"done"<<endl;
cout<<" Total events "<<nEvents<<" Will use weight "<<histWeightsH->GetSumOfWeights()<<" Norm Factor for a Lumi of "<<Lumi<<"/pb is "<<XSec*Lumi/( histWeightsH->GetSumOfWeights())<<endl;
cout<<" First content "<<CFCounter[0]<<endl;
cout<<" Run range from -----> "<<RunMin<<" to "<<RunMax<<endl;
/*
for (int i=0;i<CutNumb;++i){
CFCounter[i] *= double(XSec*Lumi/( histWeights->GetSumOfWeights()));
if (iCFCounter[i] <0.2) statUnc[i] =0;
else statUnc[i] = CFCounter[i]/sqrt(iCFCounter[i]);
}
*/
//write out cutflow
ofstream tfile;
// TString outname = argv[argc-1];
TString outname=argv[2];
TString textfilename = "cutflow_"+outname+"_"+Channel+"_"+argv[3]+".txt";
// tfile.open(textfilename);
// tfile << "########################################" << endl;
for(int ci = 0; ci < CutNumb; ci++)
{
// tfile << CutList[ci]<<"\t & \t"
// << CFCounter[ci] <<"\t & \t"<< statUnc[ci] <<"\t & \t"<< iCFCounter[ci] << endl;
CutFlowUnW->SetBinContent(1+ci,0);
CutFlow->SetBinContent(1+ci,0);
CutFlowUnW->SetBinContent(1+ci,float(CFCounter[ci]) );
CFCounter[ci] *= double(XSec*Lumi/( histWeightsH->GetSumOfWeights()));
CutFlow->SetBinContent(1+ci,float(CFCounter[ci]));
cout << " i "<<ci<<" "<<iCFCounter[ci]<<" "<<XSec*Lumi/( histWeightsH->GetSumOfWeights())<<" "<<CutFlowUnW->GetBinContent(1+ci)<<" "<<CutFlow->GetBinContent(1+ci)<<endl;
if (iCFCounter[ci] <0.2) statUnc[ci] =0;
//else statUnc[i] = CFCounter[i]/sqrt(iCFCounter[i]);
else statUnc[ci] = sqrt(CFCounter[ci]);
}
//ofstream tfile1;
//TString textfile_Con = "CMG_cutflow_Con_Mu_"+outname+".txt";
//tfile1.open(textfile_Con);
//tfile1 << "########################################" << endl;
//tfile << "Cut efficiency numbers:" << endl;
// tfile << " Cut "<<"\t & \t"<<"#Evnts for "<<Lumi/1000<<" fb-1 & \t"<<" Uncertainty \t"<<" cnt\t"<<endl;
// tfile.close();
std::cout << std::endl;
int allEvents = int(inputEventsH->GetEntries());
std::cout << "Total number of input events = " << allEvents << std::endl;
std::cout << "Total number of events in Tree = " << nEvents << std::endl;
std::cout << "Total number of selected events = " << selEvents << std::endl;
std::cout << std::endl;
file->cd(Channel.c_str());
WriteTree();
hxsec->Fill(XSec);
hxsec->Write();
inputEventsH->Write();
histWeightsH->Write();
histRuns->Write();
CutFlowUnW->Write();
CutFlow->Write();
MuSF_IdIso_Mu1H->Write();
file->Write();
file->Close();
delete file;
}
void HuffmanProcessor::Compress(char* output_filename)
{
output_file = output_filename;
// Start all frequencies out at zero.
for(int i = 0; i < 255; ++i)
{
frequencies.insert(pair<unsigned char, int>((unsigned char)i, 0));
}
// Build histogram
for(int curr_byte = 0; curr_byte < size; ++curr_byte)
{
++frequencies[buffer[curr_byte]];
}
// Print the frequency table to a file for debugging
ofstream debug_output = ofstream("frequency_table.txt");
for(std::map<unsigned char, int>::iterator itr = frequencies.begin(); itr != frequencies.end(); ++itr)
{
debug_output << "Byte: " << (*itr).first << " | " << "Frequency: " << (*itr).second << "\n";
}
// For each nonzero frequency byte, create a new leaf node in the tree.
for(std::map<unsigned char, int>::iterator itr = frequencies.begin(); itr != frequencies.end(); ++itr)
{
if((*itr).second)
{
nodes.push(new Node((*itr).first, (*itr).second));
}
}
while(nodes.size() > 1)
{
Node* lowest = new Node();
Node* secondLowest = new Node();
// Find the two least frequent nodes.
lowest = nodes.top();
nodes.pop();
secondLowest = nodes.top();
nodes.pop();
// Make a new node the parent of the two we just removed where its
// frequency is the sum of the bottom two.
Node* newParent = new Node();
newParent->data = NULL;
newParent->SetChildren(lowest, secondLowest);
newParent->frequency = lowest->frequency + secondLowest->frequency;
nodes.push(newParent);
}
root = nodes.top();
output = ofstream(output_filename);
//PrintTreeInOrder(root);
std::vector<bool> none;
AccumulateBytes(root, none);
output = ofstream((output_file ? output_file : filename), ios::binary);
short begin = 0x4D4D;
output.write((char*)&begin, sizeof(short));
WriteTree(root);
short end = 0x0FFF;
//output.write((char*)&end, sizeof(short));
//WriteToFile();
output.close();
}
