summaryrefslogtreecommitdiff
path: root/tex/context/base/publ-fnd.lua
blob: ba60bdca0dd5396c2e8c67de2086137ace1db54a (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
if not modules then modules = { } end modules ['publ-fnd'] = {
    version   = 1.001,
    comment   = "this module part of publication support",
    author    = "Hans Hagen, PRAGMA-ADE, Hasselt NL",
    copyright = "PRAGMA ADE / ConTeXt Development Team",
    license   = "see context related readme files"
}

if not characters then
    dofile(resolvers.findfile("char-def.lua"))
    dofile(resolvers.findfile("char-utf.lua"))
end

-- this tracker is only for real debugging and not for the average user

local trace_match = false  trackers.register("publications.cite.match", function(v) trace_match = v end)

if not publications then
    publications = { }
end

local tonumber, next, type = tonumber, next, type
local P, R, C, Cs, Cp, Cc, Carg = lpeg.P, lpeg.R, lpeg.C, lpeg.Cs, lpeg.Cp, lpeg.Cc, lpeg.Carg
local lpegmatch, lpegpatterns = lpeg.match, lpeg.patterns
local concat = table.concat

local formatters = string.formatters
local lowercase  = characters.lower

local report  = logs.reporter("publications","match")

local colon   = P(":")
local dash    = P("-")
local lparent = P("(")
local rparent = P(")")
local space   = lpegpatterns.whitespace
local valid   = 1 - colon - space - lparent - rparent
----- key     = C(valid^1)
local key     = C(R("az","AZ")^1)
local word    = Cs(lpegpatterns.unquoted + lpegpatterns.argument + valid^1)
local number  = C(valid^1)

----- f_string_key = formatters["  local s_%s = entry[%q]"]
local f_string_key = formatters["  local s_%s = entry[%q] if s_%s then s_%s = lower(s_%s) end "]
local f_number_key = formatters["  local n_%s = tonumber(entry[%q]) or 0"]
local f_field_key  = formatters["  local f_%s = entry[%q] or ''"]

----- f_string_match = formatters["(s_%s and find(lower(s_%s),%q))"]
local f_string_match = formatters["(s_%s and find(s_%s,%q))"]
local f_number_match = formatters["(n_%s and n_%s >= %s and n_%s <= %s)"]
local f_field_match  = formatters["f_%s"]

local match  = key * (colon/"") * word * Carg(1) / function(key,_,word,keys)
 -- keys[key] = f_string_key(key,key)
    keys[key] = f_string_key(key,key,key,key,key)
    return f_string_match(key,key,lowercase(word))
end

local range  = key * (colon/"") * number * (dash/"") * number * Carg(1)  / function(key,_,first,_,last,keys)
    keys[key] = f_number_key(key,key)
    return f_number_match(key,key,tonumber(first) or 0,key,tonumber(last) or 0)
end

local field  = (P("field:")/"") * key * Carg(1) / function(_,key,keys)
    keys[key] = f_field_key(key,key)
    return f_field_match(key)
end

----- b_match = lparent
----- e_match = rparent * space^0 * P(-1)
----- pattern = Cs(b_match * ((field + range + match + space + P(1))-e_match)^1 * e_match)

local b_match = lparent
local e_match = rparent * space^0 * (#P(-1) + P(",")/" or ") -- maybe also + -> and
local p_match = b_match * ((field + range + match + space + P(1))-e_match)^1 * e_match
local pattern = Cs(Cc("(") * (P("match")/"" * space^0 * p_match)^1 * Cc(")"))

-- -- -- -- -- -- -- -- -- -- -- -- --
-- -- -- -- -- -- -- -- -- -- -- -- --

-- no longer faster
--
-- local tolower  = lpegpatterns.tolower
-- local lower    = string.lower
--
-- local allascii = R("\000\127")^1 * P(-1)
--
-- function characters.checkedlower(str)
--     return lpegmatch(allascii,str) and lower(str) or lpegmatch(tolower,str) or str
-- end

-- -- -- -- -- -- -- -- -- -- -- -- --
-- -- -- -- -- -- -- -- -- -- -- -- --

local f_template = string.formatters[ [[
local find = string.find
local lower = characters.lower
return function(entry)
%s
  return %s and true or false
end
]] ]

----- function compile(expr,start)
local function compile(expr)
    local keys        = { }
 -- local expression  = lpegmatch(pattern,expr,start,keys)
    local expression  = lpegmatch(pattern,expr,1,keys)
    if trace_match then
        report("compiling expression: %s",expr)
    end
    local definitions = { }
    for k, v in next, keys do
        definitions[#definitions+1] = v
    end
    if #definitions == 0 then
        report("invalid expression: %s",expr)
    elseif trace_match then
        for i=1,#definitions do
            report("% 3i : %s",i,definitions[i])
        end
    end
    definitions = concat(definitions,"\n")
    local code = f_template(definitions,expression)
    if trace_match then
        report("generated code: %s",code)
    end
    code = loadstring(code)
    if type(code) == "function" then
        code = code()
        if type(code) == "function" then
            return code
        end
    end
    report("invalid expression: %s",expr)
    return false
end

-- print(lpegmatch(pattern,"match ( author:cleveland and year:1993 ) "),1,{})

-- compile([[match(key:"foo bar")]])
-- compile([[match(key:'foo bar')]])
-- compile([[match(key:{foo bar})]])

local cache = { } -- todo: make weak, or just remember the last one (trial typesetting)

local check = P("match") -- * space^0 * Cp()

local function finder(expression)
    local found = cache[expression]
    if found == nil then
     -- local e = lpegmatch(check,expression)
     -- found = e and compile(expression,e) or false
        found = lpegmatch(check,expression) and compile(expression) or false
        if found then
            local okay, message = pcall(found,{})
            if not okay then
                found = false
                report("error in match: %s",message)
            end
        end
        cache[expression] = found
    end
    return found
end

-- finder("match(author:foo)")
-- finder("match(author:foo and author:bar)")
-- finder("match(author:foo or (author:bar and page:123))")
-- finder("match(author:foo),match(author:foo)")

publications.finder = finder

function publications.search(dataset,expression)
    local find = finder(expression)
    if find then
        local ordered = dataset.ordered
        local target  = { }
        for i=1,#ordered do
            local entry = ordered[i]
            if find(entry) then
                target[entry.tag] = entry
            end
        end
        return target
    else
        return { source[expression] }
    end
end

-- local dataset = publications.new()
-- publications.load(dataset,"t:/manuals/hybrid/tugboat.bib")
--
-- local n = 500
--
-- local function test(dataset,str)
--     local found
--     local start = os.clock()
--     for i=1,n do
--         found = search(dataset,str)
--     end
--     local elapsed = os.clock() - start
--     print(elapsed,elapsed/500,#table.keys(dataset.luadata),str)
--     print(table.concat(table.sortedkeys(found)," "))
--     return found
-- end
--
-- local found = test(dataset,[[match(author:hagen)]])
-- local found = test(dataset,[[match(author:hagen and author:hoekwater and year:1990-2010)]])
-- local found = test(dataset,[[match(author:"Bogusław Jackowski")]])
-- local found = test(dataset,[[match(author:"Bogusław Jackowski" and (tonumber(field:year) or 0) > 2000)]])
-- local found = test(dataset,[[Hagen:TB19-3-304]])

-- 1.328	0.002656	2710	author:hagen
-- Berdnikov:TB21-2-129 Guenther:TB5-1-24 Hagen:TB17-1-54 Hagen:TB19-3-304 Hagen:TB19-3-311 Hagen:TB19-3-317 Hagen:TB22-1-58 Hagen:TB22-3-118 Hagen:TB22-3-136 Hagen:TB22-3-160 Hagen:TB23-1-49 Hagen:TB25-1-108 Hagen:TB25-1-48 Hagen:TB26-2-152

-- 1.812	0.003624	2710	author:hagen and author:hoekwater and year:1990-2010
-- Berdnikov:TB21-2-129

-- 1.344	0.002688	2710	author:"Bogusław Jackowski"
-- Berdnikov:TB21-2-129 Jackowski:TB16-4-388 Jackowski:TB19-3-267 Jackowski:TB19-3-272 Jackowski:TB20-2-104 Jackowski:TB24-1-64 Jackowski:TB24-3-575 Nowacki:TB19-3-242 Rycko:TB14-3-171

-- 1.391	0.002782	2710	author:"Bogusław Jackowski" and (tonumber(field:year) or 0) > 2000
-- Jackowski:TB24-1-64 Jackowski:TB24-3-575