pflotran_dir = os.environ['PFLOTRAN_DIR'] except KeyError: print('PFLOTRAN_DIR must point to PFLOTRAN installation directory and be defined in system environment variables.') sys.exit(1) sys.path.append(pflotran_dir + '/src/python') import numpy as np import matplotlib import matplotlib.pyplot as plt import math import pflotran as pft path = [] path.append('.') files = pft.get_tec_filenames(range(0,36,5)) filenames = pft.get_full_paths(path,files) f = plt.figure(figsize=(6,6)) plt.subplot(1,1,1) f.suptitle("1D Variably Saturated Flow - Pulse, 2 Layer",fontsize=16) plt.xlabel('Saturation [-]') plt.ylabel('Z [m]') plt.xlim(0.,1.) #plt.ylim(0.,1.) #plt.grid(True) line_styles = [] for i in range(7): line_styles.append('-') line_styles.append('--')
print( 'PFLOTRAN_DIR must point to PFLOTRAN installation directory and be defined in system environment variables.' ) sys.exit(1) sys.path.append(pflotran_dir + '/src/python') import numpy as np import matplotlib import matplotlib.pyplot as plt import math import pflotran as pft path = [] path.append('.') files = pft.get_tec_filenames('calcite_tran_only', range(1, 6)) filenames = pft.get_full_paths(path, files) f = plt.figure(figsize=(6, 6)) plt.subplot(1, 1, 1) f.suptitle("1D Calcite", fontsize=16) plt.xlabel('X [m]') plt.ylabel('pH') #plt.xlim(0.,1.) plt.ylim(4.8, 8.2) #plt.grid(True) for ifile in range(len(filenames)): data = pft.Dataset(filenames[ifile], 1, 4) plt.plot(data.get_array('x'), data.get_array('y'), label=data.title)