Function run_tests_for_group

src/memory/processing/diffexp/mod.rs:529–724

(
    csr_matrix: &CsrMatrix<T>,
    group_indices: &[usize],
    reference_indices: &[usize],
    method: TestMethod,
    correction_method: CorrectionMethod,
    compute_lfc: bool,
    pseudocount: 

#[allow(clippy::too_many_arguments)]
fn run_tests_for_group<T>(
    csr_matrix: &CsrMatrix<T>,
    group_indices: &[usize],
    reference_indices: &[usize],
    method: TestMethod,
    correction_method: CorrectionMethod,
    compute_lfc: bool,
    pseudocount: f64,
    n_genes: usize,
    var_names: &[String],
) -> anyhow::Result<GroupTestResults>
where
    T: FloatOpsTS,
    CsrMatrix<T>: MatrixStatTests<T>,
{
    let n_cols = csr_matrix.ncols();
    let n_rows = csr_matrix.nrows();

    let group_size_f64 = group_indices.len() as f64;
    let ref_size_f64 = reference_indices.len() as f64;

    let mut scores: Vec<f64> = Vec::with_capacity(n_cols);
    let mut pvals: Vec<f64> = Vec::with_capacity(n_cols);
    let mut logfoldchanges: Vec<f64> = Vec::with_capacity(n_cols);

    let mut group_sums_f64 = vec![0.0f64; n_cols];
    let mut ref_sums_f64 = vec![0.0f64; n_cols];
    let mut group_sum_sq_f64 = vec![0.0f64; n_cols];
    let mut ref_sum_sq_f64 = vec![0.0f64; n_cols];

    let mut is_group = vec![false; n_rows];
    let mut is_ref = vec![false; n_rows];

    for &idx in group_indices {
        is_group[idx] = true;
    }
    for &idx in reference_indices {
        is_ref[idx] = true;
    }

    for row in 0..n_rows {
        let row_is_group = is_group[row];
        let row_is_ref = is_ref[row];

        if !row_is_group && !row_is_ref {
            continue;
        }

        let row_data = csr_matrix.row(row);
        for (&col, &value) in row_data.col_indices().iter().zip(row_data.values()) {
            let value_f64 = value.to_f64().unwrap_or(0.0);
            if row_is_group {
                group_sums_f64[col] += value_f64;
                group_sum_sq_f64[col] += value_f64 * value_f64;
            }
            if row_is_ref {
                ref_sums_f64[col] += value_f64;
                ref_sum_sq_f64[col] += value_f64 * value_f64;