Merge branch 'truthTagResults.xAODBTaggingEfficiency-20190111' into 'master'

xAODBTaggingEfficiency: Copy TruthTagResults.h from 21.2.

See merge request atlas/athena!20308

PhysicsAnalysis/JetTagging/JetTagPerformanceCalibration/xAODBTaggingEfficiency/xAODBTaggingEfficiency/TruthTagResults.h

+
+#ifndef ANALYSISTRUTHTAGRESULTS_H
+#define ANALYSISTRUTHTAGRESULTS_H
+
+#include <map>
+#include <vector>
+#include <string>
+
+namespace Analysis {
+
+    class TruthTagResults {
+
+    public:
+
+    //map from systematic name to vector of event weights for different number of tagged jets
+    // for examples: map_trf_weight_ex["Nominal"].at(3) is the event weight for exactly 3 tagged jets
+    // map_trf_weight_in["Nominal"].at(2) is the event weight for 2 or more tagged jets
+    std::map<std::string,std::vector<double> > map_trf_weight_ex;
+    std::map<std::string,std::vector<double> > map_trf_weight_in;
+
+    //chosen permutation (does not depend on the systematic variation)
+    // trf_chosen_perm_ex.at(3) is the chosen permutation for exactly 3 tagged jets
+    std::vector<std::vector<bool> > trf_chosen_perm_ex;
+    std::vector<std::vector<bool> > trf_chosen_perm_in;
+
+    //the tag weight bin, similar configuration to the chosen permutation vectors.
+    std::vector<std::vector<int> > trf_bin_ex;
+    std::vector<std::vector<int> > trf_bin_in;
+
+    //random tag weights generated for the chosen quantiles.
+    std::vector<std::vector<double> > trf_bin_score_ex;
+    std::vector<std::vector<double> > trf_bin_score_in;
+
+    //direct tagging results
+    std::vector<bool> is_tagged;
+    std::map<std::string,double > map_SF;
+
+    std::vector<std::string> syst_names;
+
+    std::vector<float> tagweight_bin_edges;
+    // tagweight_bin_edges gives the cut values for the different MV2 or DL1 scores
+    // Cheatsheet:
+    // 3  cut value 60%
+    // 2  cut value 70%
+    // 1  cut value 77%
+    // 0  cut value 85%
+    ////////////////////
+
+    void clear(){
+
+        map_trf_weight_ex.clear();
+        map_trf_weight_in.clear();
+        trf_chosen_perm_ex.clear();
+        trf_chosen_perm_in.clear();
+        trf_bin_ex.clear();
+        trf_bin_in.clear();
+        tagweight_bin_edges.clear();
+
+    }
+
+    double getEvtDirectTagSF(std::string syst_name="Nominal")
+    {
+        return(map_SF[syst_name]);
+    }
+
+    std::vector<bool> getDirectTaggedJets()
+    {
+        return(is_tagged);
+
+    }
+
+    double getEventWeight(int nbtag,bool Ex,std::string syst_name="Nominal")
+    {
+        if(Ex)
+        {
+            return map_trf_weight_ex[syst_name].at(nbtag);
+        }
+        else
+        {
+            return map_trf_weight_in[syst_name].at(nbtag);
+        }
+    }
+
+    std::vector<bool> getEventPermutation(int nbtag,bool Ex )
+    {
+        if(Ex)
+        {
+            return(trf_chosen_perm_ex.at(nbtag));
+        }
+
+        else
+        {
+            return(trf_chosen_perm_in.at(nbtag));
+        }
+    }
+
+    std::vector<int> getEventQuantiles(int nbtag,bool Ex )
+    {
+        if(Ex)
+        {
+            return(trf_bin_ex.at(nbtag));
+        }
+
+        else
+        {
+            return(trf_bin_in.at(nbtag));
+        }
+    }
+
+
+    std::vector<double> getRandomTaggerScores(int nbtag,bool Ex )
+    {
+        if(Ex)
+        {
+            return(trf_bin_score_ex.at(nbtag));
+        }
+        else
+        {
+            return(trf_bin_score_in.at(nbtag));
+        }
+    }
+
+
+    };
+
+
+}
+
+#endif // ANALYSISTRUTHTAGRESULTS_H
+