hiro/bachelor_thesis
1
0
Fork 0
mirror of https://github.com/vale981/bachelor_thesis synced 2025-03-05 09:31:42 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

make tex_value smarter to support values without unit and integers

Browse source
This commit is contained in:
hiro98 2020-04-08 13:21:58 +02:00
parent f69ae2151b
commit fcaf5c79e6
1 changed files with 16 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
prog/python/utility.py
View file

@ -27,7 +27,7 @@ def η_to_θ(η):
def η_to_pt(η, p):
return p/np.cosh(η)
def tex_value(val, err=None, unit='', prefix='', prec=10, save=None):
def tex_value(val, err=None, unit=None, prefix='', prec=0, save=None):
"""Generates LaTeX output of a value with units and error."""
if err:
@ -35,19 +35,28 @@ def tex_value(val, err=None, unit='', prefix='', prec=10, save=None):
else:
val = np.round(val, prec)
val_string = fr'\({prefix}\SI{{{val:.{prec}f}'
if prec == 0:
val = int(val)
if err:
err = int(err)
val_string = fr'{val:.{prec}f}' if prec > 0 else str(val)
if err:
val_string += fr'\pm {err:.{prec}f}'
val_string += fr'\pm {err:.{prec}f}' if prec > 0 else str(err)
val_string += fr'}}{{{unit}}}\)'
ret_string = ''
if save:
if unit is None:
ret_string += fr'\({prefix}{val_string}\)'
else:
ret_string += fr'\({prefix}\SI{{{val_string}}}{{{unit}}}\)'
if save is not None:
os.makedirs(save[0], exist_ok=True)
with open(f'{save[0]}/{save[1]}', 'w') as f:
f.write(val_string)
f.write(ret_string)
return val_string
return ret_string
###############################################################################
# Plot Porn #