pyxllib 0.3.197__py3-none-any.whl → 0.3.200__py3-none-any.whl

pyxllib/cv/imfile.py

@@ -1,159 +1,159 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-# @Author : 陈坤泽
-# @Email  : 877362867@qq.com
-# @Date   : 2020/06/02 16:00
-
-
-from collections import defaultdict
-import concurrent.futures
-import os
-import re
-import subprocess
-
-from pyxllib.file.specialist import File
-
-
-def magick(infile, *, outfile=None, if_exists='error', transparent=None, trim=False, density=None, other_args=None):
-    """ 调用iamge magick的magick.exe工具
-
-    :param infile: 处理对象文件
-    :param outfile: 输出文件，可以不写，默认原地操作（只设置透明度、裁剪时可能会原地操作）
-    :param if_exists: 如果目标文件已存在要怎么处理
-    :param transparent:
-        True: 将白底设置为透明底
-        可以传入颜色参数，控制要设置的为透明底的背景色，默认为'white'
-        注意也可以设置rgb：'rgb(164,192,167)'
-    :param trim: 裁剪掉四周空白
-    :param density: 设置图片的dpi值
-    :param other_args: 其他参数，输入格式如：['-quality', 100]
-    :return:
-        False：infile不存在或者不支持的文件扩展名
-        返回生成的文件名（outfile）
-    """
-    # 1 条件判断，有些情况下不用处理
-    if not outfile:
-        outfile = infile
-
-    # 2
-    # 200914周一20:40，这有个相对路径的bug，修复了下，否则 test/a.png 会变成 test/test/a.png
-    if File(outfile).exist_preprcs(if_exists):
-        # 2.1 判断是否是支持的输入文件类型
-        ext = os.path.splitext(infile)[1].lower()
-        if not File(infile) or not ext in ('.png', '.eps', '.pdf', '.jpg', '.jpeg', '.wmf', '.emf'):
-            return False
-
-        # 2.2 生成需要执行的参数
-        cmd = ['magick.exe']
-        # 透明、裁剪、density都是可以重复操作的
-        if density: cmd.extend(['-density', str(density)])
-        cmd.append(infile)
-        if transparent:
-            if not isinstance(transparent, str):
-                transparent_ = 'white'
-            else:
-                transparent_ = transparent
-            cmd.extend(['-transparent', transparent_])
-        if trim: cmd.append('-trim')
-        if other_args: cmd.extend(other_args)
-        cmd.append(outfile)
-
-        # 2.3 生成目标png图片
-        cmd = [x.replace('\\', '/') for x in cmd]
-        print(' '.join(cmd))
-        subprocess.run(cmd, stderr=subprocess.PIPE, shell=True)  # 看不懂magick出错写的是啥，关了
-
-    return outfile
-
-
-def ensure_pngs(folder, *, if_exists='skip',
-                transparent=None, trim=False,
-                density=None, epsdensity=None,
-                max_workers=None):
-    """ 确保一个目录下的所有图片都有一个png版本格式的文件
-
-    :param folder: 目录名，会遍历直接目录下所有没png的stem名称生成png
-    :param if_exists: 如果文件已存在，要进行的操作
-        'replace'，直接替换
-        'backup'，备份后生成新文件
-        'ignore'，不写入
-
-    :param transparent: 设置转换后的图片是否要变透明
-    :param trim: 是否裁剪边缘空白
-    :param density: 缩放尺寸
-        TODO magick 和 inkscape 的dpi参考值是不同的，inkscape是真的dpi，magick有个比例差，我还没搞明白
-    :param epsdensity: eps转png时默认放大的比例，注意默认100%是72，写144等于长宽各放大一倍
-    :param max_workers: 并行的最大线程数
-    """
-
-    # 1 提取字典d，key<str>: 文件stem名称， value<set>：含有的扩展名
-    d = defaultdict(set)
-    for file in os.listdir(folder):
-        if file.endswith('-eps-converted-to.pdf'): continue
-        name, ext = os.path.splitext(file)
-        ext = ext.lower()
-        if ext in ('.png', '.eps', '.pdf', '.jpg', '.jpeg', '.wmf', '.emf', '.svg'):
-            d[name].add(ext)
-
-    # 2 遍历处理每个stem的图片
-    executor = concurrent.futures.ThreadPoolExecutor(max_workers)
-    for name, exts in d.items():
-        # 已经存在png格式的图片时，看if_exists参数
-        if '.png' in exts:
-            if if_exists == 'ignore':
-                continue
-            elif if_exists == 'backup':
-                File(name, folder, suffix='.png').backup(move=True)
-            elif if_exists == 'replace':
-                pass
-            else:
-                raise ValueError
-
-        # 注意这里必须按照指定的类型优先级顺序，找到母图后替换，不能用找到的文件类型顺序
-        for t in ('.eps', '.pdf', '.jpg', '.jpeg', '.wmf', '.emf', '.svg'):
-            if t in exts:
-                filename = os.path.join(folder, name)
-                if t == '.svg':  # svg用inkscape软件单独处理
-                    cmd = ['inkscape.exe', '-f', filename + t]  # 使用inkscape把svg转png
-                    if trim: cmd.append('-D')  # 裁剪参数
-                    if density: cmd.extend(['-d', str(density)])  # 设置dpi参数
-                    cmd.extend(['-e', filename + '.png'])
-                    executor.submit(subprocess.run, cmd)
-                elif t == '.eps':
-                    executor.submit(magick, filename + t, outfile=filename + '.png',
-                                    transparent=transparent, trim=trim, density=epsdensity)
-                else:
-                    executor.submit(magick, filename + t, outfile=filename + '.png',
-                                    transparent=transparent, trim=trim, density=density)
-                break
-    executor.shutdown()
-
-
-def zoomsvg(file, scale=1):
-    """ 缩放svg文件
-
-    :param file:
-        如果输入一个目录，会处理目录下所有的svg图片
-        否则只处理指定的文件
-        如果是文本文件，则处理完文本后返回
-    :param scale: 缩放的比例，默认100%不调整
-    :return:
-    """
-    if scale == 1: return
-
-    def func(m):
-        def g(m): return m.group(1) + str(float(m.group(2)) * scale)
-
-        return re.sub(r'((?:height|width)=")(\d+(?:\.\d+)?)', g, m.group())
-
-    if os.path.isfile(file):
-        s = re.sub(r'<svg .+?>', func, File(file).read(), flags=re.DOTALL)
-        File(file).write(s, if_exists='replace')
-    elif os.path.isdir(file):
-        for f in os.listdir(file):
-            if not f.endswith('.svg'): continue
-            f = os.path.join(file, f)
-            s = re.sub(r'<svg\s+.+?>', func, File(f).read(), flags=re.DOTALL)
-            File(file).write(s, if_exists='replace')
-    elif isinstance(file, str) and '<svg ' in file:  # 输入svg的代码文本
-        return re.sub(r'<svg .+?>', func, file, flags=re.DOTALL)
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# @Author : 陈坤泽
+# @Email  : 877362867@qq.com
+# @Date   : 2020/06/02 16:00
+
+
+from collections import defaultdict
+import concurrent.futures
+import os
+import re
+import subprocess
+
+from pyxllib.file.specialist import File
+
+
+def magick(infile, *, outfile=None, if_exists='error', transparent=None, trim=False, density=None, other_args=None):
+    """ 调用iamge magick的magick.exe工具
+
+    :param infile: 处理对象文件
+    :param outfile: 输出文件，可以不写，默认原地操作（只设置透明度、裁剪时可能会原地操作）
+    :param if_exists: 如果目标文件已存在要怎么处理
+    :param transparent:
+        True: 将白底设置为透明底
+        可以传入颜色参数，控制要设置的为透明底的背景色，默认为'white'
+        注意也可以设置rgb：'rgb(164,192,167)'
+    :param trim: 裁剪掉四周空白
+    :param density: 设置图片的dpi值
+    :param other_args: 其他参数，输入格式如：['-quality', 100]
+    :return:
+        False：infile不存在或者不支持的文件扩展名
+        返回生成的文件名（outfile）
+    """
+    # 1 条件判断，有些情况下不用处理
+    if not outfile:
+        outfile = infile
+
+    # 2
+    # 200914周一20:40，这有个相对路径的bug，修复了下，否则 test/a.png 会变成 test/test/a.png
+    if File(outfile).exist_preprcs(if_exists):
+        # 2.1 判断是否是支持的输入文件类型
+        ext = os.path.splitext(infile)[1].lower()
+        if not File(infile) or not ext in ('.png', '.eps', '.pdf', '.jpg', '.jpeg', '.wmf', '.emf'):
+            return False
+
+        # 2.2 生成需要执行的参数
+        cmd = ['magick.exe']
+        # 透明、裁剪、density都是可以重复操作的
+        if density: cmd.extend(['-density', str(density)])
+        cmd.append(infile)
+        if transparent:
+            if not isinstance(transparent, str):
+                transparent_ = 'white'
+            else:
+                transparent_ = transparent
+            cmd.extend(['-transparent', transparent_])
+        if trim: cmd.append('-trim')
+        if other_args: cmd.extend(other_args)
+        cmd.append(outfile)
+
+        # 2.3 生成目标png图片
+        cmd = [x.replace('\\', '/') for x in cmd]
+        print(' '.join(cmd))
+        subprocess.run(cmd, stderr=subprocess.PIPE, shell=True)  # 看不懂magick出错写的是啥，关了
+
+    return outfile
+
+
+def ensure_pngs(folder, *, if_exists='skip',
+                transparent=None, trim=False,
+                density=None, epsdensity=None,
+                max_workers=None):
+    """ 确保一个目录下的所有图片都有一个png版本格式的文件
+
+    :param folder: 目录名，会遍历直接目录下所有没png的stem名称生成png
+    :param if_exists: 如果文件已存在，要进行的操作
+        'replace'，直接替换
+        'backup'，备份后生成新文件
+        'ignore'，不写入
+
+    :param transparent: 设置转换后的图片是否要变透明
+    :param trim: 是否裁剪边缘空白
+    :param density: 缩放尺寸
+        TODO magick 和 inkscape 的dpi参考值是不同的，inkscape是真的dpi，magick有个比例差，我还没搞明白
+    :param epsdensity: eps转png时默认放大的比例，注意默认100%是72，写144等于长宽各放大一倍
+    :param max_workers: 并行的最大线程数
+    """
+
+    # 1 提取字典d，key<str>: 文件stem名称， value<set>：含有的扩展名
+    d = defaultdict(set)
+    for file in os.listdir(folder):
+        if file.endswith('-eps-converted-to.pdf'): continue
+        name, ext = os.path.splitext(file)
+        ext = ext.lower()
+        if ext in ('.png', '.eps', '.pdf', '.jpg', '.jpeg', '.wmf', '.emf', '.svg'):
+            d[name].add(ext)
+
+    # 2 遍历处理每个stem的图片
+    executor = concurrent.futures.ThreadPoolExecutor(max_workers)
+    for name, exts in d.items():
+        # 已经存在png格式的图片时，看if_exists参数
+        if '.png' in exts:
+            if if_exists == 'ignore':
+                continue
+            elif if_exists == 'backup':
+                File(name, folder, suffix='.png').backup(move=True)
+            elif if_exists == 'replace':
+                pass
+            else:
+                raise ValueError
+
+        # 注意这里必须按照指定的类型优先级顺序，找到母图后替换，不能用找到的文件类型顺序
+        for t in ('.eps', '.pdf', '.jpg', '.jpeg', '.wmf', '.emf', '.svg'):
+            if t in exts:
+                filename = os.path.join(folder, name)
+                if t == '.svg':  # svg用inkscape软件单独处理
+                    cmd = ['inkscape.exe', '-f', filename + t]  # 使用inkscape把svg转png
+                    if trim: cmd.append('-D')  # 裁剪参数
+                    if density: cmd.extend(['-d', str(density)])  # 设置dpi参数
+                    cmd.extend(['-e', filename + '.png'])
+                    executor.submit(subprocess.run, cmd)
+                elif t == '.eps':
+                    executor.submit(magick, filename + t, outfile=filename + '.png',
+                                    transparent=transparent, trim=trim, density=epsdensity)
+                else:
+                    executor.submit(magick, filename + t, outfile=filename + '.png',
+                                    transparent=transparent, trim=trim, density=density)
+                break
+    executor.shutdown()
+
+
+def zoomsvg(file, scale=1):
+    """ 缩放svg文件
+
+    :param file:
+        如果输入一个目录，会处理目录下所有的svg图片
+        否则只处理指定的文件
+        如果是文本文件，则处理完文本后返回
+    :param scale: 缩放的比例，默认100%不调整
+    :return:
+    """
+    if scale == 1: return
+
+    def func(m):
+        def g(m): return m.group(1) + str(float(m.group(2)) * scale)
+
+        return re.sub(r'((?:height|width)=")(\d+(?:\.\d+)?)', g, m.group())
+
+    if os.path.isfile(file):
+        s = re.sub(r'<svg .+?>', func, File(file).read(), flags=re.DOTALL)
+        File(file).write(s, if_exists='replace')
+    elif os.path.isdir(file):
+        for f in os.listdir(file):
+            if not f.endswith('.svg'): continue
+            f = os.path.join(file, f)
+            s = re.sub(r'<svg\s+.+?>', func, File(f).read(), flags=re.DOTALL)
+            File(file).write(s, if_exists='replace')
+    elif isinstance(file, str) and '<svg ' in file:  # 输入svg的代码文本
+        return re.sub(r'<svg .+?>', func, file, flags=re.DOTALL)

