OSDN Git Service

... and more
[luatex-ja/luatexja.git] / src / ltj-setwidth.lua
index 7946829..7e264ad 100644 (file)
@@ -1,26 +1,29 @@
 --
--- luatexja/setwidth.lua
+-- src/ltj-setwidth.lua
 --
-luatexbase.provides_module({
-  name = 'luatexja.setwidth',
-  date = '2011/06/28',
-  version = '0.1',
-  description = '',
-})
-module('luatexja.setwidth', package.seeall)
-local err, warn, info, log = luatexbase.errwarinf(_NAME)
 
 luatexja.load_module('base');      local ltjb = luatexja.base
 luatexja.load_module('jfont');     local ltjf = luatexja.jfont
 
-local node_type = node.type
-local node_new = node.new
-local node_tail = node.tail
-local node_next = node.next
-local has_attr = node.has_attribute
-local set_attr = node.set_attribute
-local node_insert_before = node.insert_before
-local node_insert_after = node.insert_after
+local Dnode = node.direct or node
+
+local setfield = (Dnode ~= node) and Dnode.setfield or function(n, i, c) n[i] = c end
+local getfield = (Dnode ~= node) and Dnode.getfield or function(n, i) return n[i] end
+local getid = (Dnode ~= node) and Dnode.getid or function(n) return n.id end
+local getfont = (Dnode ~= node) and Dnode.getfont or function(n) return n.font end
+local getlist = (Dnode ~= node) and Dnode.getlist or function(n) return n.head end
+local getchar = (Dnode ~= node) and Dnode.getchar or function(n) return n.char end
+local getsubtype = (Dnode ~= node) and Dnode.getsubtype or function(n) return n.subtype end
+
+local node_traverse = Dnode.traverse
+local node_new = Dnode.new
+local node_remove = Dnode.remove
+local node_tail = Dnode.tail
+local node_next = Dnode.getnext
+local has_attr = Dnode.has_attribute
+local set_attr = Dnode.set_attribute
+local node_insert_before = Dnode.insert_before
+local node_insert_after = Dnode.insert_after
 local round = tex.round
 
 local id_glyph = node.id('glyph')
@@ -33,108 +36,146 @@ local id_math = node.id('math')
 local attr_jchar_class = luatexbase.attributes['ltj@charclass']
 local attr_curjfnt = luatexbase.attributes['ltj@curjfnt']
 local attr_yablshift = luatexbase.attributes['ltj@yablshift']
+local attr_ykblshift = luatexbase.attributes['ltj@ykblshift']
 local attr_icflag = luatexbase.attributes['ltj@icflag']
-local attr_uniqid = luatexbase.attributes['ltj@uniqid']
 
 local ltjf_font_metric_table = ltjf.font_metric_table
 
-local PACKED = 2
-
-char_data = {}
-head = nil
+local PACKED       = luatexja.icflag_table.PACKED
+local PROCESSED    = luatexja.icflag_table.PROCESSED
+local IC_PROCESSED = luatexja.icflag_table.IC_PROCESSED
+local PROCESSED_BEGIN_FLAG = luatexja.icflag_table.PROCESSED_BEGIN_FLAG
 
--- return true if and only if p is a Japanese character node
-local function is_japanese_glyph_node(p)
-   return p.font==has_attr(p, attr_curjfnt)
+local get_pr_begin_flag
+do
+   local floor = math.floor
+   get_pr_begin_flag = function (p)
+      local i = has_attr(p, attr_icflag) or 0
+      return i - i%PROCESSED_BEGIN_FLAG
+   end
 end
 
+local head, dir
+local ltjw = {} --export
+luatexja.setwidth = ltjw
+
 luatexbase.create_callback("luatexja.set_width", "data", 
                           function (fstable, fmtable, jchar_class) 
                              return fstable 
                           end)
+local call_callback = luatexbase.call_callback
 
 local fshift =  { down = 0, left = 0}
+
 -- mode: true iff p will be always encapsuled by a hbox
