Trusted, Bureaucrats, Administrators
1,660
edits
Changes
Jump to navigation
Jump to search
import scipy as scimport from scipy.ioimport loadmat
| script =
import os
import numpy as np
# Set path
#cwd = status.install_path+os.sep+'test_suite'+os.sep+'shared_data'+os.sep+'dispersion'+os.sep+'Paul_Schanda_2015_Nov'
cwd = os.getcwd()
outdir = cwd + os.sep
fields = [600, 950]
# Loop over the experiments, collect all data
for field in fields:
print("\n%s" % field,)
# Make a dic inside
# Construct the path to the data
path = cwd + os.sep + "Archive" + os.sep + "exp_%s"%field + os.sep + "matrices" + os.sep all_data['%s'%field]['path'] = path
# Collect all filename paths
for file_name in file_names:
# Create path name
file_name_path = path + "%s_%s.mat"%(field, file_name)
field_file_name_paths.append(file_name_path)
# Load the data
file_name_path_data = sc.io.loadmat(file_name_path)
# Extract as numpy
file_name_path_data_np = file_name_path_data[file_name]
all_data['%s'%field]['np_%s'%file_name] = file_name_path_data_np
print("%s %s" % (file_name, file_name_path_data_np.shape))
# Collect residues
# Write a sequence file for relax
f = open(outdir + "residues.txt", "w")
f.write("# Residue_i\n")
for res in all_res_uniq:
f.close()
f_exp = open(outdir + "exp_settings.txt", "w")
f_exp.write("# sfrq_MHz RFfield_kHz file_name\n")
# Then write the files for the rates
k = 1
for field in all_data['fields']:
resis = all_data['%s'%field]['np_residues'][0]
RFfields = all_data['%s'%field]['np_RFfields'][0]
print("\nfield: %3.3f" % field)
for i, RF_field_strength_kHz in enumerate(RFfields):
#print("RF_field_strength_kHz: %3.3f" % RF_field_strength_kHz)
# Generate file name
f_name = outdir + "sfrq_%i_MHz_RFfield_%1.3f_kHz3f_kHz_%03d.in" % (field, RF_field_strength_kHz, k)
cur_file = open(f_name, "w")
cur_file.write("# resi rate rate_err\n")
exp_string = "%11.7f %11.7f %s\n" % (field, RF_field_strength_kHz, f_name) print("%s"%exp_string,)
f_exp.write(exp_string)
cur_file.close()
k += 1
f_exp.close()
