Multilepton_Analysis/rootScripts/rootlogon.C at master · michaelmackenzie/Multilepton_Analysis · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
//------------------------------------------------------------------------------
//  rootlogon.C: a sample ROOT logon macro allowing use of ROOT script
//               compiler in CDF RunII environment. The name of this macro file
//               is defined by the .rootrc file
//------------------------------------------------------------------------------
{
  // the line below tells rootcling where to look for
  // the include files

  gInterpreter->AddIncludePath("./");
  gInterpreter->AddIncludePath("./include");
  gInterpreter->AddIncludePath(Form("%s/include",gSystem->Getenv("ROOTSYS")));
  TString hostname = gSystem->Getenv("HOSTNAME");

  const char* exec_name = gApplication->Argv(0);

  //Define useful TColors
  const int kCMSColor_1  = TColor::GetColor("#3f90da");
  const int kCMSColor_2  = TColor::GetColor("#ffa90e");
  const int kCMSColor_3  = TColor::GetColor("#bd1f01");
  const int kCMSColor_4  = TColor::GetColor("#94a4a2");
  const int kCMSColor_5  = TColor::GetColor("#832db6");
  const int kCMSColor_6  = TColor::GetColor("#a96b59");
  const int kCMSColor_7  = TColor::GetColor("#e76300");
  const int kCMSColor_8  = TColor::GetColor("#b9ac70");
  const int kCMSColor_9  = TColor::GetColor("#717581");
  const int kCMSColor_10 = TColor::GetColor("#92dadd");

  //Colors for low number of processes
  const int kCMSColor_S_1 = TColor::GetColor("#5790fc");
  const int kCMSColor_S_2 = TColor::GetColor("#f89c20");
  const int kCMSColor_S_3 = TColor::GetColor("#e42536");
  const int kCMSColor_S_4 = TColor::GetColor("#964a8b");
  const int kCMSColor_S_5 = TColor::GetColor("#9c9ca1");
  const int kCMSColor_S_6 = TColor::GetColor("#7a21dd");

  //Colors for limits
  const int kCMSColor_LimitGreen  = TColor::GetColor("#607641");
  const int kCMSColor_LimitYellow = TColor::GetColor("#F5BB54");


  if (exec_name) {
    if (strcmp(exec_name,"root.exe") == 0) {
      gInterpreter->AddIncludePath("../rootScripts");
      gInterpreter->AddIncludePath("../interface");
      gInterpreter->AddIncludePath("../scale_factors");
      gInterpreter->AddIncludePath("../weights");
                                        // print overflows/underflows in the stat box
      gStyle->SetOptStat(11111111);
                                        // print fit results in the stat box
      gStyle->SetOptFit(1110);
      TArrow::SetDefaultArrowSize(0.015);

      //-----------------------------------------------------------------------------
      // report the process ID which simplifies debugging
      //-----------------------------------------------------------------------------
      printf("process ID: %i\n",gSystem->GetPid());
      TAuthenticate::SetGlobalUser(gSystem->Getenv("USER"));
      gInterpreter->ProcessLine(".! ps | grep root");

      //-----------------------------------------------------------------------------
      // Load XGBoost and CLFVAnalysis libraries
      //-----------------------------------------------------------------------------

      TString cmssw = gSystem->Getenv("CMSSW_BASE");

      printf("Loading xgboost.so!\n");
      if(cmssw.Contains("CMSSW_10_")) {
        gSystem->Load(Form("/cvmfs/cms.cern.ch/%s/external/py2-xgboost/0.82-ghbfee2/lib/python2.7/site-packages/xgboost/lib/libxgboost.so",
                           gSystem->Getenv("SCRAM_ARCH")));
      } else if(cmssw.Contains("CMSSW_11_")) {
        gSystem->Load(Form("/cvmfs/cms.cern.ch/%s/external/py2-xgboost/0.82-llifpc/lib/python2.7/site-packages/xgboost/lib/libxgboost.so",
                           gSystem->Getenv("SCRAM_ARCH")));
      } else {
        cout << "Can't identify an XGBoost library path for CMSSW release in path " << cmssw.Data() << endl;
      }

      printf("Loading lib/libCLFVAnalysis.so!\n");
      TString path = (hostname.Contains("cms")) ? "src/CLFVAnalysis/lib/" : gSystem->Getenv("PWD");
      if(TString(gSystem->Getenv("PWD")).Contains("CLFVAnalysis_dev")) {
        path = "src/CLFVAnalysis_dev/lib/";
      }
      if(!hostname.Contains("cms")) {
        path += "/../lib/";
      } else {
        gSystem->AddDynamicPath(cmssw.Data());
        gSystem->AddDynamicPath(gSystem->Getenv("PWD"));
      }
      int status = gSystem->Load((path + "libCLFVAnalysis.so").Data());
      if(status < 0)
        status = gSystem->Load("../lib/libCLFVAnalysis.so");
      if(status < 0)
        status = gSystem->Load("../../src/CLFVAnalysis/lib/libCLFVAnalysis.so");
      if(status < 0)
        status = gSystem->Load("src/CLFVAnalysis/lib/libCLFVAnalysis.so");
      if(status < 0)
        status = gSystem->Load("libCLFVAnalysis.so");

    }
  }
}