-function capsule_glyph(p, dir, mode, met, class)
-   local h, box, q, fwidth
-   if char_data.width ~= 'prop' then
-      fwidth = char_data.width
-   else fwidth = p.width end
-   local fheight = char_data.height
-   local fdepth = char_data.depth
+local function capsule_glyph(p, met, class)
+   local char_data = met.char_type[class]
+   if not char_data then return node_next(p) end
+   local fwidth, pwidth = char_data.width, getfield(p, 'width')
+   fwidth = (fwidth ~= 'prop') and fwidth or pwidth
    fshift.down = char_data.down; fshift.left = char_data.left
-   fshift = luatexbase.call_callback("luatexja.set_width", fshift, met, class)
---   local ti = 
-   p.xoffset= p.xoffset - fshift.left
-   if mode or p.width ~= fwidth or p.height ~= fheight or p.depth ~= fdepth then
-      local y_shift = - p.yoffset + (has_attr(p,attr_yablshift) or 0)
-      p.yoffset = -fshift.down
-      head, q = node.remove(head, p)
-      local total = fwidth - p.width
-      if total == 0 then
-        h = p; p.next = nil
-      else
-        h = node_new(id_kern); h.subtype = 0
-        if char_data.align=='right' then
-           h.kern = total; p.next = nil; h.next = p
-        elseif char_data.align=='middle' then
-           h.kern = round(total*0.5); p.next = h
-           h = node_new(id_kern); h.subtype = 0
-           h.kern = total - round(total*0.5); h.next = p
-        else -- left
-           h.kern = total; p.next = h; h = p
-        end
-      end
-      box = node_new(id_hlist); 
-      box.width = fwidth; box.height = fheight; box.depth = fdepth
-      box.glue_set = 0; box.glue_order = 0; box.head = h
-      box.shift = y_shift; box.dir = dir or 'TLT'
-      set_attr(box, attr_icflag, PACKED)
-      set_attr(box, attr_uniqid, has_attr(p, attr_uniqid) or 0)
-      if q then
-        head = node_insert_before(head, q, box)
-      else
-        head = node_insert_after(head, node_tail(head), box)
-      end
+   fshift = call_callback("luatexja.set_width", fshift, met, class)
+   local fheight, fdepth = char_data.height, char_data.depth
+   if (pwidth ~= fwidth or getfield(p, 'height') ~= fheight or getfield(p, 'depth') ~= fdepth) then
+      local y_shift, ca
+         = - getfield(p, 'yoffset') + (has_attr(p,attr_ykblshift) or 0), char_data.align
+      local q
+      head, q = node_remove(head, p)
+      setfield(p, 'yoffset', -fshift.down); setfield(p, 'next', nil)
+      ca = (ca~='left') 
+        and -fshift.left + (((ca=='right') and fwidth-pwidth) or round((fwidth-pwidth)*0.5))
+        or  -fshift.left
+      setfield(p, 'xoffset', getfield(p, 'xoffset') + ca)
+      local box = node_new(id_hlist); 
+      setfield(box, 'width', fwidth)
+      setfield(box, 'height', fheight)
+      setfield(box, 'depth', fdepth)
+      setfield(box, 'head', p)
+      setfield(box, 'shift', y_shift)
+      setfield(box, 'dir', dir)
+      set_attr(box, attr_icflag, PACKED + get_pr_begin_flag(p))
+      head = q and node_insert_before(head, q, box) 
+               or node_insert_after(head, node_tail(head), box)
       return q
    else
-      p.yoffset = p.yoffset - (has_attr(p, attr_yablshift) or 0) - fshift.down
+      set_attr(p, attr_icflag, PROCESSED + get_pr_begin_flag(p))
+      setfield(p, 'xoffset', getfield(p, 'xoffset') - fshift.left)
+      setfield(p, 'yoffset', getfield(p, 'yoffset') 
+                 - (has_attr(p, attr_ykblshift) or 0) - fshift.down)
       return node_next(p)
    end
 end
