From 84987eea65fb5f910ab071b462a0ff7715a31766 Mon Sep 17 00:00:00 2001
From: Oleksii Lubynets <lubynets@lxbk0197.gsi.de>
Date: Tue, 15 Sep 2020 16:45:12 +0200
Subject: [PATCH] add macro test, fix macro

---
 macro/C2F/CMakeLists.txt                              | 10 ++++++++++
 macro/analysis/common/at_kfpf_interface/run_at_kfpf.C |  9 +++++----
 2 files changed, 15 insertions(+), 4 deletions(-)

diff --git a/macro/C2F/CMakeLists.txt b/macro/C2F/CMakeLists.txt
index 8c920c7cff..55e96e0f52 100644
--- a/macro/C2F/CMakeLists.txt
+++ b/macro/C2F/CMakeLists.txt
@@ -5,6 +5,7 @@ GENERATE_CBM_TEST_SCRIPT(${CBMROOT_SOURCE_DIR}/macro/run/run_digi.C ${MACRO_DIR}
 GENERATE_CBM_TEST_SCRIPT(${CBMROOT_SOURCE_DIR}/macro/run/run_reco_event.C ${MACRO_DIR})
 GENERATE_CBM_TEST_SCRIPT(${CBMROOT_SOURCE_DIR}/analysis/PWGC2F/flow/DataTreeCbmInterface/macro/run_treemaker.C ${MACRO_DIR})
 GENERATE_CBM_TEST_SCRIPT(${CBMROOT_SOURCE_DIR}/macro/analysis/common/analysis_tree_converter/run_analysis_tree_maker.C ${MACRO_DIR})
+GENERATE_CBM_TEST_SCRIPT(${CBMROOT_SOURCE_DIR}/macro/analysis/common/at_kfpf_interface/run_at_kfpf.C ${MACRO_DIR})
 GENERATE_CBM_TEST_SCRIPT(${CBMROOT_SOURCE_DIR}/external/DataTreeQA/macro/PopulateQAConfigurations.C ${MACRO_DIR})
 GENERATE_CBM_TEST_SCRIPT(${CBMROOT_SOURCE_DIR}/external/DataTreeQA/macro/RunDataTreeQA.C ${MACRO_DIR})
 
@@ -71,6 +72,15 @@ If( ($ENV{ctest_model} MATCHES Nightly) OR ($ENV{ctest_model} MATCHES Weekly) )
     set_tests_properties(${testname} PROPERTIES FIXTURES_REQUIRED ${fixture_c2f_ana})
     Set(fixture_c2f_analysistree fixture_c2f_analysistree_${testname})
     set_tests_properties(${testname} PROPERTIES FIXTURES_SETUP ${fixture_c2f_analysistree})
+    
+    # --- AnalysisTree to KFPF interface
+    Set(testname at_kfpf_${setup})
+    Add_Test(${testname} ${MACRO_DIR}/run_at_kfpf.sh ${NumEvents} \"data/${setup}_test\")
+    Set_Tests_Properties(${testname} PROPERTIES TIMEOUT "300")
+    Set_Tests_Properties(${testname} PROPERTIES PASS_REGULAR_EXPRESSION "Test Passed;All ok")
+    set_tests_properties(${testname} PROPERTIES FIXTURES_REQUIRED ${fixture_c2f_ana})
+    Set(fixture_c2f_at_kfpf_interface fixture_c2f_at_kfpf_interface_${testname})
+    set_tests_properties(${testname} PROPERTIES FIXTURES_SETUP ${fixture_c2f_at_kfpf_interface})
 
     # --- DataTreeQA input
     Set(testname c2f_datatreeqa_input_${setup})
diff --git a/macro/analysis/common/at_kfpf_interface/run_at_kfpf.C b/macro/analysis/common/at_kfpf_interface/run_at_kfpf.C
index 4ba2165951..e26f049127 100644
--- a/macro/analysis/common/at_kfpf_interface/run_at_kfpf.C
+++ b/macro/analysis/common/at_kfpf_interface/run_at_kfpf.C
@@ -1,9 +1,10 @@
-void run_at_kfpf(const std::string& ATFile,
-                 int nEntries              = -1,
-                 const std::string& ATTree = "aTree") {
+void run_at_kfpf(int nEntries               = -1,
+                 const std::string& dataset = "test",
+                 const std::string& ATTree  = "aTree") {
   ATKFParticleFinder man;
+  const std::string ATFile = dataset + ".analysistree.root";
   man.InitInput(ATFile.c_str(), ATTree.c_str());
-  man.InitOutput("KFPF.pwgc2f.root");
+  man.InitOutput(std::string(dataset + "kfpftree.root"));
   man.SetPIDMode(1);
 
   CutsContainer cuts;
-- 
GitLab