From 8e3802303408f6ad4b1299a59fc27f3954174992 Mon Sep 17 00:00:00 2001
From: Patrick Simianer <p@simianer.de>
Date: Sun, 21 Sep 2014 14:52:55 +0100
Subject: add_seg: option to use pre-defined index

---
 add_seg | 21 +++++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

(limited to 'add_seg')

diff --git a/add_seg b/add_seg
index 684a236..c7fd521 100755
--- a/add_seg
+++ b/add_seg
@@ -1,26 +1,35 @@
 #!/usr/bin/env ruby
 
 require 'trollop'
-
-STDIN.set_encoding 'utf-8'
-STDOUT.set_encoding 'utf-8'
-
+require 'zipf'
 
 cfg = Trollop::options do
   opt :grammar, "(Absolute) path of folder containing grammars.", :type => :string, :short => '-g', :required => true
   opt :loo, "leave one out", :type => :bool, :default => false
   opt :start_id, "start with this id", :type => :int, :default => 0, :short => '-i'
   opt :nogz, "grammar files not gzipped", :type => :bool, :default => false
+  opt :index, "number according to index", :type => :string, :default => nil
+end
+
+index = []
+if cfg[:index]
+  index = ReadFile.readlines_strip(cfg[:index]).map{ |i| i.to_i }
 end
 
 i = cfg[:start_id]
+j = 0
 while line = STDIN.gets
   ext = '.gz'
   ext = '' if cfg[:nogz]
   s = "<seg"
   if cfg[:loo] then s += " exclude=\"#{i}\"" end
   if cfg[:grammar] then s += " grammar=\"#{cfg[:grammar]}/grammar.#{i}#{ext}\"" end
-  puts s + " id=\"#{i}\"> #{line.strip} </seg>"
-  i+=1
+  if index.size > 0
+    puts s + " id=\"#{index[j]}\"> #{line.strip} </seg>"
+  else
+    puts s + " id=\"#{i}\"> #{line.strip} </seg>"
+  end
+  i += 1
+  j += 1
 end
 
-- 
cgit v1.2.3