+luatexja.setwidth.capsule_glyph = capsule_glyph
+
+local function capsule_glyph_math(p, met, class)
+   local char_data = met.char_type[class]
+   if not char_data then return nil end
+   local fwidth, pwidth = char_data.width, getfield(p, 'width')
+   fwidth = (fwidth ~= 'prop') and fwidth or pwidth
+   fshift.down = char_data.down; fshift.left = char_data.left
+   fshift = call_callback("luatexja.set_width", fshift, met, class)
+   local fheight, fdepth = char_data.height, char_data.depth
+   local y_shift, ca
+      = - getfield(p, 'yoffset') + (has_attr(p,attr_ykblshift) or 0), char_data.align
+   setfield(p, 'yoffset', -fshift.down)
+   if ca~='left'  then
+      setfield(p, 'xoffset', getfield(p, 'xoffset') - fshift.left
+                 + (((ca=='right') and fwidth - pwidth) or round((fwidth - pwidth)*0.5)))
+   else
+      setfield(p, 'xoffset', getfield(p, 'xoffset') - fshift.left)
+   end
+   local box = node_new(id_hlist); 
+   setfield(box, 'width', fwidth)
+   setfield(box, 'height', fheight)
+   setfield(box, 'depth', fdepth)
+   setfield(box, 'head', p)
+   setfield(box, 'shift', y_shift)
+   setfield(box, 'dir', tex.mathdir)
+   set_attr(box, attr_icflag, PACKED + get_pr_begin_flag(p))
+   return box
+end
+luatexja.setwidth.capsule_glyph_math = capsule_glyph_math
 
-function set_ja_width(ahead, dir)
-   local p = ahead; head  = ahead
+function luatexja.setwidth.set_ja_width(ahead, adir)
+   local p = ahead; head  = p; dir = adir or 'TLT'
    local m = false -- is in math mode?
    while p do
-      if p.id==id_glyph then
-        if is_japanese_glyph_node(p) then
-           local met = ltjf_font_metric_table[p.font]
-           local class = has_attr(p, attr_jchar_class)
-           char_data = ltjf.metrics[met.jfm].size_cache[met.size].char_type[class]
-            if char_data then
-               p = capsule_glyph(p, dir, false, met, class)
-            else
-               p = node_next(p)
-            end
-        else 
-           p.yoffset = p.yoffset - (has_attr(p,attr_yablshift) or 0); p = node_next(p)
+      local pid = getid(p)
+      if (pid==id_glyph) 
+      and ((has_attr(p, attr_icflag) or 0)%PROCESSED_BEGIN_FLAG)<=0 then
+         local pf = getfont(p)
+        if pf == has_attr(p, attr_curjfnt) then
+           p = capsule_glyph(p, ltjf_font_metric_table[pf], 
+                             has_attr(p, attr_jchar_class))
+        else
+           set_attr(p, attr_icflag, PROCESSED + get_pr_begin_flag(p))
+           setfield(p, 'yoffset',
+                    getfield(p, 'yoffset') - (has_attr(p,attr_yablshift) or 0))
+           p = node_next(p)
         end
-      elseif p.id==id_math then
-        m = (p.subtype==0); p = node_next(p)
+      elseif pid==id_math then
+        m = (getsubtype(p)==0); p = node_next(p)
       else
         if m then
-           if p.id==id_hlist or p.id==id_vlist then
-              p.shift = p.shift + (has_attr(p,attr_yablshift) or 0)
-           elseif p.id==id_rule then
-              local v = has_attr(p,attr_yablshift) or 0
-              p.height = p.height - v; p.depth = p.depth + v 
+            -- 数式の位置補正
+           if pid==id_hlist or pid==id_vlist then
+               if (has_attr(p, attr_icflag) or 0) ~= PROCESSED then
+                  setfield(p, 'shift', getfield(p, 'shift') +  (has_attr(p,attr_yablshift) or 0))
+               end
+           elseif pid==id_rule then
+              if (has_attr(p, attr_icflag) or 0) ~= PROCESSED then
+                 local v = has_attr(p,attr_yablshift) or 0
+                  setfield(p, 'height', getfield(p, 'height')-v)
+                  setfield(p, 'depth', getfield(p, 'depth')+v)
+                 set_attr(p, attr_icflag, PROCESSED + get_pr_begin_flag(p))
+              end
            end
         end
         p = node_next(p)
       end
    end
    -- adjust attr_icflag
-   tex.attribute[attr_icflag] = -(0x7FFFFFFF)
+   tex.setattribute('global', attr_icflag, 0)
    return head
 end
+