123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134 |
- #ifndef CERES_PUBLIC_INTERNAL_VARIADIC_EVALUATE_H_
- #define CERES_PUBLIC_INTERNAL_VARIADIC_EVALUATE_H_
- #include <cstddef>
- #include <type_traits>
- #include <utility>
- #include "ceres/cost_function.h"
- #include "ceres/internal/parameter_dims.h"
- namespace ceres::internal {
- template <typename Functor, typename T, int... Indices>
- inline bool VariadicEvaluateImpl(const Functor& functor,
- T const* const* input,
- T* output,
- std::false_type ,
- std::integer_sequence<int, Indices...>) {
- static_assert(sizeof...(Indices) > 0,
- "Invalid number of parameter blocks. At least one parameter "
- "block must be specified.");
- return functor(input[Indices]..., output);
- }
- template <typename Functor, typename T>
- inline bool VariadicEvaluateImpl(const Functor& functor,
- T const* const* input,
- T* output,
- std::true_type ,
- std::integer_sequence<int>) {
- return functor(input, output);
- }
- template <typename ParameterDims, typename Functor, typename T>
- inline bool VariadicEvaluateImpl(const Functor& functor,
- T const* const* input,
- T* output,
- const void* ) {
- using ParameterBlockIndices =
- std::make_integer_sequence<int, ParameterDims::kNumParameterBlocks>;
- using IsDynamic = std::integral_constant<bool, ParameterDims::kIsDynamic>;
- return VariadicEvaluateImpl(
- functor, input, output, IsDynamic(), ParameterBlockIndices());
- }
- template <typename ParameterDims, typename Functor, typename T>
- inline bool VariadicEvaluateImpl(const Functor& functor,
- T const* const* input,
- T* output,
- const CostFunction* ) {
- return functor.Evaluate(input, output, nullptr);
- }
- template <typename ParameterDims, typename Functor, typename T>
- inline bool VariadicEvaluate(const Functor& functor,
- T const* const* input,
- T* output) {
- return VariadicEvaluateImpl<ParameterDims>(functor, input, output, &functor);
- }
- template <typename Functor>
- class FirstOrderFunctorAdapter {
- public:
- explicit FirstOrderFunctorAdapter(const Functor& functor)
- : functor_(functor) {}
- bool operator()(double const* const* parameters, double* cost) const {
- return functor_(*parameters, cost);
- }
- private:
- const Functor& functor_;
- };
- }
- #endif
|