GopherProxy

	tadd fluid pressure plotting functions - sphere - GPU-based 3D discrete element…
	git clone git://src.adamsgaard.dk/sphere
	Log
	Files
	Refs
	LICENSE
	---
	commit dfa8cd6359c3fe3253c253fa633d028c24ca48dd
	parent 26cb3f873fc2d80c30a815172c1f80ea5a48b84e
	Author: Anders Damsgaard <[email protected]>
	Date: Thu, 8 Jan 2015 14:34:48 +0100

	add fluid pressure plotting functions

	Diffstat:
	M python/sphere.py \| 47 +++++++++++++++++++++++++++++…

	1 file changed, 45 insertions(+), 2 deletions(-)
	---
	diff --git a/python/sphere.py b/python/sphere.py
	t@@ -5361,7 +5361,8 @@ class sim:
	method.

	:param method: The type of plot to render. Possible values are 'energy…
	- 'walls', 'triaxial' and 'shear'
	+ 'walls', 'triaxial', 'mean-fluid-pressure', 'fluid-pressure' and
	+ 'shear'
	:type method: str
	:param savefig: Save the image instead of showing it on screen
	:type savefig: bool
	t@@ -5726,7 +5727,7 @@ class sim:
	if fh is not None:
	fh.close()

	- elif method == 'fluid-pressure':
	+ elif method == 'mean-fluid-pressure':

	# Read pressure values from simulation binaries
	for i in range(lastfile+1):
	t@@ -5751,6 +5752,48 @@ class sim:
	#ax1.legend()
	ax1.grid()

	+ elif method == 'fluid-pressure':
	+
	+ sb.readfirst(verbose=False)
	+
	+ # cell midpoint cell positions
	+ zpos_c = numpy.zeros(sb.num[2])
	+ dz = sb.L[2]/sb.num[2]
	+ for i in numpy.arange(sb.num[2]):
	+ zpos_c[i] = idz + 0.5dz
	+
	+ shear_strain = numpy.zeros(sb.status())
	+ pres = numpy.zeros((sb.num[2], sb.status()))
	+
	+ # Read pressure values from simulation binaries
	+ for i in numpy.arange(sb.status()):
	+ sb.readstep(i, verbose = False)
	+ pres[:,i] = numpy.average(numpy.average(sb.p_f, axis=0), axis=…
	+
	+ shear_strain[i] = sb.shearStrain()
	+
	+ t = numpy.linspace(0.0, sb.time_current, lastfile+1)
	+
	+ # Plotting
	+ if (outformat != 'txt'):
	+
	+ ax = plt.subplot(1,1,1)
	+
	+ im1 = ax.pcolormesh(shear_strain, zpos_c, pres/1000.0,
	+ rasterized=True)
	+ ax.set_xlim([0, numpy.max(shear_strain)])
	+ ax.set_ylim([zpos_c[0], sim.w_x[0]])
	+ ax.set_xlabel('Shear strain $\\gamma$ [-]')
	+ ax.set_ylabel('Vertical position $z$ [m]')
	+
	+ ax.set_title(sb.id())
	+
	+ cb = plt.colorbar(im1)
	+ cb.set_label('$p_\\text{f}$ [kPa]')
	+ cb.solids.set_rasterized(True)
	+ plt.tight_layout()
	+ plt.subplots_adjust(wspace = .05)
	+
	else :
	print("Visualization type '" + method + "' not understood")
	return