Source code for oemof.solph.constraints

# -*- coding: utf-8 -*-

"""Additional constraints to be used in an oemof energy model.
This file is part of project oemof ( It's copyrighted
by the contributors recorded in the version control history of the file,
available from its original location oemof/oemof/solph/

SPDX-License-Identifier: GPL-3.0-or-later

import pyomo.environ as po

[docs]def investment_limit(model, limit=None): """ Set an absolute limit for the total investment costs of an investment optimization problem: .. math:: \sum_{investment\_costs} \leq limit Parameters ---------- model : oemof.solph.Model Model to which the constraint is added limit : float Absolute limit of the investment (i.e. RHS of constraint) """ def investment_rule(m): expr = 0 if hasattr(m, "InvestmentFlow"): expr += m.InvestmentFlow.investment_costs if hasattr(m, "GenericInvestmentStorageBlock"): expr += m.GenericInvestmentStorageBlock.investment_costs return expr <= limit model.investment_limit = po.Constraint(rule=investment_rule) return model
[docs]def emission_limit(om, flows=None, limit=None): """Set a global limit for emissions. The emission attribute has to be added to every flow you want to take into account. Parameters ---------- om : oemof.solph.Model Model to which constraints are added. flows : dict Dictionary holding the flows that should be considered in constraint. Keys are (source, target) objects of the Flow. If no dictionary is given all flows containing the 'emission' attribute will be used. limit : numeric Absolute emission limit. Note ---- Flow objects required an emission attribute! """ if flows is None: flows = {} for (i, o) in om.flows: if hasattr(om.flows[i, o], 'emission'): flows[(i, o)] = om.flows[i, o] else: for (i, o) in flows: if not hasattr(flows[i, o], 'emission'): raise ValueError(('Flow with source: {0} and target: {1} ' 'has no attribute emission.').format(i.label, o.label)) def emission_rule(m): return (sum(m.flow[inflow, outflow, t] * m.timeincrement[t] * flows[inflow, outflow].emission for (inflow, outflow) in flows for t in m.TIMESTEPS) <= limit) om.emission_limit = po.Constraint(rule=emission_rule) return om
[docs]def equate_variables(model, var1, var2, factor1=1, name=None): r""" Adds a constraint to the given model that set two variables to equal adaptable by a factor. **The following constraints are build:** .. math:: var\textit{1} \cdot factor\textit{1} = var\textit{2} Parameters ---------- var1 : pyomo.environ.Var First variable, to be set to equal with Var2 and multiplied with factor1. var2 : pyomo.environ.Var Second variable, to be set equal to (Var1 * factor1). factor1 : float Factor to define the proportion between the variables. name : str Optional name for the equation e.g. in the LP file. By default the name is: equate + string representation of var1 and var2. model : oemof.solph.Model Model to which the constraint is added. Examples -------- The following example shows how to define a transmission line in the investment mode by connecting both investment variables. Note that the equivalent periodical costs (epc) of the line are 40. You could also add them to one line and set them to 0 for the other line. >>> import pandas as pd >>> from oemof import solph >>> date_time_index = pd.date_range('1/1/2012', periods=5, freq='H') >>> energysystem = solph.EnergySystem(timeindex=date_time_index) >>> bel1 = solph.Bus(label='electricity1') >>> bel2 = solph.Bus(label='electricity2') >>> energysystem.add(bel1, bel2) >>> energysystem.add(solph.Transformer( ... label='powerline_1_2', ... inputs={bel1: solph.Flow()}, ... outputs={bel2: solph.Flow( ... investment=solph.Investment(ep_costs=20))})) >>> energysystem.add(solph.Transformer( ... label='powerline_2_1', ... inputs={bel2: solph.Flow()}, ... outputs={bel1: solph.Flow( ... investment=solph.Investment(ep_costs=20))})) >>> om = solph.Model(energysystem) >>> line12 = energysystem.groups['powerline_1_2'] >>> line21 = energysystem.groups['powerline_2_1'] >>> solph.constraints.equate_variables( ... om, ... om.InvestmentFlow.invest[line12, bel2], ... om.InvestmentFlow.invest[line21, bel1]) """ if name is None: name = '_'.join(["equate", str(var1), str(var2)]) def equate_variables_rule(m): return var1 * factor1 == var2 setattr(model, name, po.Constraint(rule=equate_variables_rule))