added NNS sorting system

FultonBrowne · Nov 23, 2019 · 26e327b · 26e327b
1 parent 0f36724
commit 26e327b
Show file tree

Hide file tree

Showing 2 changed files with 54 additions and 2 deletions.
diff --git a/src/main/java/com/andromeda/araserver/Run.java b/src/main/java/com/andromeda/araserver/Run.java
@@ -57,7 +57,8 @@ private Run(int port) throws IOException {
         model = new ParserModel(is);
         parser = ParserFactory.create(model);
         keyWord = new KeyWord(is);
-        //System.out.println(new SortWords(keyWord, "where is a starbucks near me").getTopics(parser));
+        System.out.println(" reeeeesut");
+        System.out.println("                                                "  +new SortWords(keyWord, "where is a starbucks near me").getNNS(parser));
 
     }
     // Static function, to be run on start.

diff --git a/src/main/java/com/andromeda/araserver/util/SortWords.kt b/src/main/java/com/andromeda/araserver/util/SortWords.kt
@@ -52,14 +52,65 @@ class SortWords(keyWord: KeyWord, mainVal: String) {
     private fun sortForNoun(graph: Parse): ArrayList<WordGraph> {
         val toReturn = ArrayList<WordGraph>()
         for (i in graph.children!!) {
-            print(i.coveredText+ " " + i.type)
             if (i.type == "NN" || i.type == "JJ"|| i.type == "PRP"|| i.type == "IN") {
                 toReturn.add(WordGraph(i.coveredText, i.type))
             }
             if (i.childCount > 0) toReturn.addAll(sortForNoun(i))
         }
         return toReturn
 
+    }
+    fun getNNS(parse: Parser): String {
+        var toReturn = "qwertyuio"
+
+        var graph = key.getKeyWords(mainText,parse)?.get(0)
+        graph?.show()
+
+
+        var working = true
+        if (graph != null) {
+            graph.show()
+
+            while (working) {
+                when {
+                    graph?.childCount == 1 -> {
+                        graph = graph.children?.get(0)
+                    }
+                    graph?.childCount == 0 -> {
+                        working = false
+                    }
+                    else -> {
+                        if (graph != null) {
+                            val toTest = sortForNNS(graph)
+                            if(toTest != "")toReturn = toTest
+                        }
+                        working = false
+                    }
+
+                }
+            }
+        } else {
+            print("null")
+            println("fail")
+        }
+        return toReturn
+
+
+    }
+
+    private fun sortForNNS(graph: Parse): String {
+        var toReturn = ""
+        for (i in graph.children!!) {
+            print(i.coveredText+ " " + i.type)
+            if (i.type == "NNS" ) {
+                toReturn = i.coveredText
+            }
+            if (i.childCount > 0){ val toTest = sortForNNS(i)
+                if(toTest != "")toReturn = toTest
+            }
+        }
+        return toReturn
+
     }
 
     fun getActions() {