pessimisticPatternMatcherUtils.h

Go to the documentation of this file.

// -*- LSST-C++ -*-
 
/*
 * This file is part of meas_astrom.
 *
 * Developed for the LSST Data Management System.
 * This product includes software developed by the LSST Project
 * (https://www.lsst.org).
 * See the COPYRIGHT file at the top-level directory of this distribution
 * for details of code ownership.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
 
#include <Eigen/Dense>
#include <ndarray.h>
#include <cmath>
 
namespace lsst {
namespace meas {
namespace astrom {

struct PatternResult {
    PatternResult() : candidate_pairs(), shift_rot_matrix(), cos_shift(), sin_rot(), success(false) {}
    PatternResult(std::vector<std::pair<uint16_t, uint16_t>> const& candidate_pairs,
                  Eigen::Matrix3d const& shift_rot_matrix, double cos_shift, double sin_rot)
            : candidate_pairs(candidate_pairs),
              shift_rot_matrix(shift_rot_matrix),
              cos_shift(cos_shift),
              sin_rot(sin_rot),
              success(true) {}

    std::vector<std::pair<uint16_t, uint16_t>> candidate_pairs;
    Eigen::Matrix3d shift_rot_matrix;
    double cos_shift;
    double sin_rot;
    bool success;
};

struct RotationTestResult {
    RotationTestResult() : cos_rot_sq(), proj_ref_ctr_delta(), shift_matrix(), success(false) {}
    RotationTestResult(double cos_rot_sq, Eigen::Vector3d const& proj_ref_ctr_delta,
                       Eigen::Matrix3d const& shift_matrix)
            : cos_rot_sq(cos_rot_sq),
              proj_ref_ctr_delta(proj_ref_ctr_delta),
              shift_matrix(shift_matrix),
              success(true) {}

    double cos_rot_sq;
    Eigen::Vector3d proj_ref_ctr_delta;
    Eigen::Matrix3d shift_matrix;
    bool success;
};

struct SortedArrayResult {
    std::vector<float> dists;
    std::vector<uint16_t> ids;
};

struct ShiftRotMatrixResult {
    double sin_rot;
    Eigen::Matrix3d shift_rot_matrix;
};

PatternResult construct_pattern_and_shift_rot_matrix(
        ndarray::Array<double, 2, 1> src_pattern_array, ndarray::Array<double, 2, 1> src_delta_array,
        ndarray::Array<double, 1, 1> src_dist_array, ndarray::Array<float, 1, 1> dist_array,
        ndarray::Array<uint16_t, 2, 1> id_array, ndarray::Array<double, 2, 1> reference_array, size_t n_match,
        double max_cos_theta_shift, double max_cos_rot_sq, double max_dist_rad);


std::pair<size_t, size_t> find_candidate_reference_pair_range(
        float src_dist, ndarray::Array<float, 1, 1> const& ref_dist_array, double max_dist_rad);
std::pair<size_t, size_t> find_candidate_reference_pair_range(float src_dist,
                                                              std::vector<float> const& ref_dist_array,
                                                              double max_dist_rad);

RotationTestResult test_rotation(ndarray::Array<double, 1, 1> const& src_center,
                                 ndarray::Array<double, 1, 1> const& ref_center,
                                 ndarray::Array<double, 1, 1> const& src_delta,
                                 ndarray::Array<double, 1, 1> const& ref_delta, double cos_shift,
                                 double max_cos_rot_sq);

SortedArrayResult create_sorted_arrays(ndarray::Array<double, 1, 1> const& ref_center,
                                       ndarray::Array<double, 2, 1> const& reference_array);

std::vector<std::pair<size_t, size_t>> create_pattern_spokes(
        ndarray::Array<double, 1, 1> const& src_ctr, ndarray::Array<double, 2, 1> const& src_delta_array,
        ndarray::Array<double, 1, 1> const& src_dist_array, ndarray::Array<double, 1, 1> const& ref_ctr,
        Eigen::Vector3d const& proj_ref_ctr_delta, std::vector<float> const& ref_dist_array,
        std::vector<uint16_t> const& ref_id_array, ndarray::Array<double, 2, 1> const& reference_array,
        double max_dist_rad, size_t n_match);

ShiftRotMatrixResult create_shift_rot_matrix(double cos_rot_sq, Eigen::Matrix3d const& shift_matrix,
                                             ndarray::Array<double, 1, 1> const& src_delta,
                                             ndarray::Array<double, 1, 1> const& ref_ctr,
                                             ndarray::Array<double, 1, 1> const& ref_delta);

Eigen::Matrix3d create_spherical_rotation_matrix(Eigen::Vector3d const& rot_axis, double cos_rotation,
                                                 double sin_rotion);

bool intermediate_verify_comparison(std::vector<Eigen::Vector3d> const& src_pattern,
                                    std::vector<Eigen::Vector3d> const& ref_pattern,
                                    Eigen::Matrix3d const& shift_rot_matrix, double max_dist_rad);

int check_spoke(double cos_theta_src, double sin_theta_src, ndarray::Array<double, 1, 1> const& ref_ctr,
                Eigen::Vector3d const& proj_ref_ctr_delta, double proj_ref_ctr_dist_sq,
                std::pair<size_t, size_t> const& candidate_range, std::vector<uint16_t> const& ref_id_array,
                ndarray::Array<double, 2, 1> const& reference_array, double src_sin_tol);
 
}  // namespace astrom
}  // namespace meas
}  // namespace lsst