Python实现生成bmp图像的方法
之前使用过c、java、go语言实现过生成纯色BMP图片的功能。
现在由python语言完成该功能。
from array import array class bmp: """ bmp data structure """ def __init__(self, w=1080, h=1920, color = 0xffffff): self.w = w self.h = h self.gen_bmp_header() self.paint_bgcolor(color) def calc_data_size (self): if((self.w*3)%4 == 0): self.dataSize = self.w * 3 * self.h else: self.dataSize = (((self.w * 3) // 4 + 1) * 4) * self.h self.fileSize = self.dataSize + 54 def conv2byte(self, l, num, len): tmp = num for i in range(len): l.append(tmp & 0x000000ff) tmp >>= 8 def gen_bmp_header (self): self.calc_data_size(); self.bmp_header = [0x42, 0x4d] self.conv2byte(self.bmp_header, self.fileSize, 4) #file size self.conv2byte(self.bmp_header, 0, 2) self.conv2byte(self.bmp_header, 0, 2) self.conv2byte(self.bmp_header, 54, 4) #rgb data offset self.conv2byte(self.bmp_header, 40, 4) #info block size self.conv2byte(self.bmp_header, self.w, 4) self.conv2byte(self.bmp_header, self.h, 4) self.conv2byte(self.bmp_header, 1, 2) self.conv2byte(self.bmp_header, 24, 2) #888 self.conv2byte(self.bmp_header, 0, 4) #no compression self.conv2byte(self.bmp_header, self.dataSize, 4) #rgb data size self.conv2byte(self.bmp_header, 0, 4) self.conv2byte(self.bmp_header, 0, 4) self.conv2byte(self.bmp_header, 0, 4) self.conv2byte(self.bmp_header, 0, 4) def print_bmp_header (self): length = len(self.bmp_header) for i in range(length): print("{:0>2x}".format(self.bmp_header[i]), end=' ') if i%16 == 15: print('') print('') def paint_bgcolor(self, color=0xffffff): self.rgbData = [] for r in range(self.h): self.rgbDataRow = [] for c in range(self.w): self.rgbDataRow.append(color) self.rgbData.append(self.rgbDataRow) def paint_line(self, x1, y1, x2, y2, color): k = (y2 - y1) / (x2 - x1) for x in range(x1, x2+1): y = int(k * (x - x1) + y1) self.rgbData[y][x] = color def paint_rect(self, x1, y1, w, h, color): for x in range(x1, x1+w): for y in range(y1, y1+h): self.rgbData[y][x] = color def paint_point(self, x, y, color=0x000000): self.rgbData[y][x] = color def save_image(self, name="save.bmp"): f = open(name, 'wb') #write bmp header f.write(array('B', self.bmp_header).tobytes()) #write rgb data zeroBytes = self.dataSize // self.h - self.w * 3 for r in range(self.h): l = [] for i in range(len(self.rgbData[r])): p = self.rgbData[r][i] l.append(p & 0x0000ff) p >>= 8 l.append(p & 0x0000ff) p >>= 8 l.append(p & 0x0000ff) f.write(array('B', l).tobytes()) for i in range(zeroBytes): f.write(bytes([0x00])) #close file f.close() if __name__ == '__main__': image = bmp(35, 35) for i in range(35): image.paint_point(i, i, 0xff0000) image.save_image("save1.bmp") import os os.system("save1.bmp")
到此这篇关于Python实现生成bmp图像的方法的文章就介绍到这了,更多相关Python生成bmp图像内容请搜索我们以前的文章或继续浏览下面的相关文章希望大家以后多多支持我们!
赞 (0)