1 files changed, 23 insertions, 24 deletions
diff --git a/gi/pyp-topics/scripts/spans2labels.py b/gi/pyp-topics/scripts/spans2labels.py
index 409fda92..3221dbf0 100755
--- a/gi/pyp-topics/scripts/spans2labels.py
+++ b/gi/pyp-topics/scripts/spans2labels.py
@@ -3,44 +3,43 @@
 import sys
 from operator import itemgetter
 
-if len(sys.argv) != 4:
-  print "Usage: spans2labels.py phrase_index context_index phrase_context_index"
+if len(sys.argv) <= 2:
+  print "Usage: spans2labels.py phrase_context_index [order]"
   exit(1)
 
-phrase_index = dict(map(lambda x: (x[1].strip(),x[0]), enumerate(file(sys.argv[1], 'r').readlines())))
-context_index = dict(map(lambda x: (x[1].strip(),x[0]), enumerate(file(sys.argv[2], 'r').readlines())))
+order=1
+if len(sys.argv) > 2:
+  order = int(sys.argv[2])
 
 phrase_context_index = {}
-for i,line in enumerate(file(sys.argv[3], 'r').readlines()):
-  for c,l in map(lambda x: x.split(':'), line.split()[1:]):
-    phrase_context_index[(int(i),int(c))] = l
+for line in file(sys.argv[1], 'r'):
+  phrase,tail= line.split('\t')
+  contexts = tail.split(" ||| ")
+  assert len(contexts) % 2 == 0
+  for i in range(0, len(contexts), 2):
+    category = contexts[i+1].split("=")[1].strip()
+    phrase_context_index[(phrase,contexts[i])] = category
+#   print (phrase,contexts[i]), category
 
 for line in sys.stdin:
   line_segments = line.split('|||')
-  source = ['<s>'] + line_segments[0].split() + ['</s>']
-  target = ['<s>'] + line_segments[1].split() + ['</s>']
+  source = ['<s>' for x in range(order)] + line_segments[0].split() + ['</s>' for x in range(order)]
+  target = ['<s>' for x in range(order)] + line_segments[1].split() + ['</s>' for x in range(order)]
   phrases = [ [int(i) for i in x.split('-')] for x in line_segments[2].split()]
 
-# for x in source[1:-1]: 
-#   print x,
-# print "|||",
-# for x in target[1:-1]: 
-#   print x,
   print "|||",
 
   for s1,s2,t1,t2 in phrases:
-    s1 += 1
-    s2 += 1
-    t1 += 1
-    t2 += 1
+    s1 += order
+    s2 += order
+    t1 += order
+    t2 += order
 
     phrase = reduce(lambda x, y: x+y+" ", target[t1:t2], "").strip()
-    context = "%s <PHRASE> %s" % (target[t1-1], target[t2])
+    left_context = reduce(lambda x, y: x+y+" ", target[t1-order:t1], "")
+    right_context = reduce(lambda x, y: x+y+" ", target[t2:t2+order], "").strip()
+    context = "%s<PHRASE> %s" % (left_context, right_context)
 
-    pi = phrase_index[phrase]
-    ci = context_index[context]
-    label = phrase_context_index[(pi,ci)]
+    label = phrase_context_index[(phrase,context)]
     print "%s-%s:X%s" % (t1-1,t2-1,label),
-#   print phrase, pi, context, ci
-#   print phrase_context_index[(pi,ci)]
   print