home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / pyth_os2.zip / python-1.0.2 / Lib / sgi / readcd.py < prev    next >
Text File  |  1993-12-17  |  6KB  |  233 lines
  1. # Class interface to the CD module.
  2.  
  3. import cd, CD
  4.  
  5. Error = 'Readcd.Error'
  6. _Stop = 'Readcd.Stop'
  7.  
  8. def _doatime(self, cb_type, data):
  9.     if ((data[0] * 60) + data[1]) * 75 + data[2] > self.end:
  10. ##        print 'done with list entry',`self.listindex`
  11.         raise _Stop
  12.     func, arg = self.callbacks[cb_type]
  13.     if func:
  14.         func(arg, cb_type, data)
  15.  
  16. def _dopnum(self, cb_type, data):
  17.     if data > self.end:
  18. ##        print 'done with list entry',`self.listindex`
  19.         raise _Stop
  20.     func, arg = self.callbacks[cb_type]
  21.     if func:
  22.         func(arg, cb_type, data)
  23.  
  24. class Readcd:
  25.     def __init__(self, *arg):
  26.         if len(arg) == 0:
  27.             self.player = cd.open()
  28.         elif len(arg) == 1:
  29.             self.player = cd.open(arg[0])
  30.         elif len(arg) == 2:
  31.             self.player = cd.open(arg[0], arg[1])
  32.         else:
  33.             raise Error, 'bad __init__ call'
  34.         self.list = []
  35.         self.callbacks = [(None, None)] * 8
  36.         self.parser = cd.createparser()
  37.         self.playing = 0
  38.         self.end = 0
  39.         self.status = None
  40.         self.trackinfo = None
  41.  
  42.     def eject(self):
  43.         self.player.eject()
  44.         self.list = []
  45.         self.end = 0
  46.         self.listindex = 0
  47.         self.status = None
  48.         self.trackinfo = None
  49.         if self.playing:
  50. ##            print 'stop playing from eject'
  51.             raise _Stop
  52.  
  53.     def pmsf2msf(self, track, min, sec, frame):
  54.         if not self.status:
  55.             self.cachestatus()
  56.         if track < self.status[5] or track > self.status[6]:
  57.             raise Error, 'track number out of range'
  58.         if not self.trackinfo:
  59.             self.cacheinfo()
  60.         start, total = self.trackinfo[track]
  61.         start = ((start[0] * 60) + start[1]) * 75 + start[2]
  62.         total = ((total[0] * 60) + total[1]) * 75 + total[2]
  63.         block = ((min * 60) + sec) * 75 + frame
  64.         if block > total:
  65.             raise Error, 'out of range'
  66.         block = start + block
  67.         min, block = divmod(block, 75*60)
  68.         sec, frame = divmod(block, 75)
  69.         return min, sec, frame
  70.  
  71.     def reset(self):
  72.         self.list = []
  73.  
  74.     def appendtrack(self, track):
  75.         self.appendstretch(track, track)
  76.                 
  77.     def appendstretch(self, start, end):
  78.         if not self.status:
  79.             self.cachestatus()
  80.         if not start:
  81.             start = 1
  82.         if not end:
  83.             end = self.status[6]
  84.         if type(end) == type(0):
  85.             if end < self.status[5] or end > self.status[6]:
  86.                 raise Error, 'range error'
  87.         else:
  88.             l = len(end)
  89.             if l == 4:
  90.                 prog, min, sec, frame = end
  91.                 if prog < self.status[5] or prog > self.status[6]:
  92.                     raise Error, 'range error'
  93.                 end = self.pmsf2msf(prog, min, sec, frame)
  94.             elif l <> 3:
  95.                 raise Error, 'syntax error'
  96.         if type(start) == type(0):
  97.             if start < self.status[5] or start > self.status[6]:
  98.                 raise Error, 'range error'
  99.             if len(self.list) > 0:
  100.                 s, e = self.list[-1]
  101.                 if type(e) == type(0):
  102.                     if start == e+1:
  103.                         start = s
  104.                         del self.list[-1]
  105.         else:
  106.             l = len(start)
  107.             if l == 4:
  108.                 prog, min, sec, frame = start
  109.                 if prog < self.status[5] or prog > self.status[6]:
  110.                     raise Error, 'range error'
  111.                 start = self.pmsf2msf(prog, min, sec, frame)
  112.             elif l <> 3:
  113.                 raise Error, 'syntax error'
  114.         self.list.append((start, end))
  115.  
  116.     def settracks(self, list):
  117.         self.list = []
  118.         for track in list:
  119.             self.appendtrack(track)
  120.  
  121.     def setcallback(self, cb_type, func, arg):
  122.         if cb_type < 0 or cb_type >= 8:
  123.             raise Error, 'type out of range'
  124.         self.callbacks[cb_type] = (func, arg)
  125.         if self.playing:
  126.             start, end = self.list[self.listindex]
  127.             if type(end) == type(0):
  128.                 if cb_type <> CD.PNUM:
  129.                     self.parser.setcallback(cb_type, func, arg)
  130.             else:
  131.                 if cb_type <> CD.ATIME:
  132.                     self.parser.setcallback(cb_type, func, arg)
  133.  
  134.     def removecallback(self, cb_type):
  135.         if cb_type < 0 or cb_type >= 8:
  136.             raise Error, 'type out of range'
  137.         self.callbacks[cb_type] = (None, None)
  138.         if self.playing:
  139.             start, end = self.list[self.listindex]
  140.             if type(end) == type(0):
  141.                 if cb_type <> CD.PNUM:
  142.                     self.parser.removecallback(cb_type)
  143.             else:
  144.                 if cb_type <> CD.ATIME:
  145.                     self.parser.removecallback(cb_type)
  146.  
  147.     def gettrackinfo(self, *arg):
  148.         if not self.status:
  149.             self.cachestatus()
  150.         if not self.trackinfo:
  151.             self.cacheinfo()
  152.         if len(arg) == 0:
  153.             return self.trackinfo[self.status[5]:self.status[6]+1]
  154.         result = []
  155.         for i in arg:
  156.             if i < self.status[5] or i > self.status[6]:
  157.                 raise Error, 'range error'
  158.             result.append(self.trackinfo[i])
  159.         return result
  160.  
  161.     def cacheinfo(self):
  162.         if not self.status:
  163.             self.cachestatus()
  164.         self.trackinfo = []
  165.         for i in range(self.status[5]):
  166.             self.trackinfo.append(None)
  167.         for i in range(self.status[5], self.status[6]+1):
  168.             self.trackinfo.append(self.player.gettrackinfo(i))
  169.  
  170.     def cachestatus(self):
  171.         self.status = self.player.getstatus()
  172.         if self.status[0] == CD.NODISC:
  173.             self.status = None
  174.             raise Error, 'no disc in player'
  175.  
  176.     def getstatus(self):
  177.         return self.player.getstatus()
  178.  
  179.     def play(self):
  180.         if not self.status:
  181.             self.cachestatus()
  182.         size = self.player.bestreadsize()
  183.         self.listindex = 0
  184.         self.playing = 0
  185.         for i in range(8):
  186.             func, arg = self.callbacks[i]
  187.             if func:
  188.                 self.parser.setcallback(i, func, arg)
  189.             else:
  190.                 self.parser.removecallback(i)
  191.         if len(self.list) == 0:
  192.             for i in range(self.status[5], self.status[6]+1):
  193.                 self.appendtrack(i)
  194.         while 1:
  195.             if not self.playing:
  196.                 if self.listindex >= len(self.list):
  197.                     return
  198.                 start, end = self.list[self.listindex]
  199. ##                print 'starting with',`(start, end)`
  200.                 if type(start) == type(0):
  201.                     dummy = self.player.seektrack(start)
  202.                 else:
  203.                     min, sec, frame = start
  204.                     dummy = self.player.seek(min, sec, frame)
  205.                 if type(end) == type(0):
  206.                     self.parser.setcallback(CD.PNUM, _dopnum, self)
  207.                     self.end = end
  208.                     func, arg = self.callbacks[CD.ATIME]
  209.                     if func:
  210.                         self.parser.setcallback(CD.ATIME, func, arg)
  211.                     else:
  212.                         self.parser.removecallback(CD.ATIME)
  213.                 else:
  214.                     min, sec, frame = end
  215.                     self.parser.setcallback(CD.ATIME, _doatime, self)
  216.                     self.end = (min * 60 + sec) * 75 + frame
  217.                     func, arg = self.callbacks[CD.PNUM]
  218.                     if func:
  219.                         self.parser.setcallback(CD.PNUM, func, arg)
  220.                     else:
  221.                         self.parser.removecallback(CD.PNUM)
  222.                 self.playing = 1
  223.             data = self.player.readda(size)
  224.             if data == '':
  225.                 self.playing = 0
  226.                 self.listindex = self.listindex + 1
  227.                 continue
  228.             try:
  229.                 self.parser.parseframe(data)
  230.             except _Stop:
  231.                 self.playing = 0
  232.                 self.listindex = self.listindex + 1
  233.