f52732a8668d4bae1fce29f085d4616b53c99eb0
[chise/ruby.git] / 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       db_dir = CHISE::DataSource::DB_DIR
21       @ds = DataSource.new(CHISE::DataSource::Berkeley_DB, db_dir, 0, 0755)
22       @feature_db = {}
23       @ccs_db = {}
24       @byids_db = {}
25     end
26     attr_reader :ds
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(@ds, 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(@ds, 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(@ds, n) if @byids_db[n].nil?
66       @byids_db[n]
67     end
68
69     private
70     include EachEntryModule
71   end
72
73   class FeatureDB
74     include ParseValueModule
75     def initialize(ds, name)
76       @ds, @name = ds, name
77       # @feature = @ds.get_feature(@name)
78       @feature = @ds.get_feature(@name.path.escape.escape_win_filename.to_s)
79     end
80     def setup_db(w) @feature.setup_db(w); end
81     def sync() @feature.sync(); end
82     alias close sync
83     def set_value(cid, value) @feature.set_value(cid, value); end
84     def get_value(cid)
85       parse_value(@feature.get_value(cid))
86     end
87     def each_char
88       @feature.each_char {|cid, value|
89         yield(cid, parse_value(value))
90       }
91     end
92   end
93
94   class CCS_DB
95     def initialize(ds, name)
96       @ds, @name = ds, name
97       #qp @name
98       @ccs = @ds.get_ccs(@name)
99       @dsr = @ccsr = nil
100     end
101     def setup_db(w) @ccs.setup_db(w); end
102     def sync() @ccs.sync(); end
103     alias close sync
104     def set(code_point, cid) @ccs.set(code_point, cid); end
105     def decode(code_point) @ccs.decode(code_point); end
106
107     def setup_ccs_r
108       db_dir = CHISE::DataSource::DB_DIR
109       @dsr = DataSource_R.new(CHISE::DataSource::Berkeley_DB, db_dir, 0, 0755) if @dsr.nil?
110       #qp @name
111       @ccsr = CCS_R.new(@dsr, @name) if @ccsr.nil?
112     end
113
114     def each_char # call CCS_R here.
115       setup_ccs_r
116       @ccsr.each_char {|code_point, cid|
117         yield(code_point, cid)
118       }
119     end
120   end
121
122   class ByIDS_DB
123     include ChiseValue
124     include TableAccessModule
125
126     def initialize(ds, name)
127       @ds, @name = ds, name
128       @category, @keyvalue = "character", "by_ids"
129       reset
130     end
131
132     def decode(ids)
133       setup_db
134       return nil if @db.nil?
135       parse_c_string(@db.get(ids))
136     end
137
138     def set_decoded_char(ids, cid)
139       setup_db(true)
140       raise "@db is nil." if @db.nil?
141       @db.put(ids, format_char_id(cid))
142     end
143
144     def each
145       setup_db
146       raise "@db is nil." if @db.nil?
147       @db.each {|k, v|
148         yield(parse_value(k), parse_c_string(v))
149       }
150     end
151   end
152 end