1 files changed, 24 insertions, 0 deletions
diff --git a/split-kbest b/split-kbest
new file mode 100755
index 0000000..ab425b0
--- /dev/null
+++ b/split-kbest
@@ -0,0 +1,24 @@
+#!/usr/bin/env ruby
+
+require 'zipf'
+
+def write_kbest l, fn
+  f = WriteFile.new fn
+  f.write l.join("")
+  f.close
+end
+
+dir = ARGV[0]
+i = 0
+l = []
+while line = STDIN.gets
+  j = line.split.first.to_i
+  if j == 0 && l.size > 0
+    write_kbest l, "#{dir}/#{i}.gz"
+    l = []
+    i += 1
+  end
+  l << line
+end
+write_kbest l, "#{dir}/#{i}.gz" # last one
+