psynth_wrap.py

from planetsynth import PlanetSynth
import numpy as np
import math as ma

pls = PlanetSynth()

M = float(input('''
Mass of planet in Jupiter masses
 Min: 0.1
 Max: 30
(effects of deuterium burning for >13 Mj not modeled)
Input: '''))

if M > 5:
    maxZ = 0.04
elif M > 3:
    maxZ = 0.2
elif M > 1:
    maxZ = 0.5
else:
    maxZ = 0.8

Z = float(input(f'''
Bulk planet metallicity; portion of mass composed of elements heavier than helium
For your mass:
 Min: 0
 Max: {maxZ}
Input: '''))

maxZe = min(0.1, Z)

Ze = float(input(f'''
Atmospheric metallicity: portion of atmospheric mass composed of elements heavier than helium
For your bulk metallicity:
 Min: 0
 Max: {maxZe}
Input: '''))

logF = ma.log10(float(input('''
Stellar flux in Watts/meter^2
 Min: 0.01 (effectively 0)
 Max: 1000000 (1e6)
(assumed to remain constant)
Input: '''))*250)
#converts to log of ergs/s/cm^2

logt = ma.log10(float(input('''
Age of planet in billions of years
 Min: 0.01
 Max: 10
Input: '''))*1e9)
#converts to log of years

params = [M, Z, Ze, logF]

print('')
print('working...')

res = pls.predict(logt, params)

print(f'''
Results:
Radius (Earth radii):
 {res[0]*10.973}
Luminosity (relative to Sun):
 {10**res[1]}
Effective temperature (Kelvin):
 {res[2]}
Surface gravity (Earth g):
 {(10**res[3])/981}
 ''')

a = input('press enter to exit ')