improve example

Author: viktorklochkov

examples/example.yaml

@@ -1,42 +1,45 @@
 1D:
-- name: "px"
-  x_axis: &sim_px
-   title: "p_{x}, GeV/c"
-   field: "SimParticles.px"
-   bins: [100, -3., 3.]
-  cuts: "rec_pT_cut"
-- name: "py"
+# Simplest way to define plot
+- name: "sim_pT"
   x_axis:
-    field: "RecTracks.py"
-    bins_edges: [0, 0.2, 0.5, 1, 2]
+    field: "SimParticles.pT"
+    bins: [100, 0, 3]
+# Better way to define plot (cuts are defined in C++ code)
+- name: "sim_pT_y_1.5_3"
+  x_axis: &sim_pT # anchor could be reused later
+   title: "p_{T}, GeV/c"
+   field: "SimParticles.pT"
+   bins: [100, 0, 3]
+  cuts: "sim_y_cut"
+# Plot by variable name (defined in C++ code)
 - name: "pxpy"
   x_axis:
     field: "pxpy"
     bins: [100, -3, 3]
 2D:
-- name: "px_py"
-  x_axis: *sim_px
-  y_axis: &sim_py
-    title: "p_{y}, GeV/c"
-    field: "SimParticles.py"
-    bins: [100, -2, 2]
+- name: "sim_pT_y"
+  x_axis: &sim_y
+    title: "#it{y}_{Lab}"
+    field: "SimParticles.rapidity"
+    bins: [100, -2, 6]
+  y_axis: *sim_pT
 Profile:
-- name: "px_py_profile"
-  x_axis: *sim_px
-  y_axis: *sim_py
+- name: "pT_y_profile"
+  x_axis: *sim_pT
+  y_axis: *sim_y
 Integral:
 - name: "pT_integral"
-  x_axis: &sim_pT
+  x_axis: &sim_pT_integral
     title: "#Sum{p_{T}, GeV/c}"
     field: "SimParticles.pT"
     bins: [100, 0, 100]
-  cuts: "rec_pT_cut"
+  cuts: "sim_y_cut"
 Integral2D:
 - name: "pT_M_integral"
   x_axis:
     title: "M"
     field: "SimParticles.ones"
     bins: [100, 0, 2000]
-  y_axis: *sim_pT
-  y_cuts: "rec_pT_cut"
+  y_axis: *sim_pT_integral
+  y_cuts: "sim_y_cut"
 

examples/test_yaml.cpp

@@ -14,10 +14,10 @@ void example(const std::string& filelist){
   auto* task = new QA::Task;
   task->SetOutputFileName("cbm_qa.root");
 
-  Cuts* pT_cut =  new Cuts("rec_pT_cut", {RangeCut("RecTracks.pT", 1, 1.5)});
   Variable pxpy("pxpy", {{"RecTracks", "px"}, {"RecTracks", "py"}}, []( std::vector<double>& var ) { return var.at(0)*var.at(1); });
 
-  task->AddCut(pT_cut);
+  task->AddCut(new Cuts("sim_y_cut", {RangeCut("SimParticles.rapidity", 1.5, 3)}));
+  task->AddCut(new Cuts("rec_pT_cut", {RangeCut("RecTracks.pT", 1, 1.5)}));
   task->AddVariable(pxpy);
 
   QA::YamlReader r("/Users/viktor/Soft/AnalysisTreeQA/examples/example.yaml");