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
|
if not modules then modules = { } end modules ['mtx-flac'] = {
version = 1.001,
comment = "companion to mtxrun.lua",
author = "Hans Hagen, PRAGMA-ADE, Hasselt NL",
copyright = "PRAGMA ADE / ConTeXt Development Team",
license = "see context related readme files"
}
local sub, match, byte, lower = string.sub, string.match, string.byte, string.lower
local readstring, readnumber = io.readstring, io.readnumber
local concat, sortedpairs = table.concat, table.sortedpairs
local tonumber = tonumber
local tobitstring = number.tobitstring
local lpegmatch = lpeg.match
local p_escaped = lpeg.patterns.xml.escaped
-- rather silly: pack info in bits while a flac file is large anyway
flac = flac or { }
flac.report = string.format
local splitter = lpeg.splitat("=")
local readers = { }
readers[0] = function(f,size,target) -- not yet ok .. todo: use bit32 lib
local info = { }
target.info = info
info.minimum_block_size = readnumber(f,-2)
info.maximum_block_size = readnumber(f,-2)
info.minimum_frame_size = readnumber(f,-3)
info.maximum_frame_size = readnumber(f,-3)
local buffer = { }
for i=1,8 do
buffer[i] = tobitstring(readnumber(f,1))
end
local bytes = concat(buffer)
info.sample_rate_in_hz = tonumber(sub(bytes, 1,20),2) -- 20
info.number_of_channels = tonumber(sub(bytes,21,23),2) -- 3
info.bits_per_sample = tonumber(sub(bytes,24,28),2) -- 5
info.samples_in_stream = tonumber(sub(bytes,29,64),2) -- 36
info.md5_signature = readstring(f,16) -- 128
end
readers[4] = function(f,size,target,banner)
local tags = { }
target.tags = tags
target.vendor = readstring(f,readnumber(f,-4))
for i=1,readnumber(f,-4) do
local key, value = lpeg.match(splitter,readstring(f,readnumber(f,-4)))
tags[lower(key)] = value
end
end
readers.default = function(f,size,target)
f:seek("cur",size)
end
function flac.getmetadata(filename)
local f = io.open(filename, "rb")
if f then
local banner = readstring(f,4)
if banner == "fLaC" then
local data = {
banner = banner,
filename = filename,
filesize = lfs.attributes(filename,"size"),
}
while true do
local flag = readnumber(f,1)
local size = readnumber(f,3)
local last = flag > 127
if last then
flag = flag - 128
end
local reader = readers[flag] or readers.default
reader(f,size,data,banner)
if last then
f:close()
return data
end
end
else
flac.report("no flac file: %s (%s)",filename,banner)
end
f:close()
else
flac.report("no file: %s",filename)
end
end
function flac.savecollection(pattern,filename)
pattern = (pattern ~= "" and pattern) or "**/*.flac"
filename = (filename ~= "" and filename) or "music-collection.xml"
flac.report("identifying files using pattern %q" ,pattern)
local files = dir.glob(pattern)
flac.report("%s files found, analyzing files",#files)
local music = { }
for i=1,#files do
local data = flac.getmetadata(files[i])
if data then
local tags = data.tags
local info = data.info
local artist = tags.artist
local album = tags.album
local albums = music[artist]
if not albums then
albums = { }
music[artist] = albums
end
local albumx = albums[album]
if not albumx then
albumx = {
year = tags.date,
tracks = { },
}
albums[album] = albumx
end
albumx.tracks[tonumber(tags.tracknumber) or 0] = {
title = tags.title,
length = math.round((info.samples_in_stream/info.sample_rate_in_hz)),
}
end
end
-- inspect(music)
local nofartists, nofalbums, noftracks, noferrors = 0, 0, 0, 0
local f = io.open(filename,"wb")
if f then
flac.report("saving data in file %q",filename)
f:write("<?xml version='1.0' standalone='yes'?>\n\n")
f:write("<collection>\n")
for artist, albums in sortedpairs(music) do
nofartists = nofartists + 1
f:write("\t<artist>\n")
f:write("\t\t<name>",lpegmatch(p_escaped,artist),"</name>\n")
f:write("\t\t<albums>\n")
for album, data in sortedpairs(albums) do
nofalbums = nofalbums + 1
f:write("\t\t\t<album year='",data.year or 0,"'>\n")
f:write("\t\t\t\t<name>",lpegmatch(p_escaped,album),"</name>\n")
f:write("\t\t\t\t<tracks>\n")
local tracks = data.tracks
for i=1,#tracks do
local track = tracks[i]
if track then
noftracks = noftracks + 1
f:write("\t\t\t\t\t<track length='",track.length,"'>",lpegmatch(p_escaped,track.title),"</track>\n")
else
noferrors = noferrors + 1
flac.report("error in album: %q of %q, no track %s",album,artist,i)
f:write("\t\t\t\t\t<error track='",i,"'/>\n")
end
end
f:write("\t\t\t\t</tracks>\n")
f:write("\t\t\t</album>\n")
end
f:write("\t\t</albums>\n")
f:write("\t</artist>\n")
end
f:write("</collection>\n")
f:close()
flac.report("%s tracks of %s albums of %s artists saved in %q (%s errors)",noftracks,nofalbums,nofartists,filename,noferrors)
else
flac.report("unable to save data in file %q",filename)
end
end
--
local helpinfo = [[
<?xml version="1.0"?>
<application>
<metadata>
<entry name="name">mtx-flac</entry>
<entry name="detail">ConTeXt Flac Helpers</entry>
<entry name="version">0.10</entry>
</metadata>
<flags>
<category name="basic">
<subcategory>
<flag name="collect"><short>collect albums in xml file</short></flag>
</subcategory>
</category>
</flags>
<examples>
<category>
<title>Example</title>
<subcategory>
<example><command>mtxrun --script flac --collect somename.flac</command></example>
<example><command>mtxrun --script flac --collect --pattern="m:/music/**")</command></example>
</subcategory>
</category>
</examples>
</application>
]]
local application = logs.application {
name = "mtx-flac",
banner = "ConTeXt Flac Helpers 0.10",
helpinfo = helpinfo,
}
flac.report = application.report
-- script code
scripts = scripts or { }
scripts.flac = scripts.flac or { }
function scripts.flac.collect()
local files = environment.files
local pattern = environment.arguments.pattern
if #files > 0 then
for i=1,#files do
local filename = files[1]
if file.suffix(filename) == "flac" then
flac.savecollection(filename,file.replacesuffix(filename,"xml"))
elseif lfs.isdir(filename) then
local pattern = filename .. "/**.flac"
flac.savecollection(pattern,file.addsuffix(file.basename(filename),"xml"))
else
flac.savecollection(file.replacesuffix(filename,"flac"),file.replacesuffix(filename,"xml"))
end
end
elseif pattern then
flac.savecollection(file.addsuffix(pattern,"flac"),"music-collection.xml")
else
flac.report("no file(s) or pattern given" )
end
end
if environment.argument("collect") then
scripts.flac.collect()
elseif environment.argument("exporthelp") then
application.export(environment.argument("exporthelp"),environment.files[1])
else
application.help()
end
|