def FrisPy_Animation(config_filename):
#Parse the configuration file
#First check for necessary files for the animation
conf = config.read_config_file(config_filename)
if 'params_file_name' not in conf:
raise Exception("Missing "+\
"params_file_name in config_name")
if 'initial_conditions_file_name' not in conf:
raise Exception("Missing "+\
"initial_conditions_file_name"+\
" in config_name")
#Read in the parameters (aka coefficients) and the initial conditions
params_file_name = conf['params_file_name']
initial_conditions_file_name = conf['initial_conditions_file_name']
params,initial_conditions = setup.read_params_and_initial_conditions\
(params_file_name,
initial_conditions_file_name)
#Pass the parameters and conditions to the driver routine
positions, n_times = driver_interface_animation.get_positions(initial_conditions,params)
#Write positions to a file. This will be commented out
#when testing is not underway
output_trajectory.write_trajectory_to_file(positions,n_times)
#Pass the positions and number of times to the animation routines
make_visualizations.make_plots(positions,n_times)
print "\nCompleting Animation\n"
return
def FrisPy_MCMC(config_filename):
#Parse config file
conf = config.read_config_file(config_filename)
if 'params_file_name' not in conf:
raise Exception("Missing "+\
"params_file_name in config_name")
if 'flight_data_file_name' not in conf:
raise Exception("Missing "+\
"flight_data_file_name in config_name")
#Read in the flight data
params_file_name = conf['params_file_name']
flight_data_file_name = conf['flight_data_file_name']
initial_conditions_file_name = conf['initial_conditions_file_name']
params,flight_data =setup.read_params_and_flight_data(params_file_name,flight_data_file_name)
initial_conditions = setup.read_initial_conditions(initial_conditions_file_name)
#flight_data is a numpy.ndarray with names,
#while params is an array, so the flight
#data names need to be extracted
names = flight_data.dtype.names
#Perform the MCMC
samples,likelihoods = start_mcmc.perform_mcmc(\
names,flight_data,initial_conditions,params)
#Create plots
mcmc_analysis.analysis(samples,likelihoods,names,flight_data,params)
print "\nCompleting MCMC\n"
return
