com.powsybl.afs.security.SubjectInfoExtensionSerializer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of powsybl-afs-security-analysis Show documentation
Security analysis runner AFS file, and security analysis service interface
The newest version!
/**
 * Copyright (c) 2018, 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.powsybl.afs.security;

import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.core.JsonToken;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.DeserializationContext;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.google.auto.service.AutoService;
import com.powsybl.commons.extensions.ExtensionJsonSerializer;
import com.powsybl.iidm.network.Country;
import com.powsybl.security.LimitViolation;

import java.io.IOException;
import java.util.Set;
import java.util.TreeSet;

/**
 * @author Geoffroy Jamgotchian {@literal }
 */
@AutoService(ExtensionJsonSerializer.class)
public class SubjectInfoExtensionSerializer implements ExtensionJsonSerializer {

    @Override
    public String getExtensionName() {
        return "SubjectInfo";
    }

    @Override
    public String getCategoryName() {
        return "security-analysis";
    }

    @Override
    public Class getExtensionClass() {
        return SubjectInfoExtension.class;
    }

    @Override
    public void serialize(SubjectInfoExtension extension, JsonGenerator jsonGenerator, SerializerProvider serializerProvider) throws IOException {
        jsonGenerator.writeStartObject();

        jsonGenerator.writeFieldName("countries");
        jsonGenerator.writeStartArray();
        for (Country country : extension.getCountries()) {
            jsonGenerator.writeString(country.name());
        }
        jsonGenerator.writeEndArray();

        jsonGenerator.writeFieldName("nominalVoltages");
        jsonGenerator.writeStartArray();
        for (double nominalVoltage : extension.getNominalVoltages()) {
            jsonGenerator.writeNumber(nominalVoltage);
        }
        jsonGenerator.writeEndArray();

        jsonGenerator.writeEndObject();
    }

    @Override
    public SubjectInfoExtension deserialize(JsonParser parser, DeserializationContext deserializationContext) throws IOException {
        Set countries = null;
        Set nominalVoltages = null;

        while (parser.nextToken() != JsonToken.END_OBJECT) {
            switch (parser.getCurrentName()) {
                case "countries" -> {
                    parser.nextToken();
                    countries = parser.readValueAs(new TypeReference>() {
                    });
                }
                case "nominalVoltages" -> {
                    parser.nextToken();
                    nominalVoltages = parser.readValueAs(new TypeReference>() {
                    });
                }
                default -> throw new AssertionError("Unexpected field: " + parser.getCurrentName());
            }
        }

        return new SubjectInfoExtension(countries, nominalVoltages);
    }
}