Example usage for org.apache.lucene.spatial.prefix RecursivePrefixTreeStrategy RecursivePrefixTreeStrategy

List of usage examples for org.apache.lucene.spatial.prefix RecursivePrefixTreeStrategy RecursivePrefixTreeStrategy

  1. HOME
  2. Java
  3. org.apache.lucene
  4. org.apache.lucene.spatial.prefix.*
  5. RecursivePrefixTreeStrategy
  6. RecursivePrefixTreeStrategy

Introduction

In this page you can find the example usage for org.apache.lucene.spatial.prefix RecursivePrefixTreeStrategy RecursivePrefixTreeStrategy.

Prototype

public RecursivePrefixTreeStrategy(SpatialPrefixTree grid, String fieldName)

Source Link

Usage

From source file:com.berico.clavin.resolver.impl.lucene.LuceneComponentsFactory.java

License:Apache License

/**
 * Initialize the common stuff used by both the IndexWriter and 
 * Searcher.//  w  w w .j  a va2s .co m
 * @throws IOException
 */
protected void initializeCommon() throws IOException {

    // load the Lucene index directory from disk
    index = FSDirectory.open(indexDir);

    // index employs simple lower-casing & tokenizing on whitespace
    indexAnalyzer = new WhitespaceLowerCaseAnalyzer();

    // Retrieve the spatial context implementation from Spatial4j,
    // which will be used by Lucene to do it's Geo thing.
    spatialContext = SpatialContext.GEO;

    // Instantiate the spatial search strategy.
    spatialStrategy = new RecursivePrefixTreeStrategy(new GeohashPrefixTree(spatialContext, 11),
            FieldConstants.GEOMETRY);
}

From source file:com.orientechnologies.lucene.manager.OLuceneSpatialIndexManager.java

License:Apache License

public OLuceneSpatialIndexManager(OShapeFactory factory) {
    super();/* w w w. j  a  va 2s  . co m*/
    this.ctx = SpatialContext.GEO;
    this.factory = factory;
    SpatialPrefixTree grid = new GeohashPrefixTree(ctx, 11);
    this.strategy = new RecursivePrefixTreeStrategy(grid, "location");
}

From source file:com.orientechnologies.spatial.engine.OLuceneLegacySpatialIndexEngine.java

License:Apache License

@Override
protected SpatialStrategy createSpatialStrategy(OIndexDefinition indexDefinition, ODocument metadata) {
    return new RecursivePrefixTreeStrategy(new GeohashPrefixTree(ctx, 11), "location");
}

From source file:com.orientechnologies.spatial.factory.OSpatialStrategyFactory.java

License:Apache License

public SpatialStrategy createStrategy(SpatialContext ctx, ODatabaseDocumentInternal db,
        OIndexDefinition indexDefinition, ODocument metadata) {

    SpatialStrategy strategy;/*from   w  w w. j  av  a  2  s .  c  o  m*/

    OClass aClass = db.getMetadata().getSchema().getClass(indexDefinition.getClassName());

    OProperty property = aClass.getProperty(indexDefinition.getFields().get(0));

    OClass linkedClass = property.getLinkedClass();

    if ("OPoint".equalsIgnoreCase(linkedClass.getName())) {
        RecursivePrefixTreeStrategy recursivePrefixTreeStrategy = new RecursivePrefixTreeStrategy(
                new GeohashPrefixTree(ctx, 11), "location");
        recursivePrefixTreeStrategy.setDistErrPct(0);
        strategy = recursivePrefixTreeStrategy;

    } else {
        strategy = new BBoxStrategy(ctx, "location");
    }
    return strategy;
}

From source file:com.orientechnologies.spatial.sandbox.LuceneGeoTest.java

License:Apache License

