OSDN Git Service

fix #41020
[luatex-ja/luatexja.git] / src / ltj-adjust.lua
1 --
2 -- ltj-adjust.lua
3 --
4 luatexja.load_module 'base';      local ltjb = luatexja.base
5 luatexja.load_module 'jfont';     local ltjf = luatexja.jfont
6 luatexja.load_module 'jfmglue';   local ltjj = luatexja.jfmglue
7 luatexja.load_module 'stack';     local ltjs = luatexja.stack
8 luatexja.load_module 'direction'; local ltjd = luatexja.direction
9 luatexja.load_module 'lineskip';  local ltjl = luatexja.lineskip
10 luatexja.adjust = luatexja.adjust or {}
11
12 local to_node = node.direct.tonode
13 local to_direct = node.direct.todirect
14
15 local setfield = node.direct.setfield
16 local setglue = luatexja.setglue
17 local getfield = node.direct.getfield
18 local getlist = node.direct.getlist
19 local getid = node.direct.getid
20 local getfont = node.direct.getfont
21 local getsubtype = node.direct.getsubtype
22
23 local node_traverse_id = node.direct.traverse_id
24 local node_new = node.direct.new
25 local node_next = node.direct.getnext
26 local node_free = node.direct.free
27 local node_prev = node.direct.getprev
28 local node_tail = node.direct.tail
29 local has_attr = node.direct.has_attribute
30 local set_attr = node.direct.set_attribute
31 local insert_after = node.direct.insert_after
32
33 local id_glyph   = node.id 'glyph'
34 local id_kern    = node.id 'kern'
35 local id_hlist   = node.id 'hlist'
36 local id_glue    = node.id 'glue'
37 local id_whatsit = node.id 'whatsit'
38 local id_penalty = node.id 'penalty'
39 local attr_icflag = luatexbase.attributes['ltj@icflag']
40 local attr_jchar_class = luatexbase.attributes['ltj@charclass']
41 local lang_ja = luatexja.lang_ja
42
43 local ltjf_font_metric_table = ltjf.font_metric_table
44 local ipairs, pairs = ipairs, pairs
45
46 local PACKED       = luatexja.icflag_table.PACKED
47 local LINEEND      = luatexja.icflag_table.LINEEND
48 local FROM_JFM     = luatexja.icflag_table.FROM_JFM
49 local KANJI_SKIP   = luatexja.icflag_table.KANJI_SKIP
50 local KANJI_SKIP_JFM = luatexja.icflag_table.KANJI_SKIP_JFM
51 local XKANJI_SKIP  = luatexja.icflag_table.XKANJI_SKIP
52 local XKANJI_SKIP_JFM  = luatexja.icflag_table.XKANJI_SKIP_JFM
53
54 local get_attr_icflag
55 do
56    local PROCESSED_BEGIN_FLAG = luatexja.icflag_table.PROCESSED_BEGIN_FLAG
57    get_attr_icflag = function(p)
58       return (has_attr(p, attr_icflag) or 0) % PROCESSED_BEGIN_FLAG
59    end
60 end
61
62 local priority_num = { 0, 0 }
63 local at2pr = { {}, {} }
64 local at2pr_st, at2pr_sh = at2pr[1], at2pr[2]
65 do
66    local priority_table = {{},{}}
67    luatexja.adjust.priority_table = priority_table
68    local tmp = {}
69    local function cmp(a,b) return a[1]>b[1] end -- 大きいほうが先!
70    local function make_priority_table(glue_sign)
71       for i,_ in pairs(tmp) do tmp[i]=nil end
72       if glue_sign==2 then -- shrink
73          for i=0,63 do tmp[#tmp+1] = { (i%8)-4, FROM_JFM+i } end
74       else -- stretch
75          for i=0,63 do tmp[#tmp+1] = { math.floor(i/8)-4, FROM_JFM+i } end
76       end    
77       local pt = priority_table[glue_sign]
78       tmp[#tmp+1] = { pt[2]/10, XKANJI_SKIP }
79       tmp[#tmp+1] = { pt[2]/10, XKANJI_SKIP_JFM }
80       tmp[#tmp+1] = { pt[1]/10, KANJI_SKIP }
81       tmp[#tmp+1] = { pt[1]/10, KANJI_SKIP_JFM }
82       tmp[#tmp+1] = { pt[3]/10, -1 }
83       table.sort(tmp, cmp)
84       local a, m, n = at2pr[glue_sign], 10000000, 0
85       for i=1,#tmp do
86          if tmp[i][1]<m then n,m = n+1,tmp[i][1] end
87          a[tmp[i][2]] = n
88       end
89       local o = a[-1]
90       priority_num[glue_sign] = n
91       setmetatable(a, {__index = function () return o end })
92    end
93    luatexja.adjust.make_priority_table = make_priority_table
94 end
95
96 -- box 内で伸縮された glue の合計値を計算
97
98 local total_stsh = {{},{}}
99 local total_st, total_sh = total_stsh[1], total_stsh[2]
100 local get_total_stretched
101 do
102 local dimensions = node.direct.dimensions
103 function get_total_stretched(p)
104 -- return value: <補正値(sp)>
105    local ph = getlist(p)
106    if not ph then return 0 end
107    for i,_ in pairs(total_st) do total_st[i]=nil; total_sh[i]=nil end
108    for i=1,priority_num[1] do total_st[i]=0 end
109    for i=1,priority_num[2] do total_sh[i]=0 end
110    for i=0,4 do total_st[i*65536]=0; total_sh[i*65536]=0 end
111    for q in node_traverse_id(id_glue, ph) do
112       local a = getfield(q, 'stretch_order')
113       if a==0 then
114          local b = at2pr_st[get_attr_icflag(q)]; 
115          total_st[b] = total_st[b]+getfield(q, 'stretch')
116       end
117       total_st[a*65536] = total_st[a]+getfield(q, 'stretch')
118       local a = getfield(q, 'shrink_order')
119       if a==0 then
120          local b = at2pr_sh[get_attr_icflag(q)]; 
121          total_sh[b] = total_sh[b]+getfield(q, 'shrink')
122       end
123       total_sh[a*65536] = total_sh[a]+getfield(q, 'shrink')
124    end
125    for i=4,1,-1 do if total_st[i*65536]~=0 then total_st.order=i; break end; end
126    if not total_st.order then
127        total_st.order, total_st[-65536] = -1,0.1 -- dummy
128    end
129    for i=4,1,-1 do if total_sh[i*65536]~=0 then total_sh.order=i; break end; end
130    if not total_sh.order then
131        total_sh.order, total_sh[-65536] = -1,0.1 -- dummy
132    end
133    return getfield(p,'width') - dimensions(ph)
134 end
135 end
136
137 -- step 1: 行末に kern を挿入(句読点,中点用)
138 local abs = math.abs
139 local ltjd_glyph_from_packed = ltjd.glyph_from_packed
140 local function aw_step1(p, total)
141    local head = getlist(p)
142    local x = node_tail(head); if not x then return total, false end
143    -- x: \rightskip
144    x = node_prev(x); if not x then return total, false end
145    local xi, xc = getid(x)
146    -- x may be penalty
147    while xi==id_penalty do
148       x = node_prev(x); if not x then return total, false end
149       xi = getid(x)
150    end
151    if (total>0 and total_st.order>0) or (total<0 and total_sh.order>0) then
152        -- 無限大のグルーで処理が行われているときは処理中止.
153        return total, false
154    end
155    if xi == id_glyph and getfield(x, 'lang')==lang_ja then
156       -- 和文文字
157       xc = x
158    elseif xi == id_hlist and get_attr_icflag(x) == PACKED then
159       -- packed JAchar
160       xc = ltjd_glyph_from_packed(x)
161       while getid(xc) == id_whatsit do xc = node_next(xc) end -- これはなんのために?
162    else
163       return total, false-- それ以外は対象外.
164    end
165    local eadt = ltjf_font_metric_table[getfont(xc)]
166       .char_type[has_attr(xc, attr_jchar_class) or 0].end_adjust
167    if not eadt then 
168       return total, false
169    end
170    local eadt_ratio = {}
171    for i, v in ipairs(eadt) do
172       local t = total - v
173       if t>0 then
174          eadt_ratio[i] = {i, t/total_st[65536*total_st.order], t, v}
175       else
176          eadt_ratio[i] = {i, t/total_sh[65536*total_sh.order], t, v}
177       end
178    end
179    table.sort(eadt_ratio, 
180    function (a,b) 
181        for i=2,4 do
182            local at, bt = abs(a[i]), abs(b[i])
183            if at~=bt then return at<bt end
184        end
185        return a[4]<b[4]
186    end)
187    if eadt[eadt_ratio[1][1]]~=0 then
188       local kn = node_new(id_kern, 1)
189       setfield(kn, 'kern', eadt[eadt_ratio[1][1]]); set_attr(kn, attr_icflag, LINEEND)
190       insert_after(head, x, kn)
191       return eadt_ratio[1][3], true
192    else
193       return total, false
194    end
195 end
196
197 -- step 1 最終行用
198 local min, max = math.min, math.max
199 local function aw_step1_last(p, total)
200    local head = getlist(p)
201    local x = node_tail(head); if not x then return total, false end
202    -- x: \rightskip
203    local pf = node_prev(x); if not x then return total, false end
204    if getid(pf) ~= id_glue or getsubtype(pf) ~= 15 then return total, false end
205    x = node_prev(node_prev(pf))
206    local xi, xc = getid(x)
207    if xi == id_glyph and getfield(x, 'lang')==lang_ja then
208       -- 和文文字
209       xc = x
210    elseif xi == id_hlist and get_attr_icflag(x) == PACKED then
211       -- packed JAchar
212       xc = ltjd_glyph_from_packed(x)
213       while getid(xc) == id_whatsit do xc = node_next(xc) end -- これはなんのために?
214    else
215       return total, false-- それ以外は対象外.
216    end
217    -- 続行条件1:無限の伸縮度を持つグルーは \parfillskipのみ
218    if total>0 and total_st.order>0 then
219       if total_st.order ~= getfield(pf, 'stretch_order') then return total, false end
220       if total_st[total_st.order*65536] ~= getfield(pf, 'stretch') then return total, false end
221       for i=total_st.order-1, 1, -1 do
222          if total_st[i*65536] ~= 0 then return total, false end
223       end
224    end
225    if total<0 and total_sh.order>0 then
226       if total_sh.order ~= getfield(pf, 'shrink_order') then return total, false end
227       if total_sh[total_sh.order*65536] ~= getfield(pf, 'shrink') then return total, false end
228       for i=total_sh.order-1, 1, -1 do
229          if total_sh[i*65536] ~= 0 then return total, false end
230       end
231    end
232    local eadt = ltjf_font_metric_table[getfont(xc)]
233       .char_type[has_attr(xc, attr_jchar_class) or 0].end_adjust
234    if not eadt then 
235       return total, false
236    end
237    -- 続行条件2: min(eadt[1], 0)<= \parfillskip <= max(eadt[#eadt], 0)
238    local pfw = getfield(pf, 'width') 
239      + (total>0 and getfield(pf, 'stretch') or -getfield(pf, 'shrink')) *getfield(p, 'glue_set') 
240    if pfw<min(0,eadt[1]) or max(0,eadt[#eadt])<pfw then return total, false end
241    -- \parfillskip を 0 にする
242    total = total + getfield(pf, 'width') 
243    total_st.order, total_sh.order = 0, 0
244    if getfield(pf, 'stretch_order')==0 then 
245       local i = at2pr_st[-1] 
246       total_st[0] = total_st[0] - getfield(pf, 'stretch') 
247       total_st[i] = total_st[i] - getfield(pf, 'stretch') 
248       total_st.order = (total_st[0]==0) and -1 or 0
249    end
250    if getfield(pf, 'shrink_order')==0 then 
251       local i = at2pr_sh[-1] 
252       total_sh[0] = total_sh[0] - getfield(pf, 'shrink') 
253       total_sh[i] = total_sh[i] - getfield(pf, 'shrink') 
254       total_sh.order = (total_sh[0]==0) and -1 or 0
255    end
256    setfield(pf, 'subtype', 1); setglue(pf)
257    local eadt_ratio = {}
258    for i, v in ipairs(eadt) do
259       local t = total - v
260       if t>0 then
261          eadt_ratio[i] = {i, t/total_st[65536*total_st.order], t, v}
262       else
263          eadt_ratio[i] = {i, t/total_sh[65536*total_sh.order], t, v}
264       end
265    end
266    table.sort(eadt_ratio, 
267    function (a,b) 
268        for i=2,4 do
269            local at, bt = abs(a[i]), abs(b[i])
270            if at~=bt then return at<bt end
271        end
272        return a[4]<b[4]
273    end)
274    if eadt[eadt_ratio[1][1]]~=0 then
275       local kn = node_new(id_kern, 1)
276       setfield(kn, 'kern', eadt[eadt_ratio[1][1]]); set_attr(kn, attr_icflag, LINEEND)
277       insert_after(head, x, kn)
278       return eadt_ratio[1][3], true
279    else
280       return total, false
281    end
282 end
283
284
285 -- step 2: 行中の glue を変える
286 local aw_step2, aw_step2_dummy
287 do
288 local node_hpack = node.direct.hpack
289 local function repack(p)
290    local orig_of, orig_hfuzz, orig_hbad = tex.overfullrule, tex.hfuzz, tex.hbadness
291    tex.overfullrule=0; tex.hfuzz=1073741823; tex.hbadness=10000
292    local f = node_hpack(getlist(p), getfield(p, 'width'), 'exactly')
293    tex.overfullrule=orig_of; tex.hfuzz=orig_hfuzz; tex.hbadness=orig_hbad
294    setfield(f, 'head', nil)
295    setfield(p, 'glue_set', getfield(f, 'glue_set'))
296    setfield(p, 'glue_order', getfield(f, 'glue_order'))
297    setfield(p, 'glue_sign', getfield(f, 'glue_sign'))
298    node_free(f)
299    return
300 end
301 function aw_step2_dummy(p, _, added_flag)
302    if added_flag then return repack(p) end
303 end
304
305 local function clear_stretch(p, ind, ap, name)
306    for q in node_traverse_id(id_glue, getlist(p)) do
307       local f = ap[get_attr_icflag(q)]
308       if f == ind then
309          setfield(q, name..'_order', 0); setfield(q, name, 0)
310       end
311    end
312 end
313
314 local function set_stretch(p, after, before, ind, ap, name)
315    if before > 0 then
316       local ratio = after/before
317       for q in node_traverse_id(id_glue, getlist(p)) do
318          local f = ap[get_attr_icflag(q)]
319          if (f==ind) and getfield(q, name..'_order')==0 then
320             setfield(q, name, getfield(q, name)*ratio)
321          end
322       end
323    end
324 end
325
326 function aw_step2(p, total, added_flag)
327    local name = (total>0) and 'stretch' or 'shrink'
328    local id =  (total>0) and 1 or 2
329    local res = total_stsh[id]
330    local pnum = priority_num[id]
331    if total==0 or res.order > 0 then 
332       -- もともと伸縮の必要なしか,残りの伸縮量は無限大
333       if added_flag then return repack(p) end
334    end
335    total = abs(total)
336    for i = 1, pnum do
337       if total <= res[i] then
338          local a = at2pr[id]  
339          for j = i+1,pnum do
340             clear_stretch(p, j, a, name)
341          end
342          set_stretch(p, total, res[i], i, a, name); break
343       end
344       total = total - res[i]
345    end
346    return repack(p)
347 end
348 end
349
350 -- step 1': lineend=extended の場合(行分割時に考慮))
351 local insert_lineend_kern
352 do
353    local insert_before = node.direct.insert_before
354    local KINSOKU      = luatexja.icflag_table.KINSOKU
355    insert_lineend_kern = function (head, nq, np, Bp)
356       if nq.met then 
357          local eadt = nq.met.char_type[nq.class].end_adjust
358          if not eadt then return end
359          if eadt[1]~=0 then
360             local x = node_new(id_kern, 1)
361             setfield(x, 'kern', eadt[1]); set_attr(x, attr_icflag, LINEEND)
362             insert_before(head, np.first, x)
363          end
364          local eadt_num = #eadt
365          for i=2,eadt_num do
366             local x = node_new(id_penalty)
367             setfield(x, 'penalty', 0); set_attr(x, attr_icflag, KINSOKU)
368             insert_before(head, np.first, x); Bp[#Bp+1] = x
369             local x = node_new(id_kern, 1)
370             setfield(x, 'kern', eadt[i]-eadt[i-1]); set_attr(x, attr_icflag, LINEEND)
371             insert_before(head, np.first, x)
372          end
373          if eadt_num>1 or eadt[1]~=0 then
374             local x = node_new(id_penalty)
375             setfield(x, 'penalty', 0); set_attr(x, attr_icflag, KINSOKU)
376             insert_before(head, np.first, x); Bp[#Bp+1] = x
377             local x = node_new(id_kern, 1)
378             setfield(x, 'kern', -eadt[eadt_num]); set_attr(x, attr_icflag, LINEEND)
379             insert_before(head, np.first, x)
380             local x = node_new(id_penalty)
381             setfield(x, 'penalty', 10000); set_attr(x, attr_icflag, KINSOKU)
382             insert_before(head, np.first, x); Bp[#Bp+1] = x
383          end
384       end
385    end
386 end
387
388 local adjust_width
389 do
390    local myaw_step1, myaw_step2, myaw_step1_last
391    local dummy =  function(p,t,n) return t, false end
392    local ltjs_fast_get_stack_skip = ltjs.fast_get_stack_skip
393    function adjust_width(head)
394       if not head then return head end
395       local last_p
396       for p in node_traverse_id(id_hlist, to_direct(head)) do
397          if last_p then
398             myaw_step2(last_p, myaw_step1(last_p, get_total_stretched(last_p)))
399          end
400          last_p = p
401       end
402       if last_p then
403          myaw_step2(last_p, myaw_step1_last(last_p, get_total_stretched(last_p)))
404       end
405       return to_node(head)
406    end
407    local is_reg = false
408    local function enable_cb(status_le, status_pr, status_lp, status_ls)
409       if (status_le>0 or status_pr>0) and (not is_reg) then
410          ltjb.add_to_callback('post_linebreak_filter',
411             adjust_width, 'Adjust width', 
412             luatexbase.priority_in_callback('post_linebreak_filter', 'ltj.lineskip')-1)
413          is_reg = true
414       elseif is_reg and (status_le==0 and status_pr==0) then
415          luatexbase.remove_from_callback('post_linebreak_filter', 'Adjust width')
416          is_reg = false
417       end
418       if status_le==2 then
419          if not luatexbase.in_callback('luatexja.adjust_jfmglue', 'luatexja.adjust') then
420             ltjb.add_to_callback('luatexja.adjust_jfmglue', insert_lineend_kern, 'luatexja.adjust')
421          end
422          myaw_step1, myaw_step1_last = dummy, aw_step1_last
423       else
424          if status_le==0 then
425             myaw_step1, myaw_step1_last = dummy, dummy
426          else
427             myaw_step1, myaw_step1_last = aw_step1, aw_step1_last
428          end
429          if luatexbase.in_callback('luatexja.adjust_jfmglue', 'luatexja.adjust') then
430                luatexbase.remove_from_callback('luatexja.adjust_jfmglue', 'luatexja.adjust')
431          end
432       end
433       myaw_step2 = (status_pr>0) and aw_step2 or aw_step2_dummy
434       luatexja.lineskip.setting(
435          status_lp>0 and 'profile' or 'dummy',
436          status_ls>0 and 'step' or 'dummy'
437       )      
438    end
439    local function disable_cb() -- only for compatibility
440        enable_cs(0,0,0,0)
441    end
442    luatexja.adjust.enable_cb=enable_cb
443    luatexja.adjust.disable_cb=disable_cb
444 end
445
446 luatexja.unary_pars.adjust = function(t)
447    return is_reg and 1 or 0
448 end
449
450 -- ----------------------------------
451 local init_range
452 do
453   local max, ins, sort = math.max, table.insert, table.sort
454   local function insert(package, ind, d, b, e)
455     local bound = package[2]
456     bound[b], bound[e]=true, true
457     ins(package[1], {b,e,[ind]=d})
458   end
459   local function flatten(package)
460     local bd = {} for i,_ in pairs(package[2]) do ins(bd,{i}) end
461     sort(bd, function (a,b) return a[1]<b[1] end)
462     local bdc=#bd; local t = package[1]
463     sort(t, function (a,b) return a[1]<b[1] end)
464     local bdi =1
465     for i=1,#t do
466       while bd[bdi][1]<t[i][1] do bdi=bdi+1 end
467       local j = bdi
468       while j<bdc and bd[j+1][1]<=t[i][2] do
469         for k,w in pairs(t[i]) do
470           if k>=3 then
471             bd[j][k]=bd[j][k] and max(bd[j][k],w) or w
472           end
473         end
474         j = j + 1
475       end
476     end
477     package[2]=nil; package[1]=nil; package.flatten, package.insert=nil, nil
478     bd[#bd]=nil
479     return bd
480   end
481   init_range = function ()
482     return {{},{}, insert=insert, flatten=flatten}
483   end
484 end
485
486 -- -----------------------------------
487 luatexja.adjust.step_factor = 0.5
488 luatexja.unary_pars.linestep_factor = function(t)
489    return luatexja.adjust.step_factor
490 end
491 luatexja.adjust.profile_hgap_factor = 1
492 luatexja.unary_pars.profile_hgap_factor = function(t)
493    return luatexja.adjust.profile_hgap_factor
494 end
495 do
496   local insert = table.insert
497   local rangedimensions, max = node.direct.rangedimensions, math.max
498   local function profile_inner(box, range, ind, vmirrored, adj)
499     local w_acc, d_before = getfield(box,'shift'), 0
500     local x = getlist(box); local xn = node_next(x)
501     while x do
502       local w, h, d
503       if xn then w, h, d= rangedimensions(box,x,xn)
504       else w, h, d= rangedimensions(box,x) end
505       if vmirrored then h=d end
506       local w_new = w_acc + w
507       if w>=0 then
508         range:insert(ind, h, w_acc-adj, w_new)
509       else
510         range:insert(ind, h, w_new-adj, w_acc)
511       end
512       w_acc = w_new; x = xn; if x then xn = node_next(x) end
513     end
514   end  
515   function ltjl.p_profile(before, after, mirrored, bw)
516     local range, tls 
517       = init_range(), luatexja.adjust.profile_hgap_factor*tex.lineskip.width
518     profile_inner(before, range, 3, true,     tls)
519     profile_inner(after,  range, 4, mirrored, tls)
520     range = range:flatten()
521     do
522       local dmax, d, hmax, h, lmin = 0, 0, 0, 0, 1/0
523       for i,v in ipairs(range) do
524         d, h = (v[3] or 0), (v[4] or 0)
525         if d>dmax then dmax=d end
526         if h>hmax then hmax=h end
527         if bw-h-d<lmin then lmin=bw-h-d end
528       end
529       if lmin==1/0 then lmin = bw end
530       return lmin, 
531          bw - lmin - getfield(before, 'depth')
532             - getfield(after, mirrored and 'depth' or 'height')
533     end
534   end
535 end
536
537 do
538   local ltja = luatexja.adjust
539   local copy_glue = ltjl.copy_glue
540   local floor, max = math.floor, math.max
541   function ltjl.l_step(dist, g, adj, normal, bw, loc)
542     if loc=='alignment' then
543       return ltjl.l_dummy(dist, g, adj, normal, bw, loc)
544     end
545     if dist < tex.lineskiplimit then
546     local f = max(1, bw*ltja.step_factor)
547        copy_glue(g, tex.baselineskip, 1, normal - f * floor((dist-tex.lineskip.width)/f))
548     else
549        copy_glue(g, tex.baselineskip, 2, normal)
550     end
551   end
552 end
553
554