xref: /petsc/src/ts/tutorials/extchem.py (revision 1b37a2a7cc4a4fb30c3e967db1c694c0a1013f51)
1#!/usr/bin/env python3
2#
3#    Reads in the output trajectory data from a run of extchem and
4#  formats it for graphing
5#
6#  Make sure $PETSC_DIR/bin is in your PYTHONPATH
7#
8from __future__ import print_function
9import sys
10import PetscBinaryIOTrajectory
11
12if __name__ ==  '__main__':
13  if len(sys.argv) > 1:
14    directory = sys.argv[1]
15  else:
16    directory = 'Visualization-data'
17  (t,v,names) = PetscBinaryIOTrajectory.ReadTrajectory(directory)
18#  PetscBinaryIOTrajectory.PlotTrajectories(t,v,names,['Temp','CO','CO2','H2O','H2','O2','CH4','C2H2','N2'])
19#
20#  Code is currently hardwired to display certain species only, edit the list below to display the species you want displayed
21#
22  for i in range(0,len(t)-1):
23    print(t[i],v[i][names.index('Temp')],v[i][names.index('CH4')],v[i][names.index('O2')],v[i][names.index('N2')],v[i][names.index('CO')],v[i][names.index('CO2')],v[i][names.index('O')],v[i][names.index('OH')],v[i][names.index('H2O')])
24
25
26