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_id = Dnode.traverse_id
  20 local node_traverse = Dnode.traverse
  21 local node_new = Dnode.new
  22 local node_copy = Dnode.copy
  23 local node_remove = Dnode.remove
  24 local node_tail = Dnode.tail
  25 local node_next = (Dnode ~= node) and Dnode.getnext or node.next
  26 local has_attr = Dnode.has_attribute
  27 local set_attr = Dnode.set_attribute
  28 local node_insert_before = Dnode.insert_before
  29 local node_insert_after = Dnode.insert_after
  30 local round = tex.round
  31
  32 local id_glyph = node.id('glyph')
  33 local id_kern = node.id('kern')
  34 local id_hlist = node.id('hlist')
  35 local id_vlist = node.id('vlist')
  36 local id_rule = node.id('rule')
  37 local id_math = node.id('math')
  38 local id_whatsit = node.id('whatsit')
  39 local sid_save = node.subtype('pdf_save')
  40 local sid_restore = node.subtype('pdf_restore')
  41 local sid_matrix = node.subtype('pdf_setmatrix')
  42 local dir_tate = luatexja.dir_table.dir_tate
  43
  44 local attr_ykblshift = luatexbase.attributes['ltj@ykblshift']
  45 local attr_tkblshift = luatexbase.attributes['ltj@tkblshift']
  46 local attr_icflag = luatexbase.attributes['ltj@icflag']
  47
  48 local ltjf_font_metric_table = ltjf.font_metric_table
  49 local ltjf_font_extra_info = ltjf.font_extra_info
  50 local ltjf_get_vert_glyph = ltjf.get_vert_glyph
  51
  52 local PACKED       = luatexja.icflag_table.PACKED
  53 local PROCESSED    = luatexja.icflag_table.PROCESSED
  54 local IC_PROCESSED = luatexja.icflag_table.IC_PROCESSED
  55 local PROCESSED_BEGIN_FLAG = luatexja.icflag_table.PROCESSED_BEGIN_FLAG
  56
  57 local get_pr_begin_flag
  58 do
  59    local floor = math.floor
  60    get_pr_begin_flag = function (p)
  61       local i = has_attr(p, attr_icflag) or 0
  62       return i - i%PROCESSED_BEGIN_FLAG
  63    end
  64 end
  65
  66 local ltjw = {} --export
  67 luatexja.setwidth = ltjw
  68
  69 luatexbase.create_callback("luatexja.set_width", "data",
  70                            function (fstable, fmtable, jchar_class)
  71                               return fstable
  72                            end)
  73 local call_callback = luatexbase.call_callback
  74
  75 local fshift =  { down = 0, left = 0}
  76
  77 local min, max = math.min, math.max
  78
  79 -- 和文文字の位置補正（横）
  80 local function capsule_glyph_yoko(p, met, class, head, dir)
  81    local char_data = met.char_type[class]
  82    if not char_data then return node_next(p), head, p end
  83    -- f*: whd specified in JFM
  84    local fwidth, pwidth = char_data.width, getfield(p, 'width')
  85    fwidth = (fwidth ~= 'prop') and fwidth or pwidth
  86    fshift.down = char_data.down; fshift.left = char_data.left
  87    fshift = call_callback("luatexja.set_width", fshift, met, class)
  88    local fheight, fdepth = char_data.height, char_data.depth
  89    local kbl = has_attr(p, attr_ykblshift) or 0
  90    --
  91    local need_hbox
  92    if pwidth==fwidth then
  93       -- 補正後glyph node は ht: p.height - kbl - down, dp: p.depth + min(0, kbl+down) を持つ
  94       -- 設定されるべき寸法: ht: fheight - kbl, dp: fdepth + kbl
  95       local ht_diff = fheight + fshift.down - getfield(p, 'height')
  96       local dp_diff = fdepth  + kbl - getfield(p, 'depth') - min(kbl + fshift.down, 0)
  97       if  ht_diff == 0 and dp_diff ==0 then -- offset only
  98          set_attr(p, attr_icflag, PROCESSED)
  99          setfield(p, 'xoffset', getfield(p, 'xoffset') - fshift.left)
 100          setfield(p, 'yoffset', getfield(p, 'yoffset') - kbl - fshift.down)
 101          return node_next(p), head, p
 102       elseif ht_diff >= 0 and dp_diff >=0 then -- rule
 103          local box = node_new(id_rule)
 104          setfield(p, 'yoffset', getfield(p, 'yoffset') - kbl - fshift.down)
 105          setfield(box, 'width', 0)
 106          setfield(box, 'height', fheight - kbl)
 107          setfield(box, 'depth', fdepth + kbl)
 108          setfield(box, 'dir', dir)
 109          set_attr(box, attr_icflag, PACKED)
 110          set_attr(p, attr_icflag, PACKED)
 111          head = p and node_insert_before(head, p, box)
 112             or node_insert_after(head, node_tail(head), box)
 113          return node_next(p), head, p, box
 114       else
 115          need_hbox = true
 116       end
 117    else
 118       need_hbox = true
 119    end
 120
 121    if need_hbox then
 122       local q
 123       head, q = node_remove(head, p)
 124       local box = node_new(id_hlist)
 125       setfield(p, 'yoffset', getfield(p, 'yoffset') -fshift.down);
 126       setfield(p, 'next', nil)
 127       setfield(p, 'xoffset', getfield(p, 'xoffset')
 128                   + char_data.align*(fwidth-pwidth) - fshift.left)
 129       local box = node_new(id_hlist)
 130       setfield(box, 'width', fwidth)
 131       setfield(box, 'height', fheight)
 132       setfield(box, 'depth', fdepth)
 133       setfield(box, 'head', p)
 134       setfield(box, 'shift', kbl)
 135       setfield(box, 'dir', dir)
 136       set_attr(box, attr_icflag, PACKED)
 137       head = q and node_insert_before(head, q, box)
 138                or node_insert_after(head, node_tail(head), box)
 139       return q, head, box
 140    end
 141 end
 142
 143 luatexja.setwidth.capsule_glyph_yoko = capsule_glyph_yoko
 144
 145 -- 和文文字の位置補正（縦）
 146 local function capsule_glyph_tate(p, met, class, head, dir)
 147    local char_data = met.char_type[class]
 148    if not char_data then return node_next(p), head end
 149    local ascent, descent = met.ascent, met.descent
 150    local fwidth, pwidth = char_data.width
 151    do
 152       local pf = getfont(p)
 153       local pc = getchar(p)
 154       setfield(p, 'char', pc)
 155       pwidth = ltjf_font_extra_info[pf] and  ltjf_font_extra_info[pf][pc]
 156          and ltjf_font_extra_info[pf][pc].vwidth
 157          and ltjf_font_extra_info[pf][pc].vwidth * met.size or (ascent+descent)
 158       pwidth = pwidth + (met.v_advance[pc] or 0)
 159       ascent = met.v_origin[pc] and ascent - met.v_origin[pc] or ascent
 160    end
 161    fwidth = (fwidth ~= 'prop') and fwidth or pwidth
 162    fshift.down = char_data.down; fshift.left = char_data.left
 163    fshift = call_callback("luatexja.set_width", fshift, met, class)
 164    local fheight, fdepth = char_data.height, char_data.depth
 165
 166    local y_shift
 167       = getfield(p, 'xoffset') + (has_attr(p,attr_tkblshift) or 0)
 168    local q
 169    head, q = node_remove(head, p)
 170    local box = node_new(id_hlist)
 171    setfield(box, 'width', fwidth)
 172    setfield(box, 'height', fheight)
 173    setfield(box, 'depth', fdepth)
 174    setfield(box, 'shift', y_shift)
 175    setfield(box, 'dir', dir)
 176
 177    setfield(p, 'xoffset', - fshift.down)
 178    setfield(p, 'yoffset', getfield(p, 'yoffset') -(ascent
 179                                 + char_data.align*(fwidth-pwidth) - fshift.left) )
 180    local ws = node_new(id_whatsit, sid_save)
 181    local wm = node_new(id_whatsit, sid_matrix)
 182    setfield(wm, 'data', '0 1 -1 0')
 183    local pwnh = - 0.5*getfield(p, 'width')
 184    local k2 = node_new(id_kern); setfield(k2, 'kern', - 0.5*getfield(p, 'width'))
 185    local k3 = node_copy(k2)
 186    local wr = node_new(id_whatsit, sid_restore)
 187    setfield(box, 'head', ws)
 188    setfield(ws, 'next', wm);  setfield(wm, 'next', k2);
 189    setfield(k2, 'next', p);   setfield(p, 'next', k3);
 190    setfield(k3, 'next', wr);
 191
 192    set_attr(box, attr_icflag, PACKED)
 193    head = q and node_insert_before(head, q, box)
 194       or node_insert_after(head, node_tail(head), box)
 195    return q, head, box
 196 end
 197 luatexja.setwidth.capsule_glyph_tate = capsule_glyph_tate
 198
 199 local function capsule_glyph_math(p, met, class)
 200    local char_data = met.char_type[class]
 201    if not char_data then return nil end
 202    local fwidth, pwidth = char_data.width, getfield(p, 'width')
 203    fwidth = (fwidth ~= 'prop') and fwidth or pwidth
 204    fshift.down = char_data.down; fshift.left = char_data.left
 205    fshift = call_callback("luatexja.set_width", fshift, met, class)
 206    local fheight, fdepth = char_data.height, char_data.depth
 207    local y_shift, ca
 208       = - getfield(p, 'yoffset') + (has_attr(p,attr_ykblshift) or 0), char_data.align
 209    setfield(p, 'yoffset', -fshift.down)
 210    setfield(p, 'xoffset', getfield(p, 'xoffset') + char_data.align*(fwidth-pwidth) - fshift.left)
 211    local box = node_new(id_hlist);
 212    setfield(box, 'width', fwidth)
 213    setfield(box, 'height', fheight)
 214    setfield(box, 'depth', fdepth)
 215    setfield(box, 'head', p)
 216    setfield(box, 'shift', y_shift)
 217    setfield(box, 'dir', tex.mathdir)
 218    set_attr(box, attr_icflag, PACKED)
 219    return box
 220 end
 221 luatexja.setwidth.capsule_glyph_math = capsule_glyph_math
 222
 223 -- 数式の位置補正
 224 function luatexja.setwidth.apply_ashift_math(head, last, attr_ablshift)
 225    for p in node_traverse(head) do
 226       local pid = getid(p)
 227       if p==last then
 228          return
 229       elseif (has_attr(p, attr_icflag) or 0) ~= PROCESSED then
 230          if pid==id_hlist or pid==id_vlist then
 231             setfield(p, 'shift', getfield(p, 'shift') +  (has_attr(p,attr_ablshift) or 0))
 232          elseif pid==id_rule then
 233             local v = has_attr(p,attr_ablshift) or 0
 234             setfield(p, 'height', getfield(p, 'height')-v)
 235             setfield(p, 'depth', getfield(p, 'depth')+v)
 236             set_attr(p, attr_icflag, PROCESSED)
 237          elseif pid==id_glyph then
 238             -- 欧文文字; 和文文字は pid == id_hlist の場合で処理される
 239             -- (see conv_jchar_to_hbox_A in ltj-math.lua)
 240             setfield(p, 'yoffset',
 241                      getfield(p, 'yoffset') - (has_attr(p,attr_ablshift) or 0))
 242          end
 243       end
 244    end
 245 end
 246
 247 -- discretionary の位置補正
 248 do
 249    local attr_yablshift = luatexbase.attributes['ltj@yablshift']
 250    local attr_tablshift = luatexbase.attributes['ltj@tablshift']
 251    local attr_ablshift
 252    local disc, tex_dir
 253    local function ashift_disc_inner(field)
 254       local head = getfield(disc, field)
 255       if not head then return end
 256       local y_adjust, node_depth, adj_depth = 0, 0, 0
 257       for lp in node_traverse_id(id_glyph, head) do
 258          y_adjust = has_attr(lp,attr_ablshift) or 0
 259          node_depth = max(getfield(lp, 'depth') + min(y_adjust, 0), node_depth)
 260          adj_depth = (y_adjust>0) and max(getfield(lp, 'depth') + y_adjust, adj_depth) or adj_depth
 261          setfield(lp, 'yoffset', getfield(lp, 'yoffset') - y_adjust)
 262       end
 263       if adj_depth>node_depth then
 264          local r = node_new(id_rule)
 265          setfield(r, 'width', 0); setfield(r, 'height', 0)
 266          setfield(r, 'depth', adj_depth); setfield(r, 'dir', tex_dir)
 267          set_attr(r, attr_icflag, PROCESSED)
 268          if field=='post' then
 269             node_insert_after(head, head, r)
 270          else
 271             setfield(disc, field, (node_insert_before(head, head, r)))
 272          end
 273       end
 274    end
 275    function luatexja.setwidth.apply_ashift_disc(d, is_dir_tate, dir)
 276       attr_ablshift = is_dir_tate and attr_tablshift or attr_yablshift
 277       disc, tex_dir = d, dir
 278       ashift_disc_inner('pre')
 279       ashift_disc_inner('post')
 280       ashift_disc_inner('replace')
 281    end
 282 end