code runs smoothly except for Snapshot

.gitignore

@@ -1,4 +1,7 @@
 __init__.py
+*.png
+*.pdf
+*.root
 *.pyc
 .*.swp
 .#*

framework/RDFprocessor.py

@@ -1,35 +1,14 @@
-import ROOT
-import os
-import time
-
-# Begin code for casting
-
-# this injection can be replaced by properly having this in a header
-# included in the interpreter at framework startup
-ROOT.gInterpreter.Declare('''
-template <typename T>
-class NodeCaster {
-   public:
-   static ROOT::RDF::RNode Cast(T rdf)
-   {
-      return ROOT::RDF::RNode(rdf);
-   }
-};
-''')
-
-def CastToRNode(node):
-   return ROOT.NodeCaster(node.__cppname__).Cast(node)
-
-# end code for casting
+from header import *
 
 class RDFprocessor:
-    def __init__(self, outputFile, inputFiles, histoFile, modules=[], snapshot = False):
+    def __init__(self, outputFile, inputFiles, histoFile, keepVars='', modules=[], snapshot = False):
 
         self.outputFile = outputFile
         self.inputFiles = inputFiles
         self.modules = modules
         self.histoFile = ROOT.TFile(histoFile, "recreate")
         self.snapshot = snapshot
+        self.keepVars = keepVars
 
         RDF = ROOT.ROOT.RDataFrame
         self.d = RDF("Events", inputFiles)
@@ -52,13 +31,14 @@ def run(self):
             tmp_th2 = m.getTH2()
 
             for obj in tmp_th1:
-                self.objs.append(obj)
+                self.objs.append(ROOT.RDF.RResultPtr('TH1D')(obj))
 
             for obj in tmp_th2:
-                self.objs.append(obj)
+                self.objs.append(ROOT.RDF.RResultPtr('TH2D')(obj))
 
         if self.snapshot: 
 
+
             """ comment until we understand Snapshot bug
             ROOT.ROOT.RDataFrame("LuminosityBlocks", self.inputFiles).Snapshot("LuminosityBlocks",self.outputFile, "")
             print time.time()-t0, "first snapshot"
@@ -77,7 +57,15 @@ def run(self):
             #ROOT.ROOT.RDataFrame("ParameterSets", self.inputFiles).Snapshot("ParameterSets", self.outputFile, "", opts)
             #print time.time()-t0, "fourth loop"
 
-            self.d.Snapshot("Events", self.outputFile)
+            print self.keepVars
+            if not self.keepVars == '':
+                print "qui"
+                Parser = parser(self.keepVars)
+                self.d.Snapshot("Events", self.outputFile, Parser.parse())
+            else:
+                print "i'm snapshotting!"
+                self.d.Snapshot("Events", self.outputFile)
+
             print time.time()-start, "events snapshot"
 
         self.histoFile.cd()

framework/foo.py

@@ -0,0 +1,20 @@
+import ROOT 
+
+class parser:
+
+    def __init__(self, file):
+
+    	self.file = file
+
+        self.f = open(self.file)                      
+
+    def parse(self):
+
+        list = [x.strip() for x in self.f.readlines()] 
+        branchList = ROOT.vector('string')()
+
+        for l in list:
+        	branchList.push_back(l)
+
+        return branchList 
+

framework/header.py

@@ -0,0 +1,26 @@
+# Built-in/Generic Imports
+import os
+import sys
+import time
+import ROOT
+from foo import *
+
+# Begin code for casting
+
+# this injection can be replaced by properly having this in a header
+# included in the interpreter at framework startup
+ROOT.gInterpreter.Declare('''
+template <typename T>
+class NodeCaster {
+   public:
+   static ROOT::RDF::RNode Cast(T rdf)
+   {
+      return ROOT::RDF::RNode(rdf);
+   }
+};
+''')
+
+def CastToRNode(node):
+   return ROOT.NodeCaster(node.__cppname__).Cast(node)
+
+# end code for casting

framework/parser.py

@@ -0,0 +1,13 @@
+class parser:
+
+    def __init__(self, file):
+
+        self.file = file
+
+        self.f = open(self.file)                      
+
+    def parse(self):
+
+        print [x.strip() for x in self.f.readlines()] 
+        return [x.strip() for x in self.f.readlines()] 
+