Extend drawable particles (#90)

* Add also O2DPG to jdl

* Add other drawable particles

Author: njacazio

examples/scripts/grid/make_grid_files.py

@@ -23,6 +23,7 @@ def main(jdl_file,
          configuration_file,
          config_tag,
          delphes_version,
+         o2dpg_version,
          make_alien_directory):
     """
     Function to create the working environment for GRID job submission
@@ -48,7 +49,8 @@ def write_line(line):
                    f"--ev {number_of_events} -e {config_tag} -l -v --no-vertexing\";\n")
         write_line("Requirements = ( other.Type == \"machine\" );\n")
         write_line("Packages = {")
-        write_line(f"\t\"VO_ALICE@{delphes_version}\"")
+        write_line(f"\t\"VO_ALICE@{delphes_version}\",")
+        write_line(f"\t\"VO_ALICE@{o2dpg_version}\"")
         write_line("};\n")
         write_line("JDLVariables = {")
         write_line("\t\"Packages\",")
@@ -124,6 +126,9 @@ def write_line(line):
     parser.add_argument("--delphes_version", "--delphes",
                         type=str, default="DelphesO2::v20210409-1",
                         help="Version of DelphesO2 to use")
+    parser.add_argument("--o2dpg_version", "--o2dpg",
+                        type=str, default="O2DPG::prod-202104-27-1",
+                        help="Version of O2DPG to use")
     parser.add_argument("--no-vertexing",
                         action="store_true",
                         help="Option turning off the vertexing.")
@@ -141,4 +146,5 @@ def write_line(line):
          config_tag=args.entry,
          njobs=args.njobs,
          delphes_version=args.delphes_version,
+         o2dpg_version=args.o2dpg_version,
          make_alien_directory=args.make_alien_dir)

examples/smearing/draw.py

@@ -39,9 +39,10 @@ def main(reader_name,
 
     style()
 
-    p = {"el": "e", "pi": "#pi", "ka": "K", "pr": "p"}
+    p = {"el": "e", "pi": "#pi", "ka": "K", "pr": "p", "de": "d", "he3": "^{3}He"}
     p_colors = {"el": "#e41a1c", "pi": "#377eb8",
-                "ka": "#4daf4a", "pr": "#984ea3"}
+                "ka": "#4daf4a", "pr": "#984ea3",
+                "de": "#ff7f00", "he3": "#a65628"}
     if particles is not None:
         to_remove = []
         for i in p:
@@ -119,7 +120,7 @@ def draw_alice3_label(x=0.5, y=0.9):
                 return
             g = reader(lut, eta)
             if g.GetN() <= 0:
-                print("Skipping", g.GetName())
+                print("Skipping", g.GetName(), "because empty graph")
                 continue
             if len(g_list) == 0:
                 frame.GetXaxis().SetTitle(g.GetXaxis().GetTitle())