Python实现生成bmp图像的方法


Posted in Python onJune 13, 2021

之前使用过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图像内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
python下如何让web元素的生成更简单的分析
Jul 17 Python
Python实现Linux下守护进程的编写方法
Aug 22 Python
python实现的希尔排序算法实例
Jul 01 Python
Python 提取dict转换为xml/json/table并输出的实现代码
Aug 28 Python
Python的标准模块包json详解
Mar 13 Python
Python反射用法实例简析
Dec 22 Python
python+opencv实现高斯平滑滤波
Jul 21 Python
离线状态下在jupyter notebook中使用plotly实例
Apr 24 Python
python 识别登录验证码图片功能的实现代码(完整代码)
Jul 03 Python
Python用dilb提取照片上人脸的示例
Oct 26 Python
python tqdm库的使用
Nov 30 Python
selenium3.0+python之环境搭建的方法步骤
Feb 01 Python
Python实现随机生成迷宫并自动寻路
python中opencv实现图片文本倾斜校正
Jun 11 #Python
端午节将至,用Python爬取粽子数据并可视化,看看网友喜欢哪种粽子吧!
Python-OpenCV实现图像缺陷检测的实例
Python中OpenCV实现简单车牌字符切割
Python排序算法之插入排序及其优化方案详解
Python下opencv库的安装过程及问题汇总
Jun 11 #Python
You might like
PHP学习 变量使用总结
2011/03/24 PHP
PHP中的错误处理、异常处理机制分析
2012/05/07 PHP
确保Laravel网站不会被嵌入到其他站点中的方法
2019/10/18 PHP
PHP code 验证码生成类定义和简单使用示例
2020/05/27 PHP
PHP基于phpqrcode类生成二维码的方法示例详解
2020/08/07 PHP
JavaScript脚本性能优化注意事项
2008/11/18 Javascript
js 剪切板的用法(clipboardData.setData)与js match函数介绍
2013/11/19 Javascript
防止jQuery ajax Load使用缓存的方法小结
2014/02/22 Javascript
javascript的switch用法注意事项分析
2015/02/02 Javascript
jquery带有索引按钮且自动轮播切换特效代码分享
2015/09/15 Javascript
深入理解js中的加载事件
2017/02/08 Javascript
JavaScript基本语法_动力节点Java学院整理
2017/06/26 Javascript
JavaScript获取tr td 的三种方式全面总结(推荐)
2017/08/15 Javascript
在NPM发布自己造的轮子的方法步骤
2019/03/09 Javascript
微信小程序解析富文本过程详解
2019/07/13 Javascript
关于Vue中$refs的探索浅析
2020/11/05 Javascript
Python中unittest用法实例
2014/09/25 Python
Python的Urllib库的基本使用教程
2015/04/30 Python
简单介绍Python中的round()方法
2015/05/15 Python
详解Python中表达式i += x与i = i + x是否等价
2017/02/08 Python
python抓取网页中链接的静态图片
2018/01/29 Python
Python定义二叉树及4种遍历方法实例详解
2018/07/05 Python
Python实现最常见加密方式详解
2019/07/13 Python
python按行读取文件并找出其中指定字符串
2019/08/08 Python
python3.7环境下安装Anaconda的教程图解
2019/09/10 Python
Python PyPDF2模块安装使用解析
2020/01/19 Python
Volcom英国官方商店:美国殿堂级滑板、冲浪、滑雪服装品牌
2019/03/13 全球购物
Skyscanner新西兰:全球领先的旅游搜索网站
2019/08/26 全球购物
Pandora西班牙官方商店:PandoraShop.es
2020/10/05 全球购物
学习委员自我鉴定
2014/01/13 职场文书
财务学生的职业生涯发展
2014/02/11 职场文书
宣传口号大全
2014/06/16 职场文书
2015年全国爱眼日活动小结
2015/02/27 职场文书
会议简报格式范文
2015/07/20 职场文书
公司团队口号霸气押韵
2015/12/24 职场文书
2016大学生毕业实习心得体会
2016/01/23 职场文书