arjunsavel

"""
uses a callback function to use a tqdm progress bar with multiprocessing.
"""

def func(arg):
  return

def update_pbar(arg):
  """
  Updates a tqdm progress bar. kind of annoying to leave it as referencing a global — might want to wrap this in a class.

arjunsavel

import matplotlib.pyplot as plt
import matplotlib

# CHANGE: these planet params
PERIOD = 1.8 # days
Rp = 1.8 * 69.911e6 # meters

def calc_los(frame):
    """
    calculates LOS velocity everywhere. eq 4 of harada+21

arjunsavel

"""
Script to plot 3D T-P profiles. Anything that should be changed is noted with CHANGE.

author: @arjunsavel
"""
import pandas as pd
import matplotlib.pyplot as plt
import matplotlib.cm as cm
import matplotlib

arjunsavel

import pandas as pd
from astroquery.utils.tap.core import TapPlus

from urllib.error import HTTPError

VIZIER_TAP_URL = 'http://TAPVizieR.u-strasbg.fr/TAPVizieR/tap'
viz = TapPlus(url=VIZIER_TAP_URL)

table = "J/AJ/161/63/table3" # change the table

arjunsavel

import numpy as np
from tqdm import tqdm
import matplotlib.pyplot as plt
from scipy.interpolate import interp2d
import pandas as pd

##### CHANGE this to your absolute file path to your EOS file! #####
eos_filename = 'RT_3D_Transmission_Code/Data/eos_solar_gas_full_hires.dat'

##### CHANGE this to your absolute file path to your GCM file! #####

arjunsavel

try:
    from tqdm import tqdm
except ImportError:
    print(
        "tqdm is not installed on this system, \
            so progress bars will be unavailable. \
            If you would like the progress bar feature, \
            please see the tqdm installation docs: \
            https://github.com/tqdm/tqdm#installation"
    )

arjunsavel

import numpy as np
import matplotlib.pyplot as plt

try:
    from tqdm import tqdm
except ImportError:
    print(
        "tqdm is not installed on this system, \
            so progress bars will be unavailable. \
            If you would like the progress bar feature, \