com.powsybl.metrix.mapping.GeneratorGroupTsData.groovy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of powsybl-metrix-mapping Show documentation

There is a newer version: 2.6.0

Show newest version

/*
 * Copyright (c) 2020, RTE (http://www.rte-france.com)
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 * SPDX-License-Identifier: MPL-2.0
 */
package com.powsybl.metrix.mapping

import com.powsybl.iidm.network.Generator
import com.powsybl.iidm.network.Identifiable
import groovy.transform.CompileStatic

/**
 * @author Marianne Funfrock {@literal }
 */
class GeneratorGroupTsData extends EquipmentGroupTsData {

    Boolean withPowerType

    void withPowerType(Boolean withPowerType) {
        this.withPowerType = withPowerType
    }

    @CompileStatic
    protected static void generatorGroupTimeSeries(Binding binding, TimeSeriesMappingConfigLoader configLoader,
                                                   Closure closure, Iterable filteringContexts,
                                                   MappableEquipmentType equipmentType,
                                                   LogDslLoader logDslLoader) {
        Closure cloned = (Closure) closure.clone()
        GeneratorGroupTsData spec = new GeneratorGroupTsData()
        cloned.delegate = spec
        cloned()

        if (spec.group == null) {
            throw new TimeSeriesMappingException("provideGroupTsGenerators must define group")
        }

        // evaluate equipment filters for each variable
        Collection filteredEquipments = Filter.evaluate(binding, filteringContexts, equipmentType.scriptVariable, spec.filter)

        // for each filtered equipment, add it to the equipment time series config
        filteredEquipments.forEach({ Identifiable identifiable ->
            configLoader.addGroupGeneratorTimeSeries((Generator) identifiable, spec.group, spec.withPowerType, spec.name)
        })
    }
}