pyxllib/cv/imhash.py

@@ -1,39 +1,39 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-# @Author : 陈坤泽
-# @Email  : 877362867@qq.com
-# @Date   : 2021/06/08 22:53
-
-"""
-TODO 写一些图片相似度相关功能
-"""
-
-from pyxllib.prog.pupil import check_install_package
-
-check_install_package('imagehash', 'ImageHash')
-
-import imagehash
-import numpy as np
-
-from pyxllib.cv.xlpillib import xlpil
-
-
-def get_init_hash():
-    """ 获得一个初始、空哈希值 """
-    return imagehash.ImageHash(np.zeros([8, 8]).astype(bool))
-
-
-def phash(image, *args, **kwargs):
-    """ 修改了官方接口，这里输入的image支持泛用格式
-    """
-    im = xlpil.read(image)
-    return imagehash.phash(im, *args, **kwargs)
-
-
-def dhash(image, *args, **kwargs):
-    """ 修改了官方接口，这里输入的image支持泛用格式
-
-    官方比较推荐使用，性能速度又快的，就是dhash
-    """
-    im = xlpil.read(image)
-    return imagehash.dhash(im, *args, **kwargs)
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# @Author : 陈坤泽
+# @Email  : 877362867@qq.com
+# @Date   : 2021/06/08 22:53
+
+"""
+TODO 写一些图片相似度相关功能
+"""
+
+from pyxllib.prog.pupil import check_install_package
+
+check_install_package('imagehash', 'ImageHash')
+
+import imagehash
+import numpy as np
+
+from pyxllib.cv.xlpillib import xlpil
+
+
+def get_init_hash():
+    """ 获得一个初始、空哈希值 """
+    return imagehash.ImageHash(np.zeros([8, 8]).astype(bool))
+
+
+def phash(image, *args, **kwargs):
+    """ 修改了官方接口，这里输入的image支持泛用格式
+    """
+    im = xlpil.read(image)
+    return imagehash.phash(im, *args, **kwargs)
+
+
+def dhash(image, *args, **kwargs):
+    """ 修改了官方接口，这里输入的image支持泛用格式
+
+    官方比较推荐使用，性能速度又快的，就是dhash
+    """
+    im = xlpil.read(image)
+    return imagehash.dhash(im, *args, **kwargs)

pyxllib/cv/pupil.py

@@ -1,9 +1,9 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-# @Author : 陈坤泽
-# @Email  : 877362867@qq.com
-# @Date   : 2021/10/07 09:06
-
-# Rgb用了numpy，不再属于pupil了，而且会初始化很多颜色列表，也浪费空间，
-# 所以改成独立包，需要的时候再手动导入
-# from pyxllib.cv.rgbfmt import RgbFormatter
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# @Author : 陈坤泽
+# @Email  : 877362867@qq.com
+# @Date   : 2021/10/07 09:06
+
+# Rgb用了numpy，不再属于pupil了，而且会初始化很多颜色列表，也浪费空间，
+# 所以改成独立包，需要的时候再手动导入
+# from pyxllib.cv.rgbfmt import RgbFormatter