From fe782aaf6055b596c97567a13fae4be7f2ce737c Mon Sep 17 00:00:00 2001
From: Yannick Ulrich <yannick.ulrich@psi.ch>
Date: Sun, 16 Feb 2020 12:09:17 +0100
Subject: [PATCH] 3: added metadata and status towards #12
---
pymule/vegas.py | 25 +++++++++++++++++++++++++
1 file changed, 25 insertions(+)
diff --git a/pymule/vegas.py b/pymule/vegas.py
index aa2b41f..23b3d63 100644
--- a/pymule/vegas.py
+++ b/pymule/vegas.py
@@ -158,3 +158,28 @@ def exportvegas(dic, filename="", fp=None):
plots = sorted(list(set(dic.keys()) - {
'SHA','chi2a','iteration','msg','value','time','ndo','randy','xi'
}))
+
+ y,e = dic['value']
+
+ write_record(fp, 'c', 'v3N ')
+ write_record(fp, 'c', dic['SHA'])
+ write_record(fp, 'i', dic['iteration'])
+ write_record(fp, 'i', dic['ndo'])
+ write_record(fp, 'd', y/e**2)
+ write_record(fp, 'd', 1/e**2)
+ write_record(
+ fp, 'd',
+ dic['chi2a']*(dic['iteration']-1) + y**2/e**2
+ )
+
+ write_record(fp, 'd', dic['xi'])
+ write_record(fp, 'i', dic['randy'])
+
+ nrbins = len(dic[plots[0]])
+ nrq = len(plots)
+
+ write_record(fp, 'i', [
+ nrq,
+ nrbins,
+ max(max([len(i) for i in plots]),6)
+ ])
--
GitLab