home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / pyos2bin.zip / Demo / sgi / video / Dsend.py < prev    next >
Text File  |  1996-11-27  |  4KB  |  189 lines
  1. #! /usr/bin/env python
  2.  
  3. # Send live video UDP packets.
  4. # Usage: Vsend [-b] [-h height] [-p port] [-s size] [-t ttl] [-w width]
  5. #              [host] ..
  6.  
  7. import sys
  8. import time
  9. import struct
  10. import string
  11. import math
  12. from socket import *
  13. from SOCKET import *
  14. import gl, GL, DEVICE
  15. sys.path.append('/ufs/guido/src/video')
  16. import DisplayVideoIn
  17. import LiveVideoOut
  18. import SV
  19. import getopt
  20. from IN import *
  21.  
  22. from senddefs import *
  23.  
  24. def usage(msg):
  25.     print msg
  26.     print 'usage: Vsend [-b] [-h height] [-p port] [-s size] [-t ttl] [-c type] [-m]',
  27.     print '[-w width] [host] ...'
  28.     print '-b        : broadcast on local net'
  29.     print '-h height : window height (default ' + `DEFHEIGHT` + ')'
  30.     print '-p port   : port to use (default ' + `DEFPORT` + ')'
  31.     print '-t ttl    : time-to-live (multicast only; default 1)'
  32.     print '-s size   : max packet size (default ' + `DEFPKTMAX` + ')'
  33.     print '-S size   : use this packet size/window size'
  34.     print '-w width  : window width (default ' + `DEFWIDTH` + ')'
  35.     print '-v        : print packet rate'
  36.     print '-x xpos   : set x position'
  37.     print '-y ypos   : set y position'
  38.     print '[host] ...: host(s) to send to (default multicast to ' + \
  39.         DEFMCAST + ')'
  40.     sys.exit(2)
  41.  
  42.  
  43. def main():
  44.     sys.stdout = sys.stderr
  45.  
  46.     hosts = []
  47.     port = DEFPORT
  48.     ttl = -1
  49.     pktmax = DEFPKTMAX
  50.     width = DEFWIDTH
  51.     height = DEFHEIGHT
  52.     vtype = 'rgb'
  53.     verbose = 0
  54.     xpos = ypos = 0
  55.  
  56.     try:
  57.         opts, args = getopt.getopt(sys.argv[1:], 'bh:p:s:S:t:w:vx:y:')
  58.     except getopt.error, msg:
  59.         usage(msg)
  60.  
  61.     try:
  62.         for opt, optarg in opts:
  63.             if opt == '-p':
  64.                 port = string.atoi(optarg)
  65.             if opt == '-b':
  66.                 host = '<broadcast>'
  67.             if opt == '-t':
  68.                 ttl = string.atoi(optarg)
  69.             if opt == '-S':
  70.                 pktmax = string.atoi(optarg)
  71.                 vidmax = SV.PAL_XMAX*SV.PAL_YMAX
  72.                 if vidmax <= pktmax:
  73.                     width = SV.PAL_XMAX
  74.                     height = SV.PAL_YMAX
  75.                     pktmax = vidmax
  76.                 else:
  77.                     factor = float(vidmax)/float(pktmax)
  78.                     factor = math.sqrt(factor)
  79.                     width = int(SV.PAL_XMAX/factor)-7
  80.                     height = int(SV.PAL_YMAX/factor)-5
  81.                     print 'video:',width,'x',height,
  82.                     print 'pktsize',width*height,'..',
  83.                     print pktmax
  84.             if opt == '-s':
  85.                 pktmax = string.atoi(optarg)
  86.             if opt == '-w':
  87.                 width = string.atoi(optarg)
  88.             if opt == '-h':
  89.                 height = string.atoi(optarg)
  90.             if opt == '-c':
  91.                 vtype = optarg
  92.             if opt == '-v':
  93.                 verbose = 1
  94.             if opt == '-x':
  95.                 xpos = string.atoi(optarg)
  96.             if opt == '-y':
  97.                 ypos = string.atoi(optarg)
  98.     except string.atoi_error, msg:
  99.         usage('bad integer: ' + msg)
  100.  
  101.     for host in args:
  102.         hosts.append(gethostbyname(host))
  103.  
  104.     if not hosts:
  105.         hosts.append(gethostbyname(DEFMCAST))
  106.  
  107.     gl.foreground()
  108.     gl.prefsize(width, height)
  109.     gl.stepunit(8, 6)
  110.     wid = gl.winopen('Vsend')
  111.     gl.keepaspect(width, height)
  112.     gl.stepunit(8, 6)
  113.     gl.maxsize(SV.PAL_XMAX, SV.PAL_YMAX)
  114.     gl.winconstraints()
  115.     gl.qdevice(DEVICE.ESCKEY)
  116.     gl.qdevice(DEVICE.WINSHUT)
  117.     gl.qdevice(DEVICE.WINQUIT)
  118.     gl.qdevice(DEVICE.WINFREEZE)
  119.     gl.qdevice(DEVICE.WINTHAW)
  120.     width, height = gl.getsize()
  121.  
  122.     lvo = LiveVideoOut.LiveVideoOut(wid, width, height, vtype)
  123.  
  124.     lvi = DisplayVideoIn.DisplayVideoIn(pktmax, width, height, vtype)
  125.  
  126.     if xpos or ypos:
  127.         lvi.positionvideo(xpos, ypos)
  128.  
  129.     s = socket(AF_INET, SOCK_DGRAM)
  130.     s.setsockopt(SOL_SOCKET, SO_BROADCAST, 1)
  131.     if ttl >= 0:
  132.         s.setsockopt(IPPROTO_IP, IP_MULTICAST_TTL, chr(ttl))
  133.  
  134.     frozen = 0
  135.  
  136.     lasttime = int(time.time())
  137.     nframe = 0
  138.     while 1:
  139.  
  140.         if gl.qtest():
  141.             dev, val = gl.qread()
  142.             if dev in (DEVICE.ESCKEY, \
  143.                 DEVICE.WINSHUT, DEVICE.WINQUIT):
  144.                 break
  145.             if dev == DEVICE.WINFREEZE:
  146.                 frozen = 1
  147.             if dev == DEVICE.WINTHAW:
  148.                 frozen = 0
  149.             if dev == DEVICE.REDRAW:
  150.                 w, h = gl.getsize()
  151.                 x, y = gl.getorigin()
  152.                 if (w, h) <> (width, height):
  153.                     width, height = w, h
  154.                     lvi.resizevideo(width, height)
  155.                     lvo.resizevideo(width, height)
  156.  
  157.         rv = lvi.getnextpacket()
  158.         if not rv:
  159.             time.sleep(0.010)
  160.             continue
  161.  
  162.         pos, data = rv
  163.         print pos, len(data) # DBG
  164.  
  165.         if not frozen:
  166.             lvo.putnextpacket(pos, data)
  167.  
  168.         hdr = struct.pack('hhh', pos, width, height)
  169.         for host in hosts:
  170.             try:
  171.                 # print len(hdr+data) # DBG
  172.                 s.sendto(hdr + data, (host, port))
  173.             except error, msg: # really socket.error
  174.                 if msg[0] <> 121: # no buffer space available
  175.                     raise error, msg # re-raise it
  176.                 print 'Warning:', msg[1]
  177.         if pos == 0 and verbose:
  178.             nframe = nframe+1
  179.             if int(time.time()) <> lasttime:
  180.                 print nframe / (time.time()-lasttime), 'fps'
  181.                 nframe = 0
  182.                 lasttime = int(time.time())
  183.  
  184.     lvi.close()
  185.     lvo.close()
  186.  
  187.  
  188. main()
  189.