atomcloud.mixed_level.bimodal module

class atomcloud.mixed_level.bimodal.ComplexBimodal(scipy_length=1000.0, fixed_length=None)[source]

Bases: MixedLevelBase

Parameters:

  • scipy_length (int) –

  • fixed_length (int | None) –

bec(*args, **kwargs)
bec_sums(coords, data)[source]
fit_order = ['gsum', 'thermal1', 'thermal2', 'thermal3', 'tfsum', 'bec']
get_fit(coords, data, *args, **kwargs)[source]

Get the fit for the mixed level fit.

This function will get the fit for the mixed level fit. Each fit level will be called in the order specified by the fit_order class variable. If a fit fails, then the fit will break out of the fit sequence and return the fit information up to that point.

The user can add their own args and kwargs to the input parameters. These are then assigned to the self.args and self.kwargs class variables, which allows the user to use these parameters in the (decorated) pre-processing functions.

If save data is True, then the data, coordinates, masks and seeds will be saved for each fit level. However, the initial data and coordinates will be saved under the key ‘default’. If any other fit level uses the same data and coordinates, as the default fit data, then the data and coordinates will be None.

Parameters:

  • coords – The coordinates of the data.

  • data – The data to be fit.

  • save_data – Whether or not to save the data and coordinates

  • level. (for each fit) –

  • *args – Additional arguments to be passed to the fit.

  • **kwargs – Additional keyword arguments to be passed to the fit.

Returns:

A tuple containing a boolean that indicates if the fit passed and a dictionary of fit dictionaries.

gsum(*args, **kwargs)
tfsum(*args, **kwargs)
thermal1(*args, **kwargs)
thermal2(*args, **kwargs)
thermal3(*args, **kwargs)
class atomcloud.mixed_level.bimodal.SimpleBimodal(scipy_length=1000.0, fixed_length=None)[source]

Bases: MixedLevelBase

Parameters:

  • scipy_length (int) –

  • fixed_length (int | None) –

bec(*args, **kwargs)
fit_order = ['gsum', 'thermal', 'tf_sum', 'bec']
gsum(*args, **kwargs)
tf_sum(*args, **kwargs)
thermal(*args, **kwargs)