#!/usr/bin/python3 # -*- coding: utf-8 -*- # -*- mode: Python -*- ''' fromild v0.1.0 Read/display once an .ild animation file and quit ?? LICENCE : CC by cocoa and Sam Neurohack Heavy u-se of : ILDA.py Python module for dealing with the ILDA Image Data Transfer Format, an interchange format for laser image frames. Copyright (c) 2008 Micah Dowty Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. ''' from __future__ import print_function import time import struct import argparse import sys name="generator::fromild" def debug(*args, **kwargs): if( verbose == False ): return print(*args, file=sys.stderr, **kwargs) argsparser = argparse.ArgumentParser(description=".ild file frame generator") argsparser.add_argument("-f","--fps",help="Frame Per Second",default=30,type=int) argsparser.add_argument("-v","--verbose",action="store_true",help="Verbose output") argsparser.add_argument("-i","--ild",help=".ild file",default="book2.ild",type=str) args = argsparser.parse_args() fps=args.fps verbose=args.verbose optimal_looptime = 1 / fps debug(name+" optimal looptime "+str(optimal_looptime)) # Format codes FORMAT_3D = 0 FORMAT_2D = 1 FORMAT_COLOR_TABLE = 2 # Mapping from FORMAT_* codes to struct format strings formatTable = ( '>hhhH', '>hhH', '>BBB', ) # Header values HEADER_MAGIC = b"ILDA\0\0\0" HEADER_RESERVED = 0 HEADER_FORMAT = ">7sB16sHHHBB" HEADER_LEN = struct.calcsize(HEADER_FORMAT) # 64 default colors table : use rgb2int(colors64[ildacolor]) colors64 = [[255, 0, 0], [255, 17, 0], [255, 34, 0], [255, 51, 0], [255, 68, 0], [255, 85, 0], [255, 102, 0], [255, 119, 0], [255, 136, 0], [255, 153, 0], [255, 170, 0], [255, 187, 0], [255, 204, 0], [255, 221, 0], [255, 238, 0], [255, 255, 0], [255, 255, 0], [238, 255, 0], [204, 255, 0], [170, 255, 0], [136, 255, 0], [102, 255, 0], [68, 255, 0], [34, 255, 0], [0, 255, 0], [0, 255, 34], [0, 255, 68], [0, 255, 102], [0, 255, 136], [0, 255, 170], [0, 255, 204], [0, 255, 238], [0, 136, 255], [0, 119, 255], [0, 102, 255], [0, 102, 255], [0, 85, 255], [0, 68, 255], [0, 68, 255], [0, 34, 255], [0, 0, 255], [34, 0, 255], [68, 0, 255], [102, 0, 255], [136, 0, 255], [170, 0, 255], [204, 0, 255], [238, 0, 255], [255, 0, 255], [255, 34, 255], [255, 68, 255], [255, 102, 255], [255, 136, 255], [255, 170, 255], [255, 204, 255], [255, 238, 255], [255, 255, 255], [255, 238, 238], [255, 204, 204], [255, 170, 170], [255, 136, 136], [255, 102, 102], [255, 68, 68], [0, 34, 34]] # 256 default colors table colors256 = [[0, 0, 0], [255, 255, 255], [255, 0, 0], [255, 255, 0], [0, 255, 0], [0, 255, 255], [0, 0, 255], [255, 0, 255], [255, 128, 128], [255, 140, 128], [255, 151, 128], [255, 163, 128], [255, 174, 128], [255, 186, 128], [255, 197, 128], [255, 209, 128], [255, 220, 128], [255, 232, 128], [255, 243, 128], [255, 255, 128], [243, 255, 128], [232, 255, 128], [220, 255, 128], [209, 255, 128], [197, 255, 128], [186, 255, 128], [174, 255, 128], [163, 255, 128], [151, 255, 128], [140, 255, 128], [128, 255, 128], [128, 255, 140], [128, 255, 151], [128, 255, 163], [128, 255, 174], [128, 255, 186], [128, 255, 197], [128, 255, 209], [128, 255, 220], [128, 255, 232], [128, 255, 243], [128, 255, 255], [128, 243, 255], [128, 232, 255], [128, 220, 255], [128, 209, 255], [128, 197, 255], [128, 186, 255], [128, 174, 255], [128, 163, 255], [128, 151, 255], [128, 140, 255], [128, 128, 255], [140, 128, 255], [151, 128, 255], [163, 128, 255], [174, 128, 255], [186, 128, 255], [197, 128, 255], [209, 128, 255], [220, 128, 255], [232, 128, 255], [243, 128, 255], [255, 128, 255], [255, 128, 243], [255, 128, 232], [255, 128, 220], [255, 128, 209], [255, 128, 197], [255, 128, 186], [255, 128, 174], [255, 128, 163], [255, 128, 151], [255, 128, 140], [255, 0, 0], [255, 23, 0], [255, 46, 0], [255, 70, 0], [255, 93, 0], [255, 116, 0], [255, 139, 0], [255, 162, 0], [255, 185, 0], [255, 209, 0], [255, 232, 0], [255, 255, 0], [232, 255, 0], [209, 255, 0], [185, 255, 0], [162, 255, 0], [139, 255, 0], [116, 255, 0], [93, 255, 0], [70, 255, 0], [46, 255, 0], [23, 255, 0], [0, 255, 0], [0, 255, 23], [0, 255, 46], [0, 255, 70], [0, 255, 93], [0, 255, 116], [0, 255, 139], [0, 255, 162], [0, 255, 185], [0, 255, 209], [0, 255, 232], [0, 255, 255], [0, 232, 255], [0, 209, 255], [0, 185, 255], [0, 162, 255], [0, 139, 255], [0, 116, 255], [0, 93, 255], [0, 70, 255], [0, 46, 255], [0, 23, 255], [0, 0, 255], [23, 0, 255], [46, 0, 255], [70, 0, 255], [93, 0, 255], [116, 0, 255], [139, 0, 255], [162, 0, 255], [185, 0, 255], [209, 0, 255], [232, 0, 255], [255, 0, 255], [255, 0, 232], [255, 0, 209], [255, 0, 185], [255, 0, 162], [255, 0, 139], [255, 0, 116], [255, 0, 93], [255, 0, 70], [255, 0, 46], [255, 0, 23], [128, 0, 0], [128, 12, 0], [128, 23, 0], [128, 35, 0], [128, 47, 0], [128, 58, 0], [128, 70, 0], [128, 81, 0], [128, 93, 0], [128, 105, 0], [128, 116, 0], [128, 128, 0], [116, 128, 0], [105, 128, 0], [93, 128, 0], [81, 128, 0], [70, 128, 0], [58, 128, 0], [47, 128, 0], [35, 128, 0], [23, 128, 0], [12, 128, 0], [0, 128, 0], [0, 128, 12], [0, 128, 23], [0, 128, 35], [0, 128, 47], [0, 128, 58], [0, 128, 70], [0, 128, 81], [0, 128, 93], [0, 128, 105], [0, 128, 116], [0, 128, 128], [0, 116, 128], [0, 105, 128], [0, 93, 128], [0, 81, 128], [0, 70, 128], [0, 58, 128], [0, 47, 128], [0, 35, 128], [0, 23, 128], [0, 12, 128], [0, 0, 128], [12, 0, 128], [23, 0, 128], [35, 0, 128], [47, 0, 128], [58, 0, 128], [70, 0, 128], [81, 0, 128], [93, 0, 128], [105, 0, 128], [116, 0, 128], [128, 0, 128], [128, 0, 116], [128, 0, 105], [128, 0, 93], [128, 0, 81], [128, 0, 70], [128, 0, 58], [128, 0, 47], [128, 0, 35], [128, 0, 23], [128, 0, 12], [255, 192, 192], [255, 64, 64], [192, 0, 0], [64, 0, 0], [255, 255, 192], [255, 255, 64], [192, 192, 0], [64, 64, 0], [192, 255, 192], [64, 255, 64], [0, 192, 0], [0, 64, 0], [192, 255, 255], [64, 255, 255], [0, 192, 192], [0, 64, 64], [192, 192, 255], [64, 64, 255], [0, 0, 192], [0, 0, 64], [255, 192, 255], [255, 64, 255], [192, 0, 192], [64, 0, 64], [255, 96, 96], [255, 255, 255], [245, 245, 245], [235, 235, 235], [224, 224, 224], [213, 213, 213], [203, 203, 203], [192, 192, 192], [181, 181, 181], [171, 171, 171], [160, 160, 160], [149, 149, 149], [139, 139, 139], [128, 128, 128], [117, 117, 117], [107, 107, 107], [96, 96, 96], [85, 85, 85], [75, 75, 75], [64, 64, 64], [53, 53, 53], [43, 43, 43], [32, 32, 32], [21, 21, 21], [11, 11, 11], [0, 0, 0]] def rgb2int(rgb): return int('0x%02x%02x%02x' % tuple(rgb),0) class Table(object): """Container object for one ILDA table: either a frame (table of points) or a palette (table of colors). The 'items' list contains the data within this table. Each item is a tuple, corresponding to the raw values within that row of the table. 2D frame: (x, y, status) 3D frame: (x, y, z, status) Color: (r, g, b) """ def __init__(self, format=FORMAT_2D, name="", length=0, number=0, total=0, scanHead=0): self.__dict__.update(locals()) self.items = [] self.itemsproducer = None def __repr__(self): return ("" % (self.format, self.name, self.length, self.number, self.total, self.scanHead)) def unpackHeader(self, data): magic, self.format, self.name, self.length, \ self.number, self.total, self.scanHead, \ reserved = struct.unpack(HEADER_FORMAT, data) print(magic, HEADER_MAGIC) if magic != HEADER_MAGIC: raise ValueError("Bad ILDA header magic. Not an ILDA file?") if reserved != HEADER_RESERVED: raise ValueError("Reserved ILDA field is not zero.") def packHeader(self): return struct.pack(HEADER_FORMAT, HEADER_MAGIC, self.format, self.name, self.length, self.number, self.total, self.scanHead, HEADER_RESERVED) def readHeader(self, stream): self.unpackHeader(stream.read(HEADER_LEN)) def writeHeader(self, stream): stream.write(self.packHeader()) def _getItemFormat(self): try: return formatTable[self.format] except IndexError: raise ValueError("Unsupported format code") def read_stream(self, stream): """Read the header, then read all items in this table.""" self.readHeader(stream) if self.length: fmt = self._getItemFormat() itemSize = struct.calcsize(fmt) self.items = [struct.unpack(fmt, stream.read(itemSize)) for i in range(self.length)] self.itemsproducer = self.produce() def write(self, stream): """Write the header, then write all items in this table.""" self.writeHeader(stream) if self.length: fmt = self._getItemFormat() itemSize = struct.calcsize(fmt) stream.write(''.join([struct.pack(fmt, *item) for item in self.items])) def iterPoints(self): """Iterate over Point instances for each item in this table. Only makes sense if this is a 2D or 3D point table. """ for item in self.items: p = Point() p.decode(item) yield p def produce(self): """Iterate over Point instances for each item in this table. Only makes sense if this is a 2D or 3D point table. """ while True: for item in self.items: p = Point() p.decode(item) yield p.encode() #yield (p.x, p.y, p.z, p.color, p.blanking) def read(self, cap): """yields what dac.play_stream() needs (x, y, z, ?, ?) """ return [next(self.itemsproducer) for i in range(cap)] class Point: """Abstraction for one vector point. The Table object, for completeness and efficiency, stores raw tuples for each point. This is a higher level interface that decodes the status bits and represents coordinates in floating point. """ def __init__(self, x=0.0, y=0.0, z=0.0, color=0, blanking=False): self.__dict__.update(locals()) def __repr__(self): return "%s, %s, %s, %s, %s" % ( self.x, self.y, self.z, self.color, self.blanking) #return "" % ( # self.x, self.y, self.z, self.color, self.blanking) def encode(self): status = self.color & 0xFF if self.blanking: status |= 1 << 14 return ( int( min(0x7FFF, max(-0x7FFF, self.x * 0x7FFF)) ), int( min(0x7FFF, max(-0x7FFF, self.y * 0x7FFF)) ), int( min(0x7FFF, max(-0x7FFF, self.z * 0x7FFF)) ), int( min(0x7FFF, max(-0x7FFF, self.color * 0x7FFF)) ), int( min(0x7FFF, max(-0x7FFF, self.blanking * 0x7FFF)) ) ) def decode(self, t): #print "~~ Decoding, t of len "+ str(len(t)) +" is: " + str(t) self.x = t[0] / 0x7FFF self.y = t[1] / 0x7FFF if len(t) > 3: self.z = t[2] / 0x7FFF # self.color = t[3] & 0xFF # self.blanking = (t[3] & (1 << 14)) != 0 else: self.z = 0.0 self.color = t[-1] & 0xFF self.blanking = (t[-1] & (1 << 14)) != 0 def read(stream): """Read ILDA data from a stream until we hit the end-of-stream marker. Yields a sequence of Table objects. """ while True: t = Table() t.read_stream(stream) if not t.length: # End-of-stream break yield t def write(stream, tables): """Write a sequence of tables in ILDA format, terminated by an end-of-stream marker. """ for t in tables: t.write(stream) Table().write(stream) def readFrames(stream): """Read ILDA data from a stream, and ignore all non-frame tables. Yields only 2D or 3D point tables. """ for t in read(stream): if t.format in (FORMAT_2D, FORMAT_3D): yield t def readFirstFrame(stream): """Read only a single frame from an ILDA stream.""" for frame in readFrames(stream): return frame # f = open(args.ild, 'rb') myframe = readFirstFrame(f) while myframe.number +1< myframe.total: start = time.time() shape =[] if myframe is None: f.close() break debug(name,"Frame", myframe.number, "/",myframe.total, "length", myframe.length) for p in myframe.iterPoints(): p2 = str(p) point = p2.split(',') x = float(point[0]) y = float(point[1]) z = float(point[2]) color = int(point[3]) blanking = point[4][1:] if blanking == "True": shape.append([300+(x*300),300+(-y*300),0]) else: shape.append([300+(x*300),300+(-y*300),rgb2int(colors64[color])]) print(shape, flush=True); myframe = readFirstFrame(f) looptime = time.time() - start if( looptime < optimal_looptime ): time.sleep( optimal_looptime - looptime) debug(name+" micro sleep:"+str( optimal_looptime - looptime)) f.close() debug(name + " end of .ild animation")