[clinton/abcde.git] / examples / cue2discid

#!/usr/bin/python

"""Outputs a long-form cddb discid given a cue file as input"""

import fileinput
import re

FRAMES_PER_SECOND = 75
SAMPLES_PER_SECOND = 44100
SAMPLES_PER_FRAME = SAMPLES_PER_SECOND // FRAMES_PER_SECOND

pregap = 0
currentTrackFrame = 0
pregapRegex = re.compile(r'PREGAP (?P<min>\d+):(?P<sec>\d\d):(?P<frame>\d\d)')
startTrackRegex = re.compile(r'INDEX 01 (?P<min>\d+):(?P<sec>\d\d):(?P<frame>\d\d)')
leadInRegex = re.compile(r'REM FLAC__lead-in (?P<sample>\d+)')
leadOutRegex = re.compile(r'REM FLAC__lead-out 170 (?P<sample>\d+)')
raw_framelist = list()

numTracks = 0

def cddb_sum(trackseconds):
    def digits(number):
        tmp = number
        while tmp:
            yield tmp % 10
            tmp = tmp // 10

    return sum(digits(trackseconds))

for line in fileinput.input():
    pregapMatch = pregapRegex.search(line)
    if pregapMatch:
        pregap = (int((pregapMatch.group('min')) * 60) + int(pregapMatch.group('sec'))) * FRAMES_PER_SECOND + int(pregapMatch.group('frame'))
        continue

    startTrackMatch = startTrackRegex.search(line)
    if startTrackMatch:
        numTracks += 1
        currentTrackFrame = ((int(startTrackMatch.group('min')) * 60) + int(startTrackMatch.group('sec'))) * FRAMES_PER_SECOND + int(startTrackMatch.group('frame'))
        raw_framelist.append(currentTrackFrame)
        continue

    leadInMatch = leadInRegex.search(line)
    if leadInMatch:
        leadInFrame = long(leadInMatch.group('sample')) // SAMPLES_PER_FRAME
        continue

    leadOutMatch = leadOutRegex.search(line)
    if leadOutMatch:
        leadOutFrame_raw = long(leadOutMatch.group('sample')) // SAMPLES_PER_FRAME
        continue

leadOutFrame = leadOutFrame_raw + leadInFrame

framelist = [i + leadInFrame + pregap for i in raw_framelist]

secondslist = [i//FRAMES_PER_SECOND for i in framelist]

n = long(sum(map(cddb_sum, secondslist)))
t = leadOutFrame/FRAMES_PER_SECOND - framelist[0]/FRAMES_PER_SECOND
print "%08x" % (((n % 0xff) << 24) | (t << 8) | numTracks), numTracks, 
print " ".join([str(i) for i in framelist]), leadOutFrame/FRAMES_PER_SECOND
Commit	Line	Data
27fdb84b	1	#!/usr/bin/python
	2
	3	"""Outputs a long-form cddb discid given a cue file as input"""
	4
	5	import fileinput
	6	import re
	7
	8	FRAMES_PER_SECOND = 75
	9	SAMPLES_PER_SECOND = 44100
	10	SAMPLES_PER_FRAME = SAMPLES_PER_SECOND // FRAMES_PER_SECOND
	11
	12	pregap = 0
	13	currentTrackFrame = 0
	14	pregapRegex = re.compile(r'PREGAP (?P<min>\d+):(?P<sec>\d\d):(?P<frame>\d\d)')
	15	startTrackRegex = re.compile(r'INDEX 01 (?P<min>\d+):(?P<sec>\d\d):(?P<frame>\d\d)')
	16	leadInRegex = re.compile(r'REM FLAC__lead-in (?P<sample>\d+)')
	17	leadOutRegex = re.compile(r'REM FLAC__lead-out 170 (?P<sample>\d+)')
	18	raw_framelist = list()
	19
	20	numTracks = 0
	21
	22	def cddb_sum(trackseconds):
	23	def digits(number):
	24	tmp = number
	25	while tmp:
	26	yield tmp % 10
	27	tmp = tmp // 10
	28
	29	return sum(digits(trackseconds))
	30
	31	for line in fileinput.input():
	32	pregapMatch = pregapRegex.search(line)
	33	if pregapMatch:
	34	pregap = (int((pregapMatch.group('min')) * 60) + int(pregapMatch.group('sec'))) * FRAMES_PER_SECOND + int(pregapMatch.group('frame'))
	35	continue
	36
	37	startTrackMatch = startTrackRegex.search(line)
	38	if startTrackMatch:
	39	numTracks += 1
	40	currentTrackFrame = ((int(startTrackMatch.group('min')) * 60) + int(startTrackMatch.group('sec'))) * FRAMES_PER_SECOND + int(startTrackMatch.group('frame'))
	41	raw_framelist.append(currentTrackFrame)
	42	continue
	43
	44	leadInMatch = leadInRegex.search(line)
	45	if leadInMatch:
	46	leadInFrame = long(leadInMatch.group('sample')) // SAMPLES_PER_FRAME
	47	continue
	48
	49	leadOutMatch = leadOutRegex.search(line)
	50	if leadOutMatch:
	51	leadOutFrame_raw = long(leadOutMatch.group('sample')) // SAMPLES_PER_FRAME
	52	continue
	53
	54	leadOutFrame = leadOutFrame_raw + leadInFrame
	55
	56	framelist = [i + leadInFrame + pregap for i in raw_framelist]
	57
	58	secondslist = [i//FRAMES_PER_SECOND for i in framelist]
	59
	60	n = long(sum(map(cddb_sum, secondslist)))
	61	t = leadOutFrame/FRAMES_PER_SECOND - framelist[0]/FRAMES_PER_SECOND
	62	print "%08x" % (((n % 0xff) << 24) \| (t << 8) \| numTracks), numTracks,
	63	print " ".join([str(i) for i in framelist]), leadOutFrame/FRAMES_PER_SECOND