Examples with GeoPoint org.opensearch.common.geo.GeoPoint used on opensource projects

Example 46 with GeoPoint

use of org.opensearch.common.geo.GeoPoint in project OpenSearch by opensearch-project.

the class AbstractGeoFieldDataTestCase method assertValues.

private void assertValues(MultiGeoPointValues values, int docId, boolean missing) throws IOException {
    assertEquals(missing == false, values.advanceExact(docId));
    if (missing == false) {
        final int docCount = values.docValueCount();
        for (int i = 0; i < docCount; ++i) {
            final GeoPoint point = values.nextValue();
            assertThat(point.lat(), allOf(greaterThanOrEqualTo(GeoUtils.MIN_LAT), lessThanOrEqualTo(GeoUtils.MAX_LAT)));
            assertThat(point.lon(), allOf(greaterThanOrEqualTo(GeoUtils.MIN_LON), lessThanOrEqualTo(GeoUtils.MAX_LON)));
        }
    }
}

Example 47 with GeoPoint

use of org.opensearch.common.geo.GeoPoint in project OpenSearch by opensearch-project.

the class ScriptDocValuesGeoPointsTests method testGeoGetLatLon.

public void testGeoGetLatLon() throws IOException {
    final double lat1 = randomLat();
    final double lat2 = randomLat();
    final double lon1 = randomLon();
    final double lon2 = randomLon();
    GeoPoint[][] points = { { new GeoPoint(lat1, lon1), new GeoPoint(lat2, lon2) } };
    final MultiGeoPointValues values = wrap(points);
    final ScriptDocValues.GeoPoints script = new ScriptDocValues.GeoPoints(values);
    script.setNextDocId(1);
    assertEquals(true, script.isEmpty());
    script.setNextDocId(0);
    assertEquals(false, script.isEmpty());
    assertEquals(new GeoPoint(lat1, lon1), script.getValue());
    assertEquals(lat1, script.getLat(), 0);
    assertEquals(lon1, script.getLon(), 0);
    assertTrue(Arrays.equals(new double[] { lat1, lat2 }, script.getLats()));
    assertTrue(Arrays.equals(new double[] { lon1, lon2 }, script.getLons()));
}

Example 48 with GeoPoint

use of org.opensearch.common.geo.GeoPoint in project OpenSearch by opensearch-project.

the class GeoPointFieldMapperTests method testNullValue.

public void testNullValue() throws Exception {
    DocumentMapper mapper = createDocumentMapper(fieldMapping(b -> b.field("type", "geo_point")));
    Mapper fieldMapper = mapper.mappers().getMapper("field");
    assertThat(fieldMapper, instanceOf(GeoPointFieldMapper.class));
    ParsedDocument doc = mapper.parse(source(b -> b.nullField("field")));
    assertThat(doc.rootDoc().getField("field"), nullValue());
    assertThat(doc.rootDoc().getFields(FieldNamesFieldMapper.NAME).length, equalTo(0));
    mapper = createDocumentMapper(fieldMapping(b -> b.field("type", "geo_point").field("doc_values", false)));
    fieldMapper = mapper.mappers().getMapper("field");
    assertThat(fieldMapper, instanceOf(GeoPointFieldMapper.class));
    doc = mapper.parse(source(b -> b.nullField("field")));
    assertThat(doc.rootDoc().getField("field"), nullValue());
    assertThat(doc.rootDoc().getFields(FieldNamesFieldMapper.NAME).length, equalTo(0));
    mapper = createDocumentMapper(fieldMapping(b -> b.field("type", "geo_point").field("null_value", "1,2")));
    fieldMapper = mapper.mappers().getMapper("field");
    assertThat(fieldMapper, instanceOf(GeoPointFieldMapper.class));
    AbstractPointGeometryFieldMapper.ParsedPoint nullValue = ((GeoPointFieldMapper) fieldMapper).nullValue;
    assertThat(nullValue, equalTo(new GeoPoint(1, 2)));
    doc = mapper.parse(source(b -> b.nullField("field")));
    assertThat(doc.rootDoc().getField("field"), notNullValue());
    BytesRef defaultValue = doc.rootDoc().getBinaryValue("field");
    // Shouldn't matter if we specify the value explicitly or use null value
    doc = mapper.parse(source(b -> b.field("field", "1, 2")));
    assertThat(defaultValue, equalTo(doc.rootDoc().getBinaryValue("field")));
    doc = mapper.parse(source(b -> b.field("field", "3, 4")));
    assertThat(defaultValue, not(equalTo(doc.rootDoc().getBinaryValue("field"))));
}

Example 49 with GeoPoint

use of org.opensearch.common.geo.GeoPoint in project OpenSearch by opensearch-project.

the class ExternalMapper method parse.

@Override
public void parse(ParseContext context) throws IOException {
    byte[] bytes = "Hello world".getBytes(Charset.defaultCharset());
    binMapper.parse(context.createExternalValueContext(bytes));
    boolMapper.parse(context.createExternalValueContext(true));
    // Let's add a Dummy Point
    double lat = 42.0;
    double lng = 51.0;
    ArrayList<GeoPoint> points = new ArrayList<>();
    points.add(new GeoPoint(lat, lng));
    pointMapper.parse(context.createExternalValueContext(points));
    // Let's add a Dummy Shape
    if (shapeMapper instanceof GeoShapeFieldMapper) {
        shapeMapper.parse(context.createExternalValueContext(new Point(-100, 45)));
    } else {
        PointBuilder pb = new PointBuilder(-100, 45);
        shapeMapper.parse(context.createExternalValueContext(pb.buildS4J()));
    }
    context = context.createExternalValueContext(generatedValue);
    // Let's add a Original String
    stringMapper.parse(context);
    multiFields.parse(this, context);
}

Example 50 with GeoPoint

use of org.opensearch.common.geo.GeoPoint in project OpenSearch by opensearch-project.

the class DistanceFeatureQueryBuilderTests method doCreateTestQueryBuilder.

@Override
protected DistanceFeatureQueryBuilder doCreateTestQueryBuilder() {
    String field = randomFrom(DATE_FIELD_NAME, DATE_NANOS_FIELD_NAME, GEO_POINT_FIELD_NAME);
    Origin origin;
    String pivot;
    switch(field) {
        case GEO_POINT_FIELD_NAME:
            GeoPoint point = new GeoPoint(randomDouble(), randomDouble());
            origin = randomBoolean() ? new Origin(point) : new Origin(point.geohash());
            pivot = randomFrom(DistanceUnit.values()).toString(randomDouble());
            break;
        case DATE_FIELD_NAME:
            long randomDateMills = randomLongBetween(0, 2_000_000_000_000L);
            origin = randomBoolean() ? new Origin(randomDateMills) : new Origin(new DateTime(randomDateMills).toString());
            pivot = randomTimeValue(1, 1000, "d", "h", "ms", "s", "m");
            break;
        default:
            // DATE_NANOS_FIELD_NAME
            randomDateMills = randomLongBetween(0, 2_000_000_000_000L);
            if (randomBoolean()) {
                // nano_dates long accept milliseconds since epoch
                origin = new Origin(randomDateMills);
            } else {
                long randomNanos = randomLongBetween(0, 1_000_000L);
                Instant randomDateNanos = Instant.ofEpochMilli(randomDateMills).plusNanos(randomNanos);
                origin = new Origin(randomDateNanos.toString());
            }
            pivot = randomTimeValue(1, 100_000_000, "nanos");
            break;
    }
    return new DistanceFeatureQueryBuilder(field, origin, pivot);
}