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)