FitParameterContainerItem.cpp

Go to the documentation of this file.

//  ************************************************************************************************
//
//  BornAgain: simulate and fit reflection and scattering
//
//! @file      GUI/Model/Job/FitParameterContainerItem.cpp
//! @brief     Implements class FitParameterContainerItem
//!
//! @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 "GUI/Model/Job/FitParameterContainerItem.h"
#include "Base/Util/Assert.h"
#include "Fit/Param/Parameters.h"
#include "GUI/Model/Job/FitParameterItem.h"
#include "GUI/Model/Job/FitParameterLinkItem.h"
#include "GUI/Model/Job/ParameterTreeItems.h"
#include "GUI/Model/Model/SessionModel.h"
#include "GUI/Util/Error.h"
 
FitParameterContainerItem::FitParameterContainerItem()
    : SessionItem(M_TYPE)
{
    registerTag(T_FIT_PARAMETERS, 0, -1, {FitParameterItem::M_TYPE});
    setDefaultTag(T_FIT_PARAMETERS);
}
 
void FitParameterContainerItem::createFitParameter(ParameterItem* parameterItem)
{
    ASSERT(parameterItem);
 
    removeLink(parameterItem);
 
    auto* fitPar = model()->insertItem<FitParameterItem>(this);
    fitPar->setDisplayName("par");
    fitPar->setStartValue(parameterItem->valueOfLink());
    fitPar->initMinMaxValues(parameterItem->limitsOfLink());
    fitPar->addLink(parameterItem->titleForFitItem(), parameterItem->link());
}
 
void FitParameterContainerItem::addToFitParameter(ParameterItem* parameterItem,
                                                  const QString& fitParName)
{
    removeLink(parameterItem);
    for (FitParameterItem* fitPar : fitParameterItems()) {
        if (fitPar->displayName() == fitParName) {
            fitPar->addLink(parameterItem->titleForFitItem(), parameterItem->link());
            break;
        }
    }
}
 
//! Returns FitParameterItem for given link (path in model)
 
FitParameterItem* FitParameterContainerItem::fitParameterItem(const QString& link) const
{
    for (auto* item : fitParameterItems())
        if (item->links().contains(link))
            return item;
 
    return nullptr;
}
 
FitParameterItem*
FitParameterContainerItem::fitParameterItem(const ParameterItem* parameterItem) const
{
    return fitParameterItem(parameterItem->link());
}
 
QVector<FitParameterItem*> FitParameterContainerItem::fitParameterItems() const
{
    return items<FitParameterItem>(T_FIT_PARAMETERS);
}
 
QStringList FitParameterContainerItem::fitParameterNames() const
{
    QStringList result;
    for (FitParameterItem* item : fitParameterItems())
        result.append(item->displayName());
    return result;
}
 
void FitParameterContainerItem::removeLink(const ParameterItem* parameterItem)
{
    if (FitParameterItem* fitParItem = fitParameterItem(parameterItem))
        fitParItem->removeLink(parameterItem->link());
}
 
bool FitParameterContainerItem::isEmpty()
{
    return fitParameterItems().isEmpty();
}
 
//! Propagate values to the corresponding parameter tree items of parameterContainer.
 
void FitParameterContainerItem::setValuesInParameterContainer(
    const QVector<double>& values, ParameterContainerItem* parameterContainer)
{
    ASSERT(parameterContainer);
 
    QVector<FitParameterItem*> fitPars = items<FitParameterItem>(T_FIT_PARAMETERS);
 
    int index(0);
    for (int i = 0; i < fitPars.size(); ++i) {
        auto link_list = fitPars[i]->linkItems();
        if (link_list.empty())
            continue;
        for (FitParameterLinkItem* linkItem : link_list)
            if (auto* itemInTuningTree = parameterContainer->findParameterItem(linkItem->link()))
                itemInTuningTree->propagateValueToLink(values[index]);
 
        index++;
    }
}
 
mumufit::Parameters FitParameterContainerItem::createParameters() const
{
    mumufit::Parameters result;
 
    int index(0);
    for (auto* fitPar : fitParameterItems()) {
        if (!fitPar->isValid()) {
            throw Error(
                QString("FitParameterContainerItem::createParameters(): invalid starting value "
                        "or (min, max) range in fitting parameter par %1")
                    .arg(index));
        }
        double startValue = fitPar->startValue();
        AttLimits limits = fitPar->attLimits();
        QString name = QString("par%1").arg(index);
        result.add(mumufit::Parameter(name.toStdString(), startValue, limits));
        ++index;
    }
 
    return result;
}