int cb_check_json_number( CBFILE **cfg, unsigned char **ucsvalue, int ucsvaluelen, int *from ){
int err=CBSUCCESS, indx=0;
unsigned long int chr=0x20;
char state=0, dotcount=0, ecount=0, minuscount=0, firstzero=0, expsigncount=0, atend=0;
if( from==NULL || ucsvalue==NULL || *ucsvalue==NULL || cfg==NULL || *cfg==NULL ){ cb_clog( CBLOGDEBUG, CBERRALLOC, "\ncb_check_json_number: ucsvalue was null."); return CBERRALLOC; }
indx = *from;
if(ucsvaluelen<=0)
return CBNOTJSONNUMBER;
while( indx<ucsvaluelen && indx<CBNAMEBUFLEN && err<CBNEGATION ){
*from = indx;
err = cb_get_ucs_chr( &chr, &(*ucsvalue), &indx, ucsvaluelen);
//fprintf( stderr, "[%c]", (unsigned char) chr );
if(atend==0){
if( chr=='-' && minuscount==0 && state==0 ){ minuscount=1; state=EXPECTDIGIT; firstzero=0; continue; }
if( chr=='0' && ( state==0 || ( firstzero==0 && minuscount==1 && state==EXPECTDIGIT ) ) ){ state=EXPECTDOTOREXP; firstzero=1; continue; }
if( chr>='1' && chr<='9' && state==0 ){ state=EXPECTDIGITORDOT; firstzero=2; continue; }
if( chr>='0' && chr<='9' && state==EXPECTEND ){ continue; }
if( chr>='0' && chr<='9' && ( state==EXPECTDIGIT || state==EXPECTDIGITORDOT || state==EXPECTDIGITBEFOREEXP || state==EXPECTEXP ) ){
if(firstzero==1 && state==EXPECTDIGIT)
return CBNOTJSONNUMBER;
if( state==EXPECTDIGITBEFOREEXP )
state=EXPECTEXP;
if( state==EXPECTDIGIT )
state=EXPECTDIGITORDOT;
continue;
}
if( chr=='.' && dotcount==0 ){ state=EXPECTDIGITBEFOREEXP; dotcount=1; continue; }
if( ( chr=='e' || chr=='E' ) && ecount==0 && ( ( dotcount==1 && state!=EXPECTDIGITBEFOREEXP ) || state==EXPECTDOTOREXP || state==EXPECTDIGITORDOT ) ){
state=EXPECTEXP; ecount=1; continue;
}
if( ( chr=='+' || chr=='-' ) && state==EXPECTEXP ){ expsigncount=1; state=EXPECTEND; continue; }
if( WSP( chr ) || CR( chr ) || LF( chr ) ){ atend=1; continue; }
if( chr==',' && state!=EXPECTDIGITBEFOREEXP && state!=EXPECTEXP && state!=EXPECTDIGIT ){
//return CBSUCCESS;
return CBSUCCESSJSONNUMBER; // 22.8.2016
}
if( chr==']' && state!=EXPECTDIGITBEFOREEXP && state!=EXPECTEXP && state!=EXPECTDIGIT ){
//return CBSUCCESS;
return CBSUCCESSJSONNUMBER; // 22.8.2016
}
return CBNOTJSONNUMBER;
}else{
if( ! WSP( chr ) && ! CR( chr ) && ! LF( chr ) && chr!=']' && chr!=',' )
return CBNOTJSONNUMBER;
}
}
//return CBSUCCESS;
return CBSUCCESSJSONNUMBER; // 22.8.2016
}
/* To be used with arrays and objects. */
int cb_check_json_substring( CBFILE **cfg, unsigned char **ucsvalue, int ucsvaluelen, int *from ){
if( from==NULL || cfg==NULL || *cfg==NULL || ucsvalue==NULL || *ucsvalue==NULL ){ cb_clog( CBLOGDEBUG, CBERRALLOC, "\ncb_check_json_null: ucsvalue was null."); return CBERRALLOC; }
int indx=0, err=CBSUCCESS;
unsigned long int chr=0x20, chprev=0x20; // chprev is WSP character
char quotecount=0;
/*
* Read until the second quote and pass the value to cb_check_json_string. */
indx = *from;
while( indx<ucsvaluelen && indx<CBNAMEBUFLEN && err<CBNEGATION ){
if( chr=='\\' && chprev=='\\' )
chprev = 0x20; // any not bypass
else
chprev = chr;
err = cb_get_ucs_chr( &chr, &(*ucsvalue), &indx, ucsvaluelen);
// comment out the next:
if(err>=CBERROR){ cb_clog( CBLOGERR, err, "\ncb_check_json_substring: cb_get_ucs_chr, error %i.", err ); }
//cb_clog( CBLOGDEBUG, CBNEGATION, "\n\tcb_check_json_substring: [%c]", (char) chr );
if( (unsigned char) chr=='"' && (unsigned char) chprev!='\\' && quotecount>=1 )
++quotecount;
if( quotecount>=2 ){
return cb_check_json_string_content( &(*ucsvalue), indx, &(*from), (**cfg).cf.jsonremovebypassfromcontent ); // updates *from
}
if( quotecount==0 ){
if( (unsigned char) chr=='"' && ( ! WSP( chprev ) && ! CR( chprev ) && ! LF( chprev ) && chprev!=',' ) ){
*from = indx;
return CBNOTJSONSTRING;
}else if( (unsigned char) chr=='"' && (unsigned char) chprev!='\\' ){
++quotecount;
}
}
}
*from = indx;
return CBNOTJSONSTRING;
}
int LocalTerminal::puts(const std::string str, bool nl) {
int i = fwrite(str.c_str(), sizeof(char), str.size(), mOutDev);
if (nl) {
LF(); CR();
}
return i;
}
/*
* Function reading the white spaces away from the end.
*
* *from is the current character index.
* If *from is not used, cb_compare compares witn -2 the first names length
* and *from is updated with the name length.
*/
int cb_check_json_value( CBFILE **cfg, unsigned char **ucsvalue, int ucsvaluelen, int *from ){
int err = CBSUCCESS, wsperr = CBSUCCESS; // 22.8.2016
unsigned long int chr=0x20;
if( from==NULL || cfg==NULL || *cfg==NULL || ucsvalue==NULL || *ucsvalue==NULL ){ cb_clog( CBLOGDEBUG, CBERRALLOC, "\ncb_check_json_value: ucsvalue was null."); return CBERRALLOC; }
err = cb_check_json_value_subfunction( &(*cfg), &(*ucsvalue), ucsvaluelen, &(*from) );
/*
* Length. */
if(*from>ucsvaluelen){
return CBINDEXOUTOFBOUNDS;
}else if(*from<ucsvaluelen){
/*
* Last ',' and ']' check missing 10.11.2015
*/
/*
* Read away white spaces. */
while( *from<ucsvaluelen && ucsvaluelen>0 && *from>0 && *from<CBNAMEBUFLEN && err<CBNEGATION && wsperr<CBNEGATION && \
( WSP( chr ) || CR( chr ) || LF( chr ) ) ){
wsperr = cb_get_ucs_chr( &chr, &(*ucsvalue), &(*from), ucsvaluelen);
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_value: read away white spaces chr [%c]", (char) chr );
}
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_value: FROM: %i, UCSVALUELEN: %i, ERR: %i.", *from, ucsvaluelen, err );
if( *from<=ucsvaluelen && err<CBNEGATION ){ // 23.4.2016
//return CBSUCCESS;
return err; // 22.8.2016
}
}else if( *from==ucsvaluelen )
return err; // 22.8.2016
//return CBSUCCESS;
return CBNOTJSONVALUE;
}
/*
* count number of occurrences of pattern P
* returns range [l,r) (right-exclusive) on BWT of P
*/
pair<ulint,ulint> count(vector<char_type> P){
pair<ulint,ulint> rn = {0,size()};
for(ulint i=0;i<P.size();++i) rn = LF(rn, P[P.size()-i-1]);
return rn;
}
/*
* The only function reading white spaces away from the beginning.
*/
int cb_check_json_value_subfunction( CBFILE **cfg, unsigned char **ucsvalue, int ucsvaluelen, int *from ){
int err=CBSUCCESS;
unsigned long int chr=0x20;
char atstart=1;
unsigned char *chrposition = NULL;
if( from==NULL || cfg==NULL || *cfg==NULL || ucsvalue==NULL || *ucsvalue==NULL ){ cb_clog( CBLOGDEBUG, CBERRALLOC, "\ncb_check_json_value_subfunction: ucsvalue was null."); return CBERRALLOC; }
//cb_clog( CBLOGDEBUG, CBNEGATION, "\n\ncb_check_json_value_subfunction: attribute [");
//cb_print_ucs_chrbuf( CBLOGEMERG, &(*ucsvalue), ucsvaluelen, ucsvaluelen );
//cb_clog( CBLOGDEBUG, CBNEGATION, "], length %i, from %i.", ucsvaluelen, *from);
while( *from<ucsvaluelen && *from<CBNAMEBUFLEN && err<CBNEGATION ){
chrposition = &(*ucsvalue)[*from];
err = cb_get_ucs_chr( &chr, &(*ucsvalue), &(*from), ucsvaluelen);
if(err>=CBERROR){ cb_clog(CBLOGDEBUG, err, "\ncb_check_json_value_subfunction: cb_get_ucs_chr, error %i.", err ); return err; }
//cb_clog(CBLOGDEBUG, CBNEGATION, "\ncb_check_json_value_subfunction: chr [%c].", (char) chr );
if( atstart==1 && ( WSP( chr ) || CR( chr ) || LF( chr ) ) )
continue;
atstart=0;
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_value_subfunction: [%c]", (char) chr );
switch( (unsigned char) chr ){
case '"':
*from -= 4;
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_value_subfunction: return cb_check_json_substring()." );
return cb_check_json_substring( &(*cfg), &(*ucsvalue), ucsvaluelen, &(*from) ); // ok
case '[':
*from -= 4;
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_value_subfunction: return cb_check_json_array()." );
return cb_check_json_array( &(*cfg), &(*ucsvalue), ucsvaluelen, &(*from) ); // ok
case '{':
*from -= 4;
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_value_subfunction: return cb_check_json_object()." );
return cb_check_json_object( &(*cfg), &(*ucsvalue), ucsvaluelen, &(*from) ); // ok, 12.11.2015
case 't':
if(*from>0) *from-=4;
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_value_subfunction: return cb_check_json_true()." );
return cb_check_json_true( &(*cfg), &(*ucsvalue), ucsvaluelen, &(*from) );
case 'n':
if(*from>0) *from-=4;
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_value_subfunction: return cb_check_json_null()." );
return cb_check_json_null( &(*cfg), &(*ucsvalue), ucsvaluelen, &(*from) );
case 'f':
if(*from>0) *from-=4;
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_value_subfunction: return cb_check_json_false()." );
return cb_check_json_false( &(*cfg), &(*ucsvalue), ucsvaluelen, &(*from) );
default:
if(*from>0) *from-=4;
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_value_subfunction: return cb_check_json_number()." );
return cb_check_json_number( &(*cfg), &(*ucsvalue), ucsvaluelen, &(*from) );
}
}
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_value_subfunction: return cbnotjsonvalue." );
return CBNOTJSONVALUE;
}
int main()
{
freopen("11577.INP","r",stdin);
freopen("11577.OUT","w",stdout);
scanf("%d",&T);
for(int i=1;i<=T;i++)
{
scanf(" ");
gets(in);
LF();
}
}
int main(int argc ,char * argv[])
{
ListFile LF(".");
if(argc == 1)
LF.ls();
else
{
while(--argc)
{
printf("%s\n",*(++argv));
LF.setFile(*argv);
LF.ls();
}
}
}
int cb_check_json_array( CBFILE **cfg, unsigned char **ucsvalue, int ucsvaluelen, int *from ){
int err=CBSUCCESS;
unsigned long int chr = 0x20;
if( cfg==NULL || *cfg==NULL || ucsvalue==NULL || *ucsvalue==NULL ){ cb_clog( CBLOGDEBUG, CBERRALLOC, "\ncb_check_json_array: ucsvalue was null."); return CBERRALLOC; }
err = cb_get_ucs_chr( &chr, &(*ucsvalue), &(*from), ucsvaluelen);
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_array: first chr from *from %i is [%c], err %i", (*from-4), (char) chr, err );
if(err>=CBERROR){ cb_clog( CBLOGDEBUG, err, "\ncb_check_json_array: cb_get_ucs_chr, error %i.", err ); return err; }
if( (unsigned char) chr == '[' ){
cb_check_json_array_value:
/*
* First value. */
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_array: to cb_check_json_value_subfunction, *from %i.", *from );
err = cb_check_json_value_subfunction( &(*cfg), &(*ucsvalue), ucsvaluelen, &(*from) );
if(err>=CBERROR){ cb_clog( CBLOGDEBUG, err, "\ncb_check_json_array: cb_check_json_value_subfunction, error %i.", err ); return err; }
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_array: after cb_check_json_value_subfunction, %i.", err );
if(err>=CBNEGATION) return err;
/*
* Read comma or closing bracket. */
while( *from<ucsvaluelen && *from<CBNAMEBUFLEN && err<CBNEGATION ){
err = cb_get_ucs_chr( &chr, &(*ucsvalue), &(*from), ucsvaluelen);
if(err>=CBERROR){ cb_clog( CBLOGDEBUG, err, "\ncb_check_json_array: cb_get_ucs_chr, error %i.", err ); return err; }
//cb_clog( CBLOGDEBUG, CBNEGATION, "[ chr: %c indx: %i len: %i]", (char) chr, *from, ucsvaluelen );
if( WSP( chr ) || CR( chr ) || LF( chr ) ){
continue;
}else if( (unsigned char) chr==']' ){
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_array: bracket, CBSUCCESS." );
//return CBSUCCESS; // 10.11.2015
return CBSUCCESSJSONARRAY; // 10.11.2015, 22.8.2016
}else if( (unsigned char) chr==',' ){
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_array: comma, next value." );
goto cb_check_json_array_value;
}else
return CBNOTJSONARRAY;
}
}else{
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_array: first char was [%c], NOTJSON.", (char) chr );
return CBNOTJSONARRAY;
}
return err;
}
void HHbbbbMikeBase(int wMs,int wM, string st,string st2,string option=""){
//1=signal ,0=QCD ,2=data-----------------------------------------------------------
int nameRoot=1;
if(st2.find("QCD")!= std::string::npos)nameRoot=0;
if(st2.find("data")!= std::string::npos)nameRoot=2;
//---------------------Thea correction
TFile *f3;
f3=TFile::Open("puppiCorr.root");
TF1 * puppisd_corrGEN,* puppisd_corrRECO_cen,* puppisd_corrRECO_for;
puppisd_corrGEN=(TF1 *) f3->FindObjectAny("puppiJECcorr_gen");
puppisd_corrRECO_cen=(TF1 *) f3->FindObjectAny("puppiJECcorr_reco_0eta1v3");
puppisd_corrRECO_for=(TF1 *) f3->FindObjectAny("puppiJECcorr_reco_1v3eta2v5");
TF1* tf1[3];
tf1[0]=puppisd_corrGEN;
tf1[1]=puppisd_corrRECO_cen;
tf1[2]=puppisd_corrRECO_for;
//option-----------------------------------------------------------
int JESOption=0;
if(option.find("JESUp")!= std::string::npos)JESOption=1;
if(option.find("JESDown")!= std::string::npos)JESOption=2;
if(option.find("BtagUp")!= std::string::npos)JESOption=3;
if(option.find("BtagDown")!= std::string::npos)JESOption=4;
if(option.find("tau21Up")!= std::string::npos)JESOption=5;
if(option.find("tau21Down")!= std::string::npos)JESOption=6;
cout<<"JESOption = "<<JESOption<<endl;
bool printHighPtSubjet=0;
bool isFast=1;
//tuple tree and cutflow variables------------------------------------------------------------------------------------
TFile *f;
TTree *tree;
int nPass[20]={0},total=0,dataPassingcsc=0;
double nPassB[6]={0};
double fixScaleNum[2]={0};
//using for Btag Eff -----------------------------------------------------------------------------
string btagsystematicsType="central";
if(JESOption==3)btagsystematicsType="up";
else if(JESOption==4)btagsystematicsType="down";
BTagCalibration calib("CSVv2L", "subjet_CSVv2_ichep.csv");
BTagCalibrationReader LF(BTagEntry::OP_LOOSE,"central", {"up", "down"});
BTagCalibrationReader HFC(BTagEntry::OP_LOOSE, "central", {"up", "down"}); // other sys types
BTagCalibrationReader HF(BTagEntry::OP_LOOSE,"central",{"up", "down"}); // other sys types
LF.load(calib, BTagEntry::FLAV_UDSG, // btag flavour
"incl"); // measurement type
HFC.load(calib, BTagEntry::FLAV_C, // btag flavour
"lt"); // measurement type
HF.load(calib, BTagEntry::FLAV_B, // btag flavour
"lt"); // measurement type
TFile *f1;
if(nameRoot==2)f1=TFile::Open("btagEffSource/data.root");
else if (nameRoot!=2 && (JESOption==0||JESOption==3||JESOption==4||JESOption==5||JESOption==6))f1=TFile::Open(Form("btagEffSource/%s.root",st2.data()));
else if (nameRoot!=2 && JESOption==1)f1=TFile::Open(Form("btagEffSource/%s_JESUp.root",st2.data()));
else if (nameRoot!=2 && JESOption==2)f1=TFile::Open(Form("btagEffSource/%s_JESDown.root",st2.data()));
TH1D* th1[6];
string btaggingEff[6]={"effD_b","effN_b","effD_c","effN_c","effD_l","effN_l"};
for(int i=0;i<6;i++){
th1[i]=(TH1D*)f1->FindObjectAny(Form("%s_1d",btaggingEff[i].data()));
if(i==1||i==3||i==5)th1[i]->Divide(th1[i-1]);
}
//check for zero btagging SF----------------------------------------------------------------------------------------
TH2D* th3[6];
th3[0]=new TH2D("zeroSF_b","zeroSF_b",200,0,2000,60,-3,3);
th3[1]=new TH2D("zeroSF_c","zeroSF_c",200,0,2000,60,-3,3);
th3[2]=new TH2D("zeroSF_l","zeroSF_l",200,0,2000,60,-3,3);
th3[3]=new TH2D("SF_vs_Pt_b","SF_vs_Pt_b",120,0,1200,40,0.8,1.2);
th3[4]=new TH2D("SF_vs_Pt_c","SF_vs_Pt_c",120,0,1200,40,0.8,1.2);
th3[5]=new TH2D("SF_vs_Pt_l","SF_vs_Pt_l",120,0,1200,40,0.8,1.2);
for(int i=0;i<6;i++)th3[i]->Sumw2();
//check for high btagging SF----------------------------------------------------------------------------------------
TH1D* th4[14];
string SF_jet_sub[8]={"SF_jet0_sub0_pass","SF_jet0_sub1_pass","SF_jet1_sub0_pass","SF_jet1_sub1_pass","SF_jet0_sub0_fail","SF_jet0_sub1_fail","SF_jet1_sub0_fail","SF_jet1_sub1_fail"};
for(int i=0;i<8;i++)th4[i]=new TH1D(Form("%s",SF_jet_sub[i].data()),Form("%s",SF_jet_sub[i].data()),40,0.8,1.2);
string weightName[6]={"weight","weight_0b","weight_1b","weight_2b","weight_2b_ll","weight_2b_onel"};
for(int i=0;i<6;i++)th4[i+8]=new TH1D(Form("%s",weightName[i].data()),Form("%s",weightName[i].data()),40,0,2);
for(int i=0;i<14;i++)th4[i]->Sumw2();
//saving variables----------------------------------------------------------------------------------------
TH1D * th5[300],* th6[300];
for(int i=0;i<2;i++){
for(int j=0;j<2;j++){
for(int k=0;k<5;k++){
th5[(i*2+j)*5+k]=new TH1D(Form("Pt_j%d_sj%d_%db",i,j,k),Form("Pt_j%d_sj%d_%db",i,j,k),200,0,2000);
th5[(i*2+j)*5+k+20]=new TH1D(Form("Eta_j%d_sj%d_%db",i,j,k),Form("Eta_j%d_sj%d_%db",i,j,k),60,-3,3);
th5[(i*2+j)*5+k+85]=new TH1D(Form("subCSV_j%d_sj%d_%db",i,j,k),Form("subCSV_j%d_sj%d_%db",i,j,k),20,0,1);
th5[(i*2+j)*5+k+194]=new TH1D(Form("subCSVCut_j%d_sj%d_%db",i,j,k),Form("subCSVCut_j%d_sj%d_%db",i,j,k),20,0,1);
}
}
for(int k=0;k<5;k++){
th5[i*5+k+40]=new TH1D(Form("deltaR_j%d_%db",i,k),Form("deltaR_j%d_%db",i,k),20,0,1);
th5[i*5+k+50]=new TH1D(Form("Pt_j%d_%db",i,k),Form("Pt_j%d_%db",i,k),200,0,2000);
th5[i*5+k+60]=new TH1D(Form("Eta_j%d_%db",i,k),Form("Eta_j%d_%db",i,k),60,-3,3);
th5[i*5+k+70]=new TH1D(Form("prMassL2L3_j%d_%db",i,k),Form("prMassL2L3_j%d_%db",i,k),15,90,150);
th5[i*5+k+105]=new TH1D(Form("tau21_j%d_%db",i,k),Form("tau21_j%d_%db",i,k),25,0,1);
th5[i*5+k+120]=new TH1D(Form("PuppiSDMassL2L3_j%d_%db",i,k),Form("PuppiSDMassL2L3_j%d_%db",i,k),15,90,150);
th5[i*5+k+130]=new TH1D(Form("puppiTau21_j%d_%db",i,k),Form("puppiTau21_j%d_%db",i,k),25,0,1);
th5[i*5+k+140]=new TH1D(Form("prMass_j%d_%db",i,k),Form("prMass_j%d_%db",i,k),15,90,150);
th5[i*5+k+150]=new TH1D(Form("PuppiSDMass_j%d_%db",i,k),Form("PuppiSDMass_j%d_%db",i,k),15,90,150);
th5[i*5+k+170]=new TH1D(Form("doubleSV_j%d_%db",i,k),Form("doubleSV_j%d_%db",i,k),40,-1,1);
th5[i*5+k+184]=new TH1D(Form("FatSV_j%d_%db",i,k),Form("FatSV_j%d_%db",i,k),20,0,1);
th5[i*5+k+250]=new TH1D(Form("PuppiSDMassThea_j%d_%db",i,k),Form("PuppiSDMassThea_j%d_%db",i,k),15,90,150);
}
}
for(int k=0;k<5;k++){
th5[k+80]=new TH1D(Form("totalMass_%db",k),Form("totalMass_%db",k),200,1000,5000);
th5[k+115]=new TH1D(Form("deltaEta_%db",k),Form("deltaEta_%db",k),40,0,2);
th5[k+160]=new TH1D(Form("logPt_%db",k),Form("logPt_%db",k),70,0,7);
th5[k+165]=new TH1D(Form("totalMassRed_%db",k),Form("totalMassRed_%db",k),200,1000,5000);
}
th5[180]= new TH1D("h_nvtx","h_nvtx",60,0,60);
th5[181]= new TH1D("h_ntrue","h_ntrue",60,0,60);
th5[182]= new TH1D("h_nvtx_cut","h_nvtx_cut",60,0,60);
th5[183]= new TH1D("h_ntrue_cut","h_ntrue_cut",60,0,60);
string prMass_no[4]={"prMass","prMassL2L3","PuppiSDMass","PuppiSDMassL2L3"};
string tau21_no[2]={"tau21","puppiTau21"};
for (int i=0;i<4;i++){
th5[214+i*2]=new TH1D(Form("%s_j0_noPr_noTau21",prMass_no[i].data()),Form("%s_j0_noPr_noTau21",prMass_no[i].data()),200,0,200);
th5[215+i*2]=new TH1D(Form("%s_j1_noPr_noTau21",prMass_no[i].data()),Form("%s_j1_noPr_noTau21",prMass_no[i].data()),200,0,200);
th5[222+i*2]=new TH1D(Form("%s_j0_noPr",prMass_no[i].data()),Form("%s_j0_noPr",prMass_no[i].data()),200,0,200);
th5[223+i*2]=new TH1D(Form("%s_j1_noPr",prMass_no[i].data()),Form("%s_j1_noPr",prMass_no[i].data()),200,0,200);
th5[230+i*2]=new TH1D(Form("%s_j0_noTau21",prMass_no[i].data()),Form("%s_j0_noTau21",prMass_no[i].data()),15,90,150);
th5[231+i*2]=new TH1D(Form("%s_j1_noTau21",prMass_no[i].data()),Form("%s_j1_noTau21",prMass_no[i].data()),15,90,150);
}
for (int i=0;i<2;i++){
th5[238+i*2]=new TH1D(Form("%s_j0_noPr_noTau21",tau21_no[i].data()),Form("%s_j0_noPr_noTau21",tau21_no[i].data()),25,0,1);
th5[239+i*2]=new TH1D(Form("%s_j1_noPr_noTau21",tau21_no[i].data()),Form("%s_j1_noPr_noTau21",tau21_no[i].data()),25,0,1);
th5[242+i*2]=new TH1D(Form("%s_j0_noPr",tau21_no[i].data()),Form("%s_j0_noPr",tau21_no[i].data()),25,0,1);
th5[243+i*2]=new TH1D(Form("%s_j1_noPr",tau21_no[i].data()),Form("%s_j1_noPr",tau21_no[i].data()),25,0,1);
th5[246+i*2]=new TH1D(Form("%s_j0_noTau21",tau21_no[i].data()),Form("%s_j0_noTau21",tau21_no[i].data()),25,0,1);
th5[247+i*2]=new TH1D(Form("%s_j1_noTau21",tau21_no[i].data()),Form("%s_j1_noTau21",tau21_no[i].data()),25,0,1);
}
string flavor[4]={"bb","b","cc","udscg"};
for (int i=0;i<4;i++){
th5[260+i]=new TH1D(Form("Pt_j0_0b_%s",flavor[i].data()),Form("Pt_j0_0b_%s",flavor[i].data()),200,0,2000);
th5[264+i]=new TH1D(Form("Pt_j0_1b_%s",flavor[i].data()),Form("Pt_j0_1b_%s",flavor[i].data()),200,0,2000);
th5[268+i]=new TH1D(Form("Pt_j0_2b_%s",flavor[i].data()),Form("Pt_j0_2b_%s",flavor[i].data()),200,0,2000);
th5[272+i]=new TH1D(Form("Pt_j0_DSV_%s",flavor[i].data()),Form("Pt_j0_DSV_%s",flavor[i].data()),200,0,2000);
}
for(int i=0;i<276;i++){
th6[i]=(TH1D* )th5[i]->Clone(Form("%ss",th5[i]->GetTitle()));
th5[i]->Sumw2();
th6[i]->Sumw2();
}
//pileup uncertainty----------------------------------------------------------------------------------------
TH1D* th7[14];
th7[0]=new TH1D("totalMass","totalMass",200,1000,5000);
th7[1]=new TH1D("totalMass_pileup_up","totalMass_pileup_up",200,1000,5000);
th7[2]=new TH1D("totalMass_pileup_down","totalMass_pileup_down",200,1000,5000);
th7[3]=new TH1D("pileupEff","pileupEff",15,0.5,15.5);
double totalPileup[3]={0},passPileup[3]={0};
standalone_LumiReWeighting LumiWeights_central(0),LumiWeights_up(1),LumiWeights_down(-1);
//PDF uncertainty
th7[4]=new TH1D("PDFEff","PDFEff",101,0.5,101.5);
double passPDF[101]={0},totalPDF[101]={0};
//QCD uncertainty
for(int i=5;i<14;i++)th7[i]=new TH1D(Form("uns_QCD_%d",i-5),Form("uns_QCD_%d",i-5),200,1000,5000);
//NCUtuple loop----------------------------------------------------------------------------------------
for (int w=wMs;w<wM;w++){
if(w%20==0)cout<<w<<endl;
//Get ntuple----------------------------------------------------------------------------------------
if (nameRoot!=1)f = TFile::Open(Form("%s%d.root",st.data(),w));
else f = TFile::Open(st.data());
if (!f || !f->IsOpen())continue;
TDirectory * dir;
if (nameRoot!=1)dir = (TDirectory*)f->Get(Form("%s%d.root:/tree",st.data(),w));
else dir = (TDirectory*)f->Get(Form("%s:/tree",st.data()));
dir->GetObject("treeMaker",tree);
TreeReader data(tree);
total+=data.GetEntriesFast();
for(Long64_t jEntry=0; jEntry<data.GetEntriesFast() ;jEntry++){//event loop----------------------------------------------------------------------------------------
data.GetEntry(jEntry);
Int_t nVtx = data.GetInt("nVtx");
Float_t ntrue= data.GetFloat("pu_nTrueInt");
//Float_t* pdfscaleSysWeights= data.GetPtrFloat("pdfscaleSysWeights");
th5[180]->Fill(nVtx);
th5[181]->Fill(ntrue);
double PU_weight[3]={1,1,1};
if(nameRoot!=2){
if(ntrue<51){
PU_weight[0] = LumiWeights_central.weight(ntrue);
PU_weight[1]= LumiWeights_up.weight(ntrue);
PU_weight[2] = LumiWeights_down.weight(ntrue);
}
else {
PU_weight[0] = LumiWeights_central.weight(50);
PU_weight[1] = LumiWeights_up.weight(50);
PU_weight[2]= LumiWeights_down.weight(50);
}
}
th6[180]->Fill(nVtx,PU_weight[0]);
th6[181]->Fill(ntrue,PU_weight[0]);
fixScaleNum[0]+=PU_weight[0];
for(int i=0;i<3;i++)totalPileup[i]+=PU_weight[i];
for(int i=0;i<101;i++)totalPDF[i]+=PU_weight[0];
//1.trigger
std::string* trigName = data.GetPtrString("hlt_trigName");
vector<bool> &trigResult = *((vector<bool>*) data.GetPtr("hlt_trigResult"));
bool passTrigger=false;
for(int it=0; it< data.GetPtrStringSize(); it++){
std::string thisTrig= trigName[it];
bool results = trigResult[it];
if( ((thisTrig.find("HLT_PFHT800")!= std::string::npos
) && results==1)){
passTrigger=true;
break;
}
}
if(!passTrigger && nameRoot==2)continue;nPass[1]++;
Float_t HT= data.GetFloat("HT");
if(nameRoot!=2 && HT<800)continue;
int nFATJet = data.GetInt("FATnJet");
TClonesArray* fatjetP4 = (TClonesArray*) data.GetPtrTObject("FATjetP4");
float* FATjetCorrUncUp = data.GetPtrFloat("FATjetCorrUncUp");
float* FATjetCorrUncDown = data.GetPtrFloat("FATjetCorrUncDown");
vector<bool> &FATjetPassIDTight = *((vector<bool>*) data.GetPtr("FATjetPassIDTight"));
//2.nJets
if(nFATJet<2)continue;nPass[2]++;
TLorentzVector* thisJet ,* thatJet;
if (JESOption==1){
TLorentzVector test0= (*((TLorentzVector*)fatjetP4->At(0)))*(1+FATjetCorrUncUp[0] );
TLorentzVector test1= (*((TLorentzVector*)fatjetP4->At(1)))*(1+FATjetCorrUncUp[1] );
thisJet= &test0;
thatJet= &test1;
}
else if (JESOption==2){
TLorentzVector test0= (*((TLorentzVector*)fatjetP4->At(0)))*(1-FATjetCorrUncDown[0] );
TLorentzVector test1= (*((TLorentzVector*)fatjetP4->At(1)))*(1-FATjetCorrUncDown[1] );
thisJet= &test0;
thatJet= &test1;
}
else{
thisJet= (TLorentzVector*)fatjetP4->At(0);
thatJet = (TLorentzVector*)fatjetP4->At(1);
}
//3. Pt
if(thisJet->Pt()<200||thatJet->Pt()<200)continue;
nPass[3]++;
//4tightId-----------------------------------------
if(FATjetPassIDTight[0]==0||FATjetPassIDTight[1]==0)continue;
nPass[4]++;
//5. Eta-----------------------------------------
if(fabs(thisJet->Eta())>2.4||fabs(thatJet->Eta())>2.4)continue;
nPass[5]++;
//6. DEta-----------------------------------------
float dEta = fabs(thisJet->Eta()-thatJet->Eta());
if(dEta>1.3)continue;
nPass[6]++;
//7. Mjj-----------------------------------------
float mjj = (*thisJet+*thatJet).M();
float mjjRed = (*thisJet+*thatJet).M()+250-thisJet->M()-thatJet->M();
//if(mjjRed<1000)continue;
nPass[7]++;
//8. fatjetPRmassL2L3Corr-----------------------------------------
Float_t* fatjetPRmassL2L3Corr = data.GetPtrFloat("FATjetPRmassL2L3Corr");
vector<float> *subjetSDCSV = data.GetPtrVectorFloat("FATsubjetSDCSV");
vector<Int_t> *FATsubjetSDHadronFlavor = data.GetPtrVectorInt("FATsubjetSDHadronFlavor");
int nbtag=0,nbtag2=0;
float MaxBJetPt = 670., MaxLJetPt = 1000.;
double sf[2][2],eta[2],pt[2],dr[2],subjetPt[2][2],subjetEta[2][2],eff[2][2],btaggingscaleFactor=1;
pt[0]=thisJet->Pt();
pt[1]=thatJet->Pt();
double scaleFactor=PU_weight[0]*btaggingscaleFactor;
Float_t* fatjetTau1 = data.GetPtrFloat("FATjetTau1");
Float_t* fatjetTau2 = data.GetPtrFloat("FATjetTau2");
Float_t* FATjetPuppiTau1 = data.GetPtrFloat("FATjetPuppiTau1");
Float_t* FATjetPuppiTau2 = data.GetPtrFloat("FATjetPuppiTau2");
double tau21[2];
tau21[0]=(fatjetTau2[0]/fatjetTau1[0]),tau21[1]=(fatjetTau2[1]/fatjetTau1[1]);
double puppiTau21[2];
puppiTau21[0]=(FATjetPuppiTau2[0]/FATjetPuppiTau1[0]),puppiTau21[1]=(FATjetPuppiTau2[1]/FATjetPuppiTau1[1]);
if(fatjetPRmassL2L3Corr[0]<50||fatjetPRmassL2L3Corr[0]>300)continue;
if(fatjetPRmassL2L3Corr[1]<50||fatjetPRmassL2L3Corr[1]>300)continue;
nPass[8]++;
//9.-----------------------------------------
if(tau21[0]>0.6 || tau21[1]>0.6) continue;
nPass[9]++;
Int_t* FATjetHadronFlavor = data.GetPtrInt("FATjetHadronFlavor");
if(subjetSDCSV[0][0]<0.605 && subjetSDCSV[0][1]<0.605){
if(FATjetHadronFlavor[0]==5 && FATsubjetSDHadronFlavor[0][0]==5 && FATsubjetSDHadronFlavor[0][1]==5)th5[260]->Fill(thisJet->Pt());
else if(FATjetHadronFlavor[0]==5 && (FATsubjetSDHadronFlavor[0][0]==5 || FATsubjetSDHadronFlavor[0][1]==5))th5[261]->Fill(thisJet->Pt());
else if(FATjetHadronFlavor[0]==4 && FATsubjetSDHadronFlavor[0][0]==4 && FATsubjetSDHadronFlavor[0][1]==4)th5[262]->Fill(thisJet->Pt());
else th5[263]->Fill(thisJet->Pt());
}
else if (subjetSDCSV[0][0]>0.605 && subjetSDCSV[0][1]>0.605){
if(FATjetHadronFlavor[0]==5 && FATsubjetSDHadronFlavor[0][0]==5 && FATsubjetSDHadronFlavor[0][1]==5)th5[268]->Fill(thisJet->Pt());
else if(FATjetHadronFlavor[0]==5 && (FATsubjetSDHadronFlavor[0][0]==5 || FATsubjetSDHadronFlavor[0][1]==5))th5[269]->Fill(thisJet->Pt());
else if (FATjetHadronFlavor[0]==4 && FATsubjetSDHadronFlavor[0][0]==4 && FATsubjetSDHadronFlavor[0][1]==4)th5[270]->Fill(thisJet->Pt());
else th5[271]->Fill(thisJet->Pt());
}
else {
if(FATjetHadronFlavor[0]==5 && FATsubjetSDHadronFlavor[0][0]==5 && FATsubjetSDHadronFlavor[0][1]==5)th5[264]->Fill(thisJet->Pt());
else if(FATjetHadronFlavor[0]==5 && (FATsubjetSDHadronFlavor[0][0]==5 || FATsubjetSDHadronFlavor[0][1]==5))th5[265]->Fill(thisJet->Pt());
else if (FATjetHadronFlavor[0]==4 && FATsubjetSDHadronFlavor[0][0]==4 && FATsubjetSDHadronFlavor[0][1]==4)th5[266]->Fill(thisJet->Pt());
else th5[267]->Fill(thisJet->Pt());
}
fixScaleNum[1]+=PU_weight[0];
Float_t* ADDjet_DoubleSV = data.GetPtrFloat("ADDjet_DoubleSV");
Float_t FATjet_DoubleSV[2]={0};
Int_t ADDnJet = data.GetInt("ADDnJet");
bool matchThis=0,matchThat=0;
TClonesArray* ADDjetP4 = (TClonesArray*) data.GetPtrTObject("ADDjetP4");
for(int i=0;i<ADDnJet;i++){
TLorentzVector* thisAddJet ;
thisAddJet= (TLorentzVector*)ADDjetP4->At(i);
if(!matchThis && thisAddJet->DeltaR(*thisJet)<0.8){
matchThis=1;
FATjet_DoubleSV[0]=ADDjet_DoubleSV[i];
continue;
}
if(!matchThat && thisAddJet->DeltaR(*thatJet)<0.8){
matchThat=1;
FATjet_DoubleSV[1]=ADDjet_DoubleSV[i];
}
if(matchThis&& matchThat){
//cout<<"match"<<FATjet_DoubleSV[0]<<","<<FATjet_DoubleSV[0]<<endl;
break;
}
}
if(FATjet_DoubleSV[0]>-0.8){
if(FATjetHadronFlavor[0]==5 && FATsubjetSDHadronFlavor[0][0]==5 && FATsubjetSDHadronFlavor[0][1]==5)th5[272]->Fill(thisJet->Pt());
else if(FATjetHadronFlavor[0]==5 && (FATsubjetSDHadronFlavor[0][0]==5 || FATsubjetSDHadronFlavor[0][1]==5))th5[273]->Fill(thisJet->Pt());
else if (FATjetHadronFlavor[0]==4 && FATsubjetSDHadronFlavor[0][0]==4 && FATsubjetSDHadronFlavor[0][1]==4)th5[274]->Fill(thisJet->Pt());
else th5[275]->Fill(thisJet->Pt());
}
}//end event loop----------------------------------------------------------------------------------------
} //end ntuple loop----------------------------------------------------------------------------------------
cout<<"entries="<<total<<endl;
for(int i=0;i<6;i++)cout<<"nPassB["<<i<<"]="<<nPassB[i]<<endl;
for(int i=0;i<16;i++)cout<<"nPass["<<i<<"]="<<nPass[i]<<endl;
for(int i=0;i<3;i++)th7[3]->SetBinContent(i+1,passPileup[i]/totalPileup[i]);
for(int i=0;i<101;i++)th7[4]->SetBinContent(i+1,passPDF[i]/totalPDF[i]);
TH1D * th2o=new TH1D("Nbtagjet","Nbtagjet",5,-0.5,4.5);
for (int i=0;i<5;i++){
if(nameRoot==2 && i>2)continue;
th2o->SetBinContent(i+1,nPass[i+10]);
}
TH1D * th2ob=new TH1D("NbtagjetB","NbtagjetB",5,-0.5,4.5);
for (int i=0;i<5;i++){
if(nameRoot==2 && i>2)continue;
th2ob->SetBinContent(i+1,nPassB[i]);
}
TH1D * fixScale=new TH1D("fixScale","fixScale",2,-0.5,1.5);
fixScale->SetBinContent(1,fixScaleNum[0]);
fixScale->SetBinContent(2,fixScaleNum[1]);
TH1D * cutflow=new TH1D("cutflow","cutflow",16,0.5,16.5);
cutflow->SetBinContent(1,total);
if(nameRoot==2)for(int ii=1;ii<14;ii++)cutflow->SetBinContent(ii+1,nPass[ii-1]);
else for(int ii=1;ii<16;ii++)cutflow->SetBinContent(ii+1,nPass[ii-1]);
TFile* outFile = new TFile(Form("root_file_mike/%s.root",st2.data()),"recreate");
th2o->Write();
th2ob->Write();
fixScale->Write();
cutflow->Write();
for(int i=0;i<6;i++)th3[i]->Write();
for(int i=0;i<14;i++)th4[i]->Write();
for(int i=0;i<276;i++){
th5[i]->Write();
th6[i]->Write();
}
for(int i=0;i<14;i++)th7[i]->Write();
outFile->Close();
}
void HH4bCategoryBase_80(int wMs,int wM, string st,string st2,string option=""){
//1=signal ,0=QCD ,2=data
int nameRoot=1;
if(st2.find("QCD")!= std::string::npos)nameRoot=0;
if(st2.find("data")!= std::string::npos)nameRoot=2;
cout<<"nameRoot = "<<nameRoot<<endl;
//option-----------------------------------------------------------
int JESOption=0;
if(option.find("JESUp")!= std::string::npos)JESOption=1;
if(option.find("JESDown")!= std::string::npos)JESOption=2;
cout<<"JESOption = "<<JESOption<<endl;
//using for Btag Eff -----------------------------------------------------------------------------
string btagsystematicsType="central";
if(JESOption==3)btagsystematicsType="up";
else if(JESOption==4)btagsystematicsType="down";
BTagCalibration calib("CSVv2L", "subjet_CSVv2_ichep.csv");
BTagCalibrationReader LF(BTagEntry::OP_LOOSE,"central", {"up", "down"});
BTagCalibrationReader HFC(BTagEntry::OP_LOOSE, "central", {"up", "down"}); // other sys types
BTagCalibrationReader HF(BTagEntry::OP_LOOSE,"central",{"up", "down"}); // other sys types
LF.load(calib, BTagEntry::FLAV_UDSG, // btag flavour
"incl"); // measurement type
HFC.load(calib, BTagEntry::FLAV_C, // btag flavour
"lt"); // measurement type
HF.load(calib, BTagEntry::FLAV_B, // btag flavour
"lt"); // measurement type
TFile *f1;
if(nameRoot==2)f1=TFile::Open("btagEffSource/data.root");
else if (nameRoot!=2 && (JESOption==0||JESOption==3||JESOption==4||JESOption==5||JESOption==6))f1=TFile::Open(Form("btagEffSource/%s.root",st2.data()));
else if (nameRoot!=2 && JESOption==1)f1=TFile::Open(Form("btagEffSource/%s_JESUp.root",st2.data()));
else if (nameRoot!=2 && JESOption==2)f1=TFile::Open(Form("btagEffSource/%s_JESDown.root",st2.data()));
TH1D* th1[6];
string btaggingEff[6]={"effD_b","effN_b","effD_c","effN_c","effD_l","effN_l"};
for(int i=0;i<6;i++){
th1[i]=(TH1D*)f1->FindObjectAny(Form("%s_1d",btaggingEff[i].data()));
if(i==1||i==3||i==5)th1[i]->Divide(th1[i-1]);
}
standalone_LumiReWeighting LumiWeights_central(0),LumiWeights_up(1),LumiWeights_down(-1);
TFile *f;
TTree *tree;
int nPass[20]={0};
int total=0;
double fixScaleNum[2]={0};
double xBinsForHeavyFlavor[5]={30,140,180,240,3000};
double xBinsForLightFlavor[11]={20,100,200,300,400,500,600,700,800,900,3000};
TH1D* th5[3];
th5[0]=new TH1D("cat0","cat0",4000,1000,5000);
th5[1]=new TH1D("cat1","cat1",4000,1000,5000);
th5[2]=new TH1D("cat2","cat2",4000,1000,5000);
for(int i=0;i<3;i++){
th5[i]->Sumw2();
}
//---------------------------------
for (int w=wMs;w<wM;w++){
if(w%20==0)cout<<w<<endl;
if (nameRoot!=1)f = TFile::Open(Form("%s%d.root",st.data(),w));
else f = TFile::Open(st.data());
if (!f || !f->IsOpen())continue;
TDirectory * dir;
if (nameRoot!=1)dir = (TDirectory*)f->Get(Form("%s%d.root:/tree",st.data(),w));
else dir = (TDirectory*)f->Get(Form("%s:/tree",st.data()));
dir->GetObject("treeMaker",tree);
TreeReader data(tree);
total+=data.GetEntriesFast();
for(Long64_t jEntry=0; jEntry<data.GetEntriesFast() ;jEntry++){
data.GetEntry(jEntry);
double PU_weight[3]={1,1,1};
Float_t ntrue= data.GetFloat("pu_nTrueInt");
if(nameRoot!=2){
if(ntrue<51){
PU_weight[0] = LumiWeights_central.weight(ntrue);
PU_weight[1]= LumiWeights_up.weight(ntrue);
PU_weight[2] = LumiWeights_down.weight(ntrue);
}
else {
PU_weight[0] = LumiWeights_central.weight(50);
PU_weight[1] = LumiWeights_up.weight(50);
PU_weight[2]= LumiWeights_down.weight(50);
}
}
fixScaleNum[0]+=PU_weight[0];
Int_t nVtx = data.GetInt("nVtx");
//0. has a good vertex
if(nVtx<1)continue;
nPass[0]++;
//1.trigger
std::string* trigName = data.GetPtrString("hlt_trigName");
vector<bool> &trigResult = *((vector<bool>*) data.GetPtr("hlt_trigResult"));
bool passTrigger=false;
for(int it=0; it< data.GetPtrStringSize();it++){
std::string thisTrig= trigName[it];
bool results = trigResult[it];
if( ((thisTrig.find("HLT_PFHT800")!= std::string::npos||
thisTrig.find("HLT_PFHT650_WideJetMJJ900DEtaJJ1p5_v")!= std::string::npos||
thisTrig.find("HLT_PFHT650_WideJetMJJ950DEtaJJ1p5_v")!= std::string::npos||
thisTrig.find("HLT_AK8PFJet360_TrimMass30_v")!= std::string::npos||
thisTrig.find("HLT_AK8PFHT700_TrimR0p1PT0p03Mass50_v")!= std::string::npos
) && results==1)){
passTrigger=true;
break;
}
}
if(!passTrigger && nameRoot==2)continue;
nPass[1]++;
const int nFATJet=data.GetInt("FATnJet");
//2.nJets
if(nFATJet<2)continue;nPass[2]++;
TClonesArray* fatjetP4 = (TClonesArray*) data.GetPtrTObject("FATjetP4");
float* FATjetCorrUncUp = data.GetPtrFloat("FATjetCorrUncUp");
float* FATjetCorrUncDown = data.GetPtrFloat("FATjetCorrUncDown");
TLorentzVector* thisJet ,* thatJet;
if(JESOption==0){
thisJet=(TLorentzVector*)fatjetP4->At(0);
thatJet=(TLorentzVector*)fatjetP4->At(1);
}
else if (JESOption==1){
TLorentzVector thisJetScaleUp= (*((TLorentzVector*)fatjetP4->At(0)))*(1+FATjetCorrUncUp[0]);
TLorentzVector thatJetScaleUp= (*((TLorentzVector*)fatjetP4->At(1)))*(1+FATjetCorrUncUp[1]);
thisJet= &thisJetScaleUp;
thatJet= &thatJetScaleUp;
}
else if (JESOption==2){
TLorentzVector thisJetScaleDown= (*((TLorentzVector*)fatjetP4->At(0)))*(1-FATjetCorrUncDown[0]);
TLorentzVector thatJetScaleDown= (*((TLorentzVector*)fatjetP4->At(1)))*(1-FATjetCorrUncDown[1]);
thisJet= &thisJetScaleDown;
thatJet= &thatJetScaleDown;
}
//3. Pt
if(thisJet->Pt()<200)continue;
if(thatJet->Pt()<200)continue;
nPass[3]++;
//4tightId-----------------------------------------
vector<bool> &FATjetPassIDTight = *((vector<bool>*) data.GetPtr("FATjetPassIDTight"));
if(FATjetPassIDTight[0]==0)continue;
if(FATjetPassIDTight[1]==0)continue;
Float_t* FATjetCEmEF = data.GetPtrFloat("FATjetCEmEF");
Float_t* FATjetMuEF = data.GetPtrFloat("FATjetMuEF");
if(FATjetMuEF[0]>0.8)continue;
if(FATjetCEmEF[0]>0.9)continue;
if(FATjetMuEF[1]>0.8)continue;
if(FATjetCEmEF[1]>0.9)continue;
nPass[4]++;
//5. Eta-----------------------------------------
if(fabs(thisJet->Eta())>2.4)continue;
if(fabs(thatJet->Eta())>2.4)continue;
nPass[5]++;
//6. DEta-----------------------------------------
float dEta = fabs(thisJet->Eta()-thatJet->Eta());
if(dEta>1.3)continue;
nPass[6]++;
//7. Mjj-----------------------------------------
float mjjRed = (*thisJet+*thatJet).M()+250-thisJet->M()-thatJet->M();
if(mjjRed<1000)continue;
nPass[7]++;
//8. fatjetPRmassL2L3Corr-----------------------------------------
Float_t* fatjetPRmassL2L3Corr = data.GetPtrFloat("FATjetPRmassL2L3Corr");
if(fatjetPRmassL2L3Corr[0]<105||fatjetPRmassL2L3Corr[0]>135)continue;
if(fatjetPRmassL2L3Corr[1]<105||fatjetPRmassL2L3Corr[1]>135)continue;
nPass[8]++;
//9.-----------------------------------------
Float_t* fatjetTau1 = data.GetPtrFloat("FATjetTau1");
Float_t* fatjetTau2 = data.GetPtrFloat("FATjetTau2");
double tau21_1=(fatjetTau2[0]/fatjetTau1[0]),tau21_2=(fatjetTau2[1]/fatjetTau1[1]);
//10.btag
vector<float> *subjetSDCSV = data.GetPtrVectorFloat("FATsubjetSDCSV");
int nbtag=0;
if(subjetSDCSV[0][0]>0.46)nbtag++;
if(subjetSDCSV[0][1]>0.46)nbtag++;
if(subjetSDCSV[1][0]>0.46)nbtag++;
if(subjetSDCSV[1][1]>0.46)nbtag++;
vector<float> *subjetSDPx = data.GetPtrVectorFloat("FATsubjetSDPx");
vector<float> *subjetSDPy = data.GetPtrVectorFloat("FATsubjetSDPy");
vector<float> *subjetSDPz = data.GetPtrVectorFloat("FATsubjetSDPz");
vector<float> *subjetSDE = data.GetPtrVectorFloat("FATsubjetSDE");
vector<Int_t> *FATsubjetSDHadronFlavor = data.GetPtrVectorInt("FATsubjetSDHadronFlavor");
double sf[2][2],subjetPt[2][2],subjetEta[2][2],eff[2][2],btaggingscaleFactor=1;
TLorentzVector* subjetP4[2][2];
for(int i=0;i<2;i++){
for(int j=0;j<2;j++){
sf[i][j]=1;
subjetP4[i][j]=new TLorentzVector(0,0,0,0);
subjetP4[i][j]->SetPxPyPzE(subjetSDPx[i][j],subjetSDPy[i][j],subjetSDPz[i][j],subjetSDE[i][j]);
subjetPt[i][j]=subjetP4[i][j]->Pt();
subjetEta[i][j]=subjetP4[i][j]->Eta();
}
}
for(int i=0;i<2;i++){
for(int j=0;j<2;j++){
//get btagging eff------------------------------------------------------------
int getPtBin=1;
if(subjetPt[i][j]<140)getPtBin=1;
else if (140<=subjetPt[i][j] && subjetPt[i][j]<180)getPtBin=2;
else if (180<=subjetPt[i][j] && subjetPt[i][j]<240)getPtBin=3;
else getPtBin=4;
if(FATsubjetSDHadronFlavor[i][j]==5)eff[i][j]=th1[1]->GetBinContent(getPtBin);
else if(FATsubjetSDHadronFlavor[i][j]==4)eff[i][j]=th1[3]->GetBinContent(getPtBin);
else {
int temp=0;
if(subjetPt[i][j]>=900)temp=10;
else temp=ceil(subjetPt[i][j]/100);
bool checkBinContentIfZero=0;
while(checkBinContentIfZero==0){
if(th1[4]->GetBinContent(temp)==0){
temp--;
}
else checkBinContentIfZero=1;
}
eff[i][j]=th1[5]->GetBinContent(temp);
}
//Get SF from csv------------------------------------------------------------
if(FATsubjetSDHadronFlavor[i][j]==5){
sf[i][j]=HF.eval_auto_bounds("central",BTagEntry::FLAV_B, subjetEta[i][j],subjetPt[i][j]);
}
else if(FATsubjetSDHadronFlavor[i][j]==4){
sf[i][j]=HFC.eval_auto_bounds("central",BTagEntry::FLAV_C, subjetEta[i][j],subjetPt[i][j]);
}
else {
sf[i][j]=LF.eval_auto_bounds("central",BTagEntry::FLAV_UDSG, subjetEta[i][j],subjetPt[i][j]);
}
//get tot. btagging SF
if(subjetSDCSV[i][j]>=0.46)btaggingscaleFactor*=sf[i][j];
else btaggingscaleFactor*=((1-eff[i][j]*sf[i][j])/(1-eff[i][j]));
}
}
//if(tau21_1>0.75 || tau21_2>0.75) continue;
//if(nbtag==3 && ((tau21_1>0.6 && tau21_2<0.6)||(tau21_1<0.6 && tau21_2>0.6)))th1[2]->Fill(mjjRed);
if(tau21_1>0.6 || tau21_2>0.6) continue;
if(nbtag==4)th5[0]->Fill(mjjRed,btaggingscaleFactor*PU_weight[0]);
if(nbtag==3)th5[1]->Fill(mjjRed,btaggingscaleFactor*PU_weight[0]);
if(nbtag==2)th5[2]->Fill(mjjRed,btaggingscaleFactor*PU_weight[0]);
nPass[9]++;
fixScaleNum[1]+=PU_weight[0];
}
}
cout<<"entries="<<total<<endl;
for(int i=0;i<9;i++)cout<<"nPass["<<i<<"]="<<nPass[i]<<endl;
TH1D * fixScale=new TH1D("fixScale","fixScale",2,-0.5,1.5);
fixScale->SetBinContent(1,fixScaleNum[0]);
fixScale->SetBinContent(2,fixScaleNum[1]);
TFile* outFile ;
if(JESOption==0)outFile= new TFile(Form("category/%s.root",st2.data()),"recreate");
else if(JESOption==1)outFile= new TFile(Form("category/%s_JESUp.root",st2.data()),"recreate");
else if(JESOption==2)outFile= new TFile(Form("category/%s_JESDown.root",st2.data()),"recreate");
fixScale->Write();
for(int i=0;i<3;i++){
th5[i]->Write();
}
outFile->Close();
}
void HH4bBtagEffBase_80_v2(int wMs,int wM, string st,string st2,string option=""){
//1=signal ,0=QCD ,2=data-----------------------------------------------------------
int nameRoot=1;
if((st2.find("QCD")!= std::string::npos)||
(st2.find("bGen")!= std::string::npos)||
(st2.find("bEnriched")!= std::string::npos))nameRoot=0;
if(st2.find("data")!= std::string::npos)nameRoot=2;
//Thea correction---------------------
TFile *f3;
f3=TFile::Open("puppiCorr.root");
TF1* tf1[3];
tf1[0]=(TF1 *) f3->FindObjectAny("puppiJECcorr_gen");
tf1[1]=(TF1 *) f3->FindObjectAny("puppiJECcorr_reco_0eta1v3");
tf1[2]=(TF1 *) f3->FindObjectAny("puppiJECcorr_reco_1v3eta2v5");
//option-----------------------------------------------------------
int JESOption=0;
if(option.find("JESUp")!= std::string::npos)JESOption=1;
if(option.find("JESDown")!= std::string::npos)JESOption=2;
if(option.find("BtagUp")!= std::string::npos)JESOption=3;
if(option.find("BtagDown")!= std::string::npos)JESOption=4;
if(option.find("tau21Up")!= std::string::npos)JESOption=5;
if(option.find("tau21Down")!= std::string::npos)JESOption=6;
cout<<"JESOption = "<<JESOption<<endl;
bool printHighPtSubjet=0;
bool isFast=1;
//tuple tree and cutflow variables------------------------------------------------------------------------------------
TFile *f;
TTree *tree;
double nPass[30]={0},total=0;
double fixScaleNum[2]={0};
//using for Btag Eff -----------------------------------------------------------------------------
string btagsystematicsType="central";
if(JESOption==3)btagsystematicsType="up";
else if(JESOption==4)btagsystematicsType="down";
BTagCalibration calib("CSVv2L", "subjet_CSVv2_ichep.csv");
BTagCalibrationReader LF(BTagEntry::OP_LOOSE,"central", {"up", "down"});
BTagCalibrationReader HFC(BTagEntry::OP_LOOSE, "central", {"up", "down"}); // other sys types
BTagCalibrationReader HF(BTagEntry::OP_LOOSE,"central",{"up", "down"}); // other sys types
LF.load(calib, BTagEntry::FLAV_UDSG, // btag flavour
"incl"); // measurement type
HFC.load(calib, BTagEntry::FLAV_C, // btag flavour
"lt"); // measurement type
HF.load(calib, BTagEntry::FLAV_B, // btag flavour
"lt"); // measurement type
TFile *f1;
if(nameRoot==2)f1=TFile::Open("btagEffSource/data.root");
else if (nameRoot!=2 && (JESOption==0||JESOption==3||JESOption==4||JESOption==5||JESOption==6))f1=TFile::Open(Form("btagEffSource/%s.root",st2.data()));
else if (nameRoot!=2 && JESOption==1)f1=TFile::Open(Form("btagEffSource/%s_JESUp.root",st2.data()));
else if (nameRoot!=2 && JESOption==2)f1=TFile::Open(Form("btagEffSource/%s_JESDown.root",st2.data()));
TH1D* th1[6];
string btaggingEff[6]={"effD_b","effN_b","effD_c","effN_c","effD_l","effN_l"};
for(int i=0;i<6;i++){
th1[i]=(TH1D*)f1->FindObjectAny(Form("%s_1d",btaggingEff[i].data()));
if(i==1||i==3||i==5)th1[i]->Divide(th1[i-1]);
}
//saving variables----------------------------------------------------------------------------------------
TH1D * th5[300],* th_flavor[4][300];
for(int i=0;i<2;i++){
for(int j=0;j<2;j++){
for(int k=0;k<5;k++){
th5[(i*2+j)*5+k]=new TH1D(Form("Pt_j%d_sj%d_%db",i,j,k),Form("Pt_j%d_sj%d_%db",i,j,k),200,0,2000);
th5[(i*2+j)*5+k+20]=new TH1D(Form("Eta_j%d_sj%d_%db",i,j,k),Form("Eta_j%d_sj%d_%db",i,j,k),60,-3,3);
th5[(i*2+j)*5+k+85]=new TH1D(Form("subCSV_j%d_sj%d_%db",i,j,k),Form("subCSV_j%d_sj%d_%db",i,j,k),20,0,1);
th5[(i*2+j)*5+k+194]=new TH1D(Form("subCSVCut_j%d_sj%d_%db",i,j,k),Form("subCSVCut_j%d_sj%d_%db",i,j,k),20,0,1);
}
}
for(int k=0;k<5;k++){
th5[i*5+k+40]=new TH1D(Form("deltaR_j%d_%db",i,k),Form("deltaR_j%d_%db",i,k),20,0,1);
th5[i*5+k+50]=new TH1D(Form("Pt_j%d_%db",i,k),Form("Pt_j%d_%db",i,k),200,0,2000);
th5[i*5+k+60]=new TH1D(Form("Eta_j%d_%db",i,k),Form("Eta_j%d_%db",i,k),60,-3,3);
th5[i*5+k+70]=new TH1D(Form("prMassL2L3_j%d_%db",i,k),Form("prMassL2L3_j%d_%db",i,k),15,90,150);
th5[i*5+k+105]=new TH1D(Form("tau21_j%d_%db",i,k),Form("tau21_j%d_%db",i,k),25,0,1);
th5[i*5+k+120]=new TH1D(Form("PuppiSDMassL2L3_j%d_%db",i,k),Form("PuppiSDMassL2L3_j%d_%db",i,k),15,90,150);
th5[i*5+k+130]=new TH1D(Form("puppiTau21_j%d_%db",i,k),Form("puppiTau21_j%d_%db",i,k),25,0,1);
th5[i*5+k+140]=new TH1D(Form("prMass_j%d_%db",i,k),Form("prMass_j%d_%db",i,k),15,90,150);
th5[i*5+k+150]=new TH1D(Form("PuppiSDMass_j%d_%db",i,k),Form("PuppiSDMass_j%d_%db",i,k),15,90,150);
th5[i*5+k+170]=new TH1D(Form("doubleSV_j%d_%db",i,k),Form("doubleSV_j%d_%db",i,k),40,-1,1);
th5[i*5+k+184]=new TH1D(Form("FatSV_j%d_%db",i,k),Form("FatSV_j%d_%db",i,k),20,0,1);
th5[i*5+k+250]=new TH1D(Form("PuppiSDMassThea_j%d_%db",i,k),Form("PuppiSDMassThea_j%d_%db",i,k),15,90,150);
}
}
for(int k=0;k<5;k++){
th5[k+80]=new TH1D(Form("totalMass_%db",k),Form("totalMass_%db",k),200,1000,5000);
th5[k+115]=new TH1D(Form("deltaEta_%db",k),Form("deltaEta_%db",k),40,0,2);
th5[k+160]=new TH1D(Form("logPt_%db",k),Form("logPt_%db",k),70,0,7);
th5[k+165]=new TH1D(Form("totalMassRed_%db",k),Form("totalMassRed_%db",k),200,1000,5000);
}
th5[180]= new TH1D("h_nvtx","h_nvtx",60,0,60);
th5[181]= new TH1D("h_ntrue","h_ntrue",60,0,60);
th5[182]= new TH1D("h_nvtx_cut","h_nvtx_cut",60,0,60);
th5[183]= new TH1D("h_ntrue_cut","h_ntrue_cut",60,0,60);
string prMass_no[4]={"prMass","prMassL2L3","PuppiSDMass","PuppiSDMassL2L3"};
string tau21_no[2]={"tau21","puppiTau21"};
for (int i=0;i<4;i++){
th5[214+i*2]=new TH1D(Form("%s_j0_noPr_noTau21",prMass_no[i].data()),Form("%s_j0_noPr_noTau21",prMass_no[i].data()),200,0,200);
th5[215+i*2]=new TH1D(Form("%s_j1_noPr_noTau21",prMass_no[i].data()),Form("%s_j1_noPr_noTau21",prMass_no[i].data()),200,0,200);
th5[222+i*2]=new TH1D(Form("%s_j0_noPr",prMass_no[i].data()),Form("%s_j0_noPr",prMass_no[i].data()),200,0,200);
th5[223+i*2]=new TH1D(Form("%s_j1_noPr",prMass_no[i].data()),Form("%s_j1_noPr",prMass_no[i].data()),200,0,200);
th5[230+i*2]=new TH1D(Form("%s_j0_noTau21",prMass_no[i].data()),Form("%s_j0_noTau21",prMass_no[i].data()),15,90,150);
th5[231+i*2]=new TH1D(Form("%s_j1_noTau21",prMass_no[i].data()),Form("%s_j1_noTau21",prMass_no[i].data()),15,90,150);
}
for (int i=0;i<2;i++){
th5[238+i*2]=new TH1D(Form("%s_j0_noPr_noTau21",tau21_no[i].data()),Form("%s_j0_noPr_noTau21",tau21_no[i].data()),25,0,1);
th5[239+i*2]=new TH1D(Form("%s_j1_noPr_noTau21",tau21_no[i].data()),Form("%s_j1_noPr_noTau21",tau21_no[i].data()),25,0,1);
th5[242+i*2]=new TH1D(Form("%s_j0_noPr",tau21_no[i].data()),Form("%s_j0_noPr",tau21_no[i].data()),25,0,1);
th5[243+i*2]=new TH1D(Form("%s_j1_noPr",tau21_no[i].data()),Form("%s_j1_noPr",tau21_no[i].data()),25,0,1);
th5[246+i*2]=new TH1D(Form("%s_j0_noTau21",tau21_no[i].data()),Form("%s_j0_noTau21",tau21_no[i].data()),25,0,1);
th5[247+i*2]=new TH1D(Form("%s_j1_noTau21",tau21_no[i].data()),Form("%s_j1_noTau21",tau21_no[i].data()),25,0,1);
}
for(int i=0;i<260;i++){
th5[i]->Sumw2();
th_flavor[0][i]=(TH1D* )th5[i]->Clone(Form("%s_bb",th5[i]->GetTitle()));
th_flavor[1][i]=(TH1D* )th5[i]->Clone(Form("%s_b",th5[i]->GetTitle()));
th_flavor[2][i]=(TH1D* )th5[i]->Clone(Form("%s_cc",th5[i]->GetTitle()));
th_flavor[3][i]=(TH1D* )th5[i]->Clone(Form("%s_udcsg",th5[i]->GetTitle()));
//th_flavor[0][i]->Sumw2();
//th_flavor[1][i]->Sumw2();
//th_flavor[2][i]->Sumw2();
//th_flavor[3][i]->Sumw2();
}
//pileup uncertainty----------------------------------------------------------------------------------------
TH1D* th7[14];
th7[0]=new TH1D("totalMass","totalMass",200,1000,5000);
th7[1]=new TH1D("totalMass_pileup_up","totalMass_pileup_up",200,1000,5000);
th7[2]=new TH1D("totalMass_pileup_down","totalMass_pileup_down",200,1000,5000);
th7[3]=new TH1D("pileupEff","pileupEff",15,0.5,15.5);
double totalPileup[3]={0},passPileup[3]={0};
standalone_LumiReWeighting LumiWeights_central(0),LumiWeights_up(1),LumiWeights_down(-1);
//PDF uncertainty
th7[4]=new TH1D("PDFEff","PDFEff",101,0.5,101.5);
double passPDF[101]={0},totalPDF[101]={0};
//QCD uncertainty
for(int i=5;i<14;i++)th7[i]=new TH1D(Form("uns_QCD_%d",i-5),Form("uns_QCD_%d",i-5),200,1000,5000);
//NCUtuple loop----------------------------------------------------------------------------------------
for (int w=wMs;w<wM;w++){
if(w%20==0)cout<<w<<endl;
//Get ntuple----------------------------------------------------------------------------------------
if (nameRoot!=1)f = TFile::Open(Form("%s%d.root",st.data(),w));
else f = TFile::Open(st.data());
if (!f || !f->IsOpen())continue;
TDirectory * dir;
if (nameRoot!=1)dir = (TDirectory*)f->Get(Form("%s%d.root:/tree",st.data(),w));
else dir = (TDirectory*)f->Get(Form("%s:/tree",st.data()));
dir->GetObject("treeMaker",tree);
TreeReader data(tree);
total+=data.GetEntriesFast();
//TFile* fileFast=TFile::Open(Form("fast/%s/%d.root",st2.data(),w));
//TTree* treeFast=(TTree* )fileFast->Get("hh4bFast");
//TreeReader dataFast(treeFast);
//if(isFast)dataFast=TreeReader(treeFast);
for(Long64_t jEntry=0; jEntry<data.GetEntriesFast() ;jEntry++){//event loop----------------------------------------------------------------------------------------
data.GetEntry(jEntry);
//if(isFast)dataFast.GetEntry(jEntry);
Int_t nVtx = data.GetInt("nVtx");
Float_t ntrue= data.GetFloat("pu_nTrueInt");
//Float_t* pdfscaleSysWeights= data.GetPtrFloat("pdfscaleSysWeights");
double PU_weight[3]={1,1,1};
if(nameRoot!=2){
if(ntrue<51){
PU_weight[0] = LumiWeights_central.weight(ntrue);
PU_weight[1]= LumiWeights_up.weight(ntrue);
PU_weight[2] = LumiWeights_down.weight(ntrue);
}
else {
PU_weight[0] = LumiWeights_central.weight(50);
PU_weight[1] = LumiWeights_up.weight(50);
PU_weight[2]= LumiWeights_down.weight(50);
}
}
fixScaleNum[0]+=PU_weight[0];
for(int i=0;i<3;i++)totalPileup[i]+=PU_weight[i];
for(int i=0;i<101;i++)totalPDF[i]+=PU_weight[0];
//Int_t nPassFast;
//if(isFast)nPassFast= dataFast.GetInt("nPassB");
//cout<<nPassFast<<endl;
//if(isFast &&nPassFast<8)continue;
//0. has a good vertex
if(nVtx<1)continue;nPass[0]+=PU_weight[0];
//1.trigger
std::string* trigName = data.GetPtrString("hlt_trigName");
vector<bool> &trigResult = *((vector<bool>*) data.GetPtr("hlt_trigResult"));
bool passTrigger=false;
for(int it=0; it< data.GetPtrStringSize(); it++){
std::string thisTrig= trigName[it];
bool results = trigResult[it];
if( ((thisTrig.find("HLT_PFHT800")!= std::string::npos||
//thisTrig.find("HLT_PFHT650")!= std::string::npos||
thisTrig.find("HLT_PFHT650_WideJetMJJ900DEtaJJ1p5_v")!= std::string::npos||
thisTrig.find("HLT_PFHT650_WideJetMJJ950DEtaJJ1p5_v")!= std::string::npos||
thisTrig.find("HLT_AK8PFJet360_TrimMass30_v")!= std::string::npos||
thisTrig.find("HLT_AK8PFHT700_TrimR0p1PT0p03Mass50_v")!= std::string::npos
) && results==1)){
passTrigger=true;
break;
}
}
if(!passTrigger && nameRoot==2)continue;nPass[1]+=PU_weight[0];
int nFATJet = data.GetInt("FATnJet");
TClonesArray* fatjetP4 = (TClonesArray*) data.GetPtrTObject("FATjetP4");
float* FATjetCorrUncUp = data.GetPtrFloat("FATjetCorrUncUp");
float* FATjetCorrUncDown = data.GetPtrFloat("FATjetCorrUncDown");
vector<bool> &FATjetPassIDTight = *((vector<bool>*) data.GetPtr("FATjetPassIDTight"));
//2.nJets
if(nFATJet<2)continue;nPass[2]+=PU_weight[0];
TLorentzVector* thisJet ,* thatJet;
if (JESOption==1){
TLorentzVector test0= (*((TLorentzVector*)fatjetP4->At(0)))*(1+FATjetCorrUncUp[0] );
TLorentzVector test1= (*((TLorentzVector*)fatjetP4->At(1)))*(1+FATjetCorrUncUp[1] );
thisJet= &test0;
thatJet= &test1;
}
else if (JESOption==2){
TLorentzVector test0= (*((TLorentzVector*)fatjetP4->At(0)))*(1-FATjetCorrUncDown[0] );
TLorentzVector test1= (*((TLorentzVector*)fatjetP4->At(1)))*(1-FATjetCorrUncDown[1] );
thisJet= &test0;
thatJet= &test1;
}
else{
thisJet= (TLorentzVector*)fatjetP4->At(0);
thatJet = (TLorentzVector*)fatjetP4->At(1);
}
//3. Pt
if(thisJet->Pt()<200||thatJet->Pt()<200)continue;
nPass[3]+=PU_weight[0];
//4tightId-----------------------------------------
if(FATjetPassIDTight[0]==0||FATjetPassIDTight[1]==0)continue;
Float_t* FATjetCEmEF = data.GetPtrFloat("FATjetCEmEF");
Float_t* FATjetMuEF = data.GetPtrFloat("FATjetMuEF");
if(FATjetMuEF[0]>0.8||FATjetMuEF[1]>0.8)continue;
if(FATjetCEmEF[0]>0.9||FATjetCEmEF[1]>0.9)continue;
nPass[4]+=PU_weight[0];
//5. Eta-----------------------------------------
if(fabs(thisJet->Eta())>2.4||fabs(thatJet->Eta())>2.4)continue;
nPass[5]+=PU_weight[0];
int event_flavor=-1;
Int_t* FATjetHadronFlavor = data.GetPtrInt("FATjetHadronFlavor");
vector<Int_t> *FATsubjetSDHadronFlavor = data.GetPtrVectorInt("FATsubjetSDHadronFlavor");
if(FATjetHadronFlavor[0]==5 && FATsubjetSDHadronFlavor[0][0]==5 && FATsubjetSDHadronFlavor[0][1]==5)event_flavor=0;
else if(FATjetHadronFlavor[1]==5 && FATsubjetSDHadronFlavor[1][0]==5 && FATsubjetSDHadronFlavor[1][1]==5)event_flavor=0;
else if(FATjetHadronFlavor[0]==5 && (FATsubjetSDHadronFlavor[0][0]==5 || FATsubjetSDHadronFlavor[0][1]==5))event_flavor=1;
else if(FATjetHadronFlavor[1]==5 && (FATsubjetSDHadronFlavor[1][0]==5 || FATsubjetSDHadronFlavor[1][1]==5))event_flavor=1;
else if(FATjetHadronFlavor[0]==4 && FATsubjetSDHadronFlavor[0][0]==4 && FATsubjetSDHadronFlavor[0][1]==4)event_flavor=2;
else if(FATjetHadronFlavor[1]==4 && FATsubjetSDHadronFlavor[1][0]==4 && FATsubjetSDHadronFlavor[1][1]==4)event_flavor=2;
else event_flavor=3;
th5[180]->Fill(nVtx,PU_weight[0]);
th5[181]->Fill(ntrue,PU_weight[0]);
th_flavor[event_flavor][180]->Fill(nVtx,PU_weight[0]);
th_flavor[event_flavor][181]->Fill(ntrue,PU_weight[0]);
//6. DEta-----------------------------------------
float dEta = fabs(thisJet->Eta()-thatJet->Eta());
if(dEta>1.3)continue;
nPass[6]+=PU_weight[0];
//7. Mjj-----------------------------------------
float mjj = (*thisJet+*thatJet).M();
float mjjRed = (*thisJet+*thatJet).M()+250-thisJet->M()-thatJet->M();
if(mjjRed<1000)continue;
nPass[7]+=PU_weight[0];
//8. fatjetPRmassL2L3Corr-----------------------------------------
Float_t* fatjetPRmassL2L3Corr = data.GetPtrFloat("FATjetPRmassL2L3Corr");
Float_t* FATjetPuppiSDmassL2L3Corr = data.GetPtrFloat("FATjetPuppiSDmassL2L3Corr");
Float_t* FATjetPRmass = data.GetPtrFloat("FATjetPRmass");
Float_t* FATjetPuppiSDmass = data.GetPtrFloat("FATjetPuppiSDmass");
vector<float> *subjetSDCSV = data.GetPtrVectorFloat("FATsubjetSDCSV");
vector<float> *subjetSDPx = data.GetPtrVectorFloat("FATsubjetSDPx", nFATJet);
vector<float> *subjetSDPy = data.GetPtrVectorFloat("FATsubjetSDPy", nFATJet);
vector<float> *subjetSDPz = data.GetPtrVectorFloat("FATsubjetSDPz", nFATJet);
vector<float> *subjetSDE = data.GetPtrVectorFloat("FATsubjetSDE", nFATJet);
int nbtag=0,nbtag2=0;
float MaxBJetPt = 670., MaxLJetPt = 1000.;
double sf[2][2],eta[2],pt[2],dr[2],subjetPt[2][2],subjetEta[2][2],eff[2][2],btaggingscaleFactor=1;
pt[0]=thisJet->Pt();
pt[1]=thatJet->Pt();
TLorentzVector* subjetP4[2][2];
for(int i=0;i<2;i++){
for(int j=0;j<2;j++){
sf[i][j]=1;
subjetP4[i][j]=new TLorentzVector(0,0,0,0);
subjetP4[i][j]->SetPxPyPzE(subjetSDPx[i][j],subjetSDPy[i][j],subjetSDPz[i][j],subjetSDE[i][j]);
subjetPt[i][j]=subjetP4[i][j]->Pt();
subjetEta[i][j]=subjetP4[i][j]->Eta();
}
dr[i]=subjetP4[i][0]->DeltaR(*subjetP4[i][1]);
}
for(int i=0;i<2;i++){
for(int j=0;j<2;j++){
//get btagging eff------------------------------------------------------------
int getPtBin=1;
if(subjetPt[i][j]<140)getPtBin=1;
else if (140<=subjetPt[i][j] && subjetPt[i][j]<180)getPtBin=2;
else if (180<=subjetPt[i][j] && subjetPt[i][j]<240)getPtBin=3;
else getPtBin=4;
//if(FATsubjetSDHadronFlavor[i][j]==5)eff[i][j]=th1[1]->GetBinContent(getPtBin,ceil(subjetEta[i][j]/0.2)+15);
//else if(FATsubjetSDHadronFlavor[i][j]==4)eff[i][j]=th1[3]->GetBinContent(getPtBin,ceil(subjetEta[i][j]/0.2)+15);
//else eff[i][j]=th1[5]->GetBinContent(getPtBin,ceil(subjetEta[i][j]/0.2)+15);
if(FATsubjetSDHadronFlavor[i][j]==5)eff[i][j]=th1[1]->GetBinContent(getPtBin);
else if(FATsubjetSDHadronFlavor[i][j]==4)eff[i][j]=th1[3]->GetBinContent(getPtBin);
else {
int temp=0;
if(subjetPt[i][j]>=900)temp=10;
else temp=ceil(subjetPt[i][j]/100);
bool checkBinContentIfZero=0;
while(checkBinContentIfZero==0){
if(th1[4]->GetBinContent(temp)==0){
temp--;
}
else checkBinContentIfZero=1;
}
eff[i][j]=th1[5]->GetBinContent(temp);
}
//Get SF from csv------------------------------------------------------------
if(FATsubjetSDHadronFlavor[i][j]==5){
sf[i][j]=HF.eval_auto_bounds("central",BTagEntry::FLAV_B, subjetEta[i][j],subjetPt[i][j]);
//sf[i][j]=HF.eval(BTagEntry::FLAV_B,subjetEta[i][j],subjetPt[i][j]);
}
else if(FATsubjetSDHadronFlavor[i][j]==4){
sf[i][j]=HFC.eval_auto_bounds("central",BTagEntry::FLAV_C, subjetEta[i][j],subjetPt[i][j]);
//sf[i][j]=HF.eval(BTagEntry::FLAV_C,subjetEta[i][j],subjetPt[i][j]);
}
else {
sf[i][j]=LF.eval_auto_bounds("central",BTagEntry::FLAV_UDSG, subjetEta[i][j],subjetPt[i][j]);
//sf[i][j]=LF.eval(BTagEntry::FLAV_UDSG,subjetEta[i][j],subjetPt[i][j]);
}
//conut nbtag---------------------------------------------------------
if(subjetSDCSV[i][j]>0.46)nbtag++;
//get tot. btagging SF
if(subjetSDCSV[i][j]>=0.46)btaggingscaleFactor*=sf[i][j];
else btaggingscaleFactor*=((1-eff[i][j]*sf[i][j])/(1-eff[i][j]));
}
}
if(nameRoot==2)btaggingscaleFactor=1;
double scaleFactor=PU_weight[0]*btaggingscaleFactor;
Float_t* fatjetTau1 = data.GetPtrFloat("FATjetTau1");
Float_t* fatjetTau2 = data.GetPtrFloat("FATjetTau2");
Float_t* FATjetPuppiTau1 = data.GetPtrFloat("FATjetPuppiTau1");
Float_t* FATjetPuppiTau2 = data.GetPtrFloat("FATjetPuppiTau2");
double tau21[2];
tau21[0]=(fatjetTau2[0]/fatjetTau1[0]),tau21[1]=(fatjetTau2[1]/fatjetTau1[1]);
double puppiTau21[2];
puppiTau21[0]=(FATjetPuppiTau2[0]/FATjetPuppiTau1[0]),puppiTau21[1]=(FATjetPuppiTau2[1]/FATjetPuppiTau1[1]);
for (int i=0;i<2;i++){
th5[214+i]->Fill(FATjetPRmass[i],scaleFactor);
th5[216+i]->Fill(fatjetPRmassL2L3Corr[i],scaleFactor);
th5[218+i]->Fill(FATjetPuppiSDmass[i],scaleFactor);
th5[220+i]->Fill(FATjetPuppiSDmassL2L3Corr[i],scaleFactor);
th5[238+i]->Fill(tau21[i],scaleFactor);
th5[240+i]->Fill(puppiTau21[i],scaleFactor);
th_flavor[event_flavor][214+i]->Fill(FATjetPRmass[i],scaleFactor);
th_flavor[event_flavor][216+i]->Fill(fatjetPRmassL2L3Corr[i],scaleFactor);
th_flavor[event_flavor][218+i]->Fill(FATjetPuppiSDmass[i],scaleFactor);
th_flavor[event_flavor][220+i]->Fill(FATjetPuppiSDmassL2L3Corr[i],scaleFactor);
th_flavor[event_flavor][238+i]->Fill(tau21[i],scaleFactor);
th_flavor[event_flavor][240+i]->Fill(puppiTau21[i],scaleFactor);
}
if(!(tau21[0]>0.6 || tau21[1]>0.6)){
for (int i=0;i<2;i++){
th5[222+i]->Fill(FATjetPRmass[i],scaleFactor);
th5[224+i]->Fill(fatjetPRmassL2L3Corr[i],scaleFactor);
th5[226+i]->Fill(FATjetPuppiSDmass[i],scaleFactor);
th5[228+i]->Fill(FATjetPuppiSDmassL2L3Corr[i],scaleFactor);
th5[242+i]->Fill(tau21[i],scaleFactor);
th5[244+i]->Fill(puppiTau21[i],scaleFactor);
th_flavor[event_flavor][222+i]->Fill(FATjetPRmass[i],scaleFactor);
th_flavor[event_flavor][224+i]->Fill(fatjetPRmassL2L3Corr[i],scaleFactor);
th_flavor[event_flavor][226+i]->Fill(FATjetPuppiSDmass[i],scaleFactor);
th_flavor[event_flavor][228+i]->Fill(FATjetPuppiSDmassL2L3Corr[i],scaleFactor);
th_flavor[event_flavor][242+i]->Fill(tau21[i],scaleFactor);
th_flavor[event_flavor][244+i]->Fill(puppiTau21[i],scaleFactor);
}
}
if(fatjetPRmassL2L3Corr[0]<105||fatjetPRmassL2L3Corr[0]>135)continue;
if(fatjetPRmassL2L3Corr[1]<105||fatjetPRmassL2L3Corr[1]>135)continue;
nPass[8]+=PU_weight[0];
for (int i=0;i<2;i++){
th5[230+i]->Fill(FATjetPRmass[i],scaleFactor);
th5[232+i]->Fill(fatjetPRmassL2L3Corr[i],scaleFactor);
th5[234+i]->Fill(FATjetPuppiSDmass[i],scaleFactor);
th5[236+i]->Fill(FATjetPuppiSDmassL2L3Corr[i],scaleFactor);
th5[246+i]->Fill(tau21[i],scaleFactor);
th5[248+i]->Fill(puppiTau21[i],scaleFactor);
th_flavor[event_flavor][230+i]->Fill(FATjetPRmass[i],scaleFactor);
th_flavor[event_flavor][232+i]->Fill(fatjetPRmassL2L3Corr[i],scaleFactor);
th_flavor[event_flavor][234+i]->Fill(FATjetPuppiSDmass[i],scaleFactor);
th_flavor[event_flavor][236+i]->Fill(FATjetPuppiSDmassL2L3Corr[i],scaleFactor);
th_flavor[event_flavor][246+i]->Fill(tau21[i],scaleFactor);
th_flavor[event_flavor][248+i]->Fill(puppiTau21[i],scaleFactor);
}
//9.-----------------------------------------
if(tau21[0]>0.6 || tau21[1]>0.6) continue;
nPass[9]+=PU_weight[0];
double tau21_SF=1.031*0.881;
if(JESOption==5)tau21_SF=(1.031+0.126)*(0.881+0.49);
if(JESOption==6)tau21_SF=(1.031-0.126)*(0.881-0.49);
if(tau21[0]<0.6 && tau21[1]<0.6 ){
tau21_SF=1.031*1.031;
if(JESOption==5)tau21_SF=(1.031+0.126)*(1.031+0.126);
if(JESOption==6)tau21_SF=(1.031-0.126)*(1.031-0.126);
}
//10.btag
//uncertainty -------------------------------------
//double scaleFactor=btaggingscaleFactor*PU_weight[0]*tau21_SF;
for(int i=0;i<3;i++){
passPileup[i]+=btaggingscaleFactor*PU_weight[i]*tau21_SF;
th7[i]->Fill(mjj,btaggingscaleFactor*PU_weight[i]*tau21_SF);
}
for(int i=0;i<101;i++)passPDF[i]+=btaggingscaleFactor*PU_weight[0]*tau21_SF;
for(int i=5;i<14;i++)th7[i]->Fill(mjj,btaggingscaleFactor*PU_weight[0]*tau21_SF);
fixScaleNum[1]+=PU_weight[0];
//--------------------------------------
Float_t FATjetPuppiSDmassThea[2] ={0};
FATjetPuppiSDmassThea[0]=FATjetPuppiSDmass[0]*getPUPPIweight(thisJet->Pt(),thisJet->Eta(),tf1);
FATjetPuppiSDmassThea[1]=FATjetPuppiSDmass[1]*getPUPPIweight(thatJet->Pt(),thatJet->Eta(),tf1);
eta[0]=thisJet->Eta();
eta[1]=thatJet->Eta();
Float_t* ADDjet_DoubleSV = data.GetPtrFloat("ADDjet_DoubleSV");
Float_t FATjet_DoubleSV[2]={0};
Int_t ADDnJet = data.GetInt("ADDnJet");
bool matchThis=0,matchThat=0;
TClonesArray* ADDjetP4 = (TClonesArray*) data.GetPtrTObject("ADDjetP4");
for(int i=0;i<ADDnJet;i++){
TLorentzVector* thisAddJet ;
thisAddJet= (TLorentzVector*)ADDjetP4->At(i);
if(!matchThis && thisAddJet->DeltaR(*thisJet)<0.8){
matchThis=1;
FATjet_DoubleSV[0]=ADDjet_DoubleSV[i];
continue;
}
if(!matchThat && thisAddJet->DeltaR(*thatJet)<0.8){
matchThat=1;
FATjet_DoubleSV[1]=ADDjet_DoubleSV[i];
}
if(matchThis&& matchThat)break;
}
Float_t* FATjetCISVV2 = data.GetPtrFloat("FATjetCISVV2");
for(int i=0;i<2;i++){
for(int j=0;j<2;j++){
th5[(i*2+j)*5+nbtag]->Fill(subjetPt[i][j],scaleFactor);
th5[(i*2+j)*5+nbtag+20]->Fill(subjetEta[i][j],scaleFactor);
th5[(i*2+j)*5+nbtag+85]->Fill(subjetSDCSV[i][j],scaleFactor);
if(subjetPt[i][j]>30 && fabs(subjetEta[i][j])<2.4)th5[(i*2+j)*5+nbtag+194]->Fill(subjetSDCSV[i][j],scaleFactor);
th_flavor[event_flavor][(i*2+j)*5+nbtag]->Fill(subjetPt[i][j],scaleFactor);
th_flavor[event_flavor][(i*2+j)*5+nbtag+20]->Fill(subjetEta[i][j],scaleFactor);
th_flavor[event_flavor][(i*2+j)*5+nbtag+85]->Fill(subjetSDCSV[i][j],scaleFactor);
if(subjetPt[i][j]>30 && fabs(subjetEta[i][j])<2.4)th_flavor[event_flavor][(i*2+j)*5+nbtag+194]->Fill(subjetSDCSV[i][j],scaleFactor);
}
th5[i*5+nbtag+40]->Fill(dr[i],scaleFactor);
th5[i*5+nbtag+50]->Fill(pt[i],scaleFactor);
th5[i*5+nbtag+60]->Fill(eta[i],scaleFactor);
th5[i*5+nbtag+70]->Fill(fatjetPRmassL2L3Corr[i],scaleFactor);
th5[i*5+nbtag+105]->Fill(tau21[i],scaleFactor);
th5[i*5+nbtag+120]->Fill(FATjetPuppiSDmassL2L3Corr[i],scaleFactor);
th5[i*5+nbtag+130]->Fill(puppiTau21[i],scaleFactor);
th5[i*5+nbtag+140]->Fill(FATjetPRmass[i],scaleFactor);
th5[i*5+nbtag+150]->Fill(FATjetPuppiSDmass[i],scaleFactor);
th5[i*5+nbtag+170]->Fill(FATjet_DoubleSV[i],PU_weight[0]);
th5[i*5+nbtag+184]->Fill(FATjetCISVV2[i],scaleFactor);
th5[i*5+nbtag+250]->Fill(FATjetPuppiSDmassThea[i],scaleFactor);
th_flavor[event_flavor][i*5+nbtag+40]->Fill(dr[i],scaleFactor);
th_flavor[event_flavor][i*5+nbtag+50]->Fill(pt[i],scaleFactor);
th_flavor[event_flavor][i*5+nbtag+60]->Fill(eta[i],scaleFactor);
th_flavor[event_flavor][i*5+nbtag+70]->Fill(fatjetPRmassL2L3Corr[i],scaleFactor);
th_flavor[event_flavor][i*5+nbtag+105]->Fill(tau21[i],scaleFactor);
th_flavor[event_flavor][i*5+nbtag+120]->Fill(FATjetPuppiSDmassL2L3Corr[i],scaleFactor);
th_flavor[event_flavor][i*5+nbtag+130]->Fill(puppiTau21[i],scaleFactor);
th_flavor[event_flavor][i*5+nbtag+140]->Fill(FATjetPRmass[i],scaleFactor);
th_flavor[event_flavor][i*5+nbtag+150]->Fill(FATjetPuppiSDmass[i],scaleFactor);
th_flavor[event_flavor][i*5+nbtag+170]->Fill(FATjet_DoubleSV[i],PU_weight[0]);
th_flavor[event_flavor][i*5+nbtag+184]->Fill(FATjetCISVV2[i],scaleFactor);
th_flavor[event_flavor][i*5+nbtag+250]->Fill(FATjetPuppiSDmassThea[i],scaleFactor);
}
th5[nbtag+80]->Fill(mjj,scaleFactor);
th5[nbtag+115]->Fill(dEta,scaleFactor);
th5[nbtag+165]->Fill(mjjRed,scaleFactor);
th_flavor[event_flavor][nbtag+80]->Fill(mjj,scaleFactor);
th_flavor[event_flavor][nbtag+115]->Fill(dEta,scaleFactor);
th_flavor[event_flavor][nbtag+165]->Fill(mjjRed,scaleFactor);
nPass[nbtag+10]+=PU_weight[0]*btaggingscaleFactor;
int nDoubleSV=0;
if(FATjet_DoubleSV[0]>0.6)nDoubleSV++;
if(FATjet_DoubleSV[1]>0.6)nDoubleSV++;
if(nDoubleSV==0)nPass[15]+=PU_weight[0];
else if(nDoubleSV==2)nPass[17]+=PU_weight[0];
else {
nPass[16]+=PU_weight[0];
if((FATjet_DoubleSV[0]>0.6 && subjetSDCSV[1][0]<0.46 && subjetSDCSV[1][1]<0.46) ||
(FATjet_DoubleSV[1]>0.6 && subjetSDCSV[0][0]<0.46 && subjetSDCSV[0][1]<0.46))nPass[18]+=PU_weight[0];
else if((FATjet_DoubleSV[0]>0.6 && subjetSDCSV[1][0]>0.46 && subjetSDCSV[1][1]>0.46) ||
(FATjet_DoubleSV[1]>0.6 && subjetSDCSV[0][0]>0.46 && subjetSDCSV[0][1]>0.46))nPass[20]+=PU_weight[0];
else nPass[19]+=PU_weight[0];
}
th5[182]->Fill(nVtx,scaleFactor);
th5[183]->Fill(ntrue,scaleFactor);
th_flavor[event_flavor][182]->Fill(nVtx,scaleFactor);
th_flavor[event_flavor][183]->Fill(ntrue,scaleFactor);
}//end event loop----------------------------------------------------------------------------------------
} //end ntuple loop----------------------------------------------------------------------------------------
cout<<"entries="<<total<<endl;
for(int i=0;i<22;i++)cout<<"nPass["<<i<<"]="<<nPass[i]<<endl;
for(int i=0;i<3;i++)th7[3]->SetBinContent(i+1,passPileup[i]/totalPileup[i]);
for(int i=0;i<101;i++)th7[4]->SetBinContent(i+1,passPDF[i]/totalPDF[i]);
TH1D * th2o=new TH1D("Nbtagjet","Nbtagjet",5,-0.5,4.5);
for (int i=0;i<5;i++){
if(nameRoot==2 && i>2)continue;
th2o->SetBinContent(i+1,nPass[i+10]);
}
TH1D * fixScale=new TH1D("fixScale","fixScale",2,-0.5,1.5);
fixScale->SetBinContent(1,fixScaleNum[0]);
fixScale->SetBinContent(2,fixScaleNum[1]);
TH1D * cutflow=new TH1D("cutflow","cutflow",21,0.5,21.5);
cutflow->SetBinContent(1,fixScaleNum[0]);
if(nameRoot!=2)for(int ii=1;ii<22;ii++)cutflow->SetBinContent(ii+1,nPass[ii-1]);
else for(int ii=1;ii<16;ii++)cutflow->SetBinContent(ii+1,nPass[ii-1]);
TFile* outFile ;
if(JESOption==0)outFile= new TFile(Form("sf2/%s.root",st2.data()),"recreate");
else if(JESOption==1)outFile= new TFile(Form("sf2/%s_JESUp.root",st2.data()),"recreate");
else if(JESOption==2)outFile= new TFile(Form("sf2/%s_JESDown.root",st2.data()),"recreate");
else if(JESOption==3)outFile= new TFile(Form("sf2/%s_BtagUp.root",st2.data()),"recreate");
else if(JESOption==4)outFile= new TFile(Form("sf2/%s_BtagDown.root",st2.data()),"recreate");
else if(JESOption==5)outFile= new TFile(Form("sf2/%s_tau21Up.root",st2.data()),"recreate");
else if(JESOption==6)outFile= new TFile(Form("sf2/%s_tau21Down.root",st2.data()),"recreate");
th2o->Write();
fixScale->Write();
cutflow->Write();
for(int i=0;i<260;i++){
th5[i]->Write();
th_flavor[0][i]->Write();
th_flavor[1][i]->Write();
th_flavor[2][i]->Write();
th_flavor[3][i]->Write();
}
for(int i=0;i<14;i++)th7[i]->Write();
outFile->Close();
}
/*
=================
ServerLoadMDXM - load a Ghoul 2 Mesh file
=================
*/
qboolean ServerLoadMDXM( model_t *mod, void *buffer, const char *mod_name, qboolean &bAlreadyCached ) {
int i,l, j;
mdxmHeader_t *pinmodel, *mdxm;
mdxmLOD_t *lod;
mdxmSurface_t *surf;
int version;
int size;
//shader_t *sh;
mdxmSurfHierarchy_t *surfInfo;
#ifdef Q3_BIG_ENDIAN
int k;
mdxmTriangle_t *tri;
mdxmVertex_t *v;
int *boneRef;
mdxmLODSurfOffset_t *indexes;
mdxmVertexTexCoord_t *pTexCoords;
mdxmHierarchyOffsets_t *surfIndexes;
#endif
pinmodel= (mdxmHeader_t *)buffer;
//
// read some fields from the binary, but only LittleLong() them when we know this wasn't an already-cached model...
//
version = (pinmodel->version);
size = (pinmodel->ofsEnd);
if (!bAlreadyCached)
{
LL(version);
LL(size);
}
if (version != MDXM_VERSION) {
return qfalse;
}
mod->type = MOD_MDXM;
mod->dataSize += size;
qboolean bAlreadyFound = qfalse;
mdxm = mod->mdxm = (mdxmHeader_t*) //Hunk_Alloc( size );
RE_RegisterServerModels_Malloc(size, buffer, mod_name, &bAlreadyFound, TAG_MODEL_GLM);
assert(bAlreadyCached == bAlreadyFound); // I should probably eliminate 'bAlreadyFound', but wtf?
if (!bAlreadyFound)
{
// horrible new hackery, if !bAlreadyFound then we've just done a tag-morph, so we need to set the
// bool reference passed into this function to true, to tell the caller NOT to do an ri->FS_Freefile since
// we've hijacked that memory block...
//
// Aaaargh. Kill me now...
//
bAlreadyCached = qtrue;
assert( mdxm == buffer );
// memcpy( mdxm, buffer, size ); // and don't do this now, since it's the same thing
LL(mdxm->ident);
LL(mdxm->version);
LL(mdxm->numBones);
LL(mdxm->numLODs);
LL(mdxm->ofsLODs);
LL(mdxm->numSurfaces);
LL(mdxm->ofsSurfHierarchy);
LL(mdxm->ofsEnd);
}
// first up, go load in the animation file we need that has the skeletal animation info for this model
mdxm->animIndex = RE_RegisterServerModel(va ("%s.gla",mdxm->animName));
if (!mdxm->animIndex)
{
return qfalse;
}
mod->numLods = mdxm->numLODs -1 ; //copy this up to the model for ease of use - it wil get inced after this.
if (bAlreadyFound)
{
return qtrue; // All done. Stop, go no further, do not LittleLong(), do not pass Go...
}
surfInfo = (mdxmSurfHierarchy_t *)( (byte *)mdxm + mdxm->ofsSurfHierarchy);
#ifdef Q3_BIG_ENDIAN
surfIndexes = (mdxmHierarchyOffsets_t *)((byte *)mdxm + sizeof(mdxmHeader_t));
#endif
for ( i = 0 ; i < mdxm->numSurfaces ; i++)
{
LL(surfInfo->numChildren);
LL(surfInfo->parentIndex);
// do all the children indexs
for (j=0; j<surfInfo->numChildren; j++)
{
LL(surfInfo->childIndexes[j]);
}
// We will not be using shaders on the server.
//sh = 0;
// insert it in the surface list
surfInfo->shaderIndex = 0;
RE_RegisterModels_StoreShaderRequest(mod_name, &surfInfo->shader[0], &surfInfo->shaderIndex);
#ifdef Q3_BIG_ENDIAN
// swap the surface offset
LL(surfIndexes->offsets[i]);
assert(surfInfo == (mdxmSurfHierarchy_t *)((byte *)surfIndexes + surfIndexes->offsets[i]));
#endif
// find the next surface
surfInfo = (mdxmSurfHierarchy_t *)( (byte *)surfInfo + (intptr_t)( &((mdxmSurfHierarchy_t *)0)->childIndexes[ surfInfo->numChildren ] ));
}
// swap all the LOD's (we need to do the middle part of this even for intel, because of shader reg and err-check)
lod = (mdxmLOD_t *) ( (byte *)mdxm + mdxm->ofsLODs );
for ( l = 0 ; l < mdxm->numLODs ; l++)
{
int triCount = 0;
LL(lod->ofsEnd);
// swap all the surfaces
surf = (mdxmSurface_t *) ( (byte *)lod + sizeof (mdxmLOD_t) + (mdxm->numSurfaces * sizeof(mdxmLODSurfOffset_t)) );
for ( i = 0 ; i < mdxm->numSurfaces ; i++)
{
LL(surf->thisSurfaceIndex);
LL(surf->ofsHeader);
LL(surf->numVerts);
LL(surf->ofsVerts);
LL(surf->numTriangles);
LL(surf->ofsTriangles);
LL(surf->numBoneReferences);
LL(surf->ofsBoneReferences);
LL(surf->ofsEnd);
triCount += surf->numTriangles;
if ( surf->numVerts > SHADER_MAX_VERTEXES ) {
return qfalse;
}
if ( surf->numTriangles*3 > SHADER_MAX_INDEXES ) {
return qfalse;
}
// change to surface identifier
surf->ident = SF_MDX;
// register the shaders
#ifdef Q3_BIG_ENDIAN
// swap the LOD offset
indexes = (mdxmLODSurfOffset_t *)((byte *)lod + sizeof(mdxmLOD_t));
LL(indexes->offsets[surf->thisSurfaceIndex]);
// do all the bone reference data
boneRef = (int *) ( (byte *)surf + surf->ofsBoneReferences );
for ( j = 0 ; j < surf->numBoneReferences ; j++ )
{
LL(boneRef[j]);
}
// swap all the triangles
tri = (mdxmTriangle_t *) ( (byte *)surf + surf->ofsTriangles );
for ( j = 0 ; j < surf->numTriangles ; j++, tri++ )
{
LL(tri->indexes[0]);
LL(tri->indexes[1]);
LL(tri->indexes[2]);
}
// swap all the vertexes
v = (mdxmVertex_t *) ( (byte *)surf + surf->ofsVerts );
pTexCoords = (mdxmVertexTexCoord_t *) &v[surf->numVerts];
for ( j = 0 ; j < surf->numVerts ; j++ )
{
LF(v->normal[0]);
LF(v->normal[1]);
LF(v->normal[2]);
LF(v->vertCoords[0]);
LF(v->vertCoords[1]);
LF(v->vertCoords[2]);
LF(pTexCoords[j].texCoords[0]);
LF(pTexCoords[j].texCoords[1]);
LL(v->uiNmWeightsAndBoneIndexes);
v++;
}
#endif
// find the next surface
surf = (mdxmSurface_t *)( (byte *)surf + surf->ofsEnd );
}
// find the next LOD
lod = (mdxmLOD_t *)( (byte *)lod + lod->ofsEnd );
}
return qtrue;
}
/*
=================
ServerLoadMDXA - load a Ghoul 2 animation file
=================
*/
qboolean ServerLoadMDXA( model_t *mod, void *buffer, const char *mod_name, qboolean &bAlreadyCached ) {
mdxaHeader_t *pinmodel, *mdxa;
int version;
int size;
#ifdef Q3_BIG_ENDIAN
int j, k, i;
mdxaSkel_t *boneInfo;
mdxaSkelOffsets_t *offsets;
int maxBoneIndex = 0;
mdxaCompQuatBone_t *pCompBonePool;
unsigned short *pwIn;
mdxaIndex_t *pIndex;
#endif
pinmodel = (mdxaHeader_t *)buffer;
//
// read some fields from the binary, but only LittleLong() them when we know this wasn't an already-cached model...
//
version = (pinmodel->version);
size = (pinmodel->ofsEnd);
if (!bAlreadyCached)
{
LL(version);
LL(size);
}
if (version != MDXA_VERSION) {
return qfalse;
}
mod->type = MOD_MDXA;
mod->dataSize += size;
qboolean bAlreadyFound = qfalse;
mdxa = mod->mdxa = (mdxaHeader_t*) //Hunk_Alloc( size );
RE_RegisterServerModels_Malloc(size, buffer, mod_name, &bAlreadyFound, TAG_MODEL_GLA);
assert(bAlreadyCached == bAlreadyFound); // I should probably eliminate 'bAlreadyFound', but wtf?
if (!bAlreadyFound)
{
// horrible new hackery, if !bAlreadyFound then we've just done a tag-morph, so we need to set the
// bool reference passed into this function to true, to tell the caller NOT to do an ri->FS_Freefile since
// we've hijacked that memory block...
//
// Aaaargh. Kill me now...
//
bAlreadyCached = qtrue;
assert( mdxa == buffer );
// memcpy( mdxa, buffer, size ); // and don't do this now, since it's the same thing
LL(mdxa->ident);
LL(mdxa->version);
//LF(mdxa->fScale);
LL(mdxa->numFrames);
LL(mdxa->ofsFrames);
LL(mdxa->numBones);
LL(mdxa->ofsCompBonePool);
LL(mdxa->ofsSkel);
LL(mdxa->ofsEnd);
}
if ( mdxa->numFrames < 1 ) {
return qfalse;
}
if (bAlreadyFound)
{
return qtrue; // All done, stop here, do not LittleLong() etc. Do not pass go...
}
#ifdef Q3_BIG_ENDIAN
// swap the bone info
offsets = (mdxaSkelOffsets_t *)((byte *)mdxa + sizeof(mdxaHeader_t));
for ( i = 0; i < mdxa->numBones ; i++ )
{
LL(offsets->offsets[i]);
boneInfo = (mdxaSkel_t *)((byte *)mdxa + sizeof(mdxaHeader_t) + offsets->offsets[i]);
LL(boneInfo->flags);
LL(boneInfo->parent);
for ( j = 0; j < 3; j++ )
{
for ( k = 0; k < 4; k++)
{
LF(boneInfo->BasePoseMat.matrix[j][k]);
LF(boneInfo->BasePoseMatInv.matrix[j][k]);
}
}
LL(boneInfo->numChildren);
for (k=0; k<boneInfo->numChildren; k++)
{
LL(boneInfo->children[k]);
}
}
// find the largest index, since the actual number of compressed bone pools is not stored anywhere
for ( i = 0 ; i < mdxa->numFrames ; i++ )
{
for ( j = 0 ; j < mdxa->numBones ; j++ )
{
k = (i * mdxa->numBones * 3) + (j * 3); // iOffsetToIndex
pIndex = (mdxaIndex_t *) ((byte*) mdxa + mdxa->ofsFrames + k);
// 3 byte ints, yeah...
int tmp = pIndex->iIndex & 0xFFFFFF00;
LL(tmp);
if (maxBoneIndex < tmp)
maxBoneIndex = tmp;
}
}
// swap the compressed bones
pCompBonePool = (mdxaCompQuatBone_t *) ((byte *)mdxa + mdxa->ofsCompBonePool);
for ( i = 0 ; i <= maxBoneIndex ; i++ )
{
pwIn = (unsigned short *) pCompBonePool[i].Comp;
for ( k = 0 ; k < 7 ; k++ )
LS(pwIn[k]);
}
#endif
return qtrue;
}
void simulation(aveq *A, function *G, function *S, int k, int method) { //l'intero method serve a designare il metodo di integrazione
// 0 = FTCS
// 1 = LeapFrog
// 2 = LaxWendroff
// 3 = LaxFriedrichs
int i, t, it;
char filename[50];
FILE *pt;
for(A->cf = 0.5; A->cf <= 1; A->cf += 0.5) { //ciclo su due diversi fattori di Courant
for(A->time = 10; A->time <= 20; A->time += 10) { //ciclo su due diversi tempi di esecuzione
A->xstep = 1e-1; //step di partenza
for(it = 1; it <= 3; it++) { //ciclo sulle dimensioni dello step (e conseguentemente dell'array)
A->tstep = A->cf * A->xstep; //definizione dello step temporale
//-----------------------------definizione degli array
A->dim = (int)(XMAX/A->xstep+1); //dimensione degli array
//malloc non perchè grandi ma per evitare problemi di allocamento di memoria inutile nei cicli
double *solution = (double *)malloc((A->dim) * sizeof(double)); //soluzione
double *tsol = (double *)malloc((A->dim) * sizeof(double)); //tmp soluzione
double *tsol_1 = (double *)malloc((A->dim) * sizeof(double)); //tmp soluzione3
//associa agli elementi della struct
A->solution = solution;
A->tsol = tsol;
A->tsol_1 = tsol_1;
//-----------------------------
printf("\t-> Simulating: cf = %g, execution time = %d, spatial step = %g\n", A->cf, A->time, A->xstep);
//genera i file nelle cartelle che vengono create ed esegui in base al valore di method
switch(method) {
case 0:
if(k == 1) snprintf(filename, sizeof(filename), "FTCS_G/ftcs_cf%g_xstep%g_T%d_k%d", A->cf, A->xstep, A->time,k);
else snprintf(filename, sizeof(filename), "FTCS_S/ftcs_cf%g_xstep%g_T%d_k%d", A->cf, A->xstep, A->time,k);
pt = fopen(filename,"w");
FTCS(A, G, S, k); //FTCS
break;
case 1:
if(k == 1) snprintf(filename, sizeof(filename), "LEAPFROG_G/lpfr_cf%g_xstep%g_T%d_k%d", A->cf, A->xstep, A->time,k);
else snprintf(filename, sizeof(filename), "LEAPFROG_S/lpfr_cf%g_xstep%g_T%d_k%d", A->cf, A->xstep, A->time,k);
pt = fopen(filename,"w");
LeFr(A, G, S, k); //Leapfrog
break;
case 2:
if( k== 1) snprintf(filename, sizeof(filename), "LAXWENDROFF_G/lw_cf%g_xstep%g_T%d_k%d", A->cf, A->xstep, A->time,k);
else snprintf(filename, sizeof(filename), "LAXWENDROFF_S/lw_cf%g_xstep%g_T%d_k%d", A->cf, A->xstep, A->time,k);
pt = fopen(filename,"w");
LW(A, G, S, k); //Lax Wendroff
break;
case 3:
if( k== 1) snprintf(filename, sizeof(filename), "LAXFRIEDRICHS_G/lf_cf%g_xstep%g_T%d_k%d", A->cf, A->xstep, A->time,k);
else snprintf(filename, sizeof(filename), "LAXFRIEDRICHS_S/lf_cf%g_xstep%g_T%d_k%d", A->cf, A->xstep, A->time,k);
pt = fopen(filename,"w");
LF(A, G, S, k); //Lax Friedrichs
break;
case 4:
if( k== 1) snprintf(filename, sizeof(filename), "LEAPFROG-WAVE/lefr_cf%g_xstep%g_T%d_k%d", A->cf, A->xstep, A->time,k);
else {
printf("\t-> L'equazione delle onde può essere risolta solo con condizioni gaussiane.\n");
exit(0);
}
pt = fopen(filename,"w");
LF_WAVE(A, G, S, k); //Leapfrog per l'equazione delle onde
break;
case 5:
if( k == 1) snprintf(filename, sizeof(filename), "LAXWENDROFF-WAVE/lw_cf%g_xstep%g_T%d_k%d", A->cf, A->xstep, A->time,k);
else {
printf("\t-> L'equazione delle onde può essere risolta solo con condizioni gaussiane.\n");
exit(0);
}
pt = fopen(filename,"w");
LW_WAVE(A, G, S, k); //Lax-Wendroff per l'equazione delle onde
break;
}
//stampa finalmente i valori di solution su file
for(i = 0; i < A->dim; i++)
fprintf(pt, "%g \t %g\n", i*A->xstep, A->solution[i]);
//libera tutto
fclose(pt);
free(solution);
free(tsol);
free(tsol_1);
A->xstep /= 10.; //dividi per 10 lo step ogni volta che it viene incrementato di uno
}
}
}
}
// grid offsets
STATIC_UNIT_TESTED uint8_t highestYValueForNorth;
STATIC_UNIT_TESTED uint8_t lowestYValueForSouth;
STATIC_UNIT_TESTED uint8_t highestXValueForWest;
STATIC_UNIT_TESTED uint8_t lowestXValueForEast;
STATIC_UNIT_TESTED ledCounts_t ledCounts;
// macro for initializer
#define LF(name) LED_FUNCTION_ ## name
#define LO(name) LED_FLAG_OVERLAY(LED_OVERLAY_ ## name)
#define LD(name) LED_FLAG_DIRECTION(LED_DIRECTION_ ## name)
#ifdef USE_LED_RING_DEFAULT_CONFIG
static const ledConfig_t defaultLedStripConfig[] = {
DEFINE_LED( 2, 2, 3, 0, LF(THRUST_RING), 0, 0),
DEFINE_LED( 2, 1, 3, 0, LF(THRUST_RING), 0, 0),
DEFINE_LED( 2, 0, 3, 0, LF(THRUST_RING), 0, 0),
DEFINE_LED( 1, 0, 3, 0, LF(THRUST_RING), 0, 0),
DEFINE_LED( 0, 0, 3, 0, LF(THRUST_RING), 0, 0),
DEFINE_LED( 0, 1, 3, 0, LF(THRUST_RING), 0, 0),
DEFINE_LED( 0, 2, 3, 0, LF(THRUST_RING), 0, 0),
DEFINE_LED( 1, 2, 3, 0, LF(THRUST_RING), 0, 0),
DEFINE_LED( 1, 1, 3, 0, LF(THRUST_RING), 0, 0),
DEFINE_LED( 1, 1, 3, 0, LF(THRUST_RING), 0, 0),
DEFINE_LED( 1, 1, 3, 0, LF(THRUST_RING), 0, 0),
DEFINE_LED( 1, 1, 3, 0, LF(THRUST_RING), 0, 0),
};
#else
static const ledConfig_t defaultLedStripConfig[] = {
DEFINE_LED(15, 15, 0, LD(SOUTH) | LD(EAST), LF(ARM_STATE), LO(INDICATOR), 0),
void CLitePrint::LF(UINT nmb) {LF(GetActiveFont(),nmb);}
void EscapingPlanner::OutFeetPosi()
{
MatrixXd nextLF(3, 3), nextRF(3, 3);
for (unsigned int i = 1; i < bodyPoses.size(); i++)
{
MatrixXd LF(3, 3), RF(3, 3);
LF << -0.3, -0.3, 0.45,
0.65, -0.65, 0,
1, 1, 1;
RF << -0.45, 0.3, 0.3,
0, -0.65, 0.65,
1, 1, 1;
double theta = -M_PI / 2 + bodyPoses[i].alpha;
double xRobPos = bodyPoses[i].x;
double yRobPos = bodyPoses[i].y;
Matrix3d transT;
transT << cos(theta), -sin(theta), xRobPos,
sin(theta), cos(theta), yRobPos,
0, 0, 1;
FootHold tempFootHold = feetPoses.back();
if (leftSwing)
{
LF.row(1) += MatrixXd::Ones(1, 3) * halfStep / 2;
nextLF = transT * LF;
RF.row(1) -= MatrixXd::Ones(1, 3) * halfStep / 2;
nextRF = transT * RF;
leftSwing = !leftSwing;
tempFootHold.feetHold[0] = nextLF(0, 0);
tempFootHold.feetHold[1] = nextLF(1, 0);
tempFootHold.feetHold[4] = nextLF(0, 1);
tempFootHold.feetHold[5] = nextLF(1, 1);
tempFootHold.feetHold[8] = nextLF(0, 2);
tempFootHold.feetHold[9] = nextLF(1, 2);
}
else
{
LF.row(1) -= MatrixXd::Ones(1, 3) * halfStep / 2;
nextLF = transT * LF;
RF.row(1) += MatrixXd::Ones(1, 3) * halfStep / 2;
nextRF = transT * RF;
leftSwing = !leftSwing;
tempFootHold.feetHold[2] = nextRF(0, 0);
tempFootHold.feetHold[3] = nextRF(1, 0);
tempFootHold.feetHold[6] = nextRF(0, 2);
tempFootHold.feetHold[7] = nextRF(1, 2);
tempFootHold.feetHold[10] = nextRF(0, 1);
tempFootHold.feetHold[11] = nextRF(1, 1);
}
feetPoses.push_back(tempFootHold);
//cout << feetPoses.back().feetHold[0] << " " << feetPoses.back().feetHold[1] << " " << feetPoses.back().feetHold[2] << " " << feetPoses.back().feetHold[3] << " "
// << feetPoses.back().feetHold[4] << " " << feetPoses.back().feetHold[5] << " " << feetPoses.back().feetHold[6] << " " << feetPoses.back().feetHold[7] << " "
// << feetPoses.back().feetHold[8] << " " << feetPoses.back().feetHold[9] << " " << feetPoses.back().feetHold[10] << " " << feetPoses.back().feetHold[11] << endl;
}
}
int cb_check_json_object( CBFILE **cfg, unsigned char **ucsvalue, int ucsvaluelen, int *from ){
int openpairs=0, err=CBSUCCESS;
unsigned long int chr=0x20;
if( from==NULL || cfg==NULL || *cfg==NULL || ucsvalue==NULL || *ucsvalue==NULL ){ cb_clog( CBLOGDEBUG, CBERRALLOC, "\ncb_check_json_null: ucsvalue was null."); return CBERRALLOC; }
char state=0;
//cb_clog( CBLOGDEBUG, CBNEGATION, "\ncb_check_json_object:" );
while( *from<ucsvaluelen && *from<CBNAMEBUFLEN && err<CBNEGATION ){
err = cb_get_ucs_chr( &chr, &(*ucsvalue), &(*from), ucsvaluelen);
//cb_clog( CBLOGDEBUG, CBNEGATION, "[%c]", (char) chr );
if( ( WSP( chr ) || CR( chr ) || LF( chr ) ) ){
continue;
}else if( (unsigned char) chr == '{' && state!=EXPECTCOMMA && state!=EXPECTCOLON ){
/*
* Read until '{'.
*/
++openpairs;
cb_check_json_object_read_name:
/*
* Reads name within the quotes and update *from. Quotes are bypassed. */
err = cb_check_json_substring( &(*cfg), &(*ucsvalue), ucsvaluelen, &(*from) );
// comment out the next:
if(err>=CBERROR){ cb_clog( CBLOGERR, err, "\ncb_check_json_object: cb_check_json_substring, error %i.", err ); }
state = EXPECTCOLON;
continue;
}else if( state==EXPECTCOLON && (unsigned char) chr == ':' ){
/*
* Read colon ':'.
*/
/*
* Read Value. */
err = cb_check_json_value_subfunction( &(*cfg), &(*ucsvalue), ucsvaluelen, &(*from) );
// comment out the next:
if(err>=CBERROR){ cb_clog( CBLOGERR, err, "\ncb_check_json_object: cb_check_json_value_subfunction, error %i.", err ); }
state = EXPECTCOMMA;
continue;
}else if( ( state==EXPECTCOMMA || state==EXPECTCOMMAORNAME ) && ( (unsigned char) chr == ',' || (unsigned char) chr == '}' ) ){
/*
* Read comma ',' or '}' .
*/
if( (unsigned char) chr == '}' ){
state = EXPECTCOMMAORNAME;
--openpairs;
}else{
/*
* Read next name. */
goto cb_check_json_object_read_name;
}
continue;
}else if( state==EXPECTCOMMAORNAME && (unsigned char) chr == '"' ){
*from-=4;
goto cb_check_json_object_read_name;
}else if( (unsigned char) chr == '}' ){
--openpairs;
if( state!=EXPECTCOMMA )
err = CBNOTJSONOBJECT;
state = 0;
continue;
}else
err = CBNOTJSONOBJECT;
}
if(openpairs==0)
return err;
return CBNOTJSONOBJECT;
}
/*
=================
R_LoadMD3
=================
*/
static qboolean R_LoadMD3 (model_t *mod, int lod, void *buffer, const char *mod_name, qboolean &bAlreadyCached ) {
int i, j;
md3Header_t *pinmodel;
md3Surface_t *surf;
md3Shader_t *shader;
int version;
int size;
#ifdef Q3_BIG_ENDIAN
md3Frame_t *frame;
md3Triangle_t *tri;
md3St_t *st;
md3XyzNormal_t *xyz;
md3Tag_t *tag;
#endif
pinmodel= (md3Header_t *)buffer;
//
// read some fields from the binary, but only LittleLong() them when we know this wasn't an already-cached model...
//
version = pinmodel->version;
size = pinmodel->ofsEnd;
if (!bAlreadyCached)
{
version = LittleLong(version);
size = LittleLong(size);
}
if (version != MD3_VERSION) {
ri.Printf( PRINT_WARNING, "R_LoadMD3: %s has wrong version (%i should be %i)\n",
mod_name, version, MD3_VERSION);
return qfalse;
}
mod->type = MOD_MESH;
mod->dataSize += size;
qboolean bAlreadyFound = qfalse;
mod->md3[lod] = (md3Header_t *) RE_RegisterModels_Malloc(size, buffer, mod_name, &bAlreadyFound, TAG_MODEL_MD3);
assert(bAlreadyCached == bAlreadyFound);
if (!bAlreadyFound)
{
// horrible new hackery, if !bAlreadyFound then we've just done a tag-morph, so we need to set the
// bool reference passed into this function to true, to tell the caller NOT to do an FS_Freefile since
// we've hijacked that memory block...
//
// Aaaargh. Kill me now...
//
bAlreadyCached = qtrue;
assert( mod->md3[lod] == buffer );
// memcpy( mod->md3[lod], buffer, size ); // and don't do this now, since it's the same thing
LL(mod->md3[lod]->ident);
LL(mod->md3[lod]->version);
LL(mod->md3[lod]->numFrames);
LL(mod->md3[lod]->numTags);
LL(mod->md3[lod]->numSurfaces);
LL(mod->md3[lod]->ofsFrames);
LL(mod->md3[lod]->ofsTags);
LL(mod->md3[lod]->ofsSurfaces);
LL(mod->md3[lod]->ofsEnd);
}
if ( mod->md3[lod]->numFrames < 1 ) {
ri.Printf( PRINT_WARNING, "R_LoadMD3: %s has no frames\n", mod_name );
return qfalse;
}
if (bAlreadyFound)
{
return qtrue; // All done. Stop, go no further, do not pass Go...
}
#ifdef Q3_BIG_ENDIAN
// swap all the frames
frame = (md3Frame_t *) ( (byte *)mod->md3[lod] + mod->md3[lod]->ofsFrames );
for ( i = 0 ; i < mod->md3[lod]->numFrames ; i++, frame++) {
LF(frame->radius);
for ( j = 0 ; j < 3 ; j++ ) {
LF(frame->bounds[0][j]);
LF(frame->bounds[1][j]);
LF(frame->localOrigin[j]);
}
}
// swap all the tags
tag = (md3Tag_t *) ( (byte *)mod->md3[lod] + mod->md3[lod]->ofsTags );
for ( i = 0 ; i < mod->md3[lod]->numTags * mod->md3[lod]->numFrames ; i++, tag++) {
for ( j = 0 ; j < 3 ; j++ ) {
LF(tag->origin[j]);
LF(tag->axis[0][j]);
LF(tag->axis[1][j]);
LF(tag->axis[2][j]);
}
}
#endif
// swap all the surfaces
surf = (md3Surface_t *) ( (byte *)mod->md3[lod] + mod->md3[lod]->ofsSurfaces );
for ( i = 0 ; i < mod->md3[lod]->numSurfaces ; i++) {
LL(surf->flags);
LL(surf->numFrames);
LL(surf->numShaders);
LL(surf->numTriangles);
LL(surf->ofsTriangles);
LL(surf->numVerts);
LL(surf->ofsShaders);
LL(surf->ofsSt);
LL(surf->ofsXyzNormals);
LL(surf->ofsEnd);
if ( surf->numVerts > SHADER_MAX_VERTEXES ) {
Com_Error (ERR_DROP, "R_LoadMD3: %s has more than %i verts on a surface (%i)",
mod_name, SHADER_MAX_VERTEXES, surf->numVerts );
}
if ( surf->numTriangles*3 > SHADER_MAX_INDEXES ) {
Com_Error (ERR_DROP, "R_LoadMD3: %s has more than %i triangles on a surface (%i)",
mod_name, SHADER_MAX_INDEXES / 3, surf->numTriangles );
}
// change to surface identifier
surf->ident = SF_MD3;
// lowercase the surface name so skin compares are faster
Q_strlwr( surf->name );
// strip off a trailing _1 or _2
// this is a crutch for q3data being a mess
j = strlen( surf->name );
if ( j > 2 && surf->name[j-2] == '_' ) {
surf->name[j-2] = 0;
}
// register the shaders
shader = (md3Shader_t *) ( (byte *)surf + surf->ofsShaders );
for ( j = 0 ; j < surf->numShaders ; j++, shader++ ) {
shader_t *sh;
sh = R_FindShader( shader->name, lightmapsNone, stylesDefault, qtrue );
if ( sh->defaultShader ) {
shader->shaderIndex = 0;
} else {
shader->shaderIndex = sh->index;
}
RE_RegisterModels_StoreShaderRequest(mod_name, &shader->name[0], &shader->shaderIndex);
}
#ifdef Q3_BIG_ENDIAN
// swap all the triangles
tri = (md3Triangle_t *) ( (byte *)surf + surf->ofsTriangles );
for ( j = 0 ; j < surf->numTriangles ; j++, tri++ ) {
LL(tri->indexes[0]);
LL(tri->indexes[1]);
LL(tri->indexes[2]);
}
// swap all the ST
st = (md3St_t *) ( (byte *)surf + surf->ofsSt );
for ( j = 0 ; j < surf->numVerts ; j++, st++ ) {
LF(st->st[0]);
LF(st->st[1]);
}
// swap all the XyzNormals
xyz = (md3XyzNormal_t *) ( (byte *)surf + surf->ofsXyzNormals );
for ( j = 0 ; j < surf->numVerts * surf->numFrames ; j++, xyz++ )
{
LS(xyz->xyz[0]);
LS(xyz->xyz[1]);
LS(xyz->xyz[2]);
LS(xyz->normal);
}
#endif
// find the next surface
surf = (md3Surface_t *)( (byte *)surf + surf->ofsEnd );
}
return qtrue;
}
void HHbbbbBtagEffBase_76(int wMs,int wM, string st,string st2,string option=""){
//0=signal ,1=QCD ,2=data-----------------------------------------------------------
int nameRoot=1;
if(st2.find("QCD")!= std::string::npos)nameRoot=0;
if(st2.find("data")!= std::string::npos)nameRoot=2;
//option-----------------------------------------------------------
int JESOption=0;
if(option.find("JESUp")!= std::string::npos)JESOption=1;
if(option.find("JESDown")!= std::string::npos)JESOption=2;
if(option.find("BtagUp")!= std::string::npos)JESOption=3;
if(option.find("BtagDown")!= std::string::npos)JESOption=4;
if(option.find("tau21Up")!= std::string::npos)JESOption=5;
if(option.find("tau21Down")!= std::string::npos)JESOption=6;
cout<<"JESOption = "<<JESOption<<endl;
//tuple tree and cutflow variables------------------------------------------------------------------------------------
TFile *f;
TTree *tree;
int nPass[20]={0},total=0,dataPassingcsc=0;
double nPassB[6]={0};
//using for Btag Eff -----------------------------------------------------------------------------
string btagsystematicsType="central";
if(JESOption==3)btagsystematicsType="up";
else if(JESOption==4)btagsystematicsType="down";
BTagCalibration calib("CSVv2L", "../CSVv2_76.csv");
BTagCalibrationReader LF(&calib, // calibration instance
BTagEntry::OP_LOOSE, // operating point
"incl", // measurement type
btagsystematicsType); // systematics type
BTagCalibrationReader HF(&calib, BTagEntry::OP_LOOSE, "mujets",btagsystematicsType);
TFile *f1;
if(nameRoot==2)f1=TFile::Open("../btagEffSource/data.root");
else if (nameRoot!=2 && (JESOption==0||JESOption==3||JESOption==4||JESOption==5||JESOption==6))f1=TFile::Open(Form("../btagEffSource/%s.root",st2.data()));
else if (nameRoot!=2 && JESOption==1)f1=TFile::Open(Form("../btagEffSource/%s_JESUp.root",st2.data()));
else if (nameRoot!=2 && JESOption==2)f1=TFile::Open(Form("../btagEffSource/%s_JESDown.root",st2.data()));
TH2D* th1[6];
string btaggingEff[6]={"effD_b","effN_b","effD_c","effN_c","effD_l","effN_l"};
for(int i=0;i<6;i++){
th1[i]=(TH2D*)f1->FindObjectAny(Form("%s",btaggingEff[i].data()));
if(i==1||i==3||i==5)th1[i]->Divide(th1[i-1]);
}
//check for zero btagging SF----------------------------------------------------------------------------------------
TH2D* th3[6];
th3[0]=new TH2D("zeroSF_b","zeroSF_b",200,0,2000,60,-3,3);
th3[1]=new TH2D("zeroSF_c","zeroSF_c",200,0,2000,60,-3,3);
th3[2]=new TH2D("zeroSF_l","zeroSF_l",200,0,2000,60,-3,3);
th3[3]=new TH2D("SF_vs_Pt_b","SF_vs_Pt_b",120,0,1200,40,0.8,1.2);
th3[4]=new TH2D("SF_vs_Pt_c","SF_vs_Pt_c",120,0,1200,40,0.8,1.2);
th3[5]=new TH2D("SF_vs_Pt_l","SF_vs_Pt_l",120,0,1200,40,0.8,1.2);
for(int i=0;i<6;i++)th3[i]->Sumw2();
//check for high btagging SF----------------------------------------------------------------------------------------
TH1D* th4[14];
string SF_jet_sub[8]={"SF_jet0_sub0_pass","SF_jet0_sub1_pass","SF_jet1_sub0_pass","SF_jet1_sub1_pass","SF_jet0_sub0_fail","SF_jet0_sub1_fail","SF_jet1_sub0_fail","SF_jet1_sub1_fail"};
for(int i=0;i<8;i++)th4[i]=new TH1D(Form("%s",SF_jet_sub[i].data()),Form("%s",SF_jet_sub[i].data()),40,0.8,1.2);
string weightName[6]={"weight","weight_0b","weight_1b","weight_2b","weight_2b_ll","weight_2b_onel"};
for(int i=0;i<6;i++)th4[i+8]=new TH1D(Form("%s",weightName[i].data()),Form("%s",weightName[i].data()),40,0,2);
for(int i=0;i<14;i++)th4[i]->Sumw2();
//saving variables----------------------------------------------------------------------------------------
TH1D * th5[200],* th6[200];
for(int i=0;i<2;i++){
for(int j=0;j<2;j++){
for(int k=0;k<5;k++){
th5[(i*2+j)*5+k]=new TH1D(Form("Pt_j%d_sj%d_%db",i,j,k),Form("Pt_j%d_sj%d_%db",i,j,k),200,0,2000);
th5[(i*2+j)*5+k+20]=new TH1D(Form("Eta_j%d_sj%d_%db",i,j,k),Form("Eta_j%d_sj%d_%db",i,j,k),60,-3,3);
th5[(i*2+j)*5+k+85]=new TH1D(Form("subCSV_j%d_sj%d_%db",i,j,k),Form("subCSV_j%d_sj%d_%db",i,j,k),20,0,1);
}
}
for(int k=0;k<5;k++){
th5[i*5+k+40]=new TH1D(Form("deltaR_j%d_%db",i,k),Form("deltaR_j%d_%db",i,k),20,0,1);
th5[i*5+k+50]=new TH1D(Form("Pt_j%d_%db",i,k),Form("Pt_j%d_%db",i,k),200,0,2000);
th5[i*5+k+60]=new TH1D(Form("Eta_j%d_%db",i,k),Form("Eta_j%d_%db",i,k),60,-3,3);
th5[i*5+k+70]=new TH1D(Form("prMassL2L3_j%d_%db",i,k),Form("prMassL2L3_j%d_%db",i,k),15,90,150);
th5[i*5+k+105]=new TH1D(Form("tau21_j%d_%db",i,k),Form("tau21_j%d_%db",i,k),25,0,1);
th5[i*5+k+120]=new TH1D(Form("PuppiSDmassL2L3_j%d_%db",i,k),Form("PuppiSDmassL2L3_j%d_%db",i,k),15,90,150);
th5[i*5+k+130]=new TH1D(Form("puppiTau21_j%d_%db",i,k),Form("puppiTau21_j%d_%db",i,k),25,0,1);
th5[i*5+k+140]=new TH1D(Form("prMass_j%d_%db",i,k),Form("prMass_j%d_%db",i,k),15,90,150);
th5[i*5+k+150]=new TH1D(Form("PuppiSDmass_j%d_%db",i,k),Form("PuppiSDmass_j%d_%db",i,k),15,90,150);
}
}
for(int k=0;k<5;k++){
th5[k+80]=new TH1D(Form("totalMass_%db",k),Form("totalMass_%db",k),200,1000,5000);
th5[k+115]=new TH1D(Form("deltaEta_%db",k),Form("deltaEta_%db",k),40,0,2);
th5[k+160]=new TH1D(Form("logPt_%db",k),Form("logPt_%db",k),70,0,7);
}
for(int i=0;i<165;i++){
th6[i]=(TH1D* )th5[i]->Clone(Form("%ss",th5[i]->GetTitle()));
th5[i]->Sumw2();
th6[i]->Sumw2();
}
//pileup uncertainty----------------------------------------------------------------------------------------
TH1D* th7[14];
th7[0]=new TH1D("totalMass","totalMass",200,1000,5000);
th7[1]=new TH1D("totalMass_pileup_up","totalMass_pileup_up",200,1000,5000);
th7[2]=new TH1D("totalMass_pileup_down","totalMass_pileup_down",200,1000,5000);
th7[3]=new TH1D("pileupEff","pileupEff",15,0.5,15.5);
double totalPileup[3]={0},passPileup[3]={0};
standalone_LumiReWeighting LumiWeights_central(0),LumiWeights_up(1),LumiWeights_down(-1);
//PDF uncertainty
th7[4]=new TH1D("PDFEff","PDFEff",101,0.5,101.5);
double passPDF[101]={0},totalPDF[101]={0};
//QCD uncertainty
for(int i=5;i<14;i++)th7[i]=new TH1D(Form("uns_QCD_%d",i-5),Form("uns_QCD_%d",i-5),200,1000,5000);
//NCUtuple loop----------------------------------------------------------------------------------------
for (int w=wMs;w<wM;w++){
if(w%20==0)cout<<w<<endl;
//Get ntuple----------------------------------------------------------------------------------------
if (nameRoot!=1)f = TFile::Open(Form("%s%d.root",st.data(),w));
else f = TFile::Open(st.data());
if (!f || !f->IsOpen())continue;
TDirectory * dir;
if (nameRoot!=1)dir = (TDirectory*)f->Get(Form("%s%d.root:/tree",st.data(),w));
else dir = (TDirectory*)f->Get(Form("%s:/tree",st.data()));
dir->GetObject("treeMaker",tree);
TreeReader data(tree);
total+=data.GetEntriesFast();
for(Long64_t jEntry=0; jEntry<data.GetEntriesFast() ;jEntry++){//event loop----------------------------------------------------------------------------------------
data.GetEntry(jEntry);
Int_t nVtx = data.GetInt("nVtx");
Float_t ntrue= data.GetFloat("pu_nTrueInt");
//Float_t* pdfscaleSysWeights= data.GetPtrFloat("pdfscaleSysWeights");
double PU_weight[3]={1,1,1};
if(nameRoot!=2){
if(ntrue<51){
PU_weight[0] = LumiWeights_central.weight(ntrue);
PU_weight[1]= LumiWeights_up.weight(ntrue);
PU_weight[2] = LumiWeights_down.weight(ntrue);
}
else {
PU_weight[0] = LumiWeights_central.weight(50);
PU_weight[1] = LumiWeights_up.weight(50);
PU_weight[2]= LumiWeights_down.weight(50);
}
}
for(int i=0;i<3;i++)totalPileup[i]+=PU_weight[i];
for(int i=0;i<101;i++)totalPDF[i]+=PU_weight[0];
//0. has a good vertex
if(nVtx<1)continue;nPass[0]++;
//1.trigger
std::string* trigName = data.GetPtrString("hlt_trigName");
vector<bool> &trigResult = *((vector<bool>*) data.GetPtr("hlt_trigResult"));
bool passTrigger=false;
for(int it=0; it< data.GetPtrStringSize(); it++){
std::string thisTrig= trigName[it];
bool results = trigResult[it];
if( ((thisTrig.find("HLT_PFHT800")!= std::string::npos||
thisTrig.find("HLT_PFHT650")!= std::string::npos||
thisTrig.find("HLT_PFHT650_WideJetMJJ900DEtaJJ1p5_v")!= std::string::npos||
thisTrig.find("HLT_PFHT650_WideJetMJJ950DEtaJJ1p5_v")!= std::string::npos||
thisTrig.find("HLT_AK8PFJet360_TrimMass30_v")!= std::string::npos||
thisTrig.find("HLT_AK8PFHT700_TrimR0p1PT0p03Mass50_v")!= std::string::npos
) && results==1)){
passTrigger=true;
break;
}
}
if(!passTrigger && nameRoot==2)continue;nPass[1]++;
int nFATJet = data.GetInt("FATnJet");
TClonesArray* fatjetP4 = (TClonesArray*) data.GetPtrTObject("FATjetP4");
float* FATjetCorrUncUp = data.GetPtrFloat("FATjetCorrUncUp");
float* FATjetCorrUncDown = data.GetPtrFloat("FATjetCorrUncDown");
vector<bool> &FATjetPassIDTight = *((vector<bool>*) data.GetPtr("FATjetPassIDTight"));
//2.nJets
if(nFATJet<2)continue;nPass[2]++;
TLorentzVector* thisJet ,* thatJet;
if (JESOption==1){
TLorentzVector test0= (*((TLorentzVector*)fatjetP4->At(0)))*(1+FATjetCorrUncUp[0] );
TLorentzVector test1= (*((TLorentzVector*)fatjetP4->At(1)))*(1+FATjetCorrUncUp[1] );
thisJet= &test0;
thatJet= &test1;
}
else if (JESOption==2){
TLorentzVector test0= (*((TLorentzVector*)fatjetP4->At(0)))*(1-FATjetCorrUncDown[0] );
TLorentzVector test1= (*((TLorentzVector*)fatjetP4->At(1)))*(1-FATjetCorrUncDown[1] );
thisJet= &test0;
thatJet= &test1;
}
else{
thisJet= (TLorentzVector*)fatjetP4->At(0);
thatJet = (TLorentzVector*)fatjetP4->At(1);
}
//3. Pt
if(thisJet->Pt()<300||thatJet->Pt()<300)continue;
nPass[3]++;
//4tightId-----------------------------------------
if(FATjetPassIDTight[0]==0||FATjetPassIDTight[1]==0)continue;
Float_t* FATjetCEmEF = data.GetPtrFloat("FATjetCEmEF");
Float_t* FATjetMuEF = data.GetPtrFloat("FATjetMuEF");
if(FATjetMuEF[0]>0.8||FATjetMuEF[1]>0.8)continue;
if(FATjetCEmEF[0]>0.9||FATjetCEmEF[1]>0.9)continue;
nPass[4]++;
//5. Eta-----------------------------------------
if(fabs(thisJet->Eta())>2.4||fabs(thatJet->Eta())>2.4)continue;
nPass[5]++;
//6. DEta-----------------------------------------
float dEta = fabs(thisJet->Eta()-thatJet->Eta());
if(dEta>1.3)continue;
nPass[6]++;
//7. Mjj-----------------------------------------
float mjj = (*thisJet+*thatJet).M();
if(mjj<1000)continue;
nPass[7]++;
//8. fatjetPRmassL2L3Corr-----------------------------------------
Float_t* fatjetPRmassL2L3Corr = data.GetPtrFloat("FATjetPRmassL2L3Corr");
Float_t* FATjetPuppiSDmassL2L3Corr = data.GetPtrFloat("FATjetPuppiSDmassL2L3Corr");
Float_t* FATjetPRmass = data.GetPtrFloat("FATjetPRmass");
Float_t* FATjetPuppiSDmass = data.GetPtrFloat("FATjetPuppiSDmass");
if(fatjetPRmassL2L3Corr[0]<105||fatjetPRmassL2L3Corr[0]>135)continue;
if(fatjetPRmassL2L3Corr[1]<105||fatjetPRmassL2L3Corr[1]>135)continue;
nPass[8]++;
//9.-----------------------------------------
Float_t* fatjetTau1 = data.GetPtrFloat("FATjetTau1");
Float_t* fatjetTau2 = data.GetPtrFloat("FATjetTau2");
Float_t* FATjetPuppiTau1 = data.GetPtrFloat("FATjetPuppiTau1");
Float_t* FATjetPuppiTau2 = data.GetPtrFloat("FATjetPuppiTau2");
double tau21[2];
tau21[0]=(fatjetTau2[0]/fatjetTau1[0]),tau21[1]=(fatjetTau2[1]/fatjetTau1[1]);
double puppiTau21[2];
puppiTau21[0]=(FATjetPuppiTau2[0]/FATjetPuppiTau1[0]),puppiTau21[1]=(FATjetPuppiTau2[1]/FATjetPuppiTau1[1]);
if(tau21[0]>0.75||tau21[1]>0.75)continue;
if(tau21[0]>0.6 &&tau21[1]>0.6) continue;
nPass[9]++;
bool isHPHP=0;
double tau21_SF=1.031*0.881;
if(JESOption==5)tau21_SF=(1.031+0.126)*(0.881+0.49);
if(JESOption==6)tau21_SF=(1.031-0.126)*(0.881-0.49);
if(tau21[0]<0.6 && tau21[1]<0.6 ){
isHPHP=1;
tau21_SF=1.031*1.031;
if(JESOption==5)tau21_SF=(1.031+0.126)*(1.031+0.126);
if(JESOption==6)tau21_SF=(1.031-0.126)*(1.031-0.126);
}
//8.btag
vector<float> *subjetSDCSV = data.GetPtrVectorFloat("FATsubjetSDCSV");
vector<float> *subjetSDPx = data.GetPtrVectorFloat("FATsubjetSDPx", nFATJet);
vector<float> *subjetSDPy = data.GetPtrVectorFloat("FATsubjetSDPy", nFATJet);
vector<float> *subjetSDPz = data.GetPtrVectorFloat("FATsubjetSDPz", nFATJet);
vector<float> *subjetSDE = data.GetPtrVectorFloat("FATsubjetSDE", nFATJet);
vector<Int_t> *FATsubjetSDHadronFlavor = data.GetPtrVectorInt("FATsubjetSDHadronFlavor");
int nbtag=0,nbtag2=0;
float MaxBJetPt = 670., MaxLJetPt = 1000.;
double sf[2][2],eta[2],pt[2],dr[2],subjetPt[2][2],subjetEta[2][2],eff[2][2],btaggingscaleFactor=1;
TLorentzVector* subjetP4[2][2];
for(int i=0;i<2;i++)for(int j=0;j<2;j++)sf[i][j]=1;
for(int i=0;i<2;i++){
for(int j=0;j<2;j++){
subjetP4[i][j]=new TLorentzVector(0,0,0,0);
subjetP4[i][j]->SetPxPyPzE(subjetSDPx[i][j],subjetSDPy[i][j],subjetSDPz[i][j],subjetSDE[i][j]);
subjetPt[i][j]=subjetP4[i][j]->Pt();
subjetEta[i][j]=subjetP4[i][j]->Eta();
//get btagging eff------------------------------------------------------------
if(FATsubjetSDHadronFlavor[i][j]==5)eff[i][j]=th1[1]->GetBinContent(ceil(subjetPt[i][j]/10),ceil(subjetEta[i][j]/0.1)+30);
else if(FATsubjetSDHadronFlavor[i][j]==4)eff[i][j]=th1[3]->GetBinContent(ceil(subjetPt[i][j]/10),ceil(subjetEta[i][j]/0.1)+30);
else eff[i][j]=th1[5]->GetBinContent(ceil(subjetPt[i][j]/10),ceil(subjetEta[i][j]/0.1)+30);
//check maxPt-------------------------------------------------------------
if(FATsubjetSDHadronFlavor[i][j]!=0 && subjetPt[i][j]>MaxBJetPt )subjetPt[i][j]=MaxBJetPt-0.1;
if(FATsubjetSDHadronFlavor[i][j]==0 && subjetPt[i][j]>MaxLJetPt )subjetPt[i][j]=MaxLJetPt-0.1;
//Get SF from csv------------------------------------------------------------
if(FATsubjetSDHadronFlavor[i][j]==5){
sf[i][j]=HF.eval(BTagEntry::FLAV_B,subjetEta[i][j],subjetPt[i][j]);
th3[3]->Fill(subjetPt[i][j],sf[i][j]);
}
else if(FATsubjetSDHadronFlavor[i][j]==4){
sf[i][j]=HF.eval(BTagEntry::FLAV_C,subjetEta[i][j],subjetPt[i][j]);
th3[4]->Fill(subjetPt[i][j],sf[i][j]);
}
else {
sf[i][j]=LF.eval(BTagEntry::FLAV_UDSG,subjetEta[i][j],subjetPt[i][j]);
th3[5]->Fill(subjetPt[i][j],sf[i][j]);
}
//check zero ------------------------------------------------------------
if(sf[i][j]==0 && FATsubjetSDHadronFlavor[i][j]==5 ) th3[0]->Fill(subjetPt[i][j],subjetEta[i][j]);
if(sf[i][j]==0 && FATsubjetSDHadronFlavor[i][j]==4 ) th3[1]->Fill(subjetPt[i][j],subjetEta[i][j]);
if(sf[i][j]==0 && FATsubjetSDHadronFlavor[i][j]!=5 && FATsubjetSDHadronFlavor[i][j]!=4 ) th3[2]->Fill(subjetPt[i][j],subjetEta[i][j]);
//conut nbtag---------------------------------------------------------
if(subjetSDCSV[i][j]>0.46)nbtag++;
//get tot. btagging SF
if(subjetSDCSV[i][j]>=0.46)btaggingscaleFactor*=sf[i][j];
else btaggingscaleFactor*=((1-eff[i][j]*sf[i][j])/(1-eff[i][j]));
//##############check light jet SF##########
if(subjetSDCSV[i][j]>0.46 &&FATsubjetSDHadronFlavor[i][j]==0)nbtag2++;
if(subjetSDCSV[i][j]>0.46)th4[i*2+j]->Fill(sf[i][j]);
else th4[i*2+j+4]->Fill(sf[i][j]);
subjetPt[i][j]=subjetP4[i][j]->Pt();
}
dr[i]=subjetP4[i][0]->DeltaR(*subjetP4[i][1]);
}
th4[8]->Fill(btaggingscaleFactor);
if(nbtag2==2 && nbtag==2)th4[12]->Fill(btaggingscaleFactor);
if(nbtag2==1 && nbtag==2)th4[13]->Fill(btaggingscaleFactor);
//uncertainty -------------------------------------
double scaleFactor=btaggingscaleFactor*PU_weight[0]*tau21_SF;
for(int i=0;i<3;i++){
passPileup[i]+=btaggingscaleFactor*PU_weight[i]*tau21_SF;
th7[i]->Fill(mjj,btaggingscaleFactor*PU_weight[i]*tau21_SF);
}
for(int i=0;i<101;i++)passPDF[i]+=btaggingscaleFactor*PU_weight[0]*tau21_SF;
for(int i=5;i<14;i++)th7[i]->Fill(mjj,btaggingscaleFactor*PU_weight[0]*tau21_SF);
//--------------------------------------
pt[0]=thisJet->Pt();
pt[1]=thatJet->Pt();
eta[0]=thisJet->Eta();
eta[1]=thatJet->Eta();
for(int i=0;i<2;i++){
for(int j=0;j<2;j++){
for(int k=0;k<5;k++){
if(nbtag!=k)continue;
th5[(i*2+j)*5+k]->Fill(subjetPt[i][j]);
th5[(i*2+j)*5+k+20]->Fill(subjetEta[i][j]);
th5[(i*2+j)*5+k+85]->Fill(subjetSDCSV[i][j]);
th6[(i*2+j)*5+k]->Fill(subjetPt[i][j],scaleFactor);
th6[(i*2+j)*5+k+20]->Fill(subjetEta[i][j],scaleFactor);
th6[(i*2+j)*5+k]->Fill(subjetSDCSV[i][j],scaleFactor);
}
}
for(int k=0;k<5;k++){
if(nbtag!=k)continue;
th5[i*5+k+40]->Fill(dr[i]);
th5[i*5+k+50]->Fill(pt[i]);
th5[i*5+k+60]->Fill(eta[i]);
th5[i*5+k+70]->Fill(fatjetPRmassL2L3Corr[i]);
th5[i*5+k+105]->Fill(tau21[i]);
th6[i*5+k+40]->Fill(dr[i],scaleFactor);
th6[i*5+k+50]->Fill(pt[i],scaleFactor);
th6[i*5+k+60]->Fill(eta[i],scaleFactor);
th6[i*5+k+70]->Fill(fatjetPRmassL2L3Corr[i],scaleFactor);
th6[i*5+k+105]->Fill(tau21[i],scaleFactor);
th5[i*5+k+120]->Fill(FATjetPuppiSDmassL2L3Corr[i]);
th6[i*5+k+120]->Fill(FATjetPuppiSDmassL2L3Corr[i],scaleFactor);
th5[i*5+k+130]->Fill(puppiTau21[i]);
th6[i*5+k+130]->Fill(puppiTau21[i],scaleFactor);
th5[i*5+k+140]->Fill(FATjetPRmass[i]);
th6[i*5+k+140]->Fill(FATjetPRmass[i],scaleFactor);
th5[i*5+k+150]->Fill(FATjetPuppiSDmass[i]);
th6[i*5+k+150]->Fill(FATjetPuppiSDmass[i],scaleFactor);
}
}
for(int k=0;k<5;k++){
if(nbtag!=k)continue;
th5[k+80]->Fill(mjj);
th6[k+80]->Fill(mjj,scaleFactor);
th5[k+115]->Fill(dEta);
th6[k+115]->Fill(dEta,scaleFactor);
th5[k+160]->Fill(log10(pt[0]));
th6[k+160]->Fill(log10(pt[0]),scaleFactor);
th5[k+160]->Fill(log10(pt[1]));
th6[k+160]->Fill(log10(pt[1]),scaleFactor);
nPassB[k]+=scaleFactor;
nPass[k+10]++;
if(k<3)th4[9+k]->Fill(btaggingscaleFactor);
}
if(nbtag==3){
if(isHPHP)nPassB[5]+=scaleFactor;
if(isHPHP)nPass[15]++;
}
}//end event loop----------------------------------------------------------------------------------------
} //end ntuple loop----------------------------------------------------------------------------------------
cout<<"entries="<<total<<endl;
for(int i=0;i<6;i++)cout<<"nPassB["<<i<<"]="<<nPassB[i]<<endl;
for(int i=0;i<16;i++)cout<<"nPass["<<i<<"]="<<nPass[i]<<endl;
for(int i=0;i<3;i++)th7[3]->SetBinContent(i+1,passPileup[i]/totalPileup[i]);
for(int i=0;i<101;i++)th7[4]->SetBinContent(i+1,passPDF[i]/totalPDF[i]);
TH1D * th2o=new TH1D("Nbtagjet","Nbtagjet",6,-0.5,5.5);
for (int i=0;i<6;i++){
if(nameRoot==2 && i>2)continue;
th2o->SetBinContent(i+1,nPass[i+10]);
}
TH1D * th2ob=new TH1D("NbtagjetB","NbtagjetB",6,-0.5,5.5);
for (int i=0;i<6;i++){
if(nameRoot==2 && i>2)continue;
th2ob->SetBinContent(i+1,nPassB[i]);
}
TH1D * cutflow=new TH1D("cutflow","cutflow",17,0.5,17.5);
cutflow->SetBinContent(1,total);
if(nameRoot==2)for(int ii=1;ii<14;ii++)cutflow->SetBinContent(ii+1,nPass[ii-1]);
else for(int ii=1;ii<17;ii++)cutflow->SetBinContent(ii+1,nPass[ii-1]);
TFile* outFile ;
if(JESOption==0)outFile= new TFile(Form("../sf/%s.root",st2.data()),"recreate");
else if(JESOption==1)outFile= new TFile(Form("sf/%s_JESUp.root",st2.data()),"recreate");
else if(JESOption==2)outFile= new TFile(Form("sf/%s_JESDown.root",st2.data()),"recreate");
else if(JESOption==3)outFile= new TFile(Form("sf/%s_BtagUp.root",st2.data()),"recreate");
else if(JESOption==4)outFile= new TFile(Form("sf/%s_BtagDown.root",st2.data()),"recreate");
else if(JESOption==5)outFile= new TFile(Form("sf/%s_tau21Up.root",st2.data()),"recreate");
else if(JESOption==6)outFile= new TFile(Form("sf/%s_tau21Down.root",st2.data()),"recreate");
th2o->Write();
th2ob->Write();
cutflow->Write();
for(int i=0;i<6;i++)th3[i]->Write();
for(int i=0;i<14;i++)th4[i]->Write();
for(int i=0;i<165;i++){
th5[i]->Write();
th6[i]->Write();
}
for(int i=0;i<14;i++)th7[i]->Write();
outFile->Close();
}
