added some more needed classes

FultonBrowne · Nov 16, 2019 · ce56785 · ce56785
1 parent a8e6f9c
commit ce56785
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/src/main/java/com/andromeda/araserver/util/SortWords.kt b/src/main/java/com/andromeda/araserver/util/SortWords.kt
@@ -1,11 +1,14 @@
 package com.andromeda.araserver.util
 
 import opennlp.tools.parser.Parse
+import java.util.ArrayList
 
 class SortWords(keyWord: KeyWord, mainVal: String) {
     val key = keyWord
     val mainText = mainVal
     fun getTopics() {
+        var toReturn = ArrayList<WordGraph>()
+
         var graph = key.getKeyWords(mainText)?.get(0)
 
         var working = true
@@ -14,15 +17,13 @@ class SortWords(keyWord: KeyWord, mainVal: String) {
             while (working) {
                 when {
                     graph?.childCount == 1 -> {graph = graph.children?.get(0)
-                    println("is 1")
                     print(graph?.type)}
                     graph?.childCount == 0 -> {
                         working = false
-                        println("stop")
                     }
                     else -> {
                         if (graph != null) {
-                            sort(graph)
+                            sortNouns(graph)
                         }
                         working = false
                         }
@@ -35,13 +36,10 @@ class SortWords(keyWord: KeyWord, mainVal: String) {
 
 
     }
-    private fun sort(graph: Parse){
+    private fun sortNouns(graph: Parse){
         for (i in graph.children!!) {
-            //println("go 2")
             println(i.type)
-            if (i.childCount > 0) sort(i)
-
-
+            if (i.childCount > 0) sortNouns(i)
             }
 
     }

diff --git a/src/main/java/com/andromeda/araserver/util/WordGraph.kt b/src/main/java/com/andromeda/araserver/util/WordGraph.kt
@@ -0,0 +1,3 @@
+package com.andromeda.araserver.util
+
+data class WordGraph(val word:String, val type:Int)