Example usage for org.apache.lucene.spatial.prefix HeatmapFacetCounter calcFacets

Introduction

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

Prototype

public static Heatmap calcFacets(PrefixTreeStrategy strategy, IndexReaderContext context, Bits topAcceptDocs,
        Shape inputShape, final int facetLevel, int maxCells) throws IOException 

Source Link

Document

Calculates spatial 2D facets (aggregated counts) in a grid, sometimes called a heatmap.

Usage

From source file:com.boundlessgeo.elasticsearch.geoheatmap.GeoHeatmapAggregator.java

License:Apache License

@Override
public InternalAggregation buildAggregation(long owningBucketOrdinal) throws IOException {
    SparseFixedBitSet matchingDocs = buckets.get(owningBucketOrdinal);
    if (matchingDocs == null)
        return buildEmptyAggregation();

    Heatmap heatmap = HeatmapFacetCounter.calcFacets(strategy, parentReaderContext, matchingDocs, inputShape,
            gridLevel, maxCells);/* ww  w .  j av a2s .  c  o m*/

    return new InternalGeoHeatmap(name, gridLevel, heatmap.rows, heatmap.columns, heatmap.region.getMinX(),
            heatmap.region.getMinY(), heatmap.region.getMaxX(), heatmap.region.getMaxY(), heatmap.counts,
            pipelineAggregators(), metaData());
}

---

From source file:org.apache.solr.handler.component.SpatialHeatmapFacets.java

License:Apache License

/** Called by {@link org.apache.solr.request.SimpleFacets} to compute heatmap facets. */
public static NamedList<Object> getHeatmapForField(String fieldKey, String fieldName, ResponseBuilder rb,
        SolrParams params, DocSet docSet) throws IOException {
    //get the strategy from the field type
    final SchemaField schemaField = rb.req.getSchema().getField(fieldName);
    final FieldType type = schemaField.getType();
    if (!(type instanceof AbstractSpatialPrefixTreeFieldType)) {
        //FYI we support the term query one too but few people use that one
        throw new SolrException(SolrException.ErrorCode.BAD_REQUEST,
                "heatmap field needs to be of type " + SpatialRecursivePrefixTreeFieldType.class);
    }/* w  ww  .ja  va  2  s . c o  m*/
    AbstractSpatialPrefixTreeFieldType rptType = (AbstractSpatialPrefixTreeFieldType) type;
    final PrefixTreeStrategy strategy = (PrefixTreeStrategy) rptType.getStrategy(fieldName);
    final SpatialContext ctx = strategy.getSpatialContext();

    //get the bbox (query Rectangle)
    String geomStr = params.getFieldParam(fieldKey, FacetParams.FACET_HEATMAP_GEOM);
    final Shape boundsShape = geomStr == null ? ctx.getWorldBounds()
            : SpatialUtils.parseGeomSolrException(geomStr, ctx);

    //get the grid level (possibly indirectly via distErr or distErrPct)
    final int gridLevel;
    Integer gridLevelObj = params.getFieldInt(fieldKey, FacetParams.FACET_HEATMAP_LEVEL);
    final int maxGridLevel = strategy.getGrid().getMaxLevels();
    if (gridLevelObj != null) {
        gridLevel = gridLevelObj;
        if (gridLevel <= 0 || gridLevel > maxGridLevel) {
            throw new SolrException(SolrException.ErrorCode.BAD_REQUEST,
                    FacetParams.FACET_HEATMAP_LEVEL + " should be > 0 and <= " + maxGridLevel);
        }
    } else {
        //SpatialArgs has utility methods to resolve a 'distErr' from optionally set distErr & distErrPct. Arguably that
        // should be refactored to feel less weird than using it like this.
        SpatialArgs spatialArgs = new SpatialArgs(SpatialOperation.Intersects/*ignored*/,
                boundsShape == null ? ctx.getWorldBounds() : boundsShape);
        final Double distErrObj = params.getFieldDouble(fieldKey, FacetParams.FACET_HEATMAP_DIST_ERR);
        if (distErrObj != null) {
            // convert distErr units based on configured units
            spatialArgs.setDistErr(distErrObj * rptType.getDistanceUnits().multiplierFromThisUnitToDegrees());
        }
        spatialArgs.setDistErrPct(params.getFieldDouble(fieldKey, FacetParams.FACET_HEATMAP_DIST_ERR_PCT));
        double distErr = spatialArgs.resolveDistErr(ctx, DEFAULT_DIST_ERR_PCT);
        if (distErr <= 0) {
            throw new SolrException(SolrException.ErrorCode.BAD_REQUEST,
                    FacetParams.FACET_HEATMAP_DIST_ERR_PCT + " or " + FacetParams.FACET_HEATMAP_DIST_ERR
                            + " should be > 0 or instead provide " + FacetParams.FACET_HEATMAP_LEVEL + "="
                            + maxGridLevel + " if you insist on maximum detail");
        }
        //The SPT (grid) can lookup a grid level satisfying an error distance constraint
        gridLevel = strategy.getGrid().getLevelForDistance(distErr);
    }

    //Compute!
    final HeatmapFacetCounter.Heatmap heatmap;
    try {
        heatmap = HeatmapFacetCounter.calcFacets(strategy, rb.req.getSearcher().getTopReaderContext(),
                docSet.getTopFilter(), boundsShape, gridLevel,
                params.getFieldInt(fieldKey, FacetParams.FACET_HEATMAP_MAX_CELLS, 100_000) // will throw if exceeded
        );
    } catch (IllegalArgumentException e) {//e.g. too many cells
        throw new SolrException(SolrException.ErrorCode.BAD_REQUEST, e.toString(), e);
    }

    //Populate response
    NamedList<Object> result = new NamedList<>();
    result.add("gridLevel", gridLevel);
    result.add("columns", heatmap.columns);
    result.add("rows", heatmap.rows);
    result.add("minX", heatmap.region.getMinX());
    result.add("maxX", heatmap.region.getMaxX());
    result.add("minY", heatmap.region.getMinY());
    result.add("maxY", heatmap.region.getMaxY());

    boolean hasNonZero = false;
    for (int count : heatmap.counts) {
        if (count > 0) {
            hasNonZero = true;
            break;
        }
    }
    formatCountsAndAddToNL(fieldKey, rb, params, heatmap.columns, heatmap.rows,
            hasNonZero ? heatmap.counts : null, result);

    return result;
}

---