model.h

/*
** Copyright 2007-2024, RTE (https://www.rte-france.com)
** See AUTHORS.txt
** SPDX-License-Identifier: MPL-2.0
** This file is part of Antares-Simulator,
** Adequacy and Performance assessment for interconnected energy networks.
**
** Antares_Simulator is free software: you can redistribute it and/or modify
** it under the terms of the Mozilla Public Licence 2.0 as published by
** the Mozilla Foundation, either version 2 of the License, or
** (at your option) any later version.
**
** Antares_Simulator is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
** Mozilla Public Licence 2.0 for more details.
**
** You should have received a copy of the Mozilla Public Licence 2.0
** along with Antares_Simulator. If not, see <https://opensource.org/license/mpl-2-0/>.
*/
#pragma once
 
#include <map>
#include <vector>
 
#include <antares/expressions/expression.h>
 
#include "constraint.h"
#include "parameter.h"
#include "port.h"
#include "portFieldDefinition.h"
#include "variable.h"
 
namespace Antares::Study::SystemModel
{
 
// TODO: add unit tests for this class
class Model
{
public:
    Model() = default;
    Model(Model&&) = default;
    Model(const Model&) = delete;
    ~Model() = default;
 
    Model& operator=(Model&&) = default;
    Model& operator=(const Model&) = delete;
 
    const std::string& Id() const
    {
        return id_;
    }
 
    const Expression& Objective() const
    {
        return objective_;
    }
 
    const std::map<std::string, Constraint>& getConstraints() const
    {
        return constraints_;
    }
 
    const std::map<std::string, Parameter>& Parameters() const
    {
        return parameters_;
    }
 
    const std::map<std::string, Variable>& Variables() const
    {
        // TODO : convert to vector?
        return variables_;
    }
 
    const std::map<std::string, Port>& Ports() const
    {
        return ports_;
    }
 
    const std::map<std::string, PortFieldDefinition>& PortFieldDefinitions() const
    {
        return portFieldDefinitions_;
    }
 
private:
    friend class ModelBuilder;
    std::string id_;
    Expression objective_;
 
    std::map<std::string, Parameter> parameters_;
    std::map<std::string, Variable> variables_;
    std::map<std::string, Constraint> constraints_;
    std::map<std::string, Port> ports_;
    std::map<std::string, PortFieldDefinition> portFieldDefinitions_;
};
 
class ModelBuilder
{
public:
    ModelBuilder& withId(std::string_view id);
    ModelBuilder& withObjective(Expression&& objective);
    ModelBuilder& withParameters(std::vector<Parameter>&& parameters);
    ModelBuilder& withVariables(std::vector<Variable>&& variables);
    ModelBuilder& withPorts(std::vector<Port>&& ports);
    Model build();
 
    ModelBuilder& withConstraints(std::vector<Constraint>&& constraints);
    ModelBuilder& withPortFieldDefinitions(std::vector<PortFieldDefinition>&& portFieldDefinitions);
 
private:
    Model model_;
};
 
} // namespace Antares::Study::SystemModel