OSDN Git Service

ltj-setwidth.lua: removed unnecessary nodes.
[luatex-ja/luatexja.git] / src / ltj-setwidth.lua
1 --
2 -- src/ltj-setwidth.lua
3 --
4
5 luatexja.load_module('base');      local ltjb = luatexja.base
6 luatexja.load_module('stack');     local ltjs = luatexja.stack
7 luatexja.load_module('jfont');     local ltjf = luatexja.jfont
8 luatexja.load_module('direction'); local ltjd = luatexja.direction
9
10 local Dnode = node.direct or node
11 local setfield = (Dnode ~= node) and Dnode.setfield or function(n, i, c) n[i] = c end
12 local getfield = (Dnode ~= node) and Dnode.getfield or function(n, i) return n[i] end
13 local getid = (Dnode ~= node) and Dnode.getid or function(n) return n.id end
14 local getfont = (Dnode ~= node) and Dnode.getfont or function(n) return n.font end
15 local getlist = (Dnode ~= node) and Dnode.getlist or function(n) return n.head end
16 local getchar = (Dnode ~= node) and Dnode.getchar or function(n) return n.char end
17 local getsubtype = (Dnode ~= node) and Dnode.getsubtype or function(n) return n.subtype end
18
19 local node_traverse = Dnode.traverse
20 local node_new = Dnode.new
21 local node_copy = Dnode.copy
22 local node_remove = Dnode.remove
23 local node_tail = Dnode.tail
24 local node_next = (Dnode ~= node) and Dnode.getnext or node.next
25 local has_attr = Dnode.has_attribute
26 local set_attr = Dnode.set_attribute
27 local node_insert_before = Dnode.insert_before
28 local node_insert_after = Dnode.insert_after
29 local round = tex.round
30
31 local id_glyph = node.id('glyph')
32 local id_kern = node.id('kern')
33 local id_hlist = node.id('hlist')
34 local id_vlist = node.id('vlist')
35 local id_rule = node.id('rule')
36 local id_math = node.id('math')
37 local id_whatsit = node.id('whatsit')
38 local sid_save = node.subtype('pdf_save')
39 local sid_restore = node.subtype('pdf_restore')
40 local sid_matrix = node.subtype('pdf_setmatrix')
41 local dir_tate = luatexja.dir_table.dir_tate
42
43 local attr_jchar_class = luatexbase.attributes['ltj@charclass']
44 local attr_curjfnt = luatexbase.attributes['ltj@curjfnt']
45 local attr_yablshift = luatexbase.attributes['ltj@yablshift']
46 local attr_ykblshift = luatexbase.attributes['ltj@ykblshift']
47 local attr_tablshift = luatexbase.attributes['ltj@tablshift']
48 local attr_tkblshift = luatexbase.attributes['ltj@tkblshift']
49 local attr_icflag = luatexbase.attributes['ltj@icflag']
50
51 local ltjf_font_metric_table = ltjf.font_metric_table
52 local ltjf_get_vert_glyph = ltjf.get_vert_glyph
53
54 local PACKED       = luatexja.icflag_table.PACKED
55 local PROCESSED    = luatexja.icflag_table.PROCESSED
56 local IC_PROCESSED = luatexja.icflag_table.IC_PROCESSED
57 local PROCESSED_BEGIN_FLAG = luatexja.icflag_table.PROCESSED_BEGIN_FLAG
58
59 local get_pr_begin_flag
60 do
61    local floor = math.floor
62    get_pr_begin_flag = function (p)
63       local i = has_attr(p, attr_icflag) or 0
64       return i - i%PROCESSED_BEGIN_FLAG
65    end
66 end
67
68 local head, dir
69 local ltjw = {} --export
70 luatexja.setwidth = ltjw
71
72 luatexbase.create_callback("luatexja.set_width", "data",
73                            function (fstable, fmtable, jchar_class)
74                               return fstable
75                            end)
76 local call_callback = luatexbase.call_callback
77
78 local fshift =  { down = 0, left = 0}
79
80 local function capsule_glyph_yoko(p, met, class)
81    local char_data = met.char_type[class]
82    if not char_data then return node_next(p) end
83    local fwidth, pwidth = char_data.width, getfield(p, 'width')
84    fwidth = (fwidth ~= 'prop') and fwidth or pwidth
85    fshift.down = char_data.down; fshift.left = char_data.left
86    fshift = call_callback("luatexja.set_width", fshift, met, class)
87    local fheight, fdepth = char_data.height, char_data.depth
88    if (pwidth ~= fwidth or getfield(p, 'height') ~= fheight or getfield(p, 'depth') ~= fdepth) then
89       local y_shift
90          = - getfield(p, 'yoffset') + (has_attr(p,attr_ykblshift) or 0)
91       local q
92       head, q = node_remove(head, p)
93       setfield(p, 'yoffset', -fshift.down); setfield(p, 'next', nil)
94       setfield(p, 'xoffset', getfield(p, 'xoffset') + char_data.align*(fwidth-pwidth) - fshift.left)
95       local box = node_new(id_hlist)
96       setfield(box, 'width', fwidth)
97       setfield(box, 'height', fheight)
98       setfield(box, 'depth', fdepth)
99       setfield(box, 'head', p)
100       setfield(box, 'shift', y_shift)
101       setfield(box, 'dir', dir)
102       set_attr(box, attr_icflag, PACKED)
103       head = q and node_insert_before(head, q, box)
104                or node_insert_after(head, node_tail(head), box)
105       return q
106    else
107       set_attr(p, attr_icflag, PROCESSED)
108       setfield(p, 'xoffset', getfield(p, 'xoffset') - fshift.left)
109       setfield(p, 'yoffset', getfield(p, 'yoffset')
110                   - (has_attr(p, attr_ykblshift) or 0) - fshift.down)
111       return node_next(p)
112    end
113 end
114 local function capsule_glyph_tate(p, met, class)
115    local char_data = met.char_type[class]
116    if not char_data then return node_next(p) end
117    local ascent, descent = met.ascent, met.descent
118    local fwidth, pwidth = char_data.width, ascent + descent
119    fwidth = (fwidth ~= 'prop') and fwidth or pwidth
120    fshift.down = char_data.down; fshift.left = char_data.left
121    fshift = call_callback("luatexja.set_width", fshift, met, class)
122    local fheight, fdepth = char_data.height, char_data.depth
123
124    setfield(p, 'char', ltjf_get_vert_glyph(getfont(p), getchar(p)))
125
126    local y_shift
127       = - getfield(p, 'yoffset') + (has_attr(p,attr_tkblshift) or 0)
128    local q
129    head, q = node_remove(head, p)
130    local box = node_new(id_hlist)
131    setfield(box, 'width', fwidth)
132    setfield(box, 'height', fheight)
133    setfield(box, 'depth', fdepth)
134    setfield(box, 'shift', y_shift)
135    setfield(box, 'dir', dir)
136
137    --local k1 = node_new(id_kern)
138    --setfield(k1, 'kern',
139    --    getfield(p, 'xoffset') + ascent
140    --       + char_data.align*(fwidth-pwidth) - fshift.left)
141    setfield(p, 'xoffset', -fshift.down)
142    setfield(p, 'yoffset', - (getfield(p, 'xoffset') + ascent
143                                 + char_data.align*(fwidth-pwidth) - fshift.left) )
144    local ws = node_new(id_whatsit, sid_save)
145    local wm = node_new(id_whatsit, sid_matrix)
146    setfield(wm, 'data', '0 1 -1 0')
147    local pwnh = - 0.5*getfield(p, 'width')
148    local k2 = node_new(id_kern); setfield(k2, 'kern', - 0.5*getfield(p, 'width'))
149    local k3 = node_copy(k2)
150    local wr = node_new(id_whatsit, sid_restore)
151    --setfield(box, 'head', k1); setfield(k1, 'next', ws)
152    setfield(box, 'head', ws)
153    setfield(ws, 'next', wm);  setfield(wm, 'next', k2);
154    setfield(k2, 'next', p);   setfield(p, 'next', k3);
155    setfield(k3, 'next', wr);
156
157    set_attr(box, attr_icflag, PACKED)
158    head = q and node_insert_before(head, q, box)
159       or node_insert_after(head, node_tail(head), box)
160    return q
161 end
162
163 local function capsule_glyph_math(p, met, class)
164    local char_data = met.char_type[class]
165    if not char_data then return nil end
166    local fwidth, pwidth = char_data.width, getfield(p, 'width')
167    fwidth = (fwidth ~= 'prop') and fwidth or pwidth
168    fshift.down = char_data.down; fshift.left = char_data.left
169    fshift = call_callback("luatexja.set_width", fshift, met, class)
170    local fheight, fdepth = char_data.height, char_data.depth
171    local y_shift, ca
172       = - getfield(p, 'yoffset') + (has_attr(p,attr_ykblshift) or 0), char_data.align
173    setfield(p, 'yoffset', -fshift.down)
174    setfield(p, 'xoffset', getfield(p, 'xoffset') + char_data.align*(fwidth-pwidth) - fshift.left)
175    local box = node_new(id_hlist);
176    setfield(box, 'width', fwidth)
177    setfield(box, 'height', fheight)
178    setfield(box, 'depth', fdepth)
179    setfield(box, 'head', p)
180    setfield(box, 'shift', y_shift)
181    setfield(box, 'dir', tex.mathdir)
182    set_attr(box, attr_icflag, PACKED)
183    return box
184 end
185 luatexja.setwidth.capsule_glyph_math = capsule_glyph_math
186
187 local tex_set_attr = tex.setattribute
188 function luatexja.setwidth.set_ja_width(ahead, adir)
189    local p = ahead; head  = p; dir = adir or 'TLT'
190    local m = false -- is in math mode?
191    local is_dir_tate = ltjs.list_dir==dir_tate
192    local capsule_glyph = is_dir_tate and capsule_glyph_tate or capsule_glyph_yoko
193    local attr_ablshift = is_dir_tate and attr_tablshift or attr_yablshift
194    while p do
195       local pid = getid(p)
196       if (pid==id_glyph)
197       and ((has_attr(p, attr_icflag) or 0)%PROCESSED_BEGIN_FLAG)<=0 then
198          local pf = getfont(p)
199          if pf == has_attr(p, attr_curjfnt) then
200             p = capsule_glyph(p, ltjf_font_metric_table[pf],
201                               has_attr(p, attr_jchar_class))
202          else
203             -- TODO: neg. offset does not increase depth
204             --local d = getfield(p, 'yoffset') - (has_attr(p,attr_ablshift) or 0)
205             set_attr(p, attr_icflag, PROCESSED)
206             setfield(p, 'yoffset',
207                      getfield(p, 'yoffset') - (has_attr(p,attr_ablshift) or 0))
208             p = node_next(p)
209          end
210       elseif pid==id_math then
211          m = (getsubtype(p)==0); p = node_next(p)
212       else
213          if m then
214             -- 数式の位置補正
215             if pid==id_hlist or pid==id_vlist then
216                if (has_attr(p, attr_icflag) or 0) ~= PROCESSED then
217                   setfield(p, 'shift', getfield(p, 'shift') +  (has_attr(p,attr_ablshift) or 0))
218                end
219             elseif pid==id_rule then
220                if (has_attr(p, attr_icflag) or 0) ~= PROCESSED then
221                   local v = has_attr(p,attr_yablshift) or 0
222                   setfield(p, 'height', getfield(p, 'height')-v)
223                   setfield(p, 'depth', getfield(p, 'depth')+v)
224                   set_attr(p, attr_icflag, PROCESSED)
225                end
226             end
227          end
228          p = node_next(p)
229       end
230    end
231    -- adjust attr_icflag
232    tex_set_attr('global', attr_icflag, 0)
233    return head
234 end