/afs/hep.man.ac.uk/u/markowen/ATLAS/SFrameProof/2011EPS/topUtils/src/mmTool.h

#ifndef MMTOOL_h
#define MMTOOL_h

#include <iostream>
#include <cstdlib>
#include <cmath>

//Declaration of the function
float get_MM_weight(bool isLoose, bool isTight, float var, bool ele, bool tagged);

//Get the weight for the QCD
float get_MM_weight(bool isLoose, bool isTight, float var, bool ele, bool tagged) {
        int     Nbins     = 0;
        float   wmm       = 0.0;
        double  eff_real  = -1;
        double  eff_fake  = -1;
        float*  eff_real_a = NULL;
        float*  eff_fake_a = NULL; 
        
        float   binning_ele[10] = {0.0, 0.3, 0.6, 0.9, 1.2, 1.37, 1.52, 1.8, 2.1, 2.47};
        float   binning_mu[9]   = {0.0, 0.3, 0.6, 0.9, 1.2, 1.5,  1.8,  2.1, 2.5};
        
        float   eff_fake_ele_tagged[9]   = {0.722, 0.660, 0.696, 0.673, 0.529, 0, 0.424, 0.294, 0.274};
        float   eff_fake_ele_untagged[9] = {0.515, 0.501, 0.487, 0.538, 0.517, 0, 0.405, 0.246, 0.222};
        float   eff_real_ele[9]          = {0.947, 0.945, 0.946, 0.927, 0.913, 0, 0.868, 0.852, 0.770};
        
        // Muons updated Dec,9th 2010
        float   eff_fake_mu_tagged[8]    = {0.291, 0.286, 0.307, 0.319, 0.350, 0.285, 0.279, 0.248};
        float   eff_fake_mu_untagged[8]  = {0.348, 0.348, 0.345, 0.372, 0.398, 0.351, 0.305, 0.285};
        float   eff_real_mu[8]           = {0.988, 0.990, 0.989, 0.990, 0.993, 0.991, 0.985, 0.991};

        float* binning = binning_mu;

        // setting electron or muon variables and efficiencies

        if(ele){
                Nbins= 9;
                binning = binning_ele;
                eff_real_a = eff_real_ele;
                if(!tagged){
                        eff_fake_a = eff_fake_ele_untagged;
                }else{
                        eff_fake_a = eff_fake_ele_tagged;
                }
        }else{
                Nbins = 8;
                binning = binning_mu;
                eff_real_a = eff_real_mu;
                if(!tagged){
                        eff_fake_a = eff_fake_mu_untagged;
                }else{
                        eff_fake_a = eff_fake_mu_tagged;
                }
        }
        var = fabs(var);

        for(int i=0; i<Nbins; ++i) {
                if( (var >= binning[i]) && (var < binning[i+1]) ) {
                        eff_real = eff_real_a[i];
                        eff_fake = eff_fake_a[i];
                        break;
                }
        }


        if( var > binning[Nbins] ) {
                        eff_real = eff_real_a[Nbins-1];
                        eff_fake = eff_fake_a[Nbins-1];
        }
        
        if(eff_real>1 || eff_real<0 ) {
                std::cout << "WARNING +++ WARNING +++ WARNING +++ real efficiency not correct: " << eff_real << " | var: " << var << std::endl;
                exit(1);
        }
        if(eff_fake>1 || eff_fake<0 ) {
                std::cout << "WARNING +++ WARNING +++ WARNING +++ fake efficiency not correct: " << eff_fake << " | var: " << var << std::endl;
                exit(1);
        }
        
        

        if (isLoose) wmm = (eff_fake * eff_real )/(eff_real - eff_fake);
        if (isTight) wmm = (eff_fake * (eff_real - 1.))/(eff_real - eff_fake);
        return wmm;
}

#endif

---