post_treatment.py
ea10a261ae9cmaster

post_treatment.py
View Options

			#!/bin/env python3
			################################################################
			import BlackDynamite as BD
			import matplotlib.pyplot as plt
			################################################################
			# basic connection
			parser = BD.BDParser()
			params = parser.parseBDParameters(
			'--host lsmssrv1.epfl.ch --study mystudy'.split())
			mybase = BD.Base(**params)
			################################################################


			# function to plot things (user's job)
			def plot(run_list):
			for r, j in run_list:
			ekin = r.getScalarQuantity('ekin')
			if ekin is None:
			continue
			print(j)
			list_files = r.listFiles()
			print(list_files)
			fname = r.getFile(list_files[3])
			print(fname + ':')
			_file = open(fname)
			print(_file.read())
			plt.plot(ekin[:, 0], ekin[:, 1], 'o-',
			label='$p_2 = {0}$'.format(j['param2']))

			plt.legend(loc='best')
			plt.show()
			################################################################


			# selecting some runs
			runSelector = BD.RunSelector(mybase)
			run_list = runSelector.selectRuns(params, params)
			plot(run_list)

			# selecting some other runs
			params['run_constraints'] = ['run_name =~ test', 'state = FINISHED']
			params['job_constraints'] = ['param2 > 1']
			run_list = runSelector.selectRuns(params, params)
			plot(run_list)