bitmaps
This commit is contained in:
parent
79e3d2039d
commit
fab367cf41
|
@ -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" )
|
||||
|
Binary file not shown.
After Width: | Height: | Size: 4.8 KiB |
Loading…
Reference in New Issue