OSDN Git Service

Try to optimize...
[luatex-ja/luatexja.git] / src / ltj-setwidth.lua
1 --
2 -- luatexja/setwidth.lua
3 --
4 luatexbase.provides_module({
5   name = 'luatexja.setwidth',
6   date = '2012/07/19',
7   version = '0.2',
8   description = '',
9 })
10 module('luatexja.setwidth', package.seeall)
11 local err, warn, info, log = luatexbase.errwarinf(_NAME)
12
13 luatexja.load_module('base');      local ltjb = luatexja.base
14 luatexja.load_module('jfont');     local ltjf = luatexja.jfont
15
16 local node_type = node.type
17 local node_new = node.new
18 local node_tail = node.tail
19 local node_next = node.next
20 local has_attr = node.has_attribute
21 local set_attr = node.set_attribute
22 local node_insert_before = node.insert_before
23 local node_insert_after = node.insert_after
24 local round = tex.round
25
26
27 local id_glyph = node.id('glyph')
28 local id_kern = node.id('kern')
29 local id_hlist = node.id('hlist')
30 local id_vlist = node.id('vlist')
31 local id_rule = node.id('rule')
32 local id_math = node.id('math')
33
34 local attr_jchar_class = luatexbase.attributes['ltj@charclass']
35 local attr_curjfnt = luatexbase.attributes['ltj@curjfnt']
36 local attr_yablshift = luatexbase.attributes['ltj@yablshift']
37 local attr_ykblshift = luatexbase.attributes['ltj@ykblshift']
38 local attr_icflag = luatexbase.attributes['ltj@icflag']
39
40 local ltjf_font_metric_table = ltjf.font_metric_table
41
42 local PACKED = 2
43 local PROCESSED = 11
44 local IC_PROCESSED = 12
45 local PROCESSED_BEGIN_FLAG = 32
46
47 do
48    local floor = math.floor
49    function get_pr_begin_flag(p)
50       local i = has_attr(p, attr_icflag) or 0
51       return i - i%PROCESSED_BEGIN_FLAG
52    end
53 end
54 local get_pr_begin_flag = get_pr_begin_flag
55
56 head = nil
57
58 luatexbase.create_callback("luatexja.set_width", "data", 
59                            function (fstable, fmtable, jchar_class) 
60                               return fstable 
61                            end)
62
63 local fshift =  { down = 0, left = 0}
64
65 -- mode: true iff p will be always encapsuled by a hbox
66 function capsule_glyph(p, dir, mode, met, class)
67    local char_data = met.size_cache.char_type[class]
68    if not char_data then return node_next(p) end
69    local fwidth, pwidth = char_data.width, p.width
70    fwidth = (fwidth ~= 'prop') and fwidth or pwidth
71    local fheight, fdepth = char_data.height, char_data.depth
72    fshift.down = char_data.down; fshift.left = char_data.left
73    fshift = luatexbase.call_callback("luatexja.set_width", fshift, met, class)
74    if (mode or pwidth ~= fwidth or p.height ~= fheight or p.depth ~= fdepth) then
75       local y_shift, ca
76          = - p.yoffset + (has_attr(p,attr_ykblshift) or 0), char_data.align
77       local q; head, q = node.remove(head, p)
78       p.yoffset, p.next = -fshift.down, nil
79       if total ~= 0 and ca~='left'  then
80          p.xoffset = p.xoffset - fshift.left
81             + (((ca=='right') and fwidth - pwidth) or round((fwidth - pwidth)*0.5))
82       else
83          p.xoffset = p.xoffset - fshift.left
84       end
85       local box = node_new(id_hlist); 
86       box.width, box.height, box.depth = fwidth, fheight, fdepth
87       box.head, box.shift, box.dir = p, y_shift, (dir or 'TLT')
88       --box.glue_set, box.glue_order = 0, 0 not needed
89       set_attr(box, attr_icflag, PACKED + get_pr_begin_flag(p))
90       head = q and node_insert_before(head, q, box) 
91                or node_insert_after(head, node_tail(head), box)
92       return q
93    else
94       set_attr(p, attr_icflag, PROCESSED + get_pr_begin_flag(p))
95       p.xoffset = p.xoffset - fshift.left
96       p.yoffset = p.yoffset - (has_attr(p, attr_ykblshift) or 0) - fshift.down
97       return node_next(p)
98    end
99 end
100
101 function set_ja_width(ahead, dir)
102    local p = ahead; head  = ahead
103    local m = false -- is in math mode?
104    while p do
105       if (p.id==id_glyph) 
106       and ((has_attr(p, attr_icflag) or 0)%PROCESSED_BEGIN_FLAG)<=0 then
107       local pf = p.font
108          if pf == has_attr(p, attr_curjfnt) then
109             p = capsule_glyph(p, dir, false, ltjf_font_metric_table[pf], 
110                               has_attr(p, attr_jchar_class))
111          else
112             set_attr(p, attr_icflag, PROCESSED + get_pr_begin_flag(p))
113             p.yoffset = p.yoffset - (has_attr(p,attr_yablshift) or 0); p = node_next(p)
114          end
115       elseif p.id==id_math then
116          m = (p.subtype==0); p = node_next(p)
117       else
118          if m then
119             if p.id==id_hlist or p.id==id_vlist then
120                p.shift = p.shift + (has_attr(p,attr_yablshift) or 0)
121             elseif p.id==id_rule then
122                local v = has_attr(p,attr_yablshift) or 0
123                p.height = p.height - v; p.depth = p.depth + v 
124             end
125          end
126          p = node_next(p)
127       end
128    end
129    -- adjust attr_icflag
130    tex.setattribute('global', attr_icflag, 0)
131    return head
132 end