blob: 47112b88fd8e8efb79af8fb892576b785d3ca117 [file] [log] [blame]
Austin Schuh70cc9552019-01-21 19:46:48 -08001// Ceres Solver - A fast non-linear least squares minimizer
2// Copyright 2015 Google Inc. All rights reserved.
3// http://ceres-solver.org/
4//
5// Redistribution and use in source and binary forms, with or without
6// modification, are permitted provided that the following conditions are met:
7//
8// * Redistributions of source code must retain the above copyright notice,
9// this list of conditions and the following disclaimer.
10// * Redistributions in binary form must reproduce the above copyright notice,
11// this list of conditions and the following disclaimer in the documentation
12// and/or other materials provided with the distribution.
13// * Neither the name of Google Inc. nor the names of its contributors may be
14// used to endorse or promote products derived from this software without
15// specific prior written permission.
16//
17// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
18// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
21// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
22// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
23// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
25// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
26// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27// POSSIBILITY OF SUCH DAMAGE.
28//
29// Author: sameeragarwal@google.com (Sameer Agarwal)
30
31#ifndef CERES_INTERNAL_CALLBACKS_H_
32#define CERES_INTERNAL_CALLBACKS_H_
33
34#include <string>
Austin Schuh1d1e6ea2020-12-23 21:56:30 -080035
Austin Schuh70cc9552019-01-21 19:46:48 -080036#include "ceres/internal/port.h"
Austin Schuh1d1e6ea2020-12-23 21:56:30 -080037#include "ceres/iteration_callback.h"
Austin Schuh70cc9552019-01-21 19:46:48 -080038
39namespace ceres {
40namespace internal {
41
42class Program;
43
44// Callback for updating the externally visible state of parameter
45// blocks.
46class StateUpdatingCallback : public IterationCallback {
47 public:
48 StateUpdatingCallback(Program* program, double* parameters);
49 virtual ~StateUpdatingCallback();
Austin Schuh1d1e6ea2020-12-23 21:56:30 -080050 CallbackReturnType operator()(const IterationSummary& summary) final;
51
Austin Schuh70cc9552019-01-21 19:46:48 -080052 private:
53 Program* program_;
54 double* parameters_;
55};
56
57// Callback for updating the externally visible state of the
58// parameters vector for GradientProblemSolver.
59class GradientProblemSolverStateUpdatingCallback : public IterationCallback {
60 public:
61 GradientProblemSolverStateUpdatingCallback(int num_parameters,
62 const double* internal_parameters,
63 double* user_parameters);
64 virtual ~GradientProblemSolverStateUpdatingCallback();
Austin Schuh1d1e6ea2020-12-23 21:56:30 -080065 CallbackReturnType operator()(const IterationSummary& summary) final;
66
Austin Schuh70cc9552019-01-21 19:46:48 -080067 private:
68 int num_parameters_;
69 const double* internal_parameters_;
70 double* user_parameters_;
71};
72
73// Callback for logging the state of the minimizer to STDERR or
74// STDOUT depending on the user's preferences and logging level.
75class LoggingCallback : public IterationCallback {
76 public:
77 LoggingCallback(MinimizerType minimizer_type, bool log_to_stdout);
78 virtual ~LoggingCallback();
Austin Schuh1d1e6ea2020-12-23 21:56:30 -080079 CallbackReturnType operator()(const IterationSummary& summary) final;
Austin Schuh70cc9552019-01-21 19:46:48 -080080
81 private:
82 const MinimizerType minimizer_type;
83 const bool log_to_stdout_;
84};
85
86} // namespace internal
87} // namespace ceres
88
89#endif // CERES_INTERNAL_CALLBACKS_H_