Function _write_ica

mne/preprocessing/ica.py:2945–3023

Write an ICA object. Parameters ---------- fid: file The file descriptor ica: The instance of ICA to write

(fid, ica)

def _write_ica(fid, ica):
    """Write an ICA object.

    Parameters
    ----------
    fid: file
        The file descriptor
    ica:
        The instance of ICA to write
    """
    ica_init = dict(
        noise_cov=ica.noise_cov,
        n_components=ica.n_components,
        n_pca_components=ica.n_pca_components,
        max_pca_components=ica._max_pca_components,
        current_fit=ica.current_fit,
        allow_ref_meg=ica.allow_ref_meg,
    )

    if ica.info is not None:
        start_block(fid, FIFF.FIFFB_MEAS)
        write_id(fid, FIFF.FIFF_BLOCK_ID)
        if ica.info["meas_id"] is not None:
            write_id(fid, FIFF.FIFF_PARENT_BLOCK_ID, ica.info["meas_id"])

        # Write measurement info
        write_meas_info(fid, ica.info)
        end_block(fid, FIFF.FIFFB_MEAS)

    start_block(fid, FIFF.FIFFB_MNE_ICA)

    #   ICA interface params
    write_string(fid, FIFF.FIFF_MNE_ICA_INTERFACE_PARAMS, _serialize(ica_init))

    #   Channel names
    if ica.ch_names is not None:
        write_name_list(fid, FIFF.FIFF_MNE_ROW_NAMES, ica.ch_names)

    # samples on fit
    n_samples = getattr(ica, "n_samples_", None)
    ica_misc = {
        "n_samples_": (None if n_samples is None else int(n_samples)),
        "labels_": getattr(ica, "labels_", None),
        "method": getattr(ica, "method", None),
        "n_iter_": getattr(ica, "n_iter_", None),
        "fit_params": getattr(ica, "fit_params", None),
    }

    #   ICA misc params
    write_string(fid, FIFF.FIFF_MNE_ICA_MISC_PARAMS, _serialize(ica_misc))

    #   Whitener
    write_double_matrix(fid, FIFF.FIFF_MNE_ICA_WHITENER, ica.pre_whitener_)

    #   PCA components_
    write_double_matrix(fid, FIFF.FIFF_MNE_ICA_PCA_COMPONENTS, ica.pca_components_)

    #   PCA mean_