Draft: add extra_persistreco_locations

Adding extra_persisted_locations function, to be used in Moore!1776 (closed)

PyConf/python/PyConf/packing.py

@@ -29,8 +29,15 @@ _reco_locations = {
     "RecSummary": ("/Event/Rec/Summary", "RecSummary")
 }
 
+_extra_reco_locations = {"VeloTracks": ("/Event/Rec/Track/Velo", "Tracks")}
+
 _default_persisted_locations = {k: v[0] for k, v in _reco_locations.items()}
 
+_extra_persisted_locations = {
+    k: v[0]
+    for k, v in _extra_reco_locations.items()
+}
+
 _pp2mcp_locations = {
     "ChargedPP2MC": ("/Event/Relations/ChargedPP2MCP", "PP2MCPRelations"),
     "NeutralPP2MC": ("/Event/Relations/NeutralPP2MCP", "PP2MCPRelations"),
@@ -50,6 +57,14 @@ def reco_locations(stream="/Event/HLT2"):
     }
 
 
+@configurable
+def extra_reco_locations(stream="/Event/HLT2"):
+    return {
+        k: (prefix(v[0], stream), v[1])
+        for k, v in _extra_reco_locations.items()
+    }
+
+
 @configurable
 def pp2mcp_locations(stream="/Event/HLT2"):
     return {
@@ -65,8 +80,19 @@ def default_persisted_locations(locations=_default_persisted_locations,
 
 
 @configurable
-def persisted_location(k, force=True, locations=default_persisted_locations):
-    return force_location(locations()[k]) if force else None
+def extra_persisted_locations(locations=_extra_persisted_locations,
+                              stream="/Event"):
+    return {k: prefix(v, stream) for k, v in locations.items()}
+
+
+@configurable
+def persisted_location(k, force=True, locations=None):
+
+    if not locations:
+        locations = default_persisted_locations()
+        locations.update(extra_persisted_locations())
+
+    return force_location(locations[k]) if force else None
 
 
 def packers_map():