bitmaps

client/bmp.py

@@ -0,0 +1,206 @@
+# -*- coding: utf-8 -*-
+"""
+bmp.py - module for constructing simple BMP graphics files
+
+ 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.
+
+"""
+__version__ = "0.3"
+__about =  "bmp module, version %s, written by Paul McGuire, October, 2003, updated by Margus Laak, September, 2009" % __version__ 
+
+from math import ceil, hypot
+
+
+def shortToString(i):
+  hi = (i & 0xff00) >> 8
+  lo = i & 0x00ff
+  return chr(lo) + chr(hi)
+
+def longToString(i):
+  hi = (long(i) & 0x7fff0000) >> 16
+  lo = long(i) & 0x0000ffff
+  return shortToString(lo) + shortToString(hi)
+
+def long24ToString(i):
+  return chr(i & 0xff) + chr(i >> 8 & 0xff) + chr(i >> 16 & 0xff)
+
+def stringToLong(input_string, offset):
+  return ord(input_string[offset+3]) << 24 | ord(input_string[offset+2]) << 16 | ord(input_string[offset+1]) << 8 | ord(input_string[offset])
+
+def stringToLong24(input_string, offset):
+  return ord(input_string[offset+2]) << 16 | ord(input_string[offset+1]) << 8 | ord(input_string[offset])
+
+class Color(object):
+  """class for specifying colors while drawing BitMap elements"""
+  __slots__ = [ 'red', 'grn', 'blu' ]
+  __shade = 32
+  
+  def __init__( self, r=0, g=0, b=0 ):
+    self.red = r
+    self.grn = g
+    self.blu = b
+
+  def __setattr__(self, name, value):
+    if hasattr(self, name):
+      raise AttributeError, "Color is immutable"
+    else:
+      object.__setattr__(self, name, value)
+
+  def __str__( self ):
+    return "R:%d G:%d B:%d" % (self.red, self.grn, self.blu )
+    
+  def __hash__( self ):
+    return ( ( long(self.blu) ) + 
+              ( long(self.grn) <<  8 ) + 
+              ( long(self.red) << 16 ) )
+  
+  def __eq__( self, other ):
+    return (self is other) or (self.toLong == other.toLong)
+
+  def lighten( self ):
+    return Color( 
+      min( self.red + Color.__shade, 255), 
+      min( self.grn + Color.__shade, 255), 
+      min( self.blu + Color.__shade, 255)  
+      )
+  
+  def darken( self ):
+    return Color( 
+      max( self.red - Color.__shade, 0), 
+      max( self.grn - Color.__shade, 0), 
+      max( self.blu - Color.__shade, 0)  
+      )
+       
+  def toLong( self ):
+    return self.__hash__()
+    
+  def fromLong( l ):
+    b = l & 0xff
+    l = l >> 8
+    g = l & 0xff
+    l = l >> 8
+    r = l & 0xff
+    return Color( r, g, b )
+  fromLong = staticmethod(fromLong)
+
+# define class constants for common colors
+Color.BLACK    = Color(   0,   0,   0 )
+Color.RED      = Color( 255,   0,   0 )
+Color.GREEN    = Color(   0, 255,   0 )
+Color.BLUE     = Color(   0,   0, 255 )
+Color.CYAN     = Color(   0, 255, 255 )
+Color.MAGENTA  = Color( 255,   0, 255 )
+Color.YELLOW   = Color( 255, 255,   0 )
+Color.WHITE    = Color( 255, 255, 255 )
+Color.DKRED    = Color( 128,   0,   0 )
+Color.DKGREEN  = Color(   0, 128,   0 )
+Color.DKBLUE   = Color(   0,   0, 128 )
+Color.TEAL     = Color(   0, 128, 128 )
+Color.PURPLE   = Color( 128,   0, 128 )
+Color.BROWN    = Color( 128, 128,   0 )
+Color.GRAY     = Color( 128, 128, 128 )
+
+
+class BitMap(object):
+  """class for drawing and saving simple Windows bitmap files"""
+  
+  LINE_SOLID  = 0
+  LINE_DASHED = 1
+  LINE_DOTTED = 2
+  LINE_DOT_DASH=3
+  _DASH_LEN = 12.0
+  _DOT_LEN = 6.0
+  _DOT_DASH_LEN = _DOT_LEN + _DASH_LEN
+  
+  def __init__( self, width, height, 
+                 bkgd = Color.WHITE, frgd = Color.BLACK ):
+    self.wd = int( ceil(width) )
+    self.ht = int( ceil(height) )
+    self.bgcolor = 0
+    self.fgcolor = 1
+    self.palette = []
+    self.palette.append( bkgd.toLong() )
+    self.palette.append( frgd.toLong() )
+    self.currentPen = self.fgcolor
+
+    tmparray = [ self.bgcolor ] * self.wd
+    self.bitarray = [ tmparray[:] for i in range( self.ht ) ]
+    self.currentPen = 1
+    
+
+  def plotPoint( self, x, y ):
+    if ( 0 <= x < self.wd and 0 <= y < self.ht ):
+      x = int(x)
+      y = int(y)
+      self.bitarray[y][x] = self.currentPen
+      
+
+  def _saveBitMapNoCompression( self ):
+    line_padding = (4 - (self.wd % 4)) % 4
+    
+    # write bitmap header
+    _bitmap = "BM"
+    _bitmap += longToString( 54 + self.ht*(self.wd*3 + line_padding) )   # DWORD size in bytes of the file
+    _bitmap += longToString( 0 )    # DWORD 0
+    _bitmap += longToString( 54  )
+    _bitmap += longToString( 40 )    # DWORD header size = 40
+    _bitmap += longToString( self.wd )    # DWORD image width
+    _bitmap += longToString( self.ht )    # DWORD image height
+    _bitmap += shortToString( 1 )    # WORD planes = 1
+    _bitmap += shortToString( 24 )    # WORD bits per pixel = 8
+    _bitmap += longToString( 0 )    # DWORD compression = 0
+    _bitmap += longToString( self.ht * (self.wd * 3 + line_padding) )    # DWORD sizeimage = size in bytes of the bitmap = width * height
+    _bitmap += longToString( 0 )    # DWORD horiz pixels per meter (?)
+    _bitmap += longToString( 0 )    # DWORD ver pixels per meter (?)
+    _bitmap += longToString( 0 )    # DWORD number of colors used = 256
+    _bitmap += longToString( 0 )    # DWORD number of "import colors = len( self.palette )
+
+    # write pixels
+    self.bitarray.reverse()
+    for row in self.bitarray:
+      for pixel in row:
+        c = self.palette[pixel]
+        _bitmap += long24ToString(c)
+      for i in range(line_padding):
+        _bitmap += chr( 0 )
+
+    return _bitmap
+
+    
+    
+  def saveFile( self, filename):
+    _b = self._saveBitMapNoCompression( )
+    
+    f = file(filename, 'wb')
+    f.write(_b)
+    f.close()
+  
+
+
+
+  
+    
+if __name__ == "__main__":
+  
+  bmp = BitMap( 10, 10 )
+  bmp.plotPoint( 5, 5 )
+  bmp.plotPoint( 0, 0 )
+  bmp.saveFile( "test.bmp" )
+