/*
    * Copyright 2022 the original author or authors.
    *
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    *      http://www.apache.org/licenses/LICENSE-2.0
    *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  
  package org.bremersee.geojson.converter.serialization;
  
  import static java.util.Collections.unmodifiableList;
  import static java.util.Collections.unmodifiableMap;
  import static java.util.Objects.isNull;
  
  import java.io.Serial;
  import java.io.Serializable;
  import java.util.ArrayList;
  import java.util.Arrays;
  import java.util.LinkedHashMap;
  import java.util.List;
  import java.util.Map;
  import java.util.Optional;
  import java.util.stream.Collectors;
  import org.bremersee.geojson.GeoJsonConstants;
  import org.bremersee.geojson.GeoJsonGeometryFactory;
  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;
  
  /**
   * The geometry to json converter.
   *
   * @author Christian Bremer
   */
  public class GeometryToJsonConverter implements Serializable {
  
    @Serial
    private static final long serialVersionUID = 1L;
  
    /**
     * The point converter.
     */
    private final PointToJsonConverter pointConverter;
  
    /**
     * The Line string converter.
     */
    private final LineStringToJsonConverter lineStringConverter;
  
    /**
     * The Polygon converter.
     */
    private final PolygonToJsonConverter polygonConverter;
  
    /**
     * The Multi point converter.
     */
    private final MultiPointToJsonConverter multiPointConverter;
  
    /**
     * The Multi line string converter.
     */
    private final MultiLineStringToJsonConverter multiLineStringConverter;
  
    /**
     * The Multi polygon converter.
     */
    private final MultiPolygonToJsonConverter multiPolygonConverter;
  
    /**
     * Specifies whether a bounding box should be generated or not.
     */
    private final boolean withBoundingBox;
  
    /**
     * Instantiates a new geometry to json converter.
     */
    public GeometryToJsonConverter() {
      this(false, false);
    }
  
    /**
     * Instantiates a new geometry to json converter.
     *
     * @param withBoundingBox with bounding box
     * @param useBigDecimal use big decimal
    */
   public GeometryToJsonConverter(boolean withBoundingBox, boolean useBigDecimal) {
     CoordinateToListConverter coordinateConverter = new CoordinateToListConverter(useBigDecimal);
     CoordinateSequenceToListConverter coordinateSequenceConverter
         = new CoordinateSequenceToListConverter(coordinateConverter);
 
     pointConverter = new PointToJsonConverter(coordinateConverter, withBoundingBox);
     lineStringConverter = new LineStringToJsonConverter(
         coordinateSequenceConverter,
         withBoundingBox);
     polygonConverter = new PolygonToJsonConverter(coordinateSequenceConverter, withBoundingBox);
 
     multiPointConverter = new MultiPointToJsonConverter(pointConverter, withBoundingBox);
     multiLineStringConverter = new MultiLineStringToJsonConverter(
         lineStringConverter,
         withBoundingBox);
     multiPolygonConverter = new MultiPolygonToJsonConverter(polygonConverter, withBoundingBox);
 
     this.withBoundingBox = withBoundingBox;
   }
 
   /**
    * Convert map.
    *
    * @param source the source
    * @return the map
    */
   public Map<String, Object> convert(Geometry source) {
     if (isNull(source)) {
       return null;
     }
     if (source instanceof Point) {
       return pointConverter.convert((Point) source);
     }
 
     if (source instanceof LineString) {
       return lineStringConverter.convert((LineString) source);
     }
 
     if (source instanceof Polygon) {
       return polygonConverter.convert((Polygon) source);
     }
 
     if (source instanceof MultiPoint) {
       return multiPointConverter.convert((MultiPoint) source);
     }
 
     if (source instanceof MultiLineString) {
       return multiLineStringConverter.convert((MultiLineString) source);
     }
 
     if (source instanceof MultiPolygon) {
       return multiPolygonConverter.convert((MultiPolygon) source);
     }
 
     if (source instanceof GeometryCollection) {
       List<Map<String, Object>> geometries = new ArrayList<>(source.getNumGeometries());
       for (int i = 0; i < source.getNumGeometries(); i++) {
         geometries.add(convert(source.getGeometryN(i)));
       }
       Map<String, Object> map = new LinkedHashMap<>();
       map.put(GeoJsonConstants.TYPE, GeoJsonConstants.GEOMETRY_COLLECTION);
       if (withBoundingBox) {
         Optional.ofNullable(GeoJsonGeometryFactory.getBoundingBox(source))
             .map(bbox -> Arrays.stream(bbox).boxed().collect(Collectors.toList()))
             .ifPresent(bbox -> map.put(GeoJsonConstants.BBOX, bbox));
       }
       map.put(GeoJsonConstants.GEOMETRIES, unmodifiableList(geometries));
       return unmodifiableMap(map);
     }
 
     throw new IllegalArgumentException("Geometry [" + source + "] is unsupported. It must be "
         + "an instance of Point, LineString, Polygon, MultiPoint, MultiLineString, MultiPolygon "
         + "or GeometryCollection.");
   }
 }