Function get_equality_graph

ciphercore-base/src/mpc/mpc_psi.rs:274–305

(
    context: Context,
    type1: Type,
    type2: Type,
    key_header: String,
    is_input1_private: bool,
    is_input2_private: bool,
)

}

fn get_equality_graph(
    context: Context,
    type1: Type,
    type2: Type,
    key_header: String,
    is_input1_private: bool,
    is_input2_private: bool,
) -> Result<Graph> {
    let eq_context = simple_context(|g| {
        let i0 = g.input(type1)?;
        let i1 = g.input(type2)?;

        let key_columns_0 = i0.named_tuple_get(key_header.clone())?;
        let key_columns_1 = i1.named_tuple_get(key_header)?;

        let eq_bits = g.custom_op(
            CustomOperation::new(Equal {}),
            vec![key_columns_0, key_columns_1],
        )?;

        let null_0 = i0.named_tuple_get(NULL_HEADER.to_owned())?;
        let null_1 = i1.named_tuple_get(NULL_HEADER.to_owned())?;

        null_0.multiply(null_1)?.multiply(eq_bits)
    })?;

    convert_main_graph_to_mpc(
        eq_context,
        context,
        vec![is_input1_private, is_input2_private],
    )
}

// Extracts the null column and the column masks of key headers, and computes their product.
// The Boolean output indicates whether PRF keys are needed.