Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
CalystaT committed Aug 18, 2023
1 parent 7b37969 commit 3a627ca
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions macrodensity/plotting.py
Original file line number Diff line number Diff line change
Expand Up @@ -383,8 +383,8 @@ def plot_planar_average(lattice_vector: float,input_file: str=' ',output_file: s

elif 'gulp' in input_file or '.out' in input_file:
interpolated_potential = []
output_file = 'PlanarGulp.csv'
img_file = 'PlanarGulp.png'
output_file = 'GulpPotential.csv'
img_file = 'GulpPotential.png'

pot, NGX, NGY, NGZ, Lattice = read_gulp_potential(input_file)
vector_a, vector_b, vector_c, av, bv, cv = matrix_2_abc(Lattice)
Expand Down Expand Up @@ -418,7 +418,7 @@ def plot_planar_average(lattice_vector: float,input_file: str=' ',output_file: s
df = df.transpose()
df.to_csv(output_file)

return planar, macro, fig, interpolated_potential
return planar, macro, interpolated_potential, fig

elif 'vasp' in input_file or 'LOCPOT' in input_file:
output_file = 'PlanarAverage.csv'
Expand Down

0 comments on commit 3a627ca

Please sign in to comment.