1 /***********************************************************************
2  * Copyright (c) 2013-2025 General Atomics Integrated Intelligence, Inc.
3  * All rights reserved. This program and the accompanying materials
4  * are made available under the terms of the Apache License, Version 2.0
5  * which accompanies this distribution and is available at
6  * https://www.apache.org/licenses/LICENSE-2.0
7  ***********************************************************************/
8 
9 package org.locationtech.geomesa.convert.all
10 
11 import com.typesafe.config.Config
12 import org.geotools.api.feature.simple.SimpleFeatureType
13 import org.locationtech.geomesa.convert.avro.AvroConverterFactory
14 import org.locationtech.geomesa.convert.json.JsonConverterFactory
15 import org.locationtech.geomesa.convert.parquet.ParquetConverterFactory
16 import org.locationtech.geomesa.convert.shp.ShapefileConverterFactory
17 import org.locationtech.geomesa.convert.text.DelimitedTextConverterFactory
18 import org.locationtech.geomesa.convert.xml.XmlConverterFactory
19 import org.locationtech.geomesa.convert2.{SimpleFeatureConverter, SimpleFeatureConverterFactory}
20 
21 import java.io.InputStream
22 import java.util.Locale
23 import scala.util.Try
24 
25 object TypeAwareInference {
26 
27   private val mappings = Map[String, Class[_ <: SimpleFeatureConverterFactory]](
28     "avro"    -> classOf[AvroConverterFactory],
29     "json"    -> classOf[JsonConverterFactory],
30     "csv"     -> classOf[DelimitedTextConverterFactory],
31     "tsv"     -> classOf[DelimitedTextConverterFactory],
32     "parquet" -> classOf[ParquetConverterFactory],
33     "shp"     -> classOf[ShapefileConverterFactory],
34     "xml"     -> classOf[XmlConverterFactory]
35   )
36 
37   /**
38    * Infer a converter based on a data sample
39    *
40    * @param format data format (e.g. csv, avro, json, etc)
41    * @param is input stream to convert
42    * @param sft simple feature type, if known
43    * @param hints implementation specific hints about the input
44    * @return
45    */
46   def infer(
47       format: String,
48       is: () => InputStream,
49       sft: Option[SimpleFeatureType],
50       hints: Map[String, AnyRef]): Try[(SimpleFeatureType, Config)] = {
51     val priority = mappings.get(format.toLowerCase(Locale.US)).map { clas =>
52       Ordering.by[SimpleFeatureConverterFactory, Boolean](f => !clas.isAssignableFrom(f.getClass)) // note: false sorts first
53     }
54     SimpleFeatureConverter.infer(is, sft, hints, priority)
55   }
56 }