File: MinimizerFactory.h

package info (click to toggle)
ausaxs 1.1.8-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 72,592 kB
  • sloc: cpp: 49,853; ansic: 6,901; python: 730; makefile: 18
file content (85 lines) | stat: -rw-r--r-- 3,301 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
// SPDX-License-Identifier: LGPL-3.0-or-later
// Author: Kristian Lytje

#pragma once

#include <mini/dlibMinimizer.h>
#include <mini/Golden.h>
#include <mini/MinimumExplorer.h>
#include <mini/Scan.h>
#include <mini/LimitedScan.h>

#include <memory>
#include <functional>

namespace ausaxs::mini {
    namespace detail {
        inline std::shared_ptr<Minimizer> create_minimizer(algorithm t) {
            switch (t) {
                #if defined(DLIB_AVAILABLE)
                    case algorithm::DLIB_GLOBAL:
                        return std::make_shared<dlibMinimizer<algorithm::DLIB_GLOBAL>>();
                    case algorithm::BFGS:
                        return std::make_shared<dlibMinimizer<algorithm::BFGS>>();
                #endif
                case algorithm::GOLDEN:
                    return std::make_shared<Golden>();
                case algorithm::MINIMUM_EXPLORER:
                    return std::make_shared<MinimumExplorer>();
                case algorithm::SCAN:
                    return std::make_shared<Scan>();
                case algorithm::LIMITED_SCAN:
                    return std::make_shared<LimitedScan>();
                default:
                    throw std::invalid_argument("all::create_minimizer: Unknown minimizer type.");
            }
        }

        [[maybe_unused]] inline std::shared_ptr<Minimizer> create_minimizer(algorithm t, std::function<double(std::vector<double>)>&& func) {
            auto minimizer = create_minimizer(t);
            minimizer->set_function(std::move(func));
            return minimizer;
        }
    }

    /**
     * @brief Create a new minimizer.
     *
     * @param t The algorithm to use.
     * @param func The function to minimize.
     * @param param The first parameter.
     */
    [[maybe_unused]] inline std::shared_ptr<Minimizer> create_minimizer(algorithm t, std::function<double(std::vector<double>)> func, const Parameter& param) {
        auto minimizer = detail::create_minimizer(t, std::move(func));
        minimizer->add_parameter(param);
        return minimizer;
    }

    /**
     * @brief Create a new minimizer.
     *
     * @param t The algorithm to use.
     * @param func The function to minimize.
     * @param param The parameter list.
     */
    [[maybe_unused]] inline std::shared_ptr<Minimizer> create_minimizer(algorithm t, std::function<double(std::vector<double>)> func, const std::vector<Parameter>& param) {
        auto minimizer = detail::create_minimizer(t, std::move(func));
        std::for_each(param.begin(), param.end(), [&](const Parameter& p) { minimizer->add_parameter(p); });
        return minimizer;
    }

    /**
     * @brief Create a new minimizer.
     *
     * @param t The algorithm to use.
     * @param func The function to minimize.
     * @param param The first parameter.
     * @param evals The number of evaluations to perform. Not supported by all minimizers.
     */
    [[maybe_unused]] inline std::shared_ptr<Minimizer> create_minimizer(algorithm t, std::function<double(std::vector<double>)> func, const Parameter& param, unsigned int evals) {
        auto minimizer = detail::create_minimizer(t, std::move(func));
        minimizer->add_parameter(param);
        minimizer->set_max_evals(evals);
        return minimizer;
    }
}