| package edu.uci.ics.asterix.runtime.evaluators.functions; |
| |
| import edu.uci.ics.asterix.common.functions.FunctionConstants; |
| import edu.uci.ics.asterix.om.types.ATypeTag; |
| import edu.uci.ics.asterix.om.types.BuiltinType; |
| import edu.uci.ics.asterix.runtime.evaluators.base.AbstractScalarFunctionDynamicDescriptor; |
| import edu.uci.ics.asterix.runtime.evaluators.common.GramTokensEvaluator; |
| import edu.uci.ics.fuzzyjoin.tokenizer.ITokenFactory; |
| import edu.uci.ics.fuzzyjoin.tokenizer.NGramUTF8StringBinaryTokenizer; |
| import edu.uci.ics.fuzzyjoin.tokenizer.UTF8NGramTokenFactory; |
| import edu.uci.ics.hyracks.algebricks.common.exceptions.AlgebricksException; |
| import edu.uci.ics.hyracks.algebricks.core.algebra.functions.FunctionIdentifier; |
| import edu.uci.ics.hyracks.algebricks.runtime.base.IEvaluator; |
| import edu.uci.ics.hyracks.algebricks.runtime.base.IEvaluatorFactory; |
| import edu.uci.ics.hyracks.dataflow.common.data.accessors.IDataOutputProvider; |
| |
| public class GramTokensDescriptor extends AbstractScalarFunctionDynamicDescriptor { |
| |
| private static final long serialVersionUID = 1L; |
| private final static FunctionIdentifier FID = new FunctionIdentifier(FunctionConstants.ASTERIX_NS, "gram-tokens", |
| 3, true); |
| |
| @Override |
| public FunctionIdentifier getIdentifier() { |
| return FID; |
| } |
| |
| @Override |
| public IEvaluatorFactory createEvaluatorFactory(final IEvaluatorFactory[] args) throws AlgebricksException { |
| return new IEvaluatorFactory() { |
| private static final long serialVersionUID = 1L; |
| |
| @Override |
| public IEvaluator createEvaluator(IDataOutputProvider output) throws AlgebricksException { |
| ITokenFactory tokenFactory = new UTF8NGramTokenFactory(ATypeTag.STRING.serialize(), |
| ATypeTag.INT32.serialize()); |
| NGramUTF8StringBinaryTokenizer tokenizer = new NGramUTF8StringBinaryTokenizer(3, true, true, true, |
| tokenFactory); |
| return new GramTokensEvaluator(args, output, tokenizer, BuiltinType.ASTRING); |
| } |
| }; |
| } |
| |
| } |