@Test
public void geoIntersectTest() throws IOException, ParseException {

    RecursivePrefixTreeStrategy strategy = new RecursivePrefixTreeStrategy(
            new GeohashPrefixTree(JtsSpatialContext.GEO, 11), "location");

    strategy.setDistErrPct(0);//from   ww  w  .  ja  v a 2 s .  co m

    IndexWriterConfig conf = new IndexWriterConfig(new StandardAnalyzer());
    final RAMDirectory directory = new RAMDirectory();
    final IndexWriter writer = new IndexWriter(directory, conf);

    Shape point = JtsSpatialContext.GEO.getWktShapeParser().parse("POINT (9.4714708 47.6819432)");

    Shape polygon = JtsSpatialContext.GEO.getWktShapeParser().parse(
            "POLYGON((9.481201171875 47.64885294675266,9.471416473388672 47.65128140482982,9.462661743164062 47.64781214443791,9.449443817138672 47.656947367880335,9.445838928222656 47.66110972448931,9.455795288085938 47.667352637215,9.469013214111328 47.67255449415724,9.477081298828125 47.679142768657066,9.490299224853516 47.678680460743834,9.506263732910156 47.679258344995326,9.51364517211914 47.68191653011071,9.518795013427734 47.677177931734406,9.526691436767578 47.679489496903706,9.53390121459961 47.67139857075435,9.50918197631836 47.66180341832901,9.50815200805664 47.6529003141482,9.51192855834961 47.64654002455372,9.504375457763672 47.64237650648966,9.49270248413086 47.649662445325035,9.48617935180664 47.65151268066222,9.481201171875 47.64885294675266))");

    Document doc = new Document();

    Assert.assertNotEquals(point.relate(polygon), SpatialRelation.INTERSECTS);
    for (IndexableField f : strategy.createIndexableFields(point)) {
        doc.add(f);
    }

    writer.addDocument(doc);
    writer.commit();

    SpatialArgs args = new SpatialArgs(SpatialOperation.Intersects, polygon.getBoundingBox());
    Filter filter = strategy.makeFilter(args);
    IndexReader reader = DirectoryReader.open(directory);

    IndexSearcher searcher = new IndexSearcher(reader);

    TopDocs search = searcher.search(new MatchAllDocsQuery(), filter, 1000);
    Assert.assertEquals(search.totalHits, 0);

    reader.close();
    writer.close();
}

From source file:com.orientechnologies.spatial.sandbox.LuceneGeoTest.java

License:Apache License

@Test
public void geoSpeedTest() throws IOException, ParseException {

    RecursivePrefixTreeStrategy strategy = new RecursivePrefixTreeStrategy(
            new GeohashPrefixTree(JtsSpatialContext.GEO, 11), "location");

    IndexWriterConfig conf = new IndexWriterConfig(new StandardAnalyzer());
    final RAMDirectory directory = new RAMDirectory();
    final IndexWriter writer = new IndexWriter(directory, conf);

    Shape multiPolygon = JtsSpatialContext.GEO.getWktShapeParser().parse(
            "MULTIPOLYGON (((15.520376 38.231155, 15.160243 37.444046, 15.309898 37.134219, 15.099988 36.619987, 14.335229 36.996631, 13.826733 37.104531, 12.431004 37.61295, 12.570944 38.126381, 13.741156 38.034966, 14.761249 38.143874, 15.520376 38.231155)), ((9.210012 41.209991, 9.809975 40.500009, 9.669519 39.177376, 9.214818 39.240473, 8.806936 38.906618, 8.428302 39.171847, 8.388253 40.378311, 8.159998 40.950007, 8.709991 40.899984, 9.210012 41.209991)), ((12.376485 46.767559, 13.806475 46.509306, 13.69811 46.016778, 13.93763 45.591016, 13.141606 45.736692, 12.328581 45.381778, 12.383875 44.885374, 12.261453 44.600482, 12.589237 44.091366, 13.526906 43.587727, 14.029821 42.761008, 15.14257 41.95514, 15.926191 41.961315, 16.169897 41.740295, 15.889346 41.541082, 16.785002 41.179606, 17.519169 40.877143, 18.376687 40.355625, 18.480247 40.168866, 18.293385 39.810774, 17.73838 40.277671, 16.869596 40.442235, 16.448743 39.795401, 17.17149 39.4247, 17.052841 38.902871, 16.635088 38.843572, 16.100961 37.985899, 15.684087 37.908849, 15.687963 38.214593, 15.891981 38.750942, 16.109332 38.964547, 15.718814 39.544072, 15.413613 40.048357, 14.998496 40.172949, 14.703268 40.60455, 14.060672 40.786348, 13.627985 41.188287, 12.888082 41.25309, 12.106683 41.704535, 11.191906 42.355425, 10.511948 42.931463, 10.200029 43.920007, 9.702488 44.036279, 8.888946 44.366336, 8.428561 44.231228, 7.850767 43.767148, 7.435185 43.693845, 7.549596 44.127901, 7.007562 44.254767, 6.749955 45.028518, 7.096652 45.333099, 6.802355 45.70858, 6.843593 45.991147, 7.273851 45.776948, 7.755992 45.82449, 8.31663 46.163642, 8.489952 46.005151, 8.966306 46.036932, 9.182882 46.440215, 9.922837 46.314899, 10.363378 46.483571, 10.442701 46.893546, 11.048556 46.751359, 11.164828 46.941579, 12.153088 47.115393, 12.376485 46.767559)))");

    Document doc = new Document();

    for (IndexableField f : strategy.createIndexableFields(multiPolygon)) {
        doc.add(f);//w  w w .  j  a va 2s.  c  om
    }

    writer.addDocument(doc);
    writer.commit();

    writer.close();
}

From source file:com.orientechnologies.spatial.sandbox.LuceneGeoTest.java

License:Apache License

