from lakesuperior.model.base cimport Key, TripleKey

cdef bint lookup_sk_cmp_fn(
        const TripleKey* spok, const Key* k1, const Key* k2
    ):
    """ Keyset lookup for S key. """
    return spok[0] == k1

cdef bint lookup_pk_cmp_fn(
        const TripleKey* spok, const Key* k1, const Key* k2
    ):
    """ Keyset lookup for P key. """
    return spok[1] == k1

cdef bint lookup_ok_cmp_fn(
        const TripleKey* spok, const Key* k1, const Key* k2
    ):
    """ Keyset lookup for O key. """
    return spok[2] == k1

cdef bint lookup_skpk_cmp_fn(
        const TripleKey* spok, const Key* k1, const Key* k2
    ):
    """ Keyset lookup for S and P keys. """
    return spok[0] == k1 and spok[1] == k2

cdef bint lookup_skok_cmp_fn(
        const TripleKey* spok, const Key* k1, const Key* k2
    ):
    """ Keyset lookup for S and O keys. """
    return spok[0] == k1 and spok[2] == k2

cdef bint lookup_pkok_cmp_fn(
        const TripleKey* spok, const Key* k1, const Key* k2
    ):
    """ Keyset lookup for P and O keys. """
    return spok[1] == k1 and spok[2] == k2