| 1 | #if 0 |
|---|
| 2 | # ----------------------------------------------------------------------- |
|---|
| 3 | # $Id: eyed3info.py,v 1.16 2005/01/01 14:18:11 dischi Exp $ |
|---|
| 4 | # ----------------------------------------------------------------------- |
|---|
| 5 | # $Log: eyed3info.py,v $ |
|---|
| 6 | # Revision 1.16 2005/01/01 14:18:11 dischi |
|---|
| 7 | # add samplerate, bitrate and mode |
|---|
| 8 | # |
|---|
| 9 | # Revision 1.15 2004/09/09 02:45:58 outlyer |
|---|
| 10 | # Add the TPOS tag for multiple disc sets. |
|---|
| 11 | # |
|---|
| 12 | # Revision 1.14 2004/07/21 18:54:36 outlyer |
|---|
| 13 | # Big bugfix. |
|---|
| 14 | # |
|---|
| 15 | # If a mp3 lacks a genre, then nothing after the genre parsing works. This breaks |
|---|
| 16 | # calculating the length, among other things because it's all in a try except, so |
|---|
| 17 | # self.length takes on a random value. I was getting track lengths like 20000 seconds |
|---|
| 18 | # until I made this change. |
|---|
| 19 | # |
|---|
| 20 | # Revision 1.13 2004/07/11 12:06:03 dischi |
|---|
| 21 | # add genre parsing |
|---|
| 22 | # |
|---|
| 23 | # Revision 1.12 2004/07/07 09:35:50 dischi |
|---|
| 24 | # remove guessing of tracknum in TCON |
|---|
| 25 | # |
|---|
| 26 | # Revision 1.11 2004/05/29 12:31:36 dischi |
|---|
| 27 | # try to find trackof in TCON or trackno |
|---|
| 28 | # |
|---|
| 29 | # Revision 1.10 2004/02/08 17:42:56 dischi |
|---|
| 30 | # reduce debug |
|---|
| 31 | # |
|---|
| 32 | # Revision 1.9 2003/08/30 09:36:22 dischi |
|---|
| 33 | # turn off some debug based on DEBUG |
|---|
| 34 | # |
|---|
| 35 | # Revision 1.8 2003/07/10 13:05:05 the_krow |
|---|
| 36 | # o id3v2 tabled added to eyed3 |
|---|
| 37 | # o type changed to MUSIC |
|---|
| 38 | # |
|---|
| 39 | # Revision 1.7 2003/07/02 20:15:42 dischi |
|---|
| 40 | # return nothing, not 0 |
|---|
| 41 | # |
|---|
| 42 | # Revision 1.6 2003/06/30 13:17:18 the_krow |
|---|
| 43 | # o Refactored mediainfo into factory, synchronizedobject |
|---|
| 44 | # o Parsers now register directly at mmpython not at mmpython.mediainfo |
|---|
| 45 | # o use mmpython.Factory() instead of mmpython.mediainfo.get_singleton() |
|---|
| 46 | # o Bugfix in PNG parser |
|---|
| 47 | # o Renamed disc.AudioInfo into disc.AudioDiscInfo |
|---|
| 48 | # o Renamed disc.DataInfo into disc.DataDiscInfo |
|---|
| 49 | # |
|---|
| 50 | # Revision 1.5 2003/06/30 11:38:22 dischi |
|---|
| 51 | # bugfix |
|---|
| 52 | # |
|---|
| 53 | # Revision 1.4 2003/06/29 18:30:14 dischi |
|---|
| 54 | # many many fixes |
|---|
| 55 | # |
|---|
| 56 | # Revision 1.3 2003/06/29 12:03:41 dischi |
|---|
| 57 | # fixed it to be _real_ eyed3 info |
|---|
| 58 | # |
|---|
| 59 | # Revision 1.2 2003/06/20 19:17:22 dischi |
|---|
| 60 | # remove filename again and use file.name |
|---|
| 61 | # |
|---|
| 62 | # Revision 1.1 2003/06/09 23:13:21 the_krow |
|---|
| 63 | # bugfix: unknown files are now resetted before trying if they are valid |
|---|
| 64 | # first rudimentary eyed3 mp3 parser added |
|---|
| 65 | # |
|---|
| 66 | # |
|---|
| 67 | # ----------------------------------------------------------------------- |
|---|
| 68 | # MMPython - Media Metadata for Python |
|---|
| 69 | # Copyright (C) 2003 Thomas Schueppel, et. al |
|---|
| 70 | # |
|---|
| 71 | # This program is free software; you can redistribute it and/or |
|---|
| 72 | # modify it under the terms of the GNU General Public License as |
|---|
| 73 | # published by the Free Software Foundation; either version 2 of the |
|---|
| 74 | # License, or (at your option) any later version. |
|---|
| 75 | # |
|---|
| 76 | # This program is distributed in the hope that it will be useful, |
|---|
| 77 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
|---|
| 78 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|---|
| 79 | # GNU General Public License for more details. |
|---|
| 80 | # |
|---|
| 81 | # You should have received a copy of the GNU General Public License |
|---|
| 82 | # along with this program; if not, write to the Free Software |
|---|
| 83 | # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 |
|---|
| 84 | # USA |
|---|
| 85 | # |
|---|
| 86 | # Most of the code of this module was taken from Vivake Guptas mp3info |
|---|
| 87 | # code. Below is his copyright notice. All credit to him. |
|---|
| 88 | # |
|---|
| 89 | # Copyright (c) 2002 Vivake Gupta (vivakeATomniscia.org). All rights reserved. |
|---|
| 90 | # This software is maintained by Vivake (vivakeATomniscia.org) and is available at: |
|---|
| 91 | # http://www.omniscia.org/~vivake/python/MP3Info.py |
|---|
| 92 | # |
|---|
| 93 | # |
|---|
| 94 | |
|---|
| 95 | |
|---|
| 96 | from mmpython import mediainfo |
|---|
| 97 | import mmpython |
|---|
| 98 | |
|---|
| 99 | from eyeD3 import tag as eyeD3_tag |
|---|
| 100 | from eyeD3 import frames as eyeD3_frames |
|---|
| 101 | |
|---|
| 102 | import os |
|---|
| 103 | import struct |
|---|
| 104 | import traceback |
|---|
| 105 | import id3 as id3info |
|---|
| 106 | |
|---|
| 107 | MP3_INFO_TABLE = { "APIC": "picture", |
|---|
| 108 | "LINK": "link", |
|---|
| 109 | "TALB": "album", |
|---|
| 110 | "TCOM": "composer", |
|---|
| 111 | "TCOP": "copyright", |
|---|
| 112 | "TDOR": "release", |
|---|
| 113 | "TYER": "date", |
|---|
| 114 | "TEXT": "text", |
|---|
| 115 | "TIT2": "title", |
|---|
| 116 | "TLAN": "language", |
|---|
| 117 | "TLEN": "length", |
|---|
| 118 | "TMED": "media_type", |
|---|
| 119 | "TPE1": "artist", |
|---|
| 120 | "TPE2": "artist", |
|---|
| 121 | "TRCK": "trackno", |
|---|
| 122 | "TPOS": "discs"} |
|---|
| 123 | |
|---|
| 124 | _bitrates = [ |
|---|
| 125 | [ # MPEG-2 & 2.5 |
|---|
| 126 | [0,32,48,56, 64, 80, 96,112,128,144,160,176,192,224,256,None], # Layer 1 |
|---|
| 127 | [0, 8,16,24, 32, 40, 48, 56, 64, 80, 96,112,128,144,160,None], # Layer 2 |
|---|
| 128 | [0, 8,16,24, 32, 40, 48, 56, 64, 80, 96,112,128,144,160,None] # Layer 3 |
|---|
| 129 | ], |
|---|
| 130 | |
|---|
| 131 | [ # MPEG-1 |
|---|
| 132 | [0,32,64,96,128,160,192,224,256,288,320,352,384,416,448,None], # Layer 1 |
|---|
| 133 | [0,32,48,56, 64, 80, 96,112,128,160,192,224,256,320,384,None], # Layer 2 |
|---|
| 134 | [0,32,40,48, 56, 64, 80, 96,112,128,160,192,224,256,320,None] # Layer 3 |
|---|
| 135 | ] |
|---|
| 136 | ] |
|---|
| 137 | |
|---|
| 138 | _samplerates = [ |
|---|
| 139 | [ 11025, 12000, 8000, None], # MPEG-2.5 |
|---|
| 140 | [ None, None, None, None], # reserved |
|---|
| 141 | [ 22050, 24000, 16000, None], # MPEG-2 |
|---|
| 142 | [ 44100, 48000, 32000, None], # MPEG-1 |
|---|
| 143 | ] |
|---|
| 144 | |
|---|
| 145 | _modes = [ "stereo", "joint stereo", "dual channel", "mono" ] |
|---|
| 146 | |
|---|
| 147 | _MP3_HEADER_SEEK_LIMIT = 4096 |
|---|
| 148 | |
|---|
| 149 | class eyeD3Info(mediainfo.MusicInfo): |
|---|
| 150 | |
|---|
| 151 | fileName = str(); |
|---|
| 152 | fileSize = int(); |
|---|
| 153 | |
|---|
| 154 | def __init__(self, file, tagVersion = eyeD3_tag.ID3_ANY_VERSION): |
|---|
| 155 | mediainfo.MusicInfo.__init__(self) |
|---|
| 156 | self.fileName = file.name; |
|---|
| 157 | self.valid = 1 |
|---|
| 158 | self.mime = 'audio/mp3' |
|---|
| 159 | |
|---|
| 160 | if not eyeD3_tag.isMp3File(file.name): |
|---|
| 161 | self.valid = 0 |
|---|
| 162 | return |
|---|
| 163 | |
|---|
| 164 | id3 = None |
|---|
| 165 | try: |
|---|
| 166 | id3 = eyeD3_tag.Mp3AudioFile(file.name) |
|---|
| 167 | except eyeD3_tag.TagException: |
|---|
| 168 | try: |
|---|
| 169 | id3 = eyeD3_tag.Mp3AudioFile(file.name) |
|---|
| 170 | except eyeD3_tag.InvalidAudioFormatException: |
|---|
| 171 | # File is not an MP3 |
|---|
| 172 | self.valid = 0 |
|---|
| 173 | return |
|---|
| 174 | except: |
|---|
| 175 | # The MP3 tag decoder crashed, assume the file is still |
|---|
| 176 | # MP3 and try to play it anyway |
|---|
| 177 | if mediainfo.DEBUG: |
|---|
| 178 | print 'music: oops, mp3 tag parsing failed!' |
|---|
| 179 | print 'music: filename = "%s"' % file.name |
|---|
| 180 | traceback.print_exc() |
|---|
| 181 | except: |
|---|
| 182 | # The MP3 tag decoder crashed, assume the file is still |
|---|
| 183 | # MP3 and try to play it anyway |
|---|
| 184 | if mediainfo.DEBUG: |
|---|
| 185 | print 'music: oops, mp3 tag parsing failed!' |
|---|
| 186 | print 'music: filename = "%s"' % file.name |
|---|
| 187 | traceback.print_exc() |
|---|
| 188 | |
|---|
| 189 | if not self.valid: |
|---|
| 190 | return |
|---|
| 191 | |
|---|
| 192 | if mediainfo.DEBUG > 1: |
|---|
| 193 | print id3.tag.frames |
|---|
| 194 | try: |
|---|
| 195 | if id3 and id3.tag: |
|---|
| 196 | for k in MP3_INFO_TABLE: |
|---|
| 197 | if id3.tag.frames[k]: |
|---|
| 198 | if k == 'APIC': |
|---|
| 199 | pass |
|---|
| 200 | else: |
|---|
| 201 | setattr(self, MP3_INFO_TABLE[k], id3.tag.frames[k][0].text) |
|---|
| 202 | if id3.tag.getYear(): |
|---|
| 203 | self.date = id3.tag.getYear() |
|---|
| 204 | tab = {} |
|---|
| 205 | for f in id3.tag.frames: |
|---|
| 206 | if f.__class__ is eyeD3_frames.TextFrame: |
|---|
| 207 | tab[f.header.id] = f.text |
|---|
| 208 | elif f.__class__ is eyeD3_frames.UserTextFrame: |
|---|
| 209 | tab[f.header.id] = f.text |
|---|
| 210 | elif f.__class__ is eyeD3_frames.DateFrame: |
|---|
| 211 | tab[f.header.id] = f.date_str |
|---|
| 212 | elif f.__class__ is eyeD3_frames.CommentFrame: |
|---|
| 213 | tab[f.header.id] = f.comment |
|---|
| 214 | elif f.__class__ is eyeD3_frames.URLFrame: |
|---|
| 215 | tab[f.header.id] = f.url |
|---|
| 216 | elif f.__class__ is eyeD3_frames.UserURLFrame: |
|---|
| 217 | tab[f.header.id] = f.url |
|---|
| 218 | elif mediainfo.DEBUG: |
|---|
| 219 | print f.__class__ |
|---|
| 220 | self.appendtable('id3v2', tab, 'en') |
|---|
| 221 | |
|---|
| 222 | if id3.tag.frames['TCON']: |
|---|
| 223 | genre = None |
|---|
| 224 | tcon = id3.tag.frames['TCON'][0].text |
|---|
| 225 | try: |
|---|
| 226 | genre = int(tcon) |
|---|
| 227 | except: |
|---|
| 228 | try: |
|---|
| 229 | genre = int(tcon[1:tcon.find(')')]) |
|---|
| 230 | except ValueError: |
|---|
| 231 | pass |
|---|
| 232 | if genre is not None: |
|---|
| 233 | try: |
|---|
| 234 | self.genre = id3info.GENRE_LIST[genre] |
|---|
| 235 | except: |
|---|
| 236 | pass |
|---|
| 237 | # and some tools store it as trackno/trackof in TRCK |
|---|
| 238 | if not self['trackof'] and self['trackno'] and self['trackno'].find('/') > 0: |
|---|
| 239 | self['trackof'] = self['trackno'][self['trackno'].find('/')+1:] |
|---|
| 240 | self['trackno'] = self['trackno'][:self['trackno'].find('/')] |
|---|
| 241 | if id3: |
|---|
| 242 | self.length = id3.getPlayTime() |
|---|
| 243 | except: |
|---|
| 244 | if mediainfo.DEBUG: |
|---|
| 245 | traceback.print_exc() |
|---|
| 246 | offset, header = self._find_header(file) |
|---|
| 247 | if offset == -1 or header is None: |
|---|
| 248 | return |
|---|
| 249 | |
|---|
| 250 | self._parse_header(header) |
|---|
| 251 | |
|---|
| 252 | |
|---|
| 253 | def _find_header(self, file): |
|---|
| 254 | file.seek(0, 0) |
|---|
| 255 | amount_read = 0 |
|---|
| 256 | |
|---|
| 257 | # see if we get lucky with the first four bytes |
|---|
| 258 | amt = 4 |
|---|
| 259 | |
|---|
| 260 | while amount_read < _MP3_HEADER_SEEK_LIMIT: |
|---|
| 261 | header = file.read(amt) |
|---|
| 262 | if len(header) < amt: |
|---|
| 263 | # awfully short file. just give up. |
|---|
| 264 | return -1, None |
|---|
| 265 | |
|---|
| 266 | amount_read = amount_read + len(header) |
|---|
| 267 | |
|---|
| 268 | # on the next read, grab a lot more |
|---|
| 269 | amt = 500 |
|---|
| 270 | |
|---|
| 271 | # look for the sync byte |
|---|
| 272 | offset = header.find(chr(255)) |
|---|
| 273 | if offset == -1: |
|---|
| 274 | continue |
|---|
| 275 | |
|---|
| 276 | # looks good, make sure we have the next 3 bytes after this |
|---|
| 277 | # because the header is 4 bytes including sync |
|---|
| 278 | if offset + 4 > len(header): |
|---|
| 279 | more = file.read(4) |
|---|
| 280 | if len(more) < 4: |
|---|
| 281 | # end of file. can't find a header |
|---|
| 282 | return -1, None |
|---|
| 283 | amount_read = amount_read + 4 |
|---|
| 284 | header = header + more |
|---|
| 285 | |
|---|
| 286 | # the sync flag is also in the next byte, the first 3 bits |
|---|
| 287 | # must also be set |
|---|
| 288 | if ord(header[offset+1]) >> 5 != 7: |
|---|
| 289 | continue |
|---|
| 290 | |
|---|
| 291 | # ok, that's it, looks like we have the header |
|---|
| 292 | return amount_read - len(header) + offset, header[offset:offset+4] |
|---|
| 293 | |
|---|
| 294 | # couldn't find the header |
|---|
| 295 | return -1, None |
|---|
| 296 | |
|---|
| 297 | |
|---|
| 298 | def _parse_header(self, header): |
|---|
| 299 | # http://mpgedit.org/mpgedit/mpeg_format/MP3Format.html |
|---|
| 300 | bytes = struct.unpack('>i', header)[0] |
|---|
| 301 | mpeg_version = (bytes >> 19) & 3 |
|---|
| 302 | layer = (bytes >> 17) & 3 |
|---|
| 303 | bitrate = (bytes >> 12) & 15 |
|---|
| 304 | samplerate = (bytes >> 10) & 3 |
|---|
| 305 | mode = (bytes >> 6) & 3 |
|---|
| 306 | |
|---|
| 307 | if mpeg_version == 0: |
|---|
| 308 | self.version = 2.5 |
|---|
| 309 | elif mpeg_version == 2: |
|---|
| 310 | self.version = 2 |
|---|
| 311 | elif mpeg_version == 3: |
|---|
| 312 | self.version = 1 |
|---|
| 313 | else: |
|---|
| 314 | return |
|---|
| 315 | |
|---|
| 316 | if layer > 0: |
|---|
| 317 | layer = 4 - layer |
|---|
| 318 | else: |
|---|
| 319 | return |
|---|
| 320 | |
|---|
| 321 | self.bitrate = _bitrates[mpeg_version & 1][layer - 1][bitrate] |
|---|
| 322 | self.samplerate = _samplerates[mpeg_version][samplerate] |
|---|
| 323 | |
|---|
| 324 | if self.bitrate is None or self.samplerate is None: |
|---|
| 325 | return |
|---|
| 326 | |
|---|
| 327 | self.mode = _modes[mode] |
|---|
| 328 | self.keys.append('mode') |
|---|
| 329 | |
|---|
| 330 | |
|---|
| 331 | mmpython.registertype( 'audio/mp3', ('mp3',), mediainfo.TYPE_MUSIC, eyeD3Info ) |
|---|
