Jungfraujoch/image_analysis/scale_merge/HKLKey.cpp

// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
// SPDX-License-Identifier: GPL-3.0-only

#include <cmath>

#include "HKLKey.h"
#include "gemmi/symmetry.hpp"

HKLKeyGenerator::HKLKeyGenerator(bool merge_friedel, const std::optional<gemmi::SpaceGroup> &sg)
: merge_friedel(merge_friedel), sg(sg) {}

HKLKey HKLKeyGenerator::operator()(const MergedReflection &r) const {
    return operator()(r.h, r.k, r.l);
}

HKLKey HKLKeyGenerator::operator()(const Reflection &r) const {
    return operator()(r.h, r.k, r.l);
}

HKLKey HKLKeyGenerator::operator()(int32_t h, int32_t k, int32_t l) const {
    HKLKey key{h, k, l, true};

    if (!sg.has_value()) {
        const HKLKey neg{-h, -k, -l, true};
        if (std::tie(key.h, key.k, key.l) < std::tie(neg.h, neg.k, neg.l)) {
            key.h = -key.h;
            key.k = -key.k;
            key.l = -key.l;
            key.plus = merge_friedel;
        }
    } else {
        const auto sg_local = sg.value();
        const auto ops = sg_local.operations();
        const gemmi::ReciprocalAsu asu(&sg_local);

        const gemmi::Op::Miller in{h, k, l};
        const auto [hkl, sign_plus] = asu.to_asu_sign(in, ops);

        key.h = hkl[0];
        key.k = hkl[1];
        key.l = hkl[2];
        key.plus = merge_friedel ? true : sign_plus;
    }
    return key;
}

bool AcceptReflection(const Reflection &r, std::optional<double> d_min_limit) {
    if (!std::isfinite(r.I))
        return false;
    if (!std::isfinite(r.d) || r.d <= 0.0f)
        return false;
    if (d_min_limit && r.d < d_min_limit)
        return false;
    if (!std::isfinite(r.rlp) || r.rlp == 0.0f)
        return false;
    return true;
}

bool AcceptReflection(const Reflection &r, double d_min_limit) {
    if (!std::isfinite(r.I))
        return false;
    if (!std::isfinite(r.d) || r.d <= 0.0f)
        return false;
    if (d_min_limit > 0.0 && r.d < d_min_limit)
        return false;
    if (!std::isfinite(r.rlp) || r.rlp == 0.0f)
        return false;
    return true;
}