104 lines
2.2 KiB
Python
104 lines
2.2 KiB
Python
|
#
|
||
|
|
# The Python Imaging Library.
|
||
|
|
# $Id$
|
||
|
|
#
|
||
|
|
# MSP file handling
|
||
|
|
#
|
||
|
|
# This is the format used by the Paint program in Windows 1 and 2.
|
||
|
|
#
|
||
|
|
# History:
|
||
|
|
# 95-09-05 fl Created
|
||
|
|
# 97-01-03 fl Read/write MSP images
|
||
|
|
#
|
||
|
|
# Copyright (c) Secret Labs AB 1997.
|
||
|
|
# Copyright (c) Fredrik Lundh 1995-97.
|
||
|
|
#
|
||
|
|
# See the README file for information on usage and redistribution.
|
||
|
|
#
|
||
|
|
|
||
|
|
|
||
|
|
__version__ = "0.1"
|
||
|
|
|
||
|
|
import Image, ImageFile
|
||
|
|
|
||
|
|
|
||
|
|
#
|
||
|
|
# read MSP files
|
||
|
|
|
||
|
|
def i16(c):
|
||
|
|
return ord(c[0]) + (ord(c[1])<<8)
|
||
|
|
|
||
|
|
def _accept(prefix):
|
||
|
|
return prefix[:4] in ["DanM", "LinS"]
|
||
|
|
|
||
|
|
##
|
||
|
|
# Image plugin for Windows MSP images. This plugin supports both
|
||
|
|
# uncompressed (Windows 1.0).
|
||
|
|
|
||
|
|
class MspImageFile(ImageFile.ImageFile):
|
||
|
|
|
||
|
|
format = "MSP"
|
||
|
|
format_description = "Windows Paint"
|
||
|
|
|
||
|
|
def _open(self):
|
||
|
|
|
||
|
|
# Header
|
||
|
|
s = self.fp.read(32)
|
||
|
|
if s[:4] not in ["DanM", "LinS"]:
|
||
|
|
raise SyntaxError, "not an MSP file"
|
||
|
|
|
||
|
|
# Header checksum
|
||
|
|
sum = 0
|
||
|
|
for i in range(0, 32, 2):
|
||
|
|
sum = sum ^ i16(s[i:i+2])
|
||
|
|
if sum != 0:
|
||
|
|
raise SyntaxError, "bad MSP checksum"
|
||
|
|
|
||
|
|
self.mode = "1"
|
||
|
|
self.size = i16(s[4:]), i16(s[6:])
|
||
|
|
|
||
|
|
if s[:4] == "DanM":
|
||
|
|
self.tile = [("raw", (0,0)+self.size, 32, ("1", 0, 1))]
|
||
|
|
else:
|
||
|
|
self.tile = [("msp", (0,0)+self.size, 32+2*self.size[1], None)]
|
||
|
|
|
||
|
|
#
|
||
|
|
# write MSP files (uncompressed only)
|
||
|
|
|
||
|
|
def o16(i):
|
||
|
|
return chr(i&255) + chr(i>>8&255)
|
||
|
|
|
||
|
|
def _save(im, fp, filename):
|
||
|
|
|
||
|
|
if im.mode != "1":
|
||
|
|
raise IOError, "cannot write mode %s as MSP" % im.mode
|
||
|
|
|
||
|
|
# create MSP header
|
||
|
|
header = [0] * 16
|
||
|
|
|
||
|
|
header[0], header[1] = i16("Da"), i16("nM") # version 1
|
||
|
|
header[2], header[3] = im.size
|
||
|
|
header[4], header[5] = 1, 1
|
||
|
|
header[6], header[7] = 1, 1
|
||
|
|
header[8], header[9] = im.size
|
||
|
|
|
||
|
|
sum = 0
|
||
|
|
for h in header:
|
||
|
|
sum = sum ^ h
|
||
|
|
header[12] = sum # FIXME: is this the right field?
|
||
|
|
|
||
|
|
# header
|
||
|
|
for h in header:
|
||
|
|
fp.write(o16(h))
|
||
|
|
|
||
|
|
# image body
|
||
|
|
ImageFile._save(im, fp, [("raw", (0,0)+im.size, 32, ("1", 0, 1))])
|
||
|
|
|
||
|
|
#
|
||
|
|
# registry
|
||
|
|
|
||
|
|
Image.register_open("MSP", MspImageFile, _accept)
|
||
|
|
Image.register_save("MSP", _save)
|
||
|
|
|
||
|
|
Image.register_extension("MSP", ".msp")
|