qml: Create VideoGroupsDisplay
[vlc.git] / share / lua / playlist / cue.lua
1 --[[
2    Parse CUE files
3
4  $Id$
5  Copyright (C) 2009 Laurent Aimar
6
7  This program is free software; you can redistribute it and/or modify
8  it under the terms of the GNU General Public License as published by
9  the Free Software Foundation; either version 2 of the License, or
10  (at your option) any later version.
11
12  This program is distributed in the hope that it will be useful,
13  but WITHOUT ANY WARRANTY; without even the implied warranty of
14  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  GNU General Public License for more details.
16
17  You should have received a copy of the GNU General Public License
18  along with this program; if not, write to the Free Software
19  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
20 --]]
21
22 -- Probe function.
23 function probe()
24     if( not string.match( vlc.path, "%.[cC][uU][eE]$" ) ) then
25         return false
26     end
27     header = vlc.peek( 2048 )
28     return string.match( header, "FILE.*WAVE%s*[\r\n]+" ) or
29            string.match( header, "FILE.*AIFF%s*[\r\n]+" ) or
30            string.match( header, "FILE.*MP3%s*[\r\n]+" )
31 end
32
33 -- Helpers
34 function is_utf8( src )
35     return vlc.strings.from_charset( "UTF-8", src ) == src
36 end
37
38 function cue_string( src )
39     if not is_utf8( src ) then
40         -- Convert to UTF-8 since it's probably Latin1
41         src = vlc.strings.from_charset( "ISO_8859-1", src )
42     end
43     local sub = string.match( src, "^\"(.*)\".*$" );
44     if( sub ) then
45         return sub
46     end
47     return string.match( src, "^(%S+).*$" )
48 end
49
50 function cue_path( src )
51     if( string.match( src, "^/" ) or
52         string.match( src, "^\\" ) or
53         string.match( src, "^[%l%u]:\\" ) ) then
54         return vlc.strings.make_uri(src)
55     end
56
57     local slash = string.find( string.reverse( vlc.path ), '/' )
58         local prefix = vlc.access .. "://" .. string.sub( vlc.path, 1, -slash )
59         -- FIXME: postfix may not be encoded correctly (esp. slashes)
60         local postfix = vlc.strings.encode_uri_component(src)
61     return prefix .. postfix
62 end
63
64 function cue_track( global, track )
65     if( track.index01 == nil ) then
66         return nil
67     end
68
69     t = {}
70     t.path = cue_path( track.file or global.file )
71     t.title = track.title
72     t.album = global.title
73     t.artist = track.performer or global.performer
74     t.genre = track.genre or global.genre
75     t.date = track.date or global.date
76     t.description = global.comment
77     t.tracknum = track.num
78     t.options = { ":start-time=" .. track.index01}
79
80     return t
81 end
82
83 function cue_append( tracks, global, track )
84     local t = cue_track( global, track )
85     if( t ~= nil ) then
86         if( #tracks > 0 ) then
87             local prev = tracks[#tracks]
88             table.insert( prev.options, ":stop-time=" .. track.index01)
89         end
90         table.insert( tracks, t )
91     end
92 end
93
94 -- Parse function.
95 function parse()
96     p = {}
97     global_data = nil
98     data = {}
99     file = nil
100
101     while true
102     do 
103         line = vlc.readline()
104         if not line then break end
105
106         cmd, arg = string.match( line, "^%s*(%S+)%s*(.*)$" )
107
108         if(  cmd == "REM" and arg ) then
109             subcmd, value = string.match( arg, "^(%S+)%s*(.*)$" )
110             if( subcmd == "GENRE" and value ) then
111                 data.genre = cue_string( value )
112             elseif( subcmd == "DATE" and value ) then
113                 data.date = cue_string( value )
114             elseif( subcmd == "COMMENT" and value ) then
115                 data.comment = cue_string( value )
116             end
117         elseif( cmd == "PERFORMER" and arg ) then
118             data.performer = cue_string( arg )
119         elseif( cmd == "TITLE" and arg ) then
120             data.title = cue_string( arg )
121         elseif( cmd == "FILE" ) then
122             file = cue_string( arg )
123         elseif( cmd == "TRACK" ) then
124             if( not global_data ) then
125                 global_data = data
126             else
127                 cue_append( p, global_data, data )
128             end
129             data =  { file = file, num = string.match( arg, "^(%d+)" ) }
130         elseif( cmd == "INDEX" ) then
131             local idx, m, s, f = string.match( arg, "(%d+)%s+(%d+):(%d+):(%d+)" )
132             if( idx == "01" and m ~= nil and s ~= nil and f ~= nil ) then
133                 data.index01 = m * 60 + s + f / 75
134             end
135         end
136     end
137
138     cue_append( p, global_data, data )
139
140     return p
141 end
142