# Directory where we can find data for plotting
data_dir = 'data/'

import numpy as np
import pandas as pd
import sys
import time
import matplotlib.pyplot as plt
import emcee  # MCMC package
import corner  # For plots later - not essential to run but great for viewing
import astropy.coordinates as coord  # For getting dust values
import astropy.units as u  # # For getting dust values
from psutil import virtual_memory  # For preventing RAM errors in emcee running
from mpl_toolkits.mplot3d import Axes3D  # 3D plots
from astropy.stats import knuth_bin_width  # Great bin width estimator
from scipy.optimize import minimize  # Used in a few places for minimisation
from scipy.stats import norm  # Fast normal distribution evaluator
from scipy.stats import iqr as interquartile_range
from scipy.signal import savgol_filter  # Smoothing algorithm
from scipy.interpolate import interp1d as interpolate  # Cubic splines
from astroquery.irsa_dust import IrsaDust  # For getting dust values

# Set some parameters
number_of_cores = 8  # Number of CPU cores to use
N_binaries = 0  # If zero, binary stars will be excluded from the model.
scale_true = 1000  # Mean distance to stars
use_metallicities = False  # Whether or not the model should use stellar FeHs
infer_for_Rv = False

# Set the type of parallax prior:
# finite_cloud, decreasing_space_density
prior_type = 'decreasing_space_density'
N_prior = 0  # Necessary for 'finite cloud' data