half of XPath

Author: wilsonli5000

xquery-engine/.idea/workspace.xml

@@ -18,4 +18,8 @@
         <description>Two eggs, bacon or sausage, toast, and our ever-popular hash browns</description>
         <calories>950</calories>
     </food>
+    <drink>
+            <name>Coco</name>
+            <price>$1.09</price>
+    </drink>
 </breakfast_menu>

xquery-engine/out/production/xquery-engine/src/main/java/Contex.class

@@ -25,6 +25,10 @@ public int getLength(){
         return nodeList.size();
     }
 
+    public Node item(int i){
+        return nodeList.get(i);
+    }
+
     public Node getCurrentNode(){
         return nodeList.peek();
     }

xquery-engine/out/production/xquery-engine/src/main/java/Main.class

@@ -15,7 +15,7 @@
  */
 public class Main {
 
-    public static String query = "document(\"pom.xml\")/*";
+    public static String query = "document(\"test.xml\")/*/drink/name/text()";
 
     public static void main(String[] args) throws Exception{
 
@@ -29,8 +29,8 @@ public static void main(String[] args) throws Exception{
         MyVisitor myVisitor = new MyVisitor();
         LinkedList<Node> res = (LinkedList<Node>) myVisitor.visit(tree);
 
-        for (int i=0; i<res.size(); i++) {
-            System.out.println(res.get(i).getNodeName());
+        for (Node node : res) {
+            System.out.println("query node list: " + node.getNodeValue());
         }
 
     }

xquery-engine/out/production/xquery-engine/src/main/java/MyVisitor.class

@@ -10,8 +10,10 @@
 import javax.xml.parsers.DocumentBuilder;
 import javax.xml.parsers.DocumentBuilderFactory;
 import java.io.File;
+import java.util.HashMap;
 import java.util.LinkedList;
 
+
 /**
  * Created by wilsonli on 5/10/16.
  */
@@ -20,11 +22,23 @@ public class MyVisitor extends XqueryBaseVisitor<Object> {
     /* use this stack to store the context for pass */
     private LinkedList<Contex> stack = new LinkedList<Contex>();
 
+    private static LinkedList<Node> unique(LinkedList<Node> nodeList){
+        System.out.println("uniquing.");
+        HashMap<Node, Boolean> hash = new HashMap<>();
+        LinkedList<Node> res = new LinkedList<>();
+        for (Node node : nodeList){
+            if (!hash.containsKey(node))
+                res.add(node);
+        }
+        return res;
+    }
+
     @Override public Object visitAPChildren(XqueryParser.APChildrenContext ctx) {
         /*
             for rule [doc(fileName)/rp]A:
             get the root node, which is fname and pass it to rp
         */
+
         String fname = (String) visit(ctx.doc());
         File xmlFile = new File(fname);
         // open the xml file //
@@ -38,11 +52,9 @@ public class MyVisitor extends XqueryBaseVisitor<Object> {
         } catch (Exception ex) {
             ex.printStackTrace();
         }
-
         return visit(ctx.rp());
     }
 
-
     @Override public Object visitAPAll(XqueryParser.APAllContext ctx) {
         /* get the  */
 
@@ -74,17 +86,73 @@ public class MyVisitor extends XqueryBaseVisitor<Object> {
         Contex current = stack.peek();
         LinkedList<Node> res = new LinkedList<>();
         for (int i = 0; i < current.getLength(); i++){
-            for (int j = 0; j < current.item(i).getChildNodes().getLength(); j++)
-                res.add(current.item(i).getChildNodes().item(j));
+            NodeList tmp = current.item(i).getChildNodes();
+            for (int j = 0; j < tmp.getLength(); j++)
+                res.add(tmp.item(j));
         }
         return res;
     }
 
-
-
     @Override public Object visitTagName(XqueryParser.TagNameContext ctx) {
+        System.out.println("TagName running.");
         LinkedList<Node> res = new LinkedList<>();
+        String tagName = (String) ctx.getText();
+
+        for (Node node : stack.peek().getChildNodes()){
+            if (tagName.equals(node.getNodeName()))
+                res.add(node);
+        }
+        System.out.println("TagName stop.");
+        return res;
+    }
+
+    @Override public Object visitRPwithPar(XqueryParser.RPwithParContext ctx) {
+        System.out.println("rp with parameter running.");
+        return visit(ctx.rp());
+    }
+
+    @Override public Object visitRPChildren(XqueryParser.RPChildrenContext ctx) {
+        System.out.println("RPChildren running.");
+        LinkedList<Node> x = (LinkedList<Node>) visit(ctx.rp(0));
+        Contex middle = new Contex();
+        for (Node node : x)
+            middle.add(node);
+        stack.push(middle);
+        LinkedList<Node> y = (LinkedList<Node>) visit(ctx.rp(1));
+        System.out.println(y.size());
+        LinkedList<Node> res = unique(y);
+        System.out.println(res.size());
+        stack.pop();
+        return res;
+    }
+
+    // i don't think this is correct. //
+    @Override public Object visitRPCondition(XqueryParser.RPConditionContext ctx) {
+        System.out.println("[rp[f]]r(n) running.");
+        LinkedList<Node> x = (LinkedList<Node>) visit(ctx.rp());
+        Contex middle = new Contex();
+        for (Node node : x)
+            middle.add(node);
+        stack.push(middle);
+        LinkedList<Node> res = (LinkedList<Node>) visit(ctx.fltr());
+        return res;
+    }
+
+    @Override public Object visitTwoRP(XqueryParser.TwoRPContext ctx) {
+        System.out.println("[rp1,rp2]r(n) running.");
+        LinkedList<Node> x = (LinkedList<Node>) visit(ctx.rp(0));
+        System.out.println(x.size());
+        LinkedList<Node> y = (LinkedList<Node>) visit(ctx.rp(1));
+        x.addLast(y.getFirst());
+        System.out.println(x.size());
+
+        return x;
+    }
+
+    @Override public Object visitText(XqueryParser.TextContext ctx) {
+        System.out.println("text() running.");
+        Contex current = stack.peek();
+        return current.getChildNodes();
 
-        return visitChildren(ctx);
     }
 }

xquery-engine/out/production/xquery-engine/src/main/resources/test.xml xquery-engine/out/production/xquery-engine/test.xml

@@ -18,4 +18,8 @@
         <description>Two eggs, bacon or sausage, toast, and our ever-popular hash browns</description>
         <calories>950</calories>
     </food>
+    <drink>
+            <name>Coco</name>
+            <price>$1.09</price>
+    </drink>
 </breakfast_menu>