From 98a18843da1d8a327d98fc17a1d2e199535f1b7f Mon Sep 17 00:00:00 2001
From: krangelov <kr.angelov@gmail.com>
Date: Thu, 13 Feb 2020 14:51:03 +0100
Subject: [PATCH] support command c-lookupCohorts

---
 src/server/PGFService.hs | 30 ++++++++++++++++++++++++++++--
 1 file changed, 28 insertions(+), 2 deletions(-)

diff --git a/src/server/PGFService.hs b/src/server/PGFService.hs
index 927d58310..fa515e018 100644
--- a/src/server/PGFService.hs
+++ b/src/server/PGFService.hs
@@ -159,6 +159,7 @@ cpgfMain qsem command (t,(pgf,pc)) =
     "c-translate"   -> withQSem qsem $
                        out t=<<join(trans # input % to % start % limit%treeopts)
     "c-lookupmorpho"-> out t=<< morpho # from1 % textInput
+    "c-lookupcohorts"->out t=<< cohorts # from1 % getInput "longest" % textInput
     "c-flush"       -> out t=<< flush
     "c-grammar"     -> out t grammar
     "c-abstrtree"   -> outputGraphviz=<< C.graphvizAbstractTree pgf C.graphvizDefaults # tree
@@ -245,9 +246,34 @@ cpgfMain qsem command (t,(pgf,pc)) =
                                     | (tree,prob) <- parses]
 
     morpho (from,concr) input =
-        showJSON [makeObj ["lemma".=l,"analysis".=a,"prob".=p]|(l,a,p)<-ms]
-      where ms = C.lookupMorpho concr input
+        showJSON [makeObj ["lemma".=l
+                          ,"analysis".=a
+                          ,"prob".=p]
+                     | (l,a,p)<-C.lookupMorpho concr input]
 
+    cohorts (from,concr) longest input =
+      showJSON [makeObj ["start" .=showJSON s
+                        ,"morpho".=showJSON [makeObj ["lemma".=l
+                                                     ,"analysis".=a
+                                                     ,"prob".=p] 
+                                                | (l,a,p)<-ms]
+                        ,"end"   .=showJSON e
+                        ]
+                   | (s,ms,e) <- (if longest==Just "true" then filterLongest else id)
+                                    (C.lookupCohorts concr input)]
+      where
+        filterLongest []       = []
+        filterLongest (an:ans) = longest an ans
+          where
+            longest prev                   [] = [prev]
+            longest prev@(start0,_,end0) (next@(start,an,end):ans)
+              | start0 == start               = longest next ans
+              | otherwise                     = prev : filter end0 (next:ans)
+
+            filter end                     [] = []
+            filter end (next@(start,_,_):ans)
+              | end <= start                  = filterLongest (next:ans)
+              | otherwise                     = filter end ans
 
     wordforword input@((from,_),_) = jsonWFW from . wordforword' input