@Override
  public ValueSource parse(FunctionQParser fqp) throws SyntaxError {

    ValueSource sourceQuery = fqp.parseValueSource();
    ValueSource feature1 = fqp.parseValueSource();
    ValueSource feature2 = fqp.parseValueSource();
    ValueSource feature3 = fqp.parseValueSource();

    return new LTRValueSource(fqp, ranker, sourceQuery, feature1, feature2, feature3);
  }
  @Override
  public ValueSource parse(FunctionQParser fp) throws SyntaxError {
    String key = fp.parseArg();
    double val = fp.parseDouble();

    AtomicInteger counter = new AtomicInteger();
    if (null != counters.putIfAbsent(key, counter)) {
      throw new IllegalArgumentException("Key has already been used: " + key);
    }
    return new CountDocsValueSource(counter, val);
  }
  @Override
  public ValueSource parse(FunctionQParser fp) throws SyntaxError {
    String fieldName = fp.parseId();
    SchemaField field = fp.getReq().getSchema().getField(fieldName);
    FieldType type = field.getType();
    if (!(type instanceof MultiPointDocValuesField))
      throw new SyntaxError(
          "This function only supports fields of type "
              + MultiPointDocValuesField.class.getName()
              + ", not "
              + type.getClass().getName());
    MultiPointDocValuesField mpdvFieldType = (MultiPointDocValuesField) type;

    double[] parsedLatLong = null;
    try {
      parsedLatLong = ParseUtils.parseLatitudeLongitude(fp.parseArg());
    } catch (InvalidShapeException e) {
      throw new SolrException(SolrException.ErrorCode.BAD_REQUEST, e);
    }

    double y = parsedLatLong[0];
    double x = parsedLatLong[1];

    SpatialContext ctx = mpdvFieldType.getCtx();
    Point point = ctx.makePoint(x, y);

    String score = fp.getLocalParams().get("score", "distance");
    ValueSource valueSource = new MultiPointDistanceValueSource(fieldName, point, ctx);

    if ("distance".equals(score)) {
      return valueSource;
    } else if ("recipDistance".equals(score)) {
      int shift = fp.getLocalParams().getInt("shift", 100);
      int maxScore = fp.getLocalParams().getInt("maxScore", 10);
      return new ReciprocalFloatFunction(valueSource, maxScore, shift, shift);
    } else {
      throw new SolrException(
          SolrException.ErrorCode.BAD_REQUEST,
          "'score' local-param must be one of 'distance', or 'recipDistance'");
    }
  }
Ejemplo n.º 4
0
 // parses avg(x)
 private AggValueSource parseStringStat(String key, String stat) throws SyntaxError {
   FunctionQParser parser =
       (FunctionQParser) QParser.getParser(stat, FunctionQParserPlugin.NAME, getSolrRequest());
   AggValueSource agg = parser.parseAgg(FunctionQParser.FLAG_DEFAULT);
   return agg;
 }