| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 1 | package de.ids_mannheim.korap.query.serialize; |
| 2 | |
| 3 | import java.lang.reflect.Method; |
| 4 | import java.util.ArrayList; |
| Joachim Bingel | 7ee0786 | 2014-04-28 15:22:41 +0000 | [diff] [blame] | 5 | import java.util.Arrays; |
| Joachim Bingel | 019ba5c | 2014-04-28 14:59:04 +0000 | [diff] [blame] | 6 | import java.util.HashMap; |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 7 | import java.util.LinkedHashMap; |
| 8 | import java.util.LinkedList; |
| 9 | import java.util.List; |
| 10 | import java.util.Map; |
| Joachim Bingel | fb9d5fd | 2014-06-25 09:32:43 +0000 | [diff] [blame] | 11 | import java.util.NoSuchElementException; |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 12 | |
| 13 | import org.antlr.v4.runtime.ANTLRInputStream; |
| 14 | import org.antlr.v4.runtime.BailErrorStrategy; |
| 15 | import org.antlr.v4.runtime.CharStream; |
| 16 | import org.antlr.v4.runtime.CommonTokenStream; |
| 17 | import org.antlr.v4.runtime.Lexer; |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 18 | import org.antlr.v4.runtime.ParserRuleContext; |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 19 | import org.antlr.v4.runtime.tree.ParseTree; |
| Joachim Bingel | 7503879 | 2014-05-19 15:12:23 +0000 | [diff] [blame] | 20 | import org.slf4j.LoggerFactory; |
| Joachim Bingel | c63f781 | 2014-07-30 09:12:25 +0000 | [diff] [blame] | 21 | import org.slf4j.Logger; |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 22 | |
| Joachim Bingel | 6003b85 | 2014-12-18 14:20:55 +0000 | [diff] [blame] | 23 | import de.ids_mannheim.korap.query.parse.annis.AqlLexer; |
| 24 | import de.ids_mannheim.korap.query.parse.annis.AqlParser; |
| Joachim Bingel | 3fa584b | 2014-12-17 13:35:43 +0000 | [diff] [blame] | 25 | import de.ids_mannheim.korap.query.serialize.util.Antlr4DescriptiveErrorListener; |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 26 | import de.ids_mannheim.korap.query.serialize.util.CqlfObjectGenerator; |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 27 | |
| 28 | /** |
| Joachim Bingel | c8a28e4 | 2014-04-24 15:06:42 +0000 | [diff] [blame] | 29 | * Map representation of ANNIS QL syntax tree as returned by ANTLR |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 30 | * @author joachim |
| 31 | * |
| 32 | */ |
| Joachim Bingel | 1faf8a5 | 2015-01-09 13:17:34 +0000 | [diff] [blame] | 33 | public class AnnisQueryProcessor extends Antlr4AbstractQueryProcessor { |
| 34 | private static Logger log = LoggerFactory.getLogger(AnnisQueryProcessor.class); |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 35 | /** |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 36 | * Flag that indicates whether token fields or meta fields are currently being processed |
| 37 | */ |
| 38 | boolean inMeta = false; |
| 39 | /** |
| Joachim Bingel | 949e3a8 | 2014-06-16 13:20:43 +0000 | [diff] [blame] | 40 | * Keeps track of operands that are to be integrated into yet uncreated objects. |
| 41 | */ |
| 42 | LinkedList<LinkedHashMap<String,Object>> operandStack = new LinkedList<LinkedHashMap<String,Object>>(); |
| 43 | /** |
| Joachim Bingel | c8a28e4 | 2014-04-24 15:06:42 +0000 | [diff] [blame] | 44 | * Keeps track of explicitly (by #-var definition) or implicitly (number as reference) introduced entities (for later reference by #-operator) |
| 45 | */ |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 46 | Map<String, LinkedHashMap<String,Object>> nodeVariables = new LinkedHashMap<String, LinkedHashMap<String,Object>>(); |
| Joachim Bingel | c8a28e4 | 2014-04-24 15:06:42 +0000 | [diff] [blame] | 47 | /** |
| 48 | * Counter for variable definitions. |
| 49 | */ |
| 50 | Integer variableCounter = 1; |
| 51 | /** |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 52 | * Marks the currently active token in order to know where to add flags (might already have been taken away from token stack). |
| 53 | */ |
| 54 | LinkedHashMap<String,Object> curToken = new LinkedHashMap<String,Object>(); |
| Joachim Bingel | 7503879 | 2014-05-19 15:12:23 +0000 | [diff] [blame] | 55 | /** |
| 56 | * Keeps track of operands lists that are to be serialised in an inverted |
| 57 | * order (e.g. the IN() operator) compared to their AST representation. |
| 58 | */ |
| 59 | private LinkedList<ArrayList<Object>> invertedOperandsLists = new LinkedList<ArrayList<Object>>(); |
| Joachim Bingel | 77ac5cb | 2014-04-22 14:03:28 +0000 | [diff] [blame] | 60 | /** |
| Joachim Bingel | fc42881 | 2014-06-18 14:50:14 +0000 | [diff] [blame] | 61 | * Keeps track of operation:class numbers. |
| 62 | */ |
| 63 | int classCounter = 0; |
| 64 | /** |
| Joachim Bingel | 9c3ddb9 | 2014-06-23 13:49:58 +0000 | [diff] [blame] | 65 | * Keeps track of numers of relations processed (important when dealing with multiple predications). |
| 66 | */ |
| 67 | int relationCounter = 0; |
| 68 | /** |
| Joachim Bingel | 019ba5c | 2014-04-28 14:59:04 +0000 | [diff] [blame] | 69 | * Keeps track of references to nodes that are operands of groups (e.g. tree relations). Those nodes appear on the top level of the parse tree |
| 70 | * but are to be integrated into the AqlTree at a later point (namely as operands of the respective group). Therefore, store references to these |
| 71 | * nodes here and exclude the operands from being written into the query map individually. |
| 72 | */ |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 73 | private List<ParseTree> globalLingTermNodes = new ArrayList<ParseTree>(); |
| Joachim Bingel | 9c3ddb9 | 2014-06-23 13:49:58 +0000 | [diff] [blame] | 74 | private int totalRelationCount; |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 75 | /** |
| 76 | * Keeps a record of reference-class-mapping, i.e. which 'class' has been assigned to which #n reference. This is important when introducing korap:reference |
| 77 | * spans to refer back to previously established classes for entities. |
| 78 | */ |
| Joachim Bingel | b001d19 | 2014-06-25 11:34:36 +0000 | [diff] [blame] | 79 | private LinkedHashMap<String, Integer> refClassMapping = new LinkedHashMap<String, Integer>(); |
| Joachim Bingel | 9c3ddb9 | 2014-06-23 13:49:58 +0000 | [diff] [blame] | 80 | private LinkedHashMap<String, Integer> nodeReferencesTotal = new LinkedHashMap<String, Integer>(); |
| 81 | private LinkedHashMap<String, Integer> nodeReferencesProcessed = new LinkedHashMap<String, Integer>(); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 82 | |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 83 | /** |
| 84 | * |
| 85 | * @param tree The syntax tree as returned by ANTLR |
| 86 | * @param parser The ANTLR parser instance that generated the parse tree |
| 87 | */ |
| Joachim Bingel | 1faf8a5 | 2015-01-09 13:17:34 +0000 | [diff] [blame] | 88 | public AnnisQueryProcessor(String query) { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 89 | CqlfObjectGenerator.setQueryProcessor(this); |
| Joachim Bingel | 3d5b69b | 2015-01-14 10:46:44 +0000 | [diff] [blame^] | 90 | process(query); |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 91 | System.out.println(">>> "+requestMap.get("query")+" <<<"); |
| 92 | } |
| 93 | |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 94 | @Override |
| Joachim Bingel | 3d5b69b | 2015-01-14 10:46:44 +0000 | [diff] [blame^] | 95 | public void process(String query) { |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 96 | ParseTree tree = parseAnnisQuery(query); |
| Joachim Bingel | c8a28e4 | 2014-04-24 15:06:42 +0000 | [diff] [blame] | 97 | if (this.parser != null) { |
| 98 | super.parser = this.parser; |
| 99 | } else { |
| 100 | throw new NullPointerException("Parser has not been instantiated!"); |
| 101 | } |
| Joachim Bingel | 84395b2 | 2014-12-18 10:46:18 +0000 | [diff] [blame] | 102 | log.info("Processing Annis query: "+query); |
| Joachim Bingel | e3fe367 | 2014-12-17 18:35:58 +0000 | [diff] [blame] | 103 | if (tree != null) { |
| 104 | log.debug("ANTLR parse tree: "+tree.toStringTree(parser)); |
| 105 | processNode(tree); |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 106 | } |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 107 | } |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 108 | |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 109 | private void processNode(ParseTree node) { |
| 110 | // Top-down processing |
| 111 | if (visited.contains(node)) return; |
| 112 | else visited.add(node); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 113 | |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 114 | String nodeCat = getNodeCat(node); |
| 115 | openNodeCats.push(nodeCat); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 116 | |
| Joachim Bingel | 77ac5cb | 2014-04-22 14:03:28 +0000 | [diff] [blame] | 117 | stackedObjects = 0; |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 118 | |
| Joachim Bingel | 77ac5cb | 2014-04-22 14:03:28 +0000 | [diff] [blame] | 119 | if (verbose) { |
| 120 | System.err.println(" "+objectStack); |
| 121 | System.out.println(openNodeCats); |
| 122 | } |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 123 | |
| 124 | /* |
| 125 | **************************************************************** |
| 126 | **************************************************************** |
| 127 | * Processing individual node categories * |
| 128 | **************************************************************** |
| 129 | **************************************************************** |
| 130 | */ |
| Joachim Bingel | c8a28e4 | 2014-04-24 15:06:42 +0000 | [diff] [blame] | 131 | if (nodeCat.equals("exprTop")) { |
| Joachim Bingel | 84395b2 | 2014-12-18 10:46:18 +0000 | [diff] [blame] | 132 | processExprTop(node); |
| Joachim Bingel | c8a28e4 | 2014-04-24 15:06:42 +0000 | [diff] [blame] | 133 | } |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 134 | |
| Joachim Bingel | c8a28e4 | 2014-04-24 15:06:42 +0000 | [diff] [blame] | 135 | if (nodeCat.equals("andTopExpr")) { |
| Joachim Bingel | 84395b2 | 2014-12-18 10:46:18 +0000 | [diff] [blame] | 136 | processAndTopExpr(node); |
| Joachim Bingel | c8a28e4 | 2014-04-24 15:06:42 +0000 | [diff] [blame] | 137 | } |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 138 | |
| Joachim Bingel | c89952e | 2014-06-06 12:08:42 +0000 | [diff] [blame] | 139 | if (nodeCat.equals("unary_linguistic_term")) { |
| Joachim Bingel | 84395b2 | 2014-12-18 10:46:18 +0000 | [diff] [blame] | 140 | processUnary_linguistic_term(node); |
| Joachim Bingel | c89952e | 2014-06-06 12:08:42 +0000 | [diff] [blame] | 141 | } |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 142 | |
| Joachim Bingel | c89952e | 2014-06-06 12:08:42 +0000 | [diff] [blame] | 143 | if (nodeCat.equals("n_ary_linguistic_term")) { |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 144 | processN_ary_linguistic_term(node); |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 145 | } |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 146 | |
| Joachim Bingel | c8a28e4 | 2014-04-24 15:06:42 +0000 | [diff] [blame] | 147 | if (nodeCat.equals("variableExpr")) { |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 148 | processVariableExpr(node); |
| Joachim Bingel | c8a28e4 | 2014-04-24 15:06:42 +0000 | [diff] [blame] | 149 | } |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 150 | |
| Joachim Bingel | 77ac5cb | 2014-04-22 14:03:28 +0000 | [diff] [blame] | 151 | objectsToPop.push(stackedObjects); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 152 | |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 153 | /* |
| 154 | **************************************************************** |
| 155 | **************************************************************** |
| 156 | * recursion until 'request' node (root of tree) is processed * |
| 157 | **************************************************************** |
| 158 | **************************************************************** |
| 159 | */ |
| 160 | for (int i=0; i<node.getChildCount(); i++) { |
| 161 | ParseTree child = node.getChild(i); |
| 162 | processNode(child); |
| 163 | } |
| Joachim Bingel | 019ba5c | 2014-04-28 14:59:04 +0000 | [diff] [blame] | 164 | |
| Joachim Bingel | 77ac5cb | 2014-04-22 14:03:28 +0000 | [diff] [blame] | 165 | /* |
| 166 | ************************************************************** |
| 167 | * Stuff that happens after processing the children of a node * |
| 168 | ************************************************************** |
| 169 | */ |
| Joachim Bingel | c8a28e4 | 2014-04-24 15:06:42 +0000 | [diff] [blame] | 170 | if (!objectsToPop.isEmpty()) { |
| 171 | for (int i=0; i<objectsToPop.pop(); i++) { |
| 172 | objectStack.pop(); |
| 173 | } |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 174 | } |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 175 | openNodeCats.pop(); |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 176 | } |
| 177 | |
| Joachim Bingel | 84395b2 | 2014-12-18 10:46:18 +0000 | [diff] [blame] | 178 | private void processAndTopExpr(ParseTree node) { |
| 179 | // Before processing any child expr node, check if it has one or more "*ary_linguistic_term" nodes. |
| 180 | // Those nodes may use references to earlier established operand nodes. |
| 181 | // Those operand nodes are not to be included into the query map individually but |
| 182 | // naturally as operands of the relations/groups introduced by the |
| 183 | // *node. For that purpose, this section mines all used references |
| 184 | // and stores them in a list for later reference. |
| 185 | for (ParseTree exprNode : getChildrenWithCat(node,"expr")) { |
| 186 | // Pre-process any 'variableExpr' such that the variableReferences map can be filled |
| 187 | List<ParseTree> definitionNodes = new ArrayList<ParseTree>(); |
| 188 | definitionNodes.addAll(getChildrenWithCat(exprNode, "variableExpr")); |
| 189 | for (ParseTree definitionNode : definitionNodes) { |
| 190 | processNode(definitionNode); |
| 191 | } |
| 192 | // Then, mine all relations between nodes |
| 193 | List<ParseTree> lingTermNodes = new ArrayList<ParseTree>(); |
| 194 | lingTermNodes.addAll(getChildrenWithCat(exprNode, "n_ary_linguistic_term")); |
| 195 | globalLingTermNodes.addAll(lingTermNodes); |
| 196 | totalRelationCount = globalLingTermNodes.size(); |
| 197 | // Traverse refOrNode nodes under *ary_linguistic_term nodes and extract references |
| 198 | for (ParseTree lingTermNode : lingTermNodes) { |
| 199 | for (ParseTree refOrNode : getChildrenWithCat(lingTermNode, "refOrNode")) { |
| 200 | String refOrNodeString = refOrNode.getChild(0).toStringTree(parser); |
| 201 | if (refOrNodeString.startsWith("#")) { |
| 202 | String ref = refOrNode.getChild(0).toStringTree(parser).substring(1); |
| 203 | if (nodeReferencesTotal.containsKey(ref)) { |
| 204 | nodeReferencesTotal.put(ref, nodeReferencesTotal.get(ref)+1); |
| 205 | } else { |
| 206 | nodeReferencesTotal.put(ref, 1); |
| 207 | nodeReferencesProcessed.put(ref, 0); |
| 208 | } |
| 209 | } |
| 210 | } |
| 211 | } |
| 212 | } |
| 213 | System.err.println(nodeVariables); |
| 214 | } |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 215 | |
| Joachim Bingel | 84395b2 | 2014-12-18 10:46:18 +0000 | [diff] [blame] | 216 | private void processUnary_linguistic_term(ParseTree node) { |
| 217 | LinkedHashMap<String, Object> unaryOperator = parseUnaryOperator(node); |
| 218 | String reference = node.getChild(0).toStringTree(parser).substring(1); |
| 219 | LinkedHashMap<String, Object> object = nodeVariables.get(reference); |
| 220 | object.putAll(unaryOperator); |
| 221 | } |
| 222 | |
| 223 | private void processExprTop(ParseTree node) { |
| 224 | List<ParseTree> andTopExprs = getChildrenWithCat(node, "andTopExpr"); |
| 225 | if (andTopExprs.size() > 1) { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 226 | LinkedHashMap<String, Object> topOr = CqlfObjectGenerator.makeGroup("or"); |
| Joachim Bingel | 84395b2 | 2014-12-18 10:46:18 +0000 | [diff] [blame] | 227 | requestMap.put("query", topOr); |
| 228 | objectStack.push(topOr); |
| 229 | } |
| 230 | } |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 231 | |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 232 | private LinkedHashMap<String, Object> processVariableExpr(ParseTree node) { |
| 233 | // simplex word or complex assignment (like qname = textSpec)? |
| 234 | String firstChildNodeCat = getNodeCat(node.getChild(0)); |
| 235 | LinkedHashMap<String, Object> object = null; |
| 236 | if (firstChildNodeCat.equals("node")) { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 237 | object = CqlfObjectGenerator.makeSpan(); |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 238 | } else if (firstChildNodeCat.equals("tok")) { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 239 | object = CqlfObjectGenerator.makeToken(); |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 240 | if (node.getChildCount() > 1) { // empty tokens do not wrap a term |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 241 | LinkedHashMap<String, Object> term = CqlfObjectGenerator.makeTerm(); |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 242 | term.put("layer", "orth"); |
| 243 | object.put("wrap", term); |
| 244 | } |
| 245 | } else if (firstChildNodeCat.equals("qName")) { // only (foundry/)?layer specified |
| 246 | // may be token or span, depending on indicated layer! (e.g. cnx/cat=NP vs mate/pos=NN) |
| 247 | // TODO generalize the list below -> look up layers associated with tokens rather than spans somewhere |
| 248 | HashMap<String, Object> qNameParse = parseQNameNode(node.getChild(0)); |
| 249 | if (Arrays.asList(new String[]{"p", "lemma", "m", "orth"}).contains(qNameParse.get("layer"))) { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 250 | object = CqlfObjectGenerator.makeToken(); |
| 251 | LinkedHashMap<String, Object> term = CqlfObjectGenerator.makeTerm(); |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 252 | object.put("wrap", term); |
| 253 | term.putAll(qNameParse); |
| 254 | } else { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 255 | object = CqlfObjectGenerator.makeSpan(); |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 256 | object.putAll(qNameParse); |
| 257 | } |
| 258 | } else if (firstChildNodeCat.equals("textSpec")) { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 259 | object = CqlfObjectGenerator.makeToken(); |
| 260 | LinkedHashMap<String, Object> term = CqlfObjectGenerator.makeTerm(); |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 261 | object.put("wrap", term); |
| 262 | term.put("layer", "orth"); |
| 263 | term.putAll(parseTextSpec(node.getChild(0))); |
| 264 | } |
| 265 | |
| 266 | if (node.getChildCount() == 3) { // (foundry/)?layer=key specification |
| 267 | if (object.get("@type").equals("korap:token")) { |
| Joachim Bingel | 84395b2 | 2014-12-18 10:46:18 +0000 | [diff] [blame] | 268 | @SuppressWarnings("unchecked") |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 269 | HashMap<String, Object> term = (HashMap<String, Object>) object.get("wrap"); |
| 270 | term.putAll(parseTextSpec(node.getChild(2))); |
| 271 | term.put("match", parseMatchOperator(getFirstChildWithCat(node, "eqOperator"))); |
| 272 | } else { |
| 273 | object.putAll(parseTextSpec(node.getChild(2))); |
| 274 | object.put("match", parseMatchOperator(getFirstChildWithCat(node, "eqOperator"))); |
| 275 | } |
| 276 | } |
| 277 | |
| 278 | if (object != null) { |
| 279 | // if (! operandOnlyNodeRefs.contains(variableCounter.toString())) { |
| 280 | // putIntoSuperObject(object); |
| 281 | // } |
| 282 | if (! getNodeCat(node.getParent().getParent()).equals("n_ary_linguistic_term")) { |
| 283 | putIntoSuperObject(object); |
| 284 | } |
| 285 | ParseTree parentsFirstChild = node.getParent().getChild(0); |
| 286 | if (getNodeCat(parentsFirstChild).endsWith("#")) { |
| 287 | nodeVariables.put(getNodeCat(parentsFirstChild).replaceAll("#", ""), object); |
| 288 | } |
| 289 | nodeVariables.put(variableCounter.toString(), object); |
| 290 | variableCounter++; |
| 291 | } |
| 292 | return object; |
| 293 | } |
| 294 | |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 295 | /** |
| 296 | * Processes an operand node, creating a map for the operand containing all its information |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 297 | * given in the node definition (referenced via '#'). If this node has been referred to and used earlier, |
| 298 | * a reference is created in its place. |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 299 | * The operand will be wrapped in a class group if necessary. |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 300 | * @param operandNode |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 301 | * @return A map object with the appropriate CQLF representation of the operand |
| 302 | */ |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 303 | private LinkedHashMap<String, Object> retrieveOperand(ParseTree operandNode) { |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 304 | LinkedHashMap<String, Object> operand = null; |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 305 | if (!getNodeCat(operandNode.getChild(0)).equals("variableExpr")) { |
| 306 | String ref = operandNode.getChild(0).toStringTree(parser).substring(1); |
| 307 | operand = nodeVariables.get(ref); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 308 | if (nodeReferencesTotal.get(ref) > 1) { |
| 309 | if (nodeReferencesProcessed.get(ref)==0) { |
| 310 | refClassMapping.put(ref, classCounter); |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 311 | operand = CqlfObjectGenerator.wrapInClass(operand, classCounter++); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 312 | } else if (nodeReferencesProcessed.get(ref)>0 && nodeReferencesTotal.get(ref)>1) { |
| 313 | try { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 314 | operand = CqlfObjectGenerator.wrapInReference(operandStack.pop(), refClassMapping.get(ref)); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 315 | } catch (NoSuchElementException e) { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 316 | operand = CqlfObjectGenerator.makeReference(refClassMapping.get(ref)); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 317 | } |
| 318 | } |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 319 | nodeReferencesProcessed.put(ref, nodeReferencesProcessed.get(ref)+1); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 320 | } |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 321 | } else { |
| 322 | operand = processVariableExpr(operandNode.getChild(0)); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 323 | } |
| 324 | return operand; |
| Joachim Bingel | 9c3ddb9 | 2014-06-23 13:49:58 +0000 | [diff] [blame] | 325 | } |
| 326 | |
| 327 | @SuppressWarnings("unchecked") |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 328 | private void processN_ary_linguistic_term(ParseTree node) { |
| 329 | relationCounter++; |
| 330 | // get operator and determine type of group (sequence/treeRelation/relation/...) |
| 331 | // It's possible in Annis QL to concatenate operators, so there may be several operators under one n_ary_linguistic_term node. |
| 332 | // Counter 'i' will iteratively point to all operator nodes (odd-numbered) under this node. |
| 333 | for (int i=1; i<node.getChildCount(); i = i+2) { |
| 334 | ParseTree operandTree1 = node.getChild(i-1); |
| 335 | ParseTree operandTree2 = node.getChild(i+1); |
| 336 | String reltype = getNodeCat(node.getChild(i).getChild(0)); |
| 337 | |
| 338 | LinkedHashMap<String,Object> group = null; |
| 339 | ArrayList<Object> operands = null; |
| 340 | // Retrieve operands. |
| 341 | LinkedHashMap<String, Object> operand1 = retrieveOperand(operandTree1); |
| 342 | LinkedHashMap<String, Object> operand2 = retrieveOperand(operandTree2); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 343 | // 'Proper' n_ary_linguistic_operators receive a considerably different serialisation than 'commonparent' and 'commonancestor'. |
| 344 | // For the latter cases, a dummy span is introduced and declared as a span class that has a dominance relation towards |
| 345 | // the two operands, one after the other, thus resulting in two nested relations! A Poliqarp+ equivalent for A $ B would be |
| 346 | // contains(focus(1:contains({1:<>},A)), B). |
| 347 | // This is modeled here... |
| 348 | if (reltype.equals("commonparent") || reltype.equals("commonancestor")) { |
| 349 | // make an (outer) group and an inner group containing the dummy node or previous relations |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 350 | group = CqlfObjectGenerator.makeGroup("relation"); |
| 351 | LinkedHashMap<String,Object> innerGroup = CqlfObjectGenerator.makeGroup("relation"); |
| 352 | LinkedHashMap<String,Object> relation = CqlfObjectGenerator.makeRelation(); |
| 353 | LinkedHashMap<String,Object> term = CqlfObjectGenerator.makeTerm(); |
| Joachim Bingel | 6289241 | 2014-08-06 15:18:59 +0000 | [diff] [blame] | 354 | term.put("layer", "c"); |
| 355 | relation.put("wrap", term); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 356 | // commonancestor is an indirect commonparent relation |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 357 | if (reltype.equals("commonancestor")) relation.put("boundary", CqlfObjectGenerator.makeBoundary(1, null)); |
| Joachim Bingel | 6289241 | 2014-08-06 15:18:59 +0000 | [diff] [blame] | 358 | group.put("relation", relation); |
| 359 | innerGroup.put("relation", relation); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 360 | // Get operands list before possible re-assignment of 'group' (see following 'if') |
| 361 | ArrayList<Object> outerOperands = (ArrayList<Object>) group.get("operands"); |
| 362 | ArrayList<Object> innerOperands = (ArrayList<Object>) innerGroup.get("operands"); |
| 363 | // for lowest level, add the underspecified node as first operand and wrap it in a class group |
| 364 | if (i == 1) { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 365 | innerOperands.add(CqlfObjectGenerator.wrapInClass(CqlfObjectGenerator.makeSpan(), classCounter)); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 366 | // add the first operand and wrap the whole group in a focusing reference |
| 367 | innerOperands.add(operand1); |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 368 | innerGroup = CqlfObjectGenerator.wrapInReference(innerGroup, classCounter); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 369 | outerOperands.add(innerGroup); |
| 370 | } else { |
| 371 | outerOperands.add(operandStack.pop()); |
| 372 | } |
| 373 | // Lookahead: if next operator is not commonparent or commonancestor, wrap in class for accessibility |
| 374 | if (i < node.getChildCount()-2 && !getNodeCat(node.getChild(i+2).getChild(0)).startsWith("common")) { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 375 | operand2 = CqlfObjectGenerator.wrapInClass(operand2, ++classCounter); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 376 | } |
| 377 | outerOperands.add(operand2); |
| 378 | |
| 379 | // Wrap in another reference object in case other relations are following |
| 380 | if (i < node.getChildCount()-2) { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 381 | group = CqlfObjectGenerator.wrapInReference(group, classCounter); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 382 | } |
| Joachim Bingel | 6289241 | 2014-08-06 15:18:59 +0000 | [diff] [blame] | 383 | // All other n-ary linguistic relations have special 'relation' attributes defined in CQLF and can be |
| 384 | // handled more easily... |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 385 | } else { |
| 386 | LinkedHashMap<String, Object> operatorGroup = parseOperatorNode(node.getChild(i).getChild(0)); |
| 387 | String groupType; |
| 388 | try { |
| 389 | groupType = (String) operatorGroup.get("groupType"); |
| 390 | } catch (ClassCastException | NullPointerException n) { |
| 391 | groupType = "relation"; |
| 392 | } |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 393 | if (groupType.equals("relation") || groupType.equals("treeRelation")) { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 394 | group = CqlfObjectGenerator.makeGroup(groupType); |
| Joachim Bingel | 6289241 | 2014-08-06 15:18:59 +0000 | [diff] [blame] | 395 | LinkedHashMap<String, Object> relation = new LinkedHashMap<String, Object>(); |
| 396 | putAllButGroupType(relation, operatorGroup); |
| Joachim Bingel | 6289241 | 2014-08-06 15:18:59 +0000 | [diff] [blame] | 397 | group.put("relation", relation); |
| Joachim Bingel | e6d73b1 | 2014-09-30 15:34:59 +0000 | [diff] [blame] | 398 | } else if (groupType.equals("sequence")) { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 399 | group = CqlfObjectGenerator.makeGroup(groupType); |
| Joachim Bingel | e6d73b1 | 2014-09-30 15:34:59 +0000 | [diff] [blame] | 400 | putAllButGroupType(group, operatorGroup); |
| 401 | } else if (groupType.equals("position")) { |
| 402 | group = new LinkedHashMap<String,Object>(); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 403 | putAllButGroupType(group, operatorGroup); |
| 404 | } |
| Joachim Bingel | e6d73b1 | 2014-09-30 15:34:59 +0000 | [diff] [blame] | 405 | |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 406 | // Get operands list before possible re-assignment of 'group' (see following 'if') |
| 407 | operands = (ArrayList<Object>) group.get("operands"); |
| Joachim Bingel | 755ada9 | 2014-12-16 13:55:37 +0000 | [diff] [blame] | 408 | |
| 409 | ParseTree leftChildSpec = getFirstChildWithCat(node.getChild(i).getChild(0), "@l"); |
| 410 | ParseTree rightChildSpec = getFirstChildWithCat(node.getChild(i).getChild(0), "@r"); |
| Joachim Bingel | ab1aff4 | 2014-12-16 16:38:00 +0000 | [diff] [blame] | 411 | if (leftChildSpec != null || rightChildSpec != null) { |
| Joachim Bingel | 755ada9 | 2014-12-16 13:55:37 +0000 | [diff] [blame] | 412 | String frame = (leftChildSpec!=null) ? "frames:startswith" : "frames:endswith"; |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 413 | LinkedHashMap<String,Object> positionGroup = CqlfObjectGenerator.makePosition(new String[]{frame}, null); |
| 414 | operand2 = CqlfObjectGenerator.wrapInClass(operand2, ++classCounter); |
| Joachim Bingel | 755ada9 | 2014-12-16 13:55:37 +0000 | [diff] [blame] | 415 | ((ArrayList<Object>) positionGroup.get("operands")).add(group); |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 416 | ((ArrayList<Object>) positionGroup.get("operands")).add(CqlfObjectGenerator.makeReference(classCounter,true)); |
| Joachim Bingel | 755ada9 | 2014-12-16 13:55:37 +0000 | [diff] [blame] | 417 | group = positionGroup; |
| 418 | } |
| 419 | |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 420 | // Wrap in reference object in case other relations are following |
| 421 | if (i < node.getChildCount()-2) { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 422 | group = CqlfObjectGenerator.wrapInReference(group, classCounter); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 423 | } |
| 424 | |
| 425 | // Inject operands. |
| 426 | // -> Case distinction: |
| 427 | if (node.getChildCount()==3) { |
| 428 | // Things are easy when there's just one operator (thus 3 children incl. operands)... |
| 429 | if (operand1 != null) operands.add(operand1); |
| 430 | if (operand2 != null) operands.add(operand2); |
| 431 | } else { |
| Joachim Bingel | 66472b8 | 2014-12-04 16:00:05 +0000 | [diff] [blame] | 432 | // ... but things get a little more complicated here. The AST is of this form: (operand1 operator1 operand2 operator2 operand3 operator3 ...) |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 433 | // but we'll have to serialize it in a nested, binary way: (((operand1 operator1 operand2) operator2 operand3) operator3 ...) |
| 434 | // the following code will do just that: |
| 435 | if (i == 1) { |
| 436 | // for the first operator, include both operands |
| 437 | if (operand1 != null) operands.add(operand1); |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 438 | if (operand2 != null) operands.add(CqlfObjectGenerator.wrapInClass(operand2, classCounter++)); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 439 | // Don't put this into the super object directly but store on operandStack |
| 440 | // (because this group will have to be an operand of a subsequent operator) |
| 441 | operandStack.push(group); |
| 442 | // for all subsequent operators, only take the 2nd operand (first was already added by previous operator) |
| 443 | } else if (i < node.getChildCount()-2) { |
| 444 | // for all intermediate operators, include other previous groups and 2nd operand. Store this on the operandStack, too. |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 445 | if (operand2 != null) operands.add(CqlfObjectGenerator.wrapInClass(operand2, classCounter++)); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 446 | operands.add(0, operandStack.pop()); |
| 447 | operandStack.push(group); |
| 448 | } else if (i == node.getChildCount()-2) { |
| 449 | // This is the last operator. Include 2nd operand only |
| 450 | if (operand2 != null) operands.add(operand2); |
| 451 | } |
| 452 | } |
| 453 | } |
| 454 | // Final step: decide what to do with the 'group' object, depending on whether all relations have been processed |
| 455 | if (i == node.getChildCount()-2 && relationCounter == totalRelationCount) { |
| 456 | putIntoSuperObject(group); |
| 457 | if (!operandStack.isEmpty()) { |
| 458 | operands.add(0, operandStack.pop()); |
| 459 | } |
| 460 | objectStack.push(group); |
| 461 | stackedObjects++; |
| 462 | } else { |
| 463 | operandStack.push(group); |
| 464 | } |
| 465 | } |
| Joachim Bingel | 9c3ddb9 | 2014-06-23 13:49:58 +0000 | [diff] [blame] | 466 | } |
| 467 | |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 468 | |
| 469 | |
| Joachim Bingel | ca4944e | 2014-06-13 13:55:10 +0000 | [diff] [blame] | 470 | /** |
| 471 | * Parses a unary_linguistic_operator node. Possible operators are: root, arity, tokenarity. |
| 472 | * Operators are embedded into a korap:term, in turn wrapped by an 'attr' property in a korap:span. |
| 473 | * @param node The unary_linguistic_operator node |
| 474 | * @return A map containing the attr key, to be inserted into korap:span |
| 475 | */ |
| 476 | private LinkedHashMap<String, Object> parseUnaryOperator(ParseTree node) { |
| 477 | LinkedHashMap<String, Object> attr = new LinkedHashMap<String, Object>(); |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 478 | LinkedHashMap<String, Object> term = CqlfObjectGenerator.makeTerm(); |
| Joachim Bingel | ca4944e | 2014-06-13 13:55:10 +0000 | [diff] [blame] | 479 | String op = node.getChild(1).toStringTree(parser).substring(1); |
| 480 | if (op.equals("arity") || op.equals("tokenarity")) { |
| 481 | LinkedHashMap<String, Object> boundary = boundaryFromRangeSpec(node.getChild(3), false); |
| 482 | term.put(op, boundary); |
| 483 | } else { |
| 484 | term.put(op, true); |
| 485 | } |
| Joachim Bingel | ca4944e | 2014-06-13 13:55:10 +0000 | [diff] [blame] | 486 | attr.put("attr", term); |
| 487 | return attr; |
| 488 | } |
| 489 | |
| Joachim Bingel | 66472b8 | 2014-12-04 16:00:05 +0000 | [diff] [blame] | 490 | @SuppressWarnings("unchecked") |
| Joachim Bingel | eee549e | 2014-04-29 11:15:37 +0000 | [diff] [blame] | 491 | private LinkedHashMap<String, Object> parseOperatorNode(ParseTree operatorNode) { |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 492 | LinkedHashMap<String, Object> relation = null; |
| Joachim Bingel | eee549e | 2014-04-29 11:15:37 +0000 | [diff] [blame] | 493 | String operator = getNodeCat(operatorNode); |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 494 | // DOMINANCE |
| Joachim Bingel | eee549e | 2014-04-29 11:15:37 +0000 | [diff] [blame] | 495 | if (operator.equals("dominance")) { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 496 | relation = CqlfObjectGenerator.makeRelation(); |
| Joachim Bingel | 13421ef | 2014-05-07 16:22:58 +0000 | [diff] [blame] | 497 | relation.put("groupType", "relation"); |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 498 | ParseTree qName = getFirstChildWithCat(operatorNode, "qName"); |
| Joachim Bingel | 66472b8 | 2014-12-04 16:00:05 +0000 | [diff] [blame] | 499 | ParseTree edgeSpecNode = getFirstChildWithCat(operatorNode, "edgeSpec"); |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 500 | ParseTree star = getFirstChildWithCat(operatorNode, "*"); |
| 501 | ParseTree rangeSpec = getFirstChildWithCat(operatorNode, "rangeSpec"); |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 502 | LinkedHashMap<String,Object> term = CqlfObjectGenerator.makeTerm(); |
| Joachim Bingel | 6289241 | 2014-08-06 15:18:59 +0000 | [diff] [blame] | 503 | term.put("layer", "c"); |
| Joachim Bingel | 6289241 | 2014-08-06 15:18:59 +0000 | [diff] [blame] | 504 | if (qName != null) term = parseQNameNode(qName); |
| Joachim Bingel | 66472b8 | 2014-12-04 16:00:05 +0000 | [diff] [blame] | 505 | if (edgeSpecNode != null) { |
| 506 | LinkedHashMap<String,Object> edgeSpec = parseEdgeSpec(edgeSpecNode); |
| 507 | String edgeSpecType = (String) edgeSpec.get("@type"); |
| 508 | if (edgeSpecType.equals("korap:termGroup")) { |
| 509 | ((ArrayList<Object>) edgeSpec.get("operands")).add(term); |
| 510 | term = edgeSpec; |
| 511 | } else { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 512 | term = CqlfObjectGenerator.makeTermGroup("and"); |
| Joachim Bingel | 66472b8 | 2014-12-04 16:00:05 +0000 | [diff] [blame] | 513 | ArrayList<Object> termGroupOperands = (ArrayList<Object>) term.get("operands"); |
| 514 | termGroupOperands.add(edgeSpec); |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 515 | LinkedHashMap<String,Object> constTerm = CqlfObjectGenerator.makeTerm(); |
| Joachim Bingel | 66472b8 | 2014-12-04 16:00:05 +0000 | [diff] [blame] | 516 | constTerm.put("layer", "c"); |
| 517 | termGroupOperands.add(constTerm); |
| 518 | } |
| 519 | } |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 520 | if (star != null) relation.put("boundary", CqlfObjectGenerator.makeBoundary(0, null)); |
| Joachim Bingel | 86195a5 | 2014-07-08 14:29:48 +0000 | [diff] [blame] | 521 | if (rangeSpec != null) relation.put("boundary", boundaryFromRangeSpec(rangeSpec)); |
| Joachim Bingel | 6289241 | 2014-08-06 15:18:59 +0000 | [diff] [blame] | 522 | relation.put("wrap", term); |
| Joachim Bingel | eee549e | 2014-04-29 11:15:37 +0000 | [diff] [blame] | 523 | } |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 524 | else if (operator.equals("pointing")) { |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 525 | // String reltype = operatorNode.getChild(1).toStringTree(parser); |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 526 | relation = CqlfObjectGenerator.makeRelation(); |
| Joachim Bingel | 13421ef | 2014-05-07 16:22:58 +0000 | [diff] [blame] | 527 | relation.put("groupType", "relation"); |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 528 | ParseTree qName = getFirstChildWithCat(operatorNode, "qName"); |
| 529 | ParseTree edgeSpec = getFirstChildWithCat(operatorNode, "edgeSpec"); |
| 530 | ParseTree star = getFirstChildWithCat(operatorNode, "*"); |
| 531 | ParseTree rangeSpec = getFirstChildWithCat(operatorNode, "rangeSpec"); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 532 | // if (qName != null) relation.putAll(parseQNameNode(qName)); |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 533 | LinkedHashMap<String,Object> term = CqlfObjectGenerator.makeTerm(); |
| Joachim Bingel | 6289241 | 2014-08-06 15:18:59 +0000 | [diff] [blame] | 534 | if (qName != null) term.putAll(parseQNameNode(qName)); |
| 535 | if (edgeSpec != null) term.putAll(parseEdgeSpec(edgeSpec)); |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 536 | if (star != null) relation.put("boundary", CqlfObjectGenerator.makeBoundary(0, null)); |
| Joachim Bingel | 86195a5 | 2014-07-08 14:29:48 +0000 | [diff] [blame] | 537 | if (rangeSpec != null) relation.put("boundary", boundaryFromRangeSpec(rangeSpec)); |
| Joachim Bingel | 6289241 | 2014-08-06 15:18:59 +0000 | [diff] [blame] | 538 | relation.put("wrap", term); |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 539 | } |
| 540 | else if (operator.equals("precedence")) { |
| Joachim Bingel | 13421ef | 2014-05-07 16:22:58 +0000 | [diff] [blame] | 541 | relation = new LinkedHashMap<String, Object>(); |
| 542 | relation.put("groupType", "sequence"); |
| 543 | ParseTree rangeSpec = getFirstChildWithCat(operatorNode, "rangeSpec"); |
| 544 | ParseTree star = getFirstChildWithCat(operatorNode, "*"); |
| 545 | ArrayList<Object> distances = new ArrayList<Object>(); |
| 546 | if (star != null) { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 547 | distances.add(CqlfObjectGenerator.makeDistance("w", 0, null)); |
| Joachim Bingel | 13421ef | 2014-05-07 16:22:58 +0000 | [diff] [blame] | 548 | relation.put("distances", distances); |
| 549 | } |
| 550 | if (rangeSpec != null) { |
| 551 | distances.add(parseDistance(rangeSpec)); |
| 552 | relation.put("distances", distances); |
| 553 | } |
| 554 | relation.put("inOrder", true); |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 555 | } |
| 556 | else if (operator.equals("spanrelation")) { |
| Joachim Bingel | e6d73b1 | 2014-09-30 15:34:59 +0000 | [diff] [blame] | 557 | // relation = makeGroup("position"); |
| 558 | // relation.put("groupType", "position"); |
| Joachim Bingel | 7503879 | 2014-05-19 15:12:23 +0000 | [diff] [blame] | 559 | String reltype = operatorNode.getChild(0).toStringTree(parser); |
| Joachim Bingel | e6d73b1 | 2014-09-30 15:34:59 +0000 | [diff] [blame] | 560 | String[] frames = new String[]{}; |
| Joachim Bingel | 7503879 | 2014-05-19 15:12:23 +0000 | [diff] [blame] | 561 | switch (reltype) { |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 562 | case "_=_": |
| Joachim Bingel | 6b491c2 | 2014-10-21 08:35:18 +0000 | [diff] [blame] | 563 | frames = new String[]{"frames:matches"}; |
| Joachim Bingel | e6d73b1 | 2014-09-30 15:34:59 +0000 | [diff] [blame] | 564 | break; |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 565 | case "_l_": |
| Joachim Bingel | 6b491c2 | 2014-10-21 08:35:18 +0000 | [diff] [blame] | 566 | frames = new String[]{"frames:startswith"}; |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 567 | break; |
| 568 | case "_r_": |
| Joachim Bingel | 6b491c2 | 2014-10-21 08:35:18 +0000 | [diff] [blame] | 569 | frames = new String[]{"frames:endswith"}; |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 570 | break; |
| 571 | case "_i_": |
| Joachim Bingel | 6b491c2 | 2014-10-21 08:35:18 +0000 | [diff] [blame] | 572 | frames = new String[]{"frames:contains"};break; |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 573 | case "_o_": |
| Joachim Bingel | 6b491c2 | 2014-10-21 08:35:18 +0000 | [diff] [blame] | 574 | frames = new String[]{"frames:overlapsLeft", "frames:overlapsRight"}; |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 575 | break; |
| 576 | case "_ol_": |
| Joachim Bingel | 6b491c2 | 2014-10-21 08:35:18 +0000 | [diff] [blame] | 577 | frames = new String[]{"frames:overlapsLeft"}; |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 578 | break; |
| 579 | case "_or_": |
| Joachim Bingel | 6b491c2 | 2014-10-21 08:35:18 +0000 | [diff] [blame] | 580 | frames = new String[]{"frames:overlapsRight"}; |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 581 | break; |
| Joachim Bingel | 7503879 | 2014-05-19 15:12:23 +0000 | [diff] [blame] | 582 | } |
| Joachim Bingel | e6d73b1 | 2014-09-30 15:34:59 +0000 | [diff] [blame] | 583 | // relation.put("frames", frames); |
| 584 | // relation.put("sharedClasses", sharedClasses); |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 585 | relation = CqlfObjectGenerator.makePosition(frames, new String[]{}); |
| Joachim Bingel | e6d73b1 | 2014-09-30 15:34:59 +0000 | [diff] [blame] | 586 | relation.put("groupType", "position"); |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 587 | } |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 588 | else if (operator.equals("identity")) { |
| Joachim Bingel | 66472b8 | 2014-12-04 16:00:05 +0000 | [diff] [blame] | 589 | //TODO since ANNIS v. 3.1.6 |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 590 | } |
| 591 | else if (operator.equals("equalvalue")) { |
| Joachim Bingel | 66472b8 | 2014-12-04 16:00:05 +0000 | [diff] [blame] | 592 | //TODO since ANNIS v. 3.1.6 |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 593 | } |
| 594 | else if (operator.equals("notequalvalue")) { |
| Joachim Bingel | 66472b8 | 2014-12-04 16:00:05 +0000 | [diff] [blame] | 595 | //TODO since ANNIS v. 3.1.6 |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 596 | } |
| 597 | return relation; |
| 598 | } |
| 599 | |
| Joachim Bingel | ceb7906 | 2014-09-22 11:50:37 +0000 | [diff] [blame] | 600 | @SuppressWarnings("unchecked") |
| Joachim Bingel | 6289241 | 2014-08-06 15:18:59 +0000 | [diff] [blame] | 601 | private LinkedHashMap<String,Object> parseEdgeSpec(ParseTree edgeSpec) { |
| 602 | List<ParseTree> annos = getChildrenWithCat(edgeSpec, "edgeAnno"); |
| 603 | if (annos.size() == 1) return parseEdgeAnno(annos.get(0)); |
| 604 | else { |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 605 | LinkedHashMap<String,Object> termGroup = CqlfObjectGenerator.makeTermGroup("and"); |
| Joachim Bingel | 6289241 | 2014-08-06 15:18:59 +0000 | [diff] [blame] | 606 | ArrayList<Object> operands = (ArrayList<Object>) termGroup.get("operands"); |
| 607 | for (ParseTree anno : annos) { |
| 608 | operands.add(parseEdgeAnno(anno)); |
| 609 | } |
| 610 | return termGroup; |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 611 | } |
| Joachim Bingel | eee549e | 2014-04-29 11:15:37 +0000 | [diff] [blame] | 612 | } |
| 613 | |
| Joachim Bingel | 66472b8 | 2014-12-04 16:00:05 +0000 | [diff] [blame] | 614 | private LinkedHashMap<String, Object> parseEdgeAnno(ParseTree edgeAnnoSpec) { |
| Joachim Bingel | eee549e | 2014-04-29 11:15:37 +0000 | [diff] [blame] | 615 | LinkedHashMap<String, Object> edgeAnno = new LinkedHashMap<String, Object>(); |
| Joachim Bingel | a07b8e7 | 2014-05-09 15:06:07 +0000 | [diff] [blame] | 616 | edgeAnno.put("@type", "korap:term"); |
| Joachim Bingel | 66472b8 | 2014-12-04 16:00:05 +0000 | [diff] [blame] | 617 | ParseTree textSpecNode = getFirstChildWithCat(edgeAnnoSpec, "textSpec"); |
| 618 | ParseTree layerNode = getFirstChildWithCat(edgeAnnoSpec, "layer"); |
| 619 | ParseTree foundryNode = getFirstChildWithCat(edgeAnnoSpec, "foundry"); |
| 620 | ParseTree matchOperatorNode = getFirstChildWithCat(edgeAnnoSpec, "eqOperator"); |
| Joachim Bingel | 639df9c | 2014-05-12 15:27:00 +0000 | [diff] [blame] | 621 | if (foundryNode!=null) edgeAnno.put("foundry", foundryNode.getChild(0).toStringTree(parser)); |
| 622 | if (layerNode!=null) edgeAnno.put("layer", layerNode.getChild(0).toStringTree(parser)); |
| Joachim Bingel | eee549e | 2014-04-29 11:15:37 +0000 | [diff] [blame] | 623 | edgeAnno.putAll(parseTextSpec(textSpecNode)); |
| 624 | edgeAnno.put("match", parseMatchOperator(matchOperatorNode)); |
| 625 | return edgeAnno; |
| 626 | } |
| 627 | |
| Joachim Bingel | a07b8e7 | 2014-05-09 15:06:07 +0000 | [diff] [blame] | 628 | private LinkedHashMap<String, Object> boundaryFromRangeSpec(ParseTree rangeSpec) { |
| Joachim Bingel | ca4944e | 2014-06-13 13:55:10 +0000 | [diff] [blame] | 629 | return boundaryFromRangeSpec(rangeSpec, true); |
| 630 | } |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 631 | |
| Joachim Bingel | ca4944e | 2014-06-13 13:55:10 +0000 | [diff] [blame] | 632 | private LinkedHashMap<String, Object> boundaryFromRangeSpec(ParseTree rangeSpec, boolean expandToMax) { |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 633 | Integer min = Integer.parseInt(rangeSpec.getChild(0).toStringTree(parser)); |
| Joachim Bingel | ca4944e | 2014-06-13 13:55:10 +0000 | [diff] [blame] | 634 | Integer max = min; |
| Joachim Bingel | 687e4d4 | 2014-07-30 09:34:18 +0000 | [diff] [blame] | 635 | if (expandToMax) max = null; |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 636 | if (rangeSpec.getChildCount()==3) |
| 637 | max = Integer.parseInt(rangeSpec.getChild(2).toStringTree(parser)); |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 638 | return CqlfObjectGenerator.makeBoundary(min, max); |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 639 | } |
| Joachim Bingel | a07b8e7 | 2014-05-09 15:06:07 +0000 | [diff] [blame] | 640 | |
| Joachim Bingel | 13421ef | 2014-05-07 16:22:58 +0000 | [diff] [blame] | 641 | private LinkedHashMap<String, Object> parseDistance(ParseTree rangeSpec) { |
| 642 | Integer min = Integer.parseInt(rangeSpec.getChild(0).toStringTree(parser)); |
| Joachim Bingel | 687e4d4 | 2014-07-30 09:34:18 +0000 | [diff] [blame] | 643 | Integer max = null; |
| Joachim Bingel | 13421ef | 2014-05-07 16:22:58 +0000 | [diff] [blame] | 644 | if (rangeSpec.getChildCount()==3) |
| 645 | max = Integer.parseInt(rangeSpec.getChild(2).toStringTree(parser)); |
| Joachim Bingel | 0e54d22 | 2015-01-12 13:22:16 +0000 | [diff] [blame] | 646 | return CqlfObjectGenerator.makeDistance("w", min, max); |
| Joachim Bingel | 13421ef | 2014-05-07 16:22:58 +0000 | [diff] [blame] | 647 | } |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 648 | |
| Joachim Bingel | eee549e | 2014-04-29 11:15:37 +0000 | [diff] [blame] | 649 | private LinkedHashMap<String, Object> parseTextSpec(ParseTree node) { |
| 650 | LinkedHashMap<String, Object> term = new LinkedHashMap<String, Object>(); |
| Joachim Bingel | 019ba5c | 2014-04-28 14:59:04 +0000 | [diff] [blame] | 651 | if (hasChild(node, "regex")) { |
| 652 | term.put("type", "type:regex"); |
| 653 | term.put("key", node.getChild(0).getChild(0).toStringTree(parser).replaceAll("/", "")); |
| 654 | } else { |
| 655 | term.put("key", node.getChild(1).toStringTree(parser)); |
| 656 | } |
| 657 | term.put("match", "match:eq"); |
| 658 | return term; |
| 659 | } |
| 660 | |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 661 | /** |
| Joachim Bingel | c8a28e4 | 2014-04-24 15:06:42 +0000 | [diff] [blame] | 662 | * Parses the match operator (= or !=) |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 663 | * @param node |
| 664 | * @return |
| 665 | */ |
| Joachim Bingel | c8a28e4 | 2014-04-24 15:06:42 +0000 | [diff] [blame] | 666 | private String parseMatchOperator(ParseTree node) { |
| Joachim Bingel | 66472b8 | 2014-12-04 16:00:05 +0000 | [diff] [blame] | 667 | if (node.getChildCount()>0) { |
| 668 | return node.getChild(0).toStringTree(parser).equals("=") ? "match:eq" : "match:ne"; |
| 669 | } |
| 670 | return null; |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 671 | } |
| Joachim Bingel | c8a28e4 | 2014-04-24 15:06:42 +0000 | [diff] [blame] | 672 | |
| 673 | private LinkedHashMap<String, Object> parseQNameNode(ParseTree node) { |
| 674 | LinkedHashMap<String, Object> fields = new LinkedHashMap<String, Object>(); |
| Joachim Bingel | d4ae5fd | 2014-04-29 15:00:16 +0000 | [diff] [blame] | 675 | ParseTree layerNode = getFirstChildWithCat(node, "layer"); |
| 676 | ParseTree foundryNode = getFirstChildWithCat(node, "foundry"); |
| 677 | if (foundryNode != null) fields.put("foundry", foundryNode.getChild(0).toStringTree(parser)); |
| Joachim Bingel | 66472b8 | 2014-12-04 16:00:05 +0000 | [diff] [blame] | 678 | String layer = layerNode.getChild(0).toStringTree(parser); |
| 679 | if (layer.equals("pos")) layer = "p"; |
| 680 | if (layer.equals("cat")) layer = "c"; |
| 681 | fields.put("layer", layer); |
| Joachim Bingel | c8a28e4 | 2014-04-24 15:06:42 +0000 | [diff] [blame] | 682 | return fields; |
| 683 | } |
| 684 | |
| Joachim Bingel | 77ac5cb | 2014-04-22 14:03:28 +0000 | [diff] [blame] | 685 | private void putIntoSuperObject(LinkedHashMap<String, Object> object) { |
| 686 | putIntoSuperObject(object, 0); |
| 687 | } |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 688 | |
| Joachim Bingel | 77ac5cb | 2014-04-22 14:03:28 +0000 | [diff] [blame] | 689 | @SuppressWarnings({ "unchecked" }) |
| 690 | private void putIntoSuperObject(LinkedHashMap<String, Object> object, int objStackPosition) { |
| Joachim Bingel | 9c3ddb9 | 2014-06-23 13:49:58 +0000 | [diff] [blame] | 691 | if (objectStack.size()>objStackPosition) { |
| Joachim Bingel | 77ac5cb | 2014-04-22 14:03:28 +0000 | [diff] [blame] | 692 | ArrayList<Object> topObjectOperands = (ArrayList<Object>) objectStack.get(objStackPosition).get("operands"); |
| Joachim Bingel | 7503879 | 2014-05-19 15:12:23 +0000 | [diff] [blame] | 693 | if (!invertedOperandsLists.contains(topObjectOperands)) { |
| 694 | topObjectOperands.add(object); |
| 695 | } else { |
| 696 | topObjectOperands.add(0, object); |
| 697 | } |
| Joachim Bingel | 77ac5cb | 2014-04-22 14:03:28 +0000 | [diff] [blame] | 698 | } else { |
| 699 | requestMap.put("query", object); |
| 700 | } |
| 701 | } |
| Joachim Bingel | 9c3ddb9 | 2014-06-23 13:49:58 +0000 | [diff] [blame] | 702 | |
| Joachim Bingel | 13421ef | 2014-05-07 16:22:58 +0000 | [diff] [blame] | 703 | private void putAllButGroupType(Map<String, Object> container, Map<String, Object> input) { |
| 704 | for (String key : input.keySet()) { |
| 705 | if (!key.equals("groupType")) { |
| 706 | container.put(key, input.get(key)); |
| 707 | } |
| 708 | } |
| 709 | } |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 710 | |
| Joachim Bingel | 3d5b69b | 2015-01-14 10:46:44 +0000 | [diff] [blame^] | 711 | private ParserRuleContext parseAnnisQuery (String query) { |
| Joachim Bingel | 3fa584b | 2014-12-17 13:35:43 +0000 | [diff] [blame] | 712 | Lexer lexer = new AqlLexer((CharStream)null); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 713 | ParserRuleContext tree = null; |
| Joachim Bingel | 3fa584b | 2014-12-17 13:35:43 +0000 | [diff] [blame] | 714 | Antlr4DescriptiveErrorListener errorListener = new Antlr4DescriptiveErrorListener(query); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 715 | // Like p. 111 |
| 716 | try { |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 717 | |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 718 | // Tokenize input data |
| Joachim Bingel | 3fa584b | 2014-12-17 13:35:43 +0000 | [diff] [blame] | 719 | ANTLRInputStream input = new ANTLRInputStream(query); |
| 720 | lexer.setInputStream(input); |
| 721 | CommonTokenStream tokens = new CommonTokenStream(lexer); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 722 | parser = new AqlParser(tokens); |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 723 | |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 724 | // Don't throw out erroneous stuff |
| 725 | parser.setErrorHandler(new BailErrorStrategy()); |
| Joachim Bingel | 3fa584b | 2014-12-17 13:35:43 +0000 | [diff] [blame] | 726 | lexer.removeErrorListeners(); |
| 727 | lexer.addErrorListener(errorListener); |
| 728 | parser.removeErrorListeners(); |
| 729 | parser.addErrorListener(errorListener); |
| Joachim Bingel | dc03c00 | 2014-04-17 13:40:40 +0000 | [diff] [blame] | 730 | |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 731 | // Get starting rule from parser |
| 732 | Method startRule = AqlParser.class.getMethod("start"); |
| 733 | tree = (ParserRuleContext) startRule.invoke(parser, (Object[])null); |
| 734 | } |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 735 | // Some things went wrong ... |
| 736 | catch (Exception e) { |
| Joachim Bingel | e3fe367 | 2014-12-17 18:35:58 +0000 | [diff] [blame] | 737 | log.error("Could not parse query. Please make sure it is well-formed."); |
| 738 | log.error(errorListener.generateFullErrorMsg().toString()); |
| 739 | addError(errorListener.generateFullErrorMsg()); |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 740 | } |
| Joachim Bingel | 1846c8c | 2014-07-08 14:13:31 +0000 | [diff] [blame] | 741 | return tree; |
| 742 | } |
| Joachim Bingel | 761d1c1 | 2014-12-17 14:02:40 +0000 | [diff] [blame] | 743 | } |