com.farao_community.farao.gridcapa_swe_commons.shift.SweD2ccShiftDispatcher Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gridcapa-swe-commons Show documentation

Common lib to gridcapa swe process

The newest version!

/*
 * Copyright (c) 2023, 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/.
 */
package com.farao_community.farao.gridcapa_swe_commons.shift;

import com.farao_community.farao.dichotomy.shift.ShiftDispatcher;
import com.farao_community.farao.gridcapa_swe_commons.dichotomy.DichotomyDirection;
import com.farao_community.farao.gridcapa_swe_commons.exception.SweInvalidDataException;
import com.farao_community.farao.gridcapa_swe_commons.resource.SweEICode;

import java.util.Map;
import java.util.TreeMap;
import java.util.stream.Collectors;

/**
 * @author Ameni Walha {@literal }
 */
public class SweD2ccShiftDispatcher implements ShiftDispatcher {
    private final Map initialNetPositions;
    private final DichotomyDirection direction;

    public SweD2ccShiftDispatcher(DichotomyDirection direction, Map initialNetPositions) {
        this.direction = direction;
        this.initialNetPositions = initialNetPositions;
    }

    @Override
    public Map dispatch(double value) {
        Map factors = new TreeMap<>();
        switch (direction) {
            case ES_FR:
                factors.put(SweEICode.PT_EIC, 0.);
                factors.put(SweEICode.ES_EIC, 1.);
                factors.put(SweEICode.FR_EIC, -1.);
                break;
            case FR_ES:
                factors.put(SweEICode.PT_EIC, 0.);
                factors.put(SweEICode.ES_EIC, -1.);
                factors.put(SweEICode.FR_EIC, 1.);
                break;
            case ES_PT:
                factors.put(SweEICode.PT_EIC, -1.);
                factors.put(SweEICode.ES_EIC, 1.);
                factors.put(SweEICode.FR_EIC, 0.);
                break;
            case PT_ES:
                factors.put(SweEICode.PT_EIC, 1.);
                factors.put(SweEICode.ES_EIC, -1.);
                factors.put(SweEICode.FR_EIC, 0.);
                break;
            default:
                throw new SweInvalidDataException(String.format("Unknown dichotomy direction for SWE: %s", direction));
        }
        return factors.entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, e -> e.getValue() * value - initialNetPositions.get(e.getKey())));
    }
}