@Test
public void geoSpeedTestInternal() throws IOException, ParseException {

    RecursivePrefixTreeStrategy strategy = new RecursivePrefixTreeStrategy(
            new GeohashPrefixTree(JtsSpatialContext.GEO, 11), "location");

    IndexWriterConfig conf = new IndexWriterConfig(new StandardAnalyzer());
    final RAMDirectory directory = new RAMDirectory();
    final IndexWriter writer = new IndexWriter(directory, conf);

    ODocument entries = loadMultiPolygon();

    OMultiPolygonShapeBuilder builder = new OMultiPolygonShapeBuilder();

    Shape multiPolygon = builder.fromDoc(entries);

    Document doc = new Document();

    for (IndexableField f : strategy.createIndexableFields(multiPolygon)) {
        doc.add(f);/*from   ww w  .j  av a  2 s .c  o  m*/
    }

    writer.addDocument(doc);
    writer.commit();

    writer.close();
}

From source file:com.orientechnologies.test.CreateCityDbIndex.java

License:Apache License

@Override
@Test(enabled = false)//from  w  ww  .j a  v  a2 s  .co  m
public void init() throws Exception {
    String buildDirectory = System.getProperty("buildDirectory", ".");
    if (buildDirectory == null)
        buildDirectory = ".";

    ZipFile zipFile = new ZipFile("files/allCountries.zip");
    Enumeration<? extends ZipEntry> entries = zipFile.entries();

    while (entries.hasMoreElements()) {

        ZipEntry entry = entries.nextElement();
        if (entry.getName().equals("allCountries.txt")) {

            InputStream stream = zipFile.getInputStream(entry);
            lineReader = new LineNumberReader(new InputStreamReader(stream));
        }
    }
    Directory dir = NIOFSDirectory.open(new File("Spatial"));
    IndexWriterConfig iwc = new IndexWriterConfig(Version.LUCENE_47, new StandardAnalyzer(Version.LUCENE_47));
    iwc.setOpenMode(IndexWriterConfig.OpenMode.CREATE_OR_APPEND);
    writer = new IndexWriter(dir, iwc);
    this.ctx = SpatialContext.GEO;
    SpatialPrefixTree grid = new GeohashPrefixTree(ctx, 11);
    this.strategy = new RecursivePrefixTreeStrategy(grid, "location");
}

From source file:com.stratio.cassandra.lucene.schema.mapping.GeoPointMapper.java

License:Apache License

/**
 * Builds a new {@link GeoPointMapper}.// w  w w.  j ava  2  s. c  om
 *
 * @param name      The name of the mapper.
 * @param latitude  The name of the column containing the latitude.
 * @param longitude The name of the column containing the longitude.
 * @param maxLevels The maximum number of levels in the tree.
 */
public GeoPointMapper(String name, String latitude, String longitude, Integer maxLevels) {
    super(name, true, false, AsciiType.instance, UTF8Type.instance, Int32Type.instance, LongType.instance,
            IntegerType.instance, FloatType.instance, DoubleType.instance, DecimalType.instance);

    if (StringUtils.isBlank(latitude)) {
        throw new IllegalArgumentException("latitude column name is required");
    }

    if (StringUtils.isBlank(longitude)) {
        throw new IllegalArgumentException("longitude column name is required");
    }

    this.latitude = latitude;
    this.longitude = longitude;
    this.maxLevels = maxLevels == null ? DEFAULT_MAX_LEVELS : maxLevels;
    SpatialPrefixTree grid = new GeohashPrefixTree(spatialContext, this.maxLevels);
    this.strategy = new RecursivePrefixTreeStrategy(grid, name);
}

From source file:com.stratio.cassandra.lucene.schema.mapping.GeoShapeMapper.java

License:Apache License

/**
 * Builds a new {@link GeoShapeMapper}./* w w w  .j a  v  a  2s .co  m*/
 *
 * @param field the name of the field
 * @param column the name of the column
 * @param validated if the field must be validated
 * @param maxLevels the maximum number of levels in the tree
 * @param transformations the sequence of operations to be applied to the specified shape
 */
public GeoShapeMapper(String field, String column, Boolean validated, Integer maxLevels,
        List<GeoTransformation> transformations) {
    super(field, column, true, false, validated, null, String.class, AsciiType.instance, UTF8Type.instance);

    this.column = column == null ? field : column;

    if (StringUtils.isWhitespace(column)) {
        throw new IndexException("Column must not be whitespace, but found '%s'", column);
    }

    this.maxLevels = GeospatialUtils.validateGeohashMaxLevels(maxLevels);
    SpatialPrefixTree grid = new GeohashPrefixTree(SPATIAL_CONTEXT, this.maxLevels);
    strategy = new RecursivePrefixTreeStrategy(grid, field);

    this.transformations = (transformations == null) ? Collections.<GeoTransformation>emptyList()
            : transformations;
}