9: user interface for loader in re #17

pymule/__init__.py

@@ -3,7 +3,8 @@ import matplotlib.pyplot as plt
 
 from vegas import importvegas
 from errortools import mergenumbers, plusnumbers, dividenumbers, timesnumbers,\
-                       mergeplots, addplots, combineNplots, scaleplot
-from loader import importreg, pattern,                                        \
+                       mergeplots, addplots, combineNplots, scaleplot,        \
+                       integratehistogram, mergebins
+from loader import importreg, pattern, setup, sigma,                          \
                    mergeset, mergeseeds, mergefks,                            \
-                   addsets
+                   addsets, scaleset

pymule/loader.py

@@ -2,6 +2,9 @@ import re
 from vegas import importvegas
 from errortools import *
 import os
+import inspect
+
+loadargs = {}
 
 
 def getplots(s):
@@ -114,3 +117,28 @@ def mergefks(*sets):
     for para in paras:
         ans.append(addsets([i[para] for i in sets]))
     return mergeset(ans)
+
+
+def callsanitised(func, **kwargs):
+    global loadargs
+    legal = inspect.getargspec(func).args
+    args = {}
+    args.update(loadargs)
+    args.update(kwargs)
+    args = {
+        i: args[i]
+        for i in args.keys()
+        if i in legal
+    }
+    return func(**args)
+
+
+def setup(**kwargs):
+    global loadargs
+    loadargs = kwargs
+
+
+def sigma(piece, **kwargs):
+    pat = callsanitised(pattern, piece=piece, **kwargs)
+    s = callsanitised(importreg, r=pat, **kwargs)
+    return callsanitised(mergeseeds, s=s, **kwargs)