ReconstructionOptionsWidget.cpp

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        CloudViewer: www.cloudViewer.org                  -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.cloudViewer.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#include "ReconstructionOptionsWidget.h"
 
#include "controllers/incremental_mapper.h"
 
namespace cloudViewer {
 
MapperGeneralOptionsWidget::MapperGeneralOptionsWidget(QWidget* parent,
                                                       OptionManager* options)
    : colmap::OptionsWidget(parent) {
    AddOptionBool(&options->mapper->multiple_models, "multiple_models");
    AddOptionInt(&options->mapper->max_num_models, "max_num_models");
    AddOptionInt(&options->mapper->max_model_overlap, "max_model_overlap");
    AddOptionInt(&options->mapper->min_model_size, "min_model_size");
    AddOptionBool(&options->mapper->extract_colors, "extract_colors");
    AddOptionInt(&options->mapper->num_threads, "num_threads", -1);
    AddOptionInt(&options->mapper->min_num_matches, "min_num_matches");
    AddOptionBool(&options->mapper->ignore_watermarks, "ignore_watermarks");
    AddOptionDirPath(&options->mapper->snapshot_path, "snapshot_path");
    AddOptionInt(&options->mapper->snapshot_images_freq, "snapshot_images_freq",
                 0);
}
 
MapperTriangulationOptionsWidget::MapperTriangulationOptionsWidget(
        QWidget* parent, OptionManager* options)
    : colmap::OptionsWidget(parent) {
    AddOptionInt(&options->mapper->triangulation.max_transitivity,
                 "max_transitivity");
    AddOptionDouble(&options->mapper->triangulation.create_max_angle_error,
                    "create_max_angle_error [deg]");
    AddOptionDouble(&options->mapper->triangulation.continue_max_angle_error,
                    "continue_max_angle_error [deg]");
    AddOptionDouble(&options->mapper->triangulation.merge_max_reproj_error,
                    "merge_max_reproj_error [px]");
    AddOptionDouble(&options->mapper->triangulation.re_max_angle_error,
                    "re_max_angle_error [deg]");
    AddOptionDouble(&options->mapper->triangulation.re_min_ratio,
                    "re_min_ratio");
    AddOptionInt(&options->mapper->triangulation.re_max_trials,
                 "re_max_trials");
    AddOptionDouble(&options->mapper->triangulation.complete_max_reproj_error,
                    "complete_max_reproj_error [px]");
    AddOptionInt(&options->mapper->triangulation.complete_max_transitivity,
                 "complete_max_transitivity");
    AddOptionDouble(&options->mapper->triangulation.min_angle,
                    "min_angle [deg]", 0, 180);
    AddOptionBool(&options->mapper->triangulation.ignore_two_view_tracks,
                  "ignore_two_view_tracks");
}
 
MapperRegistrationOptionsWidget::MapperRegistrationOptionsWidget(
        QWidget* parent, OptionManager* options)
    : colmap::OptionsWidget(parent) {
    AddOptionDouble(&options->mapper->mapper.abs_pose_max_error,
                    "abs_pose_max_error [px]");
    AddOptionInt(&options->mapper->mapper.abs_pose_min_num_inliers,
                 "abs_pose_min_num_inliers");
    AddOptionDouble(&options->mapper->mapper.abs_pose_min_inlier_ratio,
                    "abs_pose_min_inlier_ratio");
    AddOptionInt(&options->mapper->mapper.max_reg_trials, "max_reg_trials", 1);
}
 
MapperInitializationOptionsWidget::MapperInitializationOptionsWidget(
        QWidget* parent, OptionManager* options)
    : colmap::OptionsWidget(parent) {
    AddOptionInt(&options->mapper->init_image_id1, "init_image_id1", -1);
    AddOptionInt(&options->mapper->init_image_id2, "init_image_id2", -1);
    AddOptionInt(&options->mapper->init_num_trials, "init_num_trials");
    AddOptionInt(&options->mapper->mapper.init_min_num_inliers,
                 "init_min_num_inliers");
    AddOptionDouble(&options->mapper->mapper.init_max_error, "init_max_error");
    AddOptionDouble(&options->mapper->mapper.init_max_forward_motion,
                    "init_max_forward_motion");
    AddOptionDouble(&options->mapper->mapper.init_min_tri_angle,
                    "init_min_tri_angle [deg]");
    AddOptionInt(&options->mapper->mapper.init_max_reg_trials,
                 "init_max_reg_trials", 1);
}
 
MapperBundleAdjustmentOptionsWidget::MapperBundleAdjustmentOptionsWidget(
        QWidget* parent, OptionManager* options)
    : colmap::OptionsWidget(parent) {
    AddSection("Camera parameters");
    AddOptionBool(&options->mapper->ba_refine_focal_length,
                  "refine_focal_length");
    AddOptionBool(&options->mapper->ba_refine_principal_point,
                  "refine_principal_point");
    AddOptionBool(&options->mapper->ba_refine_extra_params,
                  "refine_extra_params");
 
    AddSpacer();
 
    AddSection("Local Bundle Adjustment");
    AddOptionInt(&options->mapper->ba_local_num_images, "num_images");
    AddOptionInt(&options->mapper->ba_local_max_num_iterations,
                 "max_num_iterations");
    AddOptionInt(&options->mapper->ba_local_max_refinements, "max_refinements",
                 1);
    AddOptionDouble(&options->mapper->ba_local_max_refinement_change,
                    "max_refinement_change", 0, 1, 1e-6, 6);
 
    AddSpacer();
 
    AddSection("Global Bundle Adjustment");
#ifdef PBA_ENABLED
    AddOptionBool(&options->mapper->ba_global_use_pba,
                  "use_pba\n(requires SIMPLE_RADIAL)");
    AddOptionInt(&options->mapper->ba_global_pba_gpu_index, "pba_gpu_index",
                 -1);
#endif
    AddOptionDouble(&options->mapper->ba_global_images_ratio, "images_ratio");
    AddOptionInt(&options->mapper->ba_global_images_freq, "images_freq");
    AddOptionDouble(&options->mapper->ba_global_points_ratio, "points_ratio");
    AddOptionInt(&options->mapper->ba_global_points_freq, "points_freq");
    AddOptionInt(&options->mapper->ba_global_max_num_iterations,
                 "max_num_iterations");
    AddOptionInt(&options->mapper->ba_global_max_refinements, "max_refinements",
                 1);
    AddOptionDouble(&options->mapper->ba_global_max_refinement_change,
                    "max_refinement_change", 0, 1, 1e-6, 6);
}
 
MapperFilteringOptionsWidget::MapperFilteringOptionsWidget(
        QWidget* parent, OptionManager* options)
    : colmap::OptionsWidget(parent) {
    AddOptionDouble(&options->mapper->min_focal_length_ratio,
                    "min_focal_length_ratio");
    AddOptionDouble(&options->mapper->max_focal_length_ratio,
                    "max_focal_length_ratio");
    AddOptionDouble(&options->mapper->max_extra_param, "max_extra_param");
 
    AddOptionDouble(&options->mapper->mapper.filter_max_reproj_error,
                    "filter_max_reproj_error [px]");
    AddOptionDouble(&options->mapper->mapper.filter_min_tri_angle,
                    "filter_min_tri_angle [deg]");
}
 
ReconstructionOptionsWidget::ReconstructionOptionsWidget(QWidget* parent,
                                                         OptionManager* options)
    : QWidget(parent) {
    setWindowFlags(Qt::Dialog);
    setWindowModality(Qt::ApplicationModal);
    setWindowTitle("Reconstruction options");
 
    QGridLayout* grid = new QGridLayout(this);
 
    QTabWidget* tab_widget = new QTabWidget(this);
    tab_widget->setElideMode(Qt::TextElideMode::ElideRight);
    tab_widget->addTab(new MapperGeneralOptionsWidget(this, options),
                       tr("General"));
    tab_widget->addTab(new MapperInitializationOptionsWidget(this, options),
                       tr("Init"));
    tab_widget->addTab(new MapperRegistrationOptionsWidget(this, options),
                       tr("Registration"));
    tab_widget->addTab(new MapperTriangulationOptionsWidget(this, options),
                       tr("Triangulation"));
    tab_widget->addTab(new MapperBundleAdjustmentOptionsWidget(this, options),
                       tr("Bundle"));
    tab_widget->addTab(new MapperFilteringOptionsWidget(this, options),
                       tr("Filter"));
 
    grid->addWidget(tab_widget, 0, 0);
}
 
}  // namespace cloudViewer