summaryrefslogtreecommitdiff
path: root/Master/texmf-dist/tex/luatex/luatexja/ltj-adjust.lua
blob: 4c418b39552d2a5690d8c35047ebd6f2c8a1f4be (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
--
-- luatexja/otf.lua
--
luatexbase.provides_module({
  name = 'luatexja.adjust',
  date = '2013/03/14',
  description = 'Advanced line adjustment for LuaTeX-ja',
})
module('luatexja.adjust', package.seeall)

luatexja.load_module('jfont');     local ltjf = luatexja.jfont
luatexja.load_module('jfmglue');   local ltjj = luatexja.jfmglue

local id_glyph = node.id('glyph')
local id_kern = node.id('kern')
local id_hlist = node.id('hlist')
local id_glue  = node.id('glue')
local id_glue_spec = node.id('glue_spec')
local has_attr = node.has_attribute
local set_attr = node.set_attribute
local attr_icflag = luatexbase.attributes['ltj@icflag']
local attr_jchar_class = luatexbase.attributes['ltj@charclass']
local attr_curjfnt = luatexbase.attributes['ltj@curjfnt']
local node_copy = node.copy
local node_next = node.next
local node_free = node.free

local ltjf_font_metric_table = ltjf.font_metric_table
local spec_zero_glue = ltjj.spec_zero_glue

local PACKED       = luatexja.icflag_table.PACKED
local FROM_JFM     = luatexja.icflag_table.FROM_JFM
local KANJI_SKIP   = luatexja.icflag_table.KANJI_SKIP
local XKANJI_SKIP  = luatexja.icflag_table.XKANJI_SKIP

local priority_table = {
   FROM_JFM + 2,
   FROM_JFM + 1,
   FROM_JFM,
   FROM_JFM - 1,
   FROM_JFM - 2,
   XKANJI_SKIP,
   KANJI_SKIP
}

local PROCESSED_BEGIN_FLAG = 32
local function get_attr_icflag(p)
   return (node.has_attribute(p, attr_icflag) or 0) % PROCESSED_BEGIN_FLAG
end

-- box 内で伸縮された glue の合計値を計算

local function get_stretched(q, go, gs)
   local qs = q.spec
   if not qs.writable then return 0 end
   if gs == 1 then -- stretching
      if qs.stretch_order == go then return qs.stretch end
   else -- shrinking
      if qs.shrink_order == go then return qs.shrink end
   end
end

-- local new_ks, new_xs
local function get_total_stretched(p)
   local go, gf, gs = p.glue_order, p.glue_set, p.glue_sign
   local res = {
      [0] = 0,
      glue_set = gf, name = (gs==1) and 'stretch' or 'shrink'
   }
   for i=1,#priority_table do res[priority_table[i]]=0 end
   if go ~= 0 then return nil end
   if gs ~= 1 and gs ~= 2 then return res end
   local head = p.head
   q = p.head
   --luatexja.ext_show_node_list(p.head, '>>> ', print)
   while q do 
      if q.id==id_glue then
         local a, ic = get_stretched(q, go, gs), get_attr_icflag(q)
         if   type(res[ic]) == 'number' then 
            -- kanjiskip, xkanjiskip は段落内で spec を共有しているが,
            -- それはここでは望ましくないので,各 glue ごとに異なる spec を使う.
            -- JFM グルーはそれぞれ異なる glue_spec を用いているので,問題ない.
            res[ic] = res[ic] + a
            if ic == KANJI_SKIP or ic == XKANJI_SKIP  then
               if q.spec ~= spec_zero_glue then
                  local ts, g; 
                  q.spec, ts = node_copy(q.spec), q.spec
                  g = node.copy(q); q.spec = ts
                  node.insert_before(head, q, g);
                  head = node.remove(head, q); node.free(q); q = g
               end
            end
         else 
            res[0]  = res[0]  + a
         end
      end
      q = node_next(q)
   end
   return res
end

local function clear_stretch(p, ic, name)
   --print('clear ' .. ic)
   for q in node.traverse_id(id_glue, p.head) do
      if get_attr_icflag(q) == ic then
         local qs = q.spec
         if qs.writable then
            qs[name..'_order'], qs[name] = 0, 0
         end
      end
   end
end

local set_stretch_table = {}
local function set_stretch(p, after, before, ic, name)
   if before > 0 then
      --print (ic, before, after)
      local ratio = after/before
      for i,_ in pairs(set_stretch_table) do
         set_stretch_table[i] = nil
      end
      for q in node.traverse_id(id_glue, p.head) do
         if get_attr_icflag(q) == ic then
            local qs, do_flag = q.spec, true
            for i=1,#set_stretch_table do 
               if set_stretch_table[i]==qs then do_flag = false end 
            end
            if qs.writable and qs[name..'_order'] == 0 and do_flag then
               qs[name] = qs[name]*ratio; 
               set_stretch_table[#set_stretch_table+1] = qs
            end
         end
      end
   end
end

-- step 1: 行末に kern を挿入(句読点,中点用)
local function aw_step1(p, res, total)
   local x = node.tail(p.head); if not x then return false end
   local x = node.prev(x)     ; if not x then return false end
   -- 本当の行末の node を格納
   if x.id == id_glue and x.subtype == 15 then 
      -- 段落最終行のときは,\penalty10000 \parfillskip が入るので,
      -- その前の node が本来の末尾文字となる
      x = node.prev(node.prev(x)) 
   end

   local xi, xc = x.id
   if xi == id_glyph and has_attr(x, attr_curjfnt) == x.font then
      -- 和文文字
      xc = x
   elseif xi == id_hlist and get_attr_icflag(x) == PACKED then
      -- packed JAchar
      xc = x.head
   else
     return false-- それ以外は対象外.
   end
   local xk = ltjf_font_metric_table -- 
     [xc.font].char_type[has_attr(xc, attr_jchar_class) or 0]
     ['end_' .. res.name] or 0
     --print(res.name, total, xk, unicode.utf8.char(xc.char))

   if xk>0 and total>=xk then
      --print("ADDED")
      total = total - xk
      local kn = node.new(id_kern)
      kn.kern = (res.name=='shrink' and -1 or 1) * xk
      set_attr(kn, attr_icflag, FROM_JFM)
      node.insert_after(p.head, x, kn)
      return true
   else return false
   end
end

-- step 2: 行中の glue を変える
local function aw_step2(p, res, total, added_flag)
   if total == 0 then -- もともと伸縮の必要なし
      if added_flag then -- 行末に kern 追加したので,それによる補正
	 local f = node.hpack(p.head, p.width, 'exactly')
	 f.head, p.glue_set, p.glue_sign, p.glue_order 
	    = nil, f.glue_set, f.glue_sign, f.glue_order
	 node.free(f); return
      end
   elseif total <= res[0] then -- 和文処理グルー以外で足りる
      for _,v in pairs(priority_table) do clear_stretch(p, v, res.name) end
      local f = node.hpack(p.head, p.width, 'exactly')
      f.head, p.glue_set, p.glue_sign, p.glue_order 
         = nil, f.glue_set, f.glue_sign, f.glue_order
      node.free(f)
   else
      total, i = total - res[0], 1
      while i <= #priority_table do
         local v = priority_table[i]
         if total <= res[v] then
            for j = i+1,#priority_table do
               clear_stretch(p, priority_table[j], res.name)
            end
            set_stretch(p, total, res[v], v, res.name)
            i = #priority_table + 9 -- ループから抜けさせたいため
         end
         total, i= total - res[v], i+1
      end
      if i == #priority_table + 10 or added_flag then
	 local f = node.hpack(p.head, p.width, 'exactly')
	 f.head, p.glue_set, p.glue_sign, p.glue_order 
	    = nil, f.glue_set, f.glue_sign, f.glue_order
	 node.free(f)
      end
   end
end


function adjust_width(head) 
   if not head then return head end
   for p in node.traverse_id(id_hlist, head) do
      local res = get_total_stretched(p)
      --print(table.serialize(res))
      if res then
         -- 調整量の合計
         local total = 0
         for i,v in pairs(res) do 
            if type(i)=='number' then
               total = total + v
            end
         end; total = tex.round(total * res.glue_set)
         local added_flag = aw_step1(p, res, total)
         --print(total, res[0], res[KANJI_SKIP], res[FROM_JFM])
         aw_step2(p, res, total, added_flag)
      end
   end
   return head
end

local is_reg = false
function enable_cb()
   if not is_reg then
      luatexbase.add_to_callback('post_linebreak_filter', adjust_width, 'Adjust width', 100)
      is_reg = true
   end
end
function disable_cb()
   if is_reg then
      luatexbase.remove_from_callback('post_linebreak_filter', 'Adjust width')
      is_reg = false
   end
end