process_output.py

import numpy as np

from ShockPy import Hugoniot


def read_output(index):
    DYNA_TO_GPA=1e-10
    with open(f'./OutputFiles/HugoniotData_{index}.txt') as f:
        lines = f.readlines()
        first_line = lines[1].split()
        initial_volume = float(first_line[0])
        initial_density = float(first_line[2])
        shock_velocities = []
        particle_velocities = []
        volumes = []
        pressures = []
        particle_velocities = []
        for i in range(1, len(lines)):
            line = lines[i].split()
            current_shock_velocity=float(line[4])
            shock_velocities.append(current_shock_velocity)
            volumes.append(float(line[5]))
            pressures.append(DYNA_TO_GPA * float(line[9]))
            current_density = float(line[6])
            particle_velocities.append((initial_density-current_density)/current_density*current_shock_velocity)
        hugoniot = Hugoniot(pressures=np.array(pressures),
                            particle_velocities=np.array(particle_velocities),
                            shock_velocities=np.array(shock_velocities),
                            volumes=np.array(volumes))
        return hugoniot


if __name__ == '__main__':
    read_output(0)
    # fire.Fire(read_output)