add hooks to recon-util for rebuilding/restreaming

baltzell · baltzell · commit 0cf2a4d696c1 · 2024-09-24T18:13:06.000+02:00
diff --git a/common-tools/clas-reco/src/main/java/org/jlab/clas/reco/EngineProcessor.java b/common-tools/clas-reco/src/main/java/org/jlab/clas/reco/EngineProcessor.java
@@ -55,12 +55,14 @@ private void setBackgroundFiles(String filenames) {
         findEngine(ENGINE_CLASS_BG).init();
     }
 
-    private void setPreloadFiles(String filenames) {
+    private void setPreloadFiles(String filenames, boolean restream, boolean rebuild) {
         if (findEngine(ENGINE_CLASS_PP) == null) {
             LOGGER.info("Adding PostprocEngine for -P option.");
             addEngine("BG",ENGINE_CLASS_PP);
         }
         findEngine(ENGINE_CLASS_PP).engineConfigMap.put("preloadFile", filenames);
+        findEngine(ENGINE_CLASS_PP).engineConfigMap.put("restream", String.valueOf(restream));
+        findEngine(ENGINE_CLASS_PP).engineConfigMap.put("rebuild", String.valueOf(rebuild));
         findEngine(ENGINE_CLASS_PP).init();
     }
 
@@ -354,6 +356,8 @@ public static void main(String[] args){
         parser.addOption("-S",null,"schema directory");
         parser.addOption("-B",null,"background file");
         parser.addOption("-P",null,"preload file for post-processing");
+        parser.addOption("-R","0","rebuild scalers");
+        parser.addOption("-H","0","restream helicity");
 
         parser.parse(args);
 
@@ -414,8 +418,11 @@ else if (config>0){
             proc.setBackgroundFiles(parser.getOption("-B").stringValue());
         
         // command-line filename for post-processing overrides YAML:
-        if (parser.getOption("-P").stringValue() != null)
-            proc.setPreloadFiles(parser.getOption("-P").stringValue());
+        if (parser.getOption("-P").stringValue() != null) {
+            proc.setPreloadFiles(parser.getOption("-P").stringValue(),
+                parser.getOption("-H").intValue()!=0,
+                parser.getOption("-R").intValue()!=0);
+        }
 
         proc.processFile(inputFile,outputFile,nskip,nevents);
     }
