def back_calc_r2eff(spin=None, spin_id=None, cpmg_frqs=None, spin_lock_nu1=None, store_chi2=False):
"""Back-calculation of R2eff/R1rho values for the given spin.
@keyword spin: The specific spin data container.
@type spin: SpinContainer instance
@keyword spin_id: The spin ID string for the spin container.
@type spin_id: str
@keyword cpmg_frqs: The CPMG frequencies to use instead of the user loaded values - to enable interpolation.
@type cpmg_frqs: list of lists of numpy rank-1 float arrays
@keyword spin_lock_nu1: The spin-lock field strengths to use instead of the user loaded values - to enable interpolation.
@type spin_lock_nu1: list of lists of numpy rank-1 float arrays
@keyword store_chi2: A flag which if True will cause the spin specific chi-squared value to be stored in the spin container.
@type store_chi2: bool
@return: The back-calculated R2eff/R1rho value for the given spin.
@rtype: numpy rank-1 float array
"""
# Skip protons for MMQ data.
if spin.model in MODEL_LIST_MMQ and spin.isotope == '1H':
return
# Create the initial parameter vector.
param_vector = assemble_param_vector(spins=[spin])
# Create a scaling matrix.
scaling_matrix = assemble_scaling_matrix(spins=[spin], scaling=False)
# Number of spectrometer fields.
fields = [None]
field_count = 1
if hasattr(cdp, 'spectrometer_frq_count'):
fields = cdp.spectrometer_frq_list
field_count = cdp.spectrometer_frq_count
# Initialise the data structures for the target function.
values, errors, missing, frqs, frqs_H, exp_types, relax_times = return_r2eff_arrays(spins=[spin], spin_ids=[spin_id], fields=fields, field_count=field_count)
# The offset and R1 data.
chemical_shifts, offsets, tilt_angles = return_offset_data(spins=[spin], spin_ids=[spin_id], field_count=field_count, fields=spin_lock_nu1)
r1 = return_r1_data(spins=[spin], spin_ids=[spin_id], field_count=field_count)
# The dispersion data.
recalc_tau = True
if cpmg_frqs == None and spin_lock_nu1 == None:
cpmg_frqs = return_cpmg_frqs(ref_flag=False)
spin_lock_nu1 = return_spin_lock_nu1(ref_flag=False)
# Reconstruct the structures for interpolation.
else:
recalc_tau = False
values = []
errors = []
missing = []
for exp_type, ei in loop_exp(return_indices=True):
values.append([])
errors.append([])
missing.append([])
for si in range(1):
values[ei].append([])
errors[ei].append([])
missing[ei].append([])
for frq, mi in loop_frq(return_indices=True):
values[ei][si].append([])
errors[ei][si].append([])
missing[ei][si].append([])
for offset, oi in loop_offset(exp_type=exp_type, frq=frq, return_indices=True):
if exp_type in EXP_TYPE_LIST_CPMG:
num = len(cpmg_frqs[ei][mi][oi])
else:
num = len(spin_lock_nu1[ei][mi][oi])
values[ei][si][mi].append(zeros(num, float64))
errors[ei][si][mi].append(ones(num, float64))
missing[ei][si][mi].append(zeros(num, int32))
# Initialise the relaxation dispersion fit functions.
model = Dispersion(model=spin.model, num_params=param_num(spins=[spin]), num_spins=1, num_frq=field_count, exp_types=exp_types, values=values, errors=errors, missing=missing, frqs=frqs, frqs_H=frqs_H, cpmg_frqs=cpmg_frqs, spin_lock_nu1=spin_lock_nu1, chemical_shifts=chemical_shifts, offset=offsets, tilt_angles=tilt_angles, r1=r1, relax_times=relax_times, scaling_matrix=scaling_matrix, recalc_tau=recalc_tau)
# Make a single function call. This will cause back calculation and the data will be stored in the class instance.
chi2 = model.func(param_vector)
# Store the chi-squared value.
if store_chi2:
spin.chi2 = chi2
# Return the structure.
return model.back_calc
def __init__(self, spins=None, spin_ids=None, sim_index=None, scaling_matrix=None, min_algor=None, min_options=None, func_tol=None, grad_tol=None, max_iterations=None, constraints=False, verbosity=0, lower=None, upper=None, inc=None, fields=None, param_names=None):
"""Initialise the base class, storing all the master data to be sent to the slave processor.
This method is run on the master processor whereas the run() method is run on the slave processor.
@keyword spins: The list of spin data container for the cluster. If this argument is supplied, then the spin_id argument will be ignored.
@type spins: list of SpinContainer instances
@keyword spin_ids: The list of spin ID strings corresponding to the spins argument.
@type spin_ids: list of str
@keyword sim_index: The index of the simulation to optimise. This should be None if normal optimisation is desired.
@type sim_index: None or int
@keyword scaling_matrix: The diagonal, square scaling matrix.
@type scaling_matrix: numpy diagonal matrix
@keyword min_algor: The minimisation algorithm to use.
@type min_algor: str
@keyword min_options: An array of options to be used by the minimisation algorithm.
@type min_options: array of str
@keyword func_tol: The function tolerance which, when reached, terminates optimisation. Setting this to None turns of the check.
@type func_tol: None or float
@keyword grad_tol: The gradient tolerance which, when reached, terminates optimisation. Setting this to None turns of the check.
@type grad_tol: None or float
@keyword max_iterations: The maximum number of iterations for the algorithm.
@type max_iterations: int
@keyword constraints: If True, constraints are used during optimisation.
@type constraints: bool
@keyword verbosity: The amount of information to print. The higher the value, the greater the verbosity.
@type verbosity: int
@keyword lower: The lower bounds of the grid search which must be equal to the number of parameters in the model. This optional argument is only used when doing a grid search.
@type lower: array of numbers
@keyword upper: The upper bounds of the grid search which must be equal to the number of parameters in the model. This optional argument is only used when doing a grid search.
@type upper: array of numbers
@keyword inc: The increments for each dimension of the space for the grid search. The number of elements in the array must equal to the number of parameters in the model. This argument is only used when doing a grid search.
@type inc: array of int
@keyword fields: The list of unique of spectrometer field strengths.
@type fields: int
@keyword param_names: The list of parameter names to use in printouts.
@type param_names: str
"""
# Execute the base class __init__() method.
super(Disp_minimise_command, self).__init__()
# Store the arguments needed by the run() method.
self.spins = spins
self.spin_ids = spin_ids
self.sim_index = sim_index
self.scaling_matrix = scaling_matrix
self.verbosity = verbosity
self.min_algor = min_algor
self.min_options = min_options
self.func_tol = func_tol
self.grad_tol = grad_tol
self.max_iterations = max_iterations
self.fields = fields
self.param_names = param_names
# Create the initial parameter vector.
self.param_vector = assemble_param_vector(spins=self.spins)
if len(scaling_matrix):
self.param_vector = dot(inv(scaling_matrix), self.param_vector)
# Get the grid search minimisation options.
self.lower_new, self.upper_new = None, None
if search('^[Gg]rid', min_algor):
self.grid_size, self.inc_new, self.lower_new, self.upper_new = grid_search_setup(spins=spins, spin_ids=spin_ids, param_vector=self.param_vector, lower=lower, upper=upper, inc=inc, scaling_matrix=self.scaling_matrix)
# Linear constraints.
self.A, self.b = None, None
if constraints:
self.A, self.b = linear_constraints(spins=spins, scaling_matrix=scaling_matrix)
# Test if the spectrometer frequencies have been set.
if spins[0].model in [MODEL_LM63, MODEL_CR72, MODEL_CR72_FULL, MODEL_M61, MODEL_TP02, MODEL_TAP03, MODEL_MP05] and not hasattr(cdp, 'spectrometer_frq'):
raise RelaxError("The spectrometer frequency information has not been specified.")
# The R2eff/R1rho data.
self.values, self.errors, self.missing, self.frqs, self.frqs_H, self.exp_types, self.relax_times = return_r2eff_arrays(spins=spins, spin_ids=spin_ids, fields=fields, field_count=len(fields), sim_index=sim_index)
# The offset and R1 data.
self.chemical_shifts, self.offsets, self.tilt_angles = return_offset_data(spins=spins, spin_ids=spin_ids, field_count=len(fields))
self.r1 = return_r1_data(spins=spins, spin_ids=spin_ids, field_count=len(fields), sim_index=sim_index)
# Parameter number.
self.param_num = param_num(spins=spins)
# The dispersion data.
self.dispersion_points = cdp.dispersion_points
self.cpmg_frqs = return_cpmg_frqs(ref_flag=False)
self.spin_lock_nu1 = return_spin_lock_nu1(ref_flag=False)
def __init__(self, spins=None, spin_ids=None, sim_index=None, scaling_matrix=None, min_algor=None, min_options=None, func_tol=None, grad_tol=None, max_iterations=None, constraints=False, verbosity=0, lower=None, upper=None, inc=None, fields=None, param_names=None):
"""Initialise the base class, storing all the master data to be sent to the slave processor.
This method is run on the master processor whereas the run() method is run on the slave processor.
@keyword spins: The list of spin data container for the cluster. If this argument is supplied, then the spin_id argument will be ignored.
@type spins: list of SpinContainer instances
@keyword spin_ids: The list of spin ID strings corresponding to the spins argument.
@type spin_ids: list of str
@keyword sim_index: The index of the simulation to optimise. This should be None if normal optimisation is desired.
@type sim_index: None or int
@keyword scaling_matrix: The diagonal, square scaling matrix.
@type scaling_matrix: numpy diagonal matrix
@keyword min_algor: The minimisation algorithm to use.
@type min_algor: str
@keyword min_options: An array of options to be used by the minimisation algorithm.
@type min_options: array of str
@keyword func_tol: The function tolerance which, when reached, terminates optimisation. Setting this to None turns of the check.
@type func_tol: None or float
@keyword grad_tol: The gradient tolerance which, when reached, terminates optimisation. Setting this to None turns of the check.
@type grad_tol: None or float
@keyword max_iterations: The maximum number of iterations for the algorithm.
@type max_iterations: int
@keyword constraints: If True, constraints are used during optimisation.
@type constraints: bool
@keyword verbosity: The amount of information to print. The higher the value, the greater the verbosity.
@type verbosity: int
@keyword lower: The lower bounds of the grid search which must be equal to the number of parameters in the model. This optional argument is only used when doing a grid search.
@type lower: array of numbers
@keyword upper: The upper bounds of the grid search which must be equal to the number of parameters in the model. This optional argument is only used when doing a grid search.
@type upper: array of numbers
@keyword inc: The increments for each dimension of the space for the grid search. The number of elements in the array must equal to the number of parameters in the model. This argument is only used when doing a grid search.
@type inc: array of int
@keyword fields: The list of unique of spectrometer field strengths.
@type fields: int
@keyword param_names: The list of parameter names to use in printouts.
@type param_names: str
"""
# Execute the base class __init__() method.
super(Disp_minimise_command, self).__init__()
# Store the arguments needed by the run() method.
self.spins = spins
self.spin_ids = spin_ids
self.sim_index = sim_index
self.scaling_matrix = scaling_matrix
self.verbosity = verbosity
self.min_algor = min_algor
self.min_options = min_options
self.func_tol = func_tol
self.grad_tol = grad_tol
self.max_iterations = max_iterations
self.lower = lower
self.upper = upper
self.inc = inc
self.fields = fields
self.param_names = param_names
# Create the initial parameter vector.
self.param_vector = assemble_param_vector(spins=self.spins)
if len(scaling_matrix):
self.param_vector = dot(inv(scaling_matrix), self.param_vector)
# Linear constraints.
self.A, self.b = None, None
if constraints:
self.A, self.b = linear_constraints(spins=spins, scaling_matrix=scaling_matrix)
# Test if the spectrometer frequencies have been set.
if spins[0].model in [MODEL_LM63, MODEL_CR72, MODEL_CR72_FULL, MODEL_M61, MODEL_TP02, MODEL_TAP03, MODEL_MP05] and not hasattr(cdp, 'spectrometer_frq'):
raise RelaxError("The spectrometer frequency information has not been specified.")
# The R2eff/R1rho data.
self.values, self.errors, self.missing, self.frqs, self.frqs_H, self.exp_types, self.relax_times = return_r2eff_arrays(spins=spins, spin_ids=spin_ids, fields=fields, field_count=len(fields), sim_index=sim_index)
# The offset and R1 data.
r1_setup()
self.offsets, spin_lock_fields_inter, self.chemical_shifts, self.tilt_angles, self.Delta_omega, self.w_eff = return_offset_data(spins=spins, spin_ids=spin_ids, field_count=len(fields))
self.r1 = return_r1_data(spins=spins, spin_ids=spin_ids, field_count=len(fields), sim_index=sim_index)
self.r1_fit = is_r1_optimised(spins[0].model)
# Parameter number.
self.param_num = param_num(spins=spins)
# The dispersion data.
self.dispersion_points = cdp.dispersion_points
self.cpmg_frqs = return_cpmg_frqs(ref_flag=False)
self.spin_lock_nu1 = return_spin_lock_nu1(ref_flag=False)
def back_calc_r2eff(spins=None, spin_ids=None, cpmg_frqs=None, spin_lock_offset=None, spin_lock_nu1=None, relax_times_new=None, store_chi2=False):
"""Back-calculation of R2eff/R1rho values for the given spin.
@keyword spins: The list of specific spin data container for cluster.
@type spins: List of SpinContainer instances
@keyword spin_ids: The list of spin ID strings for the spin containers in cluster.
@type spin_ids: list of str
@keyword cpmg_frqs: The CPMG frequencies to use instead of the user loaded values - to enable interpolation.
@type cpmg_frqs: list of lists of numpy rank-1 float arrays
@keyword spin_lock_offset: The spin-lock offsets to use instead of the user loaded values - to enable interpolation.
@type spin_lock_offset: list of lists of numpy rank-1 float arrays
@keyword spin_lock_nu1: The spin-lock field strengths to use instead of the user loaded values - to enable interpolation.
@type spin_lock_nu1: list of lists of numpy rank-1 float arrays
@keyword relax_times_new: The interpolated experiment specific fixed time period for relaxation (in seconds). The dimensions are {Ei, Mi, Oi, Di, Ti}.
@type relax_times_new: rank-4 list of floats
@keyword store_chi2: A flag which if True will cause the spin specific chi-squared value to be stored in the spin container.
@type store_chi2: bool
@return: The back-calculated R2eff/R1rho value for the given spin.
@rtype: numpy rank-1 float array
"""
# Create the initial parameter vector.
param_vector = assemble_param_vector(spins=spins)
# Number of spectrometer fields.
fields = [None]
field_count = 1
if hasattr(cdp, 'spectrometer_frq_count'):
fields = cdp.spectrometer_frq_list
field_count = cdp.spectrometer_frq_count
# Initialise the data structures for the target function.
values, errors, missing, frqs, frqs_H, exp_types, relax_times = return_r2eff_arrays(spins=spins, spin_ids=spin_ids, fields=fields, field_count=field_count)
# The offset and R1 data.
r1_setup()
offsets, spin_lock_fields_inter, chemical_shifts, tilt_angles, Delta_omega, w_eff = return_offset_data(spins=spins, spin_ids=spin_ids, field_count=field_count, spin_lock_offset=spin_lock_offset, fields=spin_lock_nu1)
r1 = return_r1_data(spins=spins, spin_ids=spin_ids, field_count=field_count)
r1_fit = is_r1_optimised(spins[0].model)
# The relaxation times.
if relax_times_new != None:
relax_times = relax_times_new
# The dispersion data.
recalc_tau = True
if cpmg_frqs == None and spin_lock_nu1 == None and spin_lock_offset == None:
cpmg_frqs = return_cpmg_frqs(ref_flag=False)
spin_lock_nu1 = return_spin_lock_nu1(ref_flag=False)
# Reset the cpmg_frqs if interpolating R1rho models.
elif cpmg_frqs == None and spin_lock_offset != None:
cpmg_frqs = None
spin_lock_nu1 = spin_lock_fields_inter
recalc_tau = False
values = []
errors = []
missing = []
for exp_type, ei in loop_exp(return_indices=True):
values.append([])
errors.append([])
missing.append([])
for si in range(len(spins)):
values[ei].append([])
errors[ei].append([])
missing[ei].append([])
for frq, mi in loop_frq(return_indices=True):
values[ei][si].append([])
errors[ei][si].append([])
missing[ei][si].append([])
for oi, offset in enumerate(offsets[ei][si][mi]):
num = len(spin_lock_nu1[ei][mi][oi])
values[ei][si][mi].append(zeros(num, float64))
errors[ei][si][mi].append(ones(num, float64))
missing[ei][si][mi].append(zeros(num, int32))
# Reconstruct the structures for interpolation.
else:
recalc_tau = False
values = []
errors = []
missing = []
for exp_type, ei in loop_exp(return_indices=True):
values.append([])
errors.append([])
missing.append([])
for si in range(len(spins)):
values[ei].append([])
errors[ei].append([])
missing[ei].append([])
for frq, mi in loop_frq(return_indices=True):
values[ei][si].append([])
errors[ei][si].append([])
missing[ei][si].append([])
for offset, oi in loop_offset(exp_type=exp_type, frq=frq, return_indices=True):
if exp_type in EXP_TYPE_LIST_CPMG:
num = len(cpmg_frqs[ei][mi][oi])
else:
num = len(spin_lock_nu1[ei][mi][oi])
values[ei][si][mi].append(zeros(num, float64))
errors[ei][si][mi].append(ones(num, float64))
missing[ei][si][mi].append(zeros(num, int32))
# Initialise the relaxation dispersion fit functions.
model = Dispersion(model=spins[0].model, num_params=param_num(spins=spins), num_spins=len(spins), num_frq=field_count, exp_types=exp_types, values=values, errors=errors, missing=missing, frqs=frqs, frqs_H=frqs_H, cpmg_frqs=cpmg_frqs, spin_lock_nu1=spin_lock_nu1, chemical_shifts=chemical_shifts, offset=offsets, tilt_angles=tilt_angles, r1=r1, relax_times=relax_times, recalc_tau=recalc_tau, r1_fit=r1_fit)
# Make a single function call. This will cause back calculation and the data will be stored in the class instance.
chi2 = model.func(param_vector)
# Store the chi-squared value.
if store_chi2:
for spin in spins:
spin.chi2 = chi2
# Return the structure.
return model.get_back_calc()
def back_calc_r2eff(spins=None, spin_ids=None, cpmg_frqs=None, spin_lock_offset=None, spin_lock_nu1=None, relax_times_new=None, store_chi2=False):
"""Back-calculation of R2eff/R1rho values for the given spin.
@keyword spins: The list of specific spin data container for cluster.
@type spins: List of SpinContainer instances
@keyword spin_ids: The list of spin ID strings for the spin containers in cluster.
@type spin_ids: list of str
@keyword cpmg_frqs: The CPMG frequencies to use instead of the user loaded values - to enable interpolation.
@type cpmg_frqs: list of lists of numpy rank-1 float arrays
@keyword spin_lock_offset: The spin-lock offsets to use instead of the user loaded values - to enable interpolation.
@type spin_lock_offset: list of lists of numpy rank-1 float arrays
@keyword spin_lock_nu1: The spin-lock field strengths to use instead of the user loaded values - to enable interpolation.
@type spin_lock_nu1: list of lists of numpy rank-1 float arrays
@keyword relax_times_new: The interpolated experiment specific fixed time period for relaxation (in seconds). The dimensions are {Ei, Mi, Oi, Di, Ti}.
@type relax_times_new: rank-4 list of floats
@keyword store_chi2: A flag which if True will cause the spin specific chi-squared value to be stored in the spin container.
@type store_chi2: bool
@return: The back-calculated R2eff/R1rho value for the given spin.
@rtype: numpy rank-1 float array
"""
# Create the initial parameter vector.
param_vector = assemble_param_vector(spins=spins)
# Number of spectrometer fields.
fields = [None]
field_count = 1
if hasattr(cdp, 'spectrometer_frq_count'):
fields = cdp.spectrometer_frq_list
field_count = cdp.spectrometer_frq_count
# Initialise the data structures for the target function.
values, errors, missing, frqs, frqs_H, exp_types, relax_times = return_r2eff_arrays(spins=spins, spin_ids=spin_ids, fields=fields, field_count=field_count)
# The offset and R1 data.
r1_setup()
offsets, spin_lock_fields_inter, chemical_shifts, tilt_angles, Delta_omega, w_eff = return_offset_data(spins=spins, spin_ids=spin_ids, field_count=field_count, spin_lock_offset=spin_lock_offset, fields=spin_lock_nu1)
r1 = return_r1_data(spins=spins, spin_ids=spin_ids, field_count=field_count)
r1_fit = is_r1_optimised(spins[0].model)
# The relaxation times.
if relax_times_new != None:
relax_times = relax_times_new
# The dispersion data.
recalc_tau = True
if cpmg_frqs == None and spin_lock_nu1 == None and spin_lock_offset == None:
cpmg_frqs = return_cpmg_frqs(ref_flag=False)
spin_lock_nu1 = return_spin_lock_nu1(ref_flag=False)
# Reset the cpmg_frqs if interpolating R1rho models.
elif cpmg_frqs == None and spin_lock_offset != None:
cpmg_frqs = None
spin_lock_nu1 = spin_lock_fields_inter
recalc_tau = False
values = []
errors = []
missing = []
for exp_type, ei in loop_exp(return_indices=True):
values.append([])
errors.append([])
missing.append([])
for si in range(len(spins)):
values[ei].append([])
errors[ei].append([])
missing[ei].append([])
for frq, mi in loop_frq(return_indices=True):
values[ei][si].append([])
errors[ei][si].append([])
missing[ei][si].append([])
for oi, offset in enumerate(offsets[ei][si][mi]):
num = len(spin_lock_nu1[ei][mi][oi])
values[ei][si][mi].append(zeros(num, float64))
errors[ei][si][mi].append(ones(num, float64))
missing[ei][si][mi].append(zeros(num, int32))
# Reconstruct the structures for interpolation.
else:
recalc_tau = False
values = []
errors = []
missing = []
for exp_type, ei in loop_exp(return_indices=True):
values.append([])
errors.append([])
missing.append([])
for si in range(len(spins)):
values[ei].append([])
errors[ei].append([])
missing[ei].append([])
for frq, mi in loop_frq(return_indices=True):
values[ei][si].append([])
errors[ei][si].append([])
missing[ei][si].append([])
for offset, oi in loop_offset(exp_type=exp_type, frq=frq, return_indices=True):
if exp_type in EXP_TYPE_LIST_CPMG:
num = len(cpmg_frqs[ei][mi][oi])
else:
num = len(spin_lock_nu1[ei][mi][oi])
values[ei][si][mi].append(zeros(num, float64))
errors[ei][si][mi].append(ones(num, float64))
missing[ei][si][mi].append(zeros(num, int32))
# Initialise the relaxation dispersion fit functions.
model = Dispersion(model=spins[0].model, num_params=param_num(spins=spins), num_spins=len(spins), num_frq=field_count, exp_types=exp_types, values=values, errors=errors, missing=missing, frqs=frqs, frqs_H=frqs_H, cpmg_frqs=cpmg_frqs, spin_lock_nu1=spin_lock_nu1, chemical_shifts=chemical_shifts, offset=offsets, tilt_angles=tilt_angles, r1=r1, relax_times=relax_times, recalc_tau=recalc_tau, r1_fit=r1_fit)
# Make a single function call. This will cause back calculation and the data will be stored in the class instance.
chi2 = model.func(param_vector)
# Store the chi-squared value.
if store_chi2:
for spin in spins:
spin.chi2 = chi2
# Return the structure.
return model.get_back_calc()