io.github.baozi926.geo.fastjson.GeometryAsGeoJSONSerializer Maven / Gradle / Ivy

package io.github.baozi926.geo.fastjson;

import java.io.IOException;

/**
 * @author 蔡惠民
 * @date 2022/10/18 17:19
 */

//
// Source code recreated from a .class file by IntelliJ IDEA
// (powered by FernFlower decompiler)
//


import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;

import java.io.IOException;

import org.locationtech.jts.geom.Coordinate;
import org.locationtech.jts.geom.CoordinateSequence;
import org.locationtech.jts.geom.Geometry;
import org.locationtech.jts.geom.GeometryCollection;
import org.locationtech.jts.geom.LineString;
import org.locationtech.jts.geom.MultiLineString;
import org.locationtech.jts.geom.MultiPoint;
import org.locationtech.jts.geom.MultiPolygon;
import org.locationtech.jts.geom.Point;
import org.locationtech.jts.geom.Polygon;

public class GeometryAsGeoJSONSerializer extends JsonSerializer {
    public GeometryAsGeoJSONSerializer() {
    }

    protected void write(JsonGenerator gen, Coordinate coord) throws IOException {
        gen.writeStartArray();
        gen.writeNumber(coord.x);
        gen.writeNumber(coord.y);
        gen.writeEndArray();
    }

    protected void write(JsonGenerator gen, CoordinateSequence coordseq) throws IOException {
        gen.writeStartArray();
        int dim = coordseq.getDimension();

        for (int i = 0; i < coordseq.size(); ++i) {
            gen.writeStartArray();
            gen.writeNumber(coordseq.getOrdinate(i, 0));
            gen.writeNumber(coordseq.getOrdinate(i, 1));
            if (dim > 2) {
                double v = coordseq.getOrdinate(i, 2);
                if (!Double.isNaN(v)) {
                    gen.writeNumber(v);
                }
            }

            gen.writeEndArray();
        }

        gen.writeEndArray();
    }

    protected void write(JsonGenerator gen, Coordinate[] coord) throws IOException {
        gen.writeStartArray();

        for (int i = 0; i < coord.length; ++i) {
            this.write(gen, coord[i]);
        }

        gen.writeEndArray();
    }

    protected void write(JsonGenerator gen, Polygon p) throws IOException {
        gen.writeStartArray();
        this.write(gen, p.getExteriorRing().getCoordinateSequence());

        for (int i = 0; i < p.getNumInteriorRing(); ++i) {
            this.write(gen, p.getInteriorRingN(i).getCoordinateSequence());
        }

        gen.writeEndArray();
    }

    public void serialize(Geometry geom, JsonGenerator gen, SerializerProvider serializers) throws IOException, JsonProcessingException {
        gen.writeStartObject();
        gen.writeFieldName("type");
        gen.writeString(geom.getClass().getSimpleName());
        if (geom instanceof Point) {
            Point v = (Point) geom;
            gen.writeFieldName("coordinates");
            this.write(gen, v.getCoordinate());
        } else if (geom instanceof Polygon) {
            gen.writeFieldName("coordinates");
            this.write(gen, (Polygon) geom);
        } else if (geom instanceof LineString) {
            LineString v = (LineString) geom;
            gen.writeFieldName("coordinates");
            this.write(gen, v.getCoordinateSequence());
        } else {
            if (geom instanceof MultiPoint) {
                MultiPoint v = (MultiPoint) geom;
                gen.writeFieldName("coordinates");
                this.write(gen, v.getCoordinates());
                return;
            }

            int i;
            if (geom instanceof MultiLineString) {
                MultiLineString v = (MultiLineString) geom;
                gen.writeFieldName("coordinates");
                gen.writeStartArray();

                for (i = 0; i < v.getNumGeometries(); ++i) {
                    this.write(gen, v.getGeometryN(i).getCoordinates());
                }

                gen.writeEndArray();
            } else if (geom instanceof MultiPolygon) {
                MultiPolygon v = (MultiPolygon) geom;
                gen.writeFieldName("coordinates");
                gen.writeStartArray();

                for (i = 0; i < v.getNumGeometries(); ++i) {
                    this.write(gen, (Polygon) v.getGeometryN(i));
                }

                gen.writeEndArray();
            } else {
                if (!(geom instanceof GeometryCollection)) {
                    throw new UnsupportedOperationException("unknown: " + geom);
                }

                GeometryCollection v = (GeometryCollection) geom;
                gen.writeFieldName("geometries");
                gen.writeStartArray();

                for (i = 0; i < v.getNumGeometries(); ++i) {
                    this.serialize(v.getGeometryN(i), gen, serializers);
                }

                gen.writeEndArray();
            }
        }

        gen.writeEndObject();
    }
}