org.neo4j.kernel.impl.index.schema.SpatialIndexPartReader Maven / Gradle / Ivy
/*
* Copyright (c) 2018-2020 "Graph Foundation,"
* Graph Foundation, Inc. [https://graphfoundation.org]
*
* This file is part of ONgDB.
*
* ONgDB is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see extends NativeSchemaIndexReader
{
private final SpatialLayout spatial;
private final SpaceFillingCurveConfiguration configuration;
SpatialIndexPartReader( GBPTree tree, Layout layout,
IndexSamplingConfig samplingConfig, SchemaIndexDescriptor descriptor,
SpaceFillingCurveConfiguration configuration )
{
super( tree, layout, samplingConfig, descriptor );
spatial = (SpatialLayout) layout;
this.configuration = configuration;
}
@Override
void validateQuery( IndexOrder indexOrder, IndexQuery[] predicates )
{
if ( predicates.length != 1 )
{
throw new UnsupportedOperationException( "Spatial index doesn't handle composite queries" );
}
CapabilityValidator.validateQuery( SpatialIndexProvider.CAPABILITY, indexOrder, predicates );
}
@Override
boolean initializeRangeForQuery( SpatialSchemaKey treeKeyFrom, SpatialSchemaKey treeKeyTo, IndexQuery[] predicates )
{
throw new UnsupportedOperationException( "Cannot initialize 1D range in multidimensional spatial index reader" );
}
@Override
public PrimitiveLongResourceIterator query( IndexQuery... predicates )
{
NodeValueIterator nodeValueIterator = new NodeValueIterator();
query( nodeValueIterator, IndexOrder.NONE, predicates );
return nodeValueIterator;
}
@Override
public void query( IndexProgressor.NodeValueClient cursor, IndexOrder indexOrder, IndexQuery... predicates )
{
validateQuery( indexOrder, predicates );
IndexQuery predicate = predicates[0];
switch ( predicate.type() )
{
case exists:
startSeekForExists( cursor, predicate );
break;
case exact:
startSeekForExact( cursor, ((ExactPredicate) predicate).value(), predicate );
break;
case range:
GeometryRangePredicate rangePredicate = (GeometryRangePredicate) predicate;
if ( !rangePredicate.crs().equals( spatial.crs ) )
{
throw new IllegalArgumentException(
"IndexQuery on spatial index with mismatching CoordinateReferenceSystem: " + rangePredicate.crs() + " != " + spatial.crs );
}
startSeekForRange( cursor, rangePredicate, predicates );
break;
default:
throw new IllegalArgumentException( "IndexQuery of type " + predicate.type() + " is not supported." );
}
}
@Override
public void distinctValues( IndexProgressor.NodeValueClient client, PropertyAccessor propertyAccessor )
{
// This is basically a version of the basic implementation, but with added consulting of the PropertyAccessor
// since these are lossy spatial values.
SpatialSchemaKey lowest = layout.newKey();
lowest.initialize( Long.MIN_VALUE );
lowest.initValueAsLowest();
SpatialSchemaKey highest = layout.newKey();
highest.initialize( Long.MAX_VALUE );
highest.initValueAsHighest();
try
{
RawCursor,IOException> seeker = tree.seek( lowest, highest );
SchemaLayout schemaLayout = (SchemaLayout) layout;
Comparator comparator =
new PropertyLookupFallbackComparator<>( schemaLayout, propertyAccessor, descriptor.schema().getPropertyId() );
NativeDistinctValuesProgressor progressor =
new NativeDistinctValuesProgressor( seeker, client, openSeekers, schemaLayout, comparator )
{
@Override
Value[] extractValues( SpatialSchemaKey key )
{
try
{
return new Value[]{propertyAccessor.getPropertyValue( key.getEntityId(), descriptor.schema().getPropertyId() )};
}
catch ( EntityNotFoundException e )
{
// We couldn't get the value due to the entity not being there. Concurrently deleted?
return null;
}
}
};
client.initialize( descriptor, progressor, new IndexQuery[0] );
}
catch ( IOException e )
{
throw new UncheckedIOException( e );
}
}
private void startSeekForExists( IndexProgressor.NodeValueClient client, IndexQuery... predicates )
{
SpatialSchemaKey treeKeyFrom = layout.newKey();
SpatialSchemaKey treeKeyTo = layout.newKey();
treeKeyFrom.initAsLowest();
treeKeyTo.initAsHighest();
startSeekForInitializedRange( client, treeKeyFrom, treeKeyTo, predicates, false );
}
private void startSeekForExact( IndexProgressor.NodeValueClient client, Value value, IndexQuery... predicates )
{
SpatialSchemaKey treeKeyFrom = layout.newKey();
SpatialSchemaKey treeKeyTo = layout.newKey();
treeKeyFrom.from( Long.MIN_VALUE, value );
treeKeyTo.from( Long.MAX_VALUE, value );
startSeekForInitializedRange( client, treeKeyFrom, treeKeyTo, predicates, false );
}
private void startSeekForRange( IndexProgressor.NodeValueClient client, GeometryRangePredicate rangePredicate, IndexQuery[] query )
{
try
{
BridgingIndexProgressor multiProgressor = new BridgingIndexProgressor( client, descriptor.schema().getPropertyIds() );
client.initialize( descriptor, multiProgressor, query );
SpaceFillingCurve curve = spatial.getSpaceFillingCurve();
double[] from = rangePredicate.from() == null ? null : rangePredicate.from().coordinate();
double[] to = rangePredicate.to() == null ? null : rangePredicate.to().coordinate();
List ranges = curve.getTilesIntersectingEnvelope( from, to, configuration );
for ( SpaceFillingCurve.LongRange range : ranges )
{
SpatialSchemaKey treeKeyFrom = layout.newKey();
SpatialSchemaKey treeKeyTo = layout.newKey();
treeKeyFrom.fromDerivedValue( Long.MIN_VALUE, range.min );
treeKeyTo.fromDerivedValue( Long.MAX_VALUE, range.max + 1 );
RawCursor,IOException> seeker = makeIndexSeeker( treeKeyFrom, treeKeyTo );
IndexProgressor hitProgressor = new SpatialHitIndexProgressor<>( seeker, client, openSeekers );
multiProgressor.initialize( descriptor, hitProgressor, query );
}
}
catch ( IllegalArgumentException e )
{
// Invalid query ranges will cause this state (eg. min>max)
client.initialize( descriptor, IndexProgressor.EMPTY, query );
}
catch ( IOException e )
{
throw new UncheckedIOException( e );
}
}
@Override
void startSeekForInitializedRange( IndexProgressor.NodeValueClient client, SpatialSchemaKey treeKeyFrom,
SpatialSchemaKey treeKeyTo, IndexQuery[] query, boolean needFilter )
{
if ( layout.compare( treeKeyFrom, treeKeyTo ) > 0 )
{
client.initialize( descriptor, IndexProgressor.EMPTY, query );
return;
}
try
{
RawCursor,IOException> seeker = makeIndexSeeker( treeKeyFrom, treeKeyTo );
IndexProgressor hitProgressor = new SpatialHitIndexProgressor<>( seeker, client, openSeekers );
client.initialize( descriptor, hitProgressor, query );
}
catch ( IOException e )
{
throw new UncheckedIOException( e );
}
}
@Override
public boolean hasFullValuePrecision( IndexQuery... predicates )
{
return false;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy