From 8f46a7926d3517bf5405fc5a8c723e6cc318ad58 Mon Sep 17 00:00:00 2001 From: Samo Penic <samo.penic@gmail.com> Date: Wed, 14 Aug 2019 21:34:31 +0000 Subject: [PATCH] Added missing function to wrapper.py --- trisurf/analyses.py | 12 ++++++------ 1 files changed, 6 insertions(+), 6 deletions(-) diff --git a/trisurf/analyses.py b/trisurf/analyses.py index 55d2af7..ae4dce7 100644 --- a/trisurf/analyses.py +++ b/trisurf/analyses.py @@ -31,12 +31,12 @@ # can be wrapped to specify scalar_field) @analysis('plotrunningavginteractive') -def plotrunningavginteractive(run, scalar_field='vertices_idx', **kwargs): +def plotrunningavginteractive(run, scalar_field='hbar', ylabel="1/n sum_i=niter^n(hbar_i)", **kwargs): import matplotlib.pyplot as plt from trisurf import VTKRendering as vtk import math from multiprocessing import Process - table=trisurf.Statistics(run.Dir.fullpath(),filename='data_tspoststat.csv').getTable() + table=trisurf.Statistics(run.Dir.fullpath(),filename='poststat.csv').getTable() def running_avg(col): import numpy as np avg=[] @@ -47,11 +47,11 @@ vtk.Renderer(kwargs.get('args', None),kwargs.get('host',None),run, timestep=n,scalar_field=scalar_field) fig=plt.figure(1) - ra=running_avg(table['hbar']) - l=len(table['hbar']) + ra=running_avg(table[scalar_field]) + l=len(table[scalar_field]) plt.plot(ra) plt.title('Running average') - plt.ylabel('1/n sum_i=niter^n(hbar_i)') + plt.ylabel(ylabel) plt.xlabel('n') def onclick(event): #print('button=%d, x=%d, y=%d, xdata=%f, ydata=%f' % (event.button, event.x, event.y, event.xdata, event.ydata)) @@ -66,7 +66,7 @@ # these functions should be wrapped # ------------------------------- @analysis('plotColumnFromPostProcess') -def plotColumnFromPostProcess(run, filename='data_tspoststat.csv', column='hbar', **kwargs): +def plotColumnFromPostProcess(run, filename='poststat.csv', column='hbar', **kwargs): import matplotlib.pyplot as plt def smooth(y, box_pts): -- Gitblit v1.9.3