com.powsybl.metrix.mapping.timeseries.EquipmentTimeSeriesMap 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) 2021, 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.timeseries;

import com.powsybl.iidm.network.Identifiable;
import com.powsybl.iidm.network.Network;
import com.powsybl.metrix.mapping.*;
import com.powsybl.timeseries.TimeSeriesTable;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import static com.powsybl.metrix.mapping.TimeSeriesMapper.indexMappingKey;

/**
 * @author Marianne Funfrock {@literal }
 */
public class EquipmentTimeSeriesMap {
    private final Map> equimentTimeSeries = new LinkedHashMap<>();

    public void init(EquipmentTimeSeriesMap map) {
        equimentTimeSeries.putAll(map.equimentTimeSeries);
    }

    public void convertToEquipmentTimeSeriesMap(Map> timeSerieMap, TimeSeriesTable table, Network network, TimeSeriesMappingConfig config) {
        equimentTimeSeries.clear();
        timeSerieMap.entrySet().forEach(timeSeries -> equimentTimeSeries
                .put(indexMappingKey(table, timeSeries.getKey()), mapEquipments(timeSeries.getKey(), timeSeries.getValue(), network, config)));
    }

    public List mapEquipments(MappingKey key, List equipmentIds, Network network, TimeSeriesMappingConfig config) {
        return equipmentIds.stream().map(equipmentId -> {
            Identifiable identifiable = getIdentifiable(network, equipmentId);
            DistributionKey distributionKey = config.getDistributionKey(new MappingKey(key.getMappingVariable(), equipmentId));
            return new MappedEquipment(identifiable, distributionKey);
        }).toList();
    }

    private static Identifiable getIdentifiable(Network network, String equipmentId) {
        Identifiable identifiable = network.getIdentifiable(equipmentId);
        // check equipment exists
        if (identifiable == null) {
            throw new TimeSeriesMappingException("'" + equipmentId + "' not found");
        }
        return identifiable;
    }

    public void addMappedEquipmentTimeSeries(IndexedMappingKey indexedMappingKey,
                                            List mappedEquipmentsTimeSeries) {
        equimentTimeSeries.put(indexedMappingKey, mappedEquipmentsTimeSeries);
    }

    public Map> getEquipmentTimeSeries() {
        return equimentTimeSeries;
    }

    public boolean isEmpty() {
        return equimentTimeSeries.isEmpty();
    }

    public void computeIfAbsent(IndexedMappingKey indexedMappingKey, MappedEquipment mappedEquipment) {
        equimentTimeSeries.computeIfAbsent(indexedMappingKey, i -> new ArrayList<>()).add(mappedEquipment);
    }
}