File: FitStatus.cpp

package info (click to toggle)
bornagain 23.0-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 103,936 kB
  • sloc: cpp: 423,131; python: 40,997; javascript: 11,167; awk: 630; sh: 318; ruby: 173; xml: 130; makefile: 51; ansic: 24
file content (83 lines) | stat: -rw-r--r-- 2,299 bytes parent folder | download | duplicates (2)
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
//  ************************************************************************************************
//
//  BornAgain: simulate and fit reflection and scattering
//
//! @file      Sim/Fitting/FitStatus.cpp
//! @brief     Implements class FitStatus.
//!
//! @homepage  http://www.bornagainproject.org
//! @license   GNU General Public License v3 or higher (see COPYING)
//! @copyright Forschungszentrum Jülich GmbH 2018
//! @authors   Scientific Computing Group at MLZ (see CITATION, AUTHORS)
//
//  ************************************************************************************************

#include "Sim/Fitting/FitStatus.h"
#include "Fit/Minimizer/MinimizerResult.h"
#include "Sim/Fitting/FitPrintService.h"
#include <memory>
#include <stdexcept>

FitStatus::FitStatus(const FitObjective* fit_objective)
    : m_code(IDLE)
    , m_fit_objective(fit_objective)
{
}

FitStatus::~FitStatus() = default;

void FitStatus::setInterrupted()
{
    m_code = INTERRUPTED;
}

bool FitStatus::isInterrupted() const
{
    return m_code == INTERRUPTED;
}

bool FitStatus::isCompleted() const
{
    return m_code == COMPLETED;
}

void FitStatus::update(const mumufit::Parameters& params, double chi2)
{
    if (!isInterrupted())
        m_code = RUNNING;

    m_iteration_info.update(params, chi2);

    m_observers.notify(*m_fit_objective);
}

void FitStatus::initPrint(int every_nth)
{
    m_print_service = std::make_unique<FitPrintService>();

    FitObserver<FitObjective>::observer_t callback = [&](const FitObjective& objective) {
        m_print_service->print(objective);
    };

    addObserver(every_nth, std::move(callback));
}

void FitStatus::addObserver(int every_nth, fit_observer_t&& observer)
{
    m_observers.addObserver(every_nth, std::move(observer));
}
mumufit::MinimizerResult FitStatus::minimizerResult() const
{
    if (!m_minimizer_result)
        throw std::runtime_error("FitStatus::minimizerResult -> Minimizer result wasn't set. "
                                 "Make sure that FitObjective::finalize() was called.");

    return {*m_minimizer_result};
}

void FitStatus::finalize(const mumufit::MinimizerResult& result)
{
    m_minimizer_result = std::make_unique<mumufit::MinimizerResult>(result);
    m_code = COMPLETED;
    m_observers.notify_all(*m_fit_objective);
}