s434730/RandomSec
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

adding regexp group capturing GEL function

Browse Source 
git-svn-id: http://google-refine.googlecode.com/svn/trunk@932 7d457c2a-affb-35e4-300a-418c747d4874
This commit is contained in:
Stefano Mazzocchi 2010-06-01 08:54:17 +00:00
parent 5e0acf28d0
commit 0648e8725e
2 changed files with 57 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
main/src/com/metaweb/gridworks/expr/functions/strings/Match.java Normal file
View File

@ -0,0 +1,55 @@
package com.metaweb.gridworks.expr.functions.strings;
import java.util.Properties;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import org.json.JSONException;
import org.json.JSONWriter;
import com.metaweb.gridworks.expr.EvalError;
import com.metaweb.gridworks.gel.ControlFunctionRegistry;
import com.metaweb.gridworks.gel.Function;
public class Match implements Function {
public Object call(Properties bindings, Object[] args) {
if (args.length == 2) {
Object s = args[0];
Object p = args[1];
if (s != null && p != null && (p instanceof String || p instanceof Pattern)) {
Pattern pattern = (p instanceof String) ? Pattern.compile((String) p) : (Pattern) p;
Matcher matcher = pattern.matcher(s.toString());
if (matcher.matches()) {
int count = matcher.groupCount();
String[] groups = new String[count];
for (int i = 0; i < count; i++) {
groups[i] = matcher.group(i + 1);
}
return groups;
} else {
return null;
}
}
return null;
}
return new EvalError(ControlFunctionRegistry.getFunctionName(this) + " expects a string or a regexp");
}
public void write(JSONWriter writer, Properties options)
throws JSONException {
writer.object();
writer.key("description"); writer.value("Returns an array of the groups matching the given regular expression");
writer.key("params"); writer.value("string or regexp");
writer.key("returns"); writer.value("array of strings");
writer.endObject();
}
}

2
main/src/com/metaweb/gridworks/gel/ControlFunctionRegistry.java
View File

@ -42,6 +42,7 @@ import com.metaweb.gridworks.expr.functions.strings.Fingerprint;
import com.metaweb.gridworks.expr.functions.strings.IndexOf;
import com.metaweb.gridworks.expr.functions.strings.LastIndexOf;
import com.metaweb.gridworks.expr.functions.strings.MD5;
import com.metaweb.gridworks.expr.functions.strings.Match;
import com.metaweb.gridworks.expr.functions.strings.NGram;
import com.metaweb.gridworks.expr.functions.strings.NGramFingerprint;
import com.metaweb.gridworks.expr.functions.strings.Partition;
@ -149,6 +150,7 @@ public class ControlFunctionRegistry {
registerFunction("reinterpret", new Reinterpret());
registerFunction("jsonize", new Jsonize());
registerFunction("ngram", new NGram());
registerFunction("match", new Match());
registerFunction("indexOf", new IndexOf());
registerFunction("lastIndexOf", new LastIndexOf());