com.powsybl.dataframe.network.extensions.CgmesMetadataModelDataframeAdder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pypowsybl Show documentation
A C interface to powsybl, for pypowsybl implementation
The newest version!
package com.powsybl.dataframe.network.extensions;

import com.powsybl.cgmes.extensions.CgmesMetadataModelsAdder;
import com.powsybl.cgmes.model.CgmesSubset;
import com.powsybl.dataframe.SeriesMetadata;
import com.powsybl.dataframe.network.adders.AbstractSimpleAdder;
import com.powsybl.dataframe.network.adders.SeriesUtils;
import com.powsybl.dataframe.update.IntSeries;
import com.powsybl.dataframe.update.StringSeries;
import com.powsybl.dataframe.update.UpdatingDataframe;
import com.powsybl.iidm.network.*;

import java.util.*;

/**
 * @author Naledi El Cheikh 
 */

public class CgmesMetadataModelDataframeAdder extends AbstractSimpleAdder {
    private static final List METADATA = List.of(
            SeriesMetadata.stringIndex("id"),
            SeriesMetadata.strings("cgmes_subset"),
            SeriesMetadata.strings("description"),
            SeriesMetadata.ints("version"),
            SeriesMetadata.strings("modeling_authority_set"),
            SeriesMetadata.strings("profiles"),
            SeriesMetadata.strings("dependent_on"),
            SeriesMetadata.strings("supersedes")
    );

    @Override
    public List> getMetadata() {
        return Collections.singletonList(METADATA);
    }

    private static final class CgmesMetadataSeries {
        private final StringSeries id;
        private final StringSeries subset;
        private final StringSeries description;
        private final IntSeries version;
        private final StringSeries modelingAuthoritySet;
        private final StringSeries profiles;
        private final StringSeries dependentOn;
        private final StringSeries supersedes;

        private CgmesMetadataSeries(UpdatingDataframe dataframe) {
            this.id = dataframe.getStrings("id");
            this.subset = dataframe.getStrings("cgmes_subset");
            this.description = dataframe.getStrings("description");
            this.version = dataframe.getInts("version");
            this.modelingAuthoritySet = dataframe.getStrings("modeling_authority_set");
            this.profiles = dataframe.getStrings("profiles");
            this.dependentOn = dataframe.getStrings("dependent_on");
            this.supersedes = dataframe.getStrings("supersedes");
        }

        void create(int row, CgmesMetadataModelsAdder adder) {
            CgmesMetadataModelsAdder.ModelAdder modelAdder = adder.newModel();
            SeriesUtils.applyIfPresent(id, row, modelAdder::setId);
            SeriesUtils.applyIfPresent(subset, row, subset -> modelAdder.setSubset(CgmesSubset.valueOf(subset)));
            SeriesUtils.applyIfPresent(description, row, modelAdder::setDescription);
            SeriesUtils.applyIfPresent(version, row, modelAdder::setVersion);
            SeriesUtils.applyIfPresent(modelingAuthoritySet, row, modelAdder::setModelingAuthoritySet);
            SeriesUtils.applyIfPresent(profiles, row, profile -> parseProfiles(profile, modelAdder));
            SeriesUtils.applyIfPresent(dependentOn, row, dependentOn -> parseDependentOn(dependentOn, modelAdder));
            SeriesUtils.applyIfPresent(supersedes, row, supersedes -> parseSupersedes(supersedes, modelAdder));
            modelAdder.add();

        }

        private void parseProfiles(String concatenatedProfiles, CgmesMetadataModelsAdder.ModelAdder modelAdder) {
            List profilesList = new ArrayList<>(Arrays.asList(concatenatedProfiles.split(",")));
            for (String profile : profilesList) {
                modelAdder.addProfile(profile);
            }
        }

        private void parseDependentOn(String concatenatedDependentOn, CgmesMetadataModelsAdder.ModelAdder modelAdder) {
            List dependentOnList = new ArrayList<>(Arrays.asList(concatenatedDependentOn.split(",")));
            for (String dependentOn : dependentOnList) {
                modelAdder.addDependentOn(dependentOn);
            }
        }

        private void parseSupersedes(String concatenatedSupersedes, CgmesMetadataModelsAdder.ModelAdder modelAdder) {
            List supersedesList = new ArrayList<>(Arrays.asList(concatenatedSupersedes.split(",")));
            for (String supersedes : supersedesList) {
                modelAdder.addSupersedes(supersedes);
            }
        }

    }

    @Override
    public void addElements(Network network, UpdatingDataframe dataframe) {
        CgmesMetadataModelsAdder adder = network.newExtension(CgmesMetadataModelsAdder.class);
        CgmesMetadataSeries series = new CgmesMetadataSeries(dataframe);
        for (int row = 0; row < dataframe.getRowCount(); row++) {
            series.create(row, adder);
        }
        adder.add();
    }

}