Function compute_rank_gene_groups

src/memory/processing/diffexp/mod.rs:351–419

(
    adata: &IMAnnData,
    groupby: &str,
    reference: Option<&str>,
    groups: Option<&[&str]>,
    method: TestMethod,
    n_genes: usize,
    correction_method: CorrectionMethod,
    compute_l

#[allow(clippy::too_many_arguments)]
fn compute_rank_gene_groups(
    adata: &IMAnnData,
    groupby: &str,
    reference: Option<&str>,
    groups: Option<&[&str]>,
    method: TestMethod,
    n_genes: usize,
    correction_method: CorrectionMethod,
    compute_lfc: bool,
    pseudocount: f64,
) -> anyhow::Result<RankGeneGroupsComputed> {
    let all_groups = get_unique_groups(adata, groupby)?;
    let groups_to_test = filter_groups_to_test(&all_groups, groups)?;
    let reference_group = resolve_reference_group(&all_groups, reference)?;

    let var_names = adata.var_names();

    let x = adata.x();
    let read_guard = x.0.read_inner();
    let data = read_guard.deref();

    let correction_method_owned = correction_method;

    let results = match data {
        ArrayData::CsrMatrix(matrix) => match matrix {
            DynCsrMatrix::F32(csr_matrix) => run_differential_expression(
                adata,
                csr_matrix,
                &groups_to_test,
                &reference_group,
                groupby,
                method,
                correction_method_owned.clone(),
                compute_lfc,
                pseudocount,
                n_genes,
                &var_names,
            )?,
            DynCsrMatrix::F64(csr_matrix) => run_differential_expression(
                adata,
                csr_matrix,
                &groups_to_test,
                &reference_group,
                groupby,
                method,
                correction_method_owned,
                compute_lfc,
                pseudocount,
                n_genes,
                &var_names,
            )?,
            _ => {
                return Err(anyhow::anyhow!(
                    "Unsupported matrix data type. Only F32 and F64 CSR matrices are supported."
                ));
            }
        },
        other => unimplemented!(