chise/chisedb.rb

   1 # Copyright (C) 2002-2004 Kouichirou Eto, All rights reserved.
   2
   3 require "singleton"
   4 require "chise/libchise"
   5
   6 module CHISE
   7   module ParseValueModule
   8     def parse_value(v)
   9       return nil if v.nil?
  10       return v.to_i if /\A\d+\Z/ =~ v # number?
  11       return $1 if /\A"(.+)"\Z/ =~ v # remove surrounding "
  12       v # sexp?
  13     end
  14   end
  15
  16   class ChiseDB
  17     include Singleton
  18
  19     def initialize
  20       @location = CHISE::DataSource::DB_DIR.path
  21       @ds = DataSource.new(CHISE::DataSource::Berkeley_DB, @location.to_s, 0, 0755)
  22       @feature_db = {}
  23       @ccs_db = {}
  24       @byids_db = {}
  25     end
  26     attr_reader :ds, :location
  27
  28     def close
  29       # @ds.close if @ds # do not close for now
  30     end
  31
  32     def each_feature_name()
  33       @ds.each_feature_name {|f|
  34         next if f.to_s == "." || f.to_s == ".."
  35         next if f.to_s =~ /\.txt\Z/
  36         yield(f.path.unescape_win_filename.unescape.to_s)
  37       }
  38     end
  39
  40     def each_ccs
  41       each_entry("character/by_feature") {|f| yield(f) }
  42     end
  43
  44     def get_feature(name)
  45       @feature_db[name] = FeatureDB.new(self, name) if @feature_db[name].nil?
  46       @feature_db[name]
  47     end
  48
  49     def load_feature(cid, name)
  50       feature = get_feature(name)
  51       feature.get_value(cid)
  52     end
  53
  54     def get_ccs(name)
  55       @ccs_db[name] = CCS_DB.new(self, name) if @ccs_db[name].nil?
  56       @ccs_db[name]
  57     end
  58
  59     def decode_char(ccs, code_point)
  60       ccsdb = get_ccs(ccs)
  61       ccsdb.decode(code_point)
  62     end
  63
  64     def get_by_ids_db(n)
  65       @byids_db[n] = ByIDS_DB.new(self, n) if @byids_db[n].nil?
  66       @byids_db[n]
  67     end
  68
  69     private
  70     include EachEntryModule
  71   end
  72
  73   module TableManagementModule
  74     def to_hash
  75       h = {}
  76       each_char {|k, v| h[k] = v }
  77       h
  78     end
  79
  80     def dump
  81       txt = @name.path.escape.escape_win_filename.to_s+".txt"
  82       t = @cd.location+@category+@keyvalue+txt
  83       t.open("wb"){|out|
  84         to_hash.sort.each {|k, v|
  85           out.printf("%s\t%s\n", k, v)
  86         }
  87       }
  88     end
  89   end
  90
  91   class FeatureDB
  92     include ParseValueModule
  93     include TableManagementModule
  94
  95     def initialize(cd, name)
  96       @cd, @name = cd, name
  97       @ds = @cd.ds
  98       @feature = @ds.get_feature(@name.path.escape.escape_win_filename.to_s)
  99       @category, @keyvalue = "character", "feature"
 100     end
 101     def setup_db(w) @feature.setup_db(w); end
 102     def sync() @feature.sync(); end
 103     alias close sync
 104
 105     def set_value(cid, value)
 106       @feature.set_value(cid, value)
 107     end
 108
 109     def get_value(cid)
 110       parse_value(@feature.get_value(cid))
 111     end
 112
 113     def each_char
 114       @feature.each_char {|cid, value|
 115         #qp cid, value
 116         yield(cid, parse_value(value))
 117       }
 118     end
 119   end
 120
 121   class CCS_DB
 122     include TableManagementModule
 123
 124     def initialize(cd, name)
 125       @cd, @name = cd, name
 126       @ds = @cd.ds
 127       @ccs = @ds.get_ccs(@name)
 128       @dsr = @ccsr = nil
 129       @category, @keyvalue = "character", "by_feature"
 130     end
 131     def setup_db(w) @ccs.setup_db(w); end
 132     def sync() @ccs.sync(); end
 133     alias close sync
 134     def set(code_point, cid) @ccs.set(code_point, cid); end
 135     def decode(code_point) @ccs.decode(code_point); end
 136
 137     def setup_ccs_r
 138       db_dir = CHISE::DataSource::DB_DIR
 139       @dsr = DataSource_R.new(CHISE::DataSource::Berkeley_DB, db_dir, 0, 0755) if @dsr.nil?
 140       #qp @name
 141       @ccsr = CCS_R.new(@dsr, @name) if @ccsr.nil?
 142     end
 143
 144     def each_char # call CCS_R here.
 145       setup_ccs_r
 146       @ccsr.each_char {|code_point, cid|
 147         yield(code_point, cid)
 148       }
 149     end
 150   end
 151
 152   class ByIDS_DB
 153     include ChiseValue
 154     include TableAccessModule
 155
 156     def initialize(cd, name)
 157       @cd, @name = cd, name
 158       @ds = @cd.ds
 159       @category, @keyvalue = "character", "by_ids"
 160       reset
 161     end
 162
 163     def decode(ids)
 164       setup_db
 165       return nil if @db.nil?
 166       parse_c_string(@db.get(ids))
 167     end
 168
 169     def set_decoded_char(ids, cid)
 170       setup_db(true)
 171       raise "@db is nil." if @db.nil?
 172       @db.put(ids, format_char_id(cid))
 173     end
 174
 175     def each
 176       setup_db
 177       raise "@db is nil." if @db.nil?
 178       @db.each {|k, v|
 179         yield(parse_value(k), parse_c_string(v))
 180       }
 181     end
 182   end
 183 end