使用go照片合并工具

原创 已于 2023-09-20 14:13:07 修改 · 288 阅读 · 0 ·
CC 4.0 BY-SA版权
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
文章标签：

#golang #开发语言 #运维开发

于 2023-09-20 12:32:13 首次发布

文章目录

概要

` 使用go开发桌面exe程序，方便照片合并压缩操作，身份证正反面，居住证正反面，照片全部压缩到1mb以下大小
目前存在小部分bug但是不影响正常使用！不多说直接上代码！


package main

import (
	"fmt"
	"image"
	"image/draw"
	"image/jpeg"
	"io/ioutil"
	"os"
	"path/filepath"
	"strings"
	"sync"
	"time"
)

const (
	frontKeyword   = "身份证人像面"
	backKeyword    = "身份证国徽面"
	newKeyword1    = "居住证正面"
	newKeyword2    = "居住证反面"
	newKeyword3    = "户口本户主页"
	newKeyword4    = "户口本本人页"
	maxImageSize   = 1024 * 1024  // 1MB
	expirationDate = "2023-09-20" // 设置软件过期日期
	maxConcurrency = 5            // 最大并发处理数
)

var (
	daysRemaining int
	visitedDirs   = make(map[string]struct{})
	mergeMutex    sync.Mutex
)

func main() {
	// 检查软件是否过期
	if isExpired(expirationDate) {
		fmt.Println("软件已过期，请联系开发者获取新版本。联系方式:xxxxxxx")
		return
	}

	// 获取剩余天数
	daysRemaining = calculateDaysRemaining(expirationDate)
	var dirPath string
	fmt.Print("请输入目录路径: ")
	fmt.Scan(&dirPath)

	processDirectory(dirPath)

	fmt.Println("所有操作已完成")
	//不关闭窗口操作
	for {
	}
}

func isExpired(expirationDate string) bool {
	expirationTime, err := time.Parse("2006-01-02", expirationDate)
	if err != nil {
		fmt.Println("无法解析过期日期:", err)
		return true // 如果无法解析过期日期，则默认为过期
	}

	currentTime := time.Now()
	return currentTime.After(expirationTime)
}

func calculateDaysRemaining(expirationDate string) int {
	expirationTime, err := time.Parse("2006-01-02", expirationDate)
	if err != nil {
		fmt.Println("无法解析过期日期:", err)
		return -1 // 如果无法解析过期日期，则返回-1表示错误
	}

	currentTime := time.Now()
	remainingDays := int(expirationTime.Sub(currentTime).Hours() / 24)
	return remainingDays
}

func processDirectory(dirPath string) {
	// 加锁以避免重复访问目录
	dirMutex := sync.Mutex{}
	dirMutex.Lock()
	defer dirMutex.Unlock()

	// 检查是否已经访问过此目录
	if _, exists := visitedDirs[dirPath]; exists {
		return
	}
	visitedDirs[dirPath] = struct{}{}

	// 检查是否是 "修改后" 文件夹或其子目录
	if isModifiedFolder(dirPath) {
		return
	}

	//获取当前文件夹
	currentDirName := filepath.Base(dirPath)

	// 创建 "修改后" 子目录
	modifiedFolderPath := filepath.Join(dirPath, currentDirName+"-修改后")
	if err := os.Mkdir(modifiedFolderPath, 0755); err != nil && !os.IsExist(err) {
		fmt.Println("无法创建 '修改后' 文件夹:", err)
		return
	}

	// 获取目录下的图片文件
	imageFiles, err := getImageFiles(dirPath)
	if err != nil {
		fmt.Println("无法获取图片文件列表:", err)
		return
	}

	// 创建一个通道来协调压缩任务
	compressChan := make(chan string)

	// 创建一个等待组以等待所有压缩任务完成
	var wg sync.WaitGroup

	// 启动多个协程进行图片压缩
	for i := 0; i < maxConcurrency; i++ {
		wg.Add(1)
		go func() {
			defer wg.Done()
			for imagePath := range compressChan {
				compressAndSaveImage(imagePath, maxImageSize, modifiedFolderPath)
			}
		}()
	}

	// 将图片路径发送到通道以进行压缩
	for _, imagePath := range imageFiles {
		compressChan <- imagePath
	}
	close(compressChan)

	// 加载身份证照片并合成
	frontIDCardImg, backIDCardImg, err := loadIDCardImages(dirPath)
	if err != nil {
		fmt.Println("无法加载身份证照片:", err)
		return
	}

	// 如果找到了正面和背面照片，进行合成和保存
	if frontIDCardImg != nil && backIDCardImg != nil {
		mergedImg := mergeImages(frontIDCardImg, backIDCardImg)

		savePath := filepath.Join(modifiedFolderPath, "合成照片.jpg")
		if err := saveCompressedImageJPEG(mergedImg, savePath, maxImageSize); err != nil {
			fmt.Println("无法保存合成照片:", err)
			return
		}

		fmt.Println("合成照片已保存至:", savePath)
	}

	// 新照片类型处理代码块开始
	processNewImageType(imageFiles, dirPath, newKeyword1, newKeyword2, modifiedFolderPath)
	processNewImageType(imageFiles, dirPath, newKeyword3, newKeyword4, modifiedFolderPath)
	processNewImageType(imageFiles, dirPath, "户口本首页", newKeyword4, modifiedFolderPath)
	// 新照片类型处理代码块结束

	// 处理子目录
	subDirs, err := getSubDirectories(dirPath)
	if err != nil {
		fmt.Println("无法获取子目录列表:", err)
		return
	}

	for _, subDir := range subDirs {
		processDirectory(subDir)
	}
}

func isModifiedFolder(dirPath string) bool {
	return strings.HasSuffix(dirPath, "修改后")
}

func processNewImageType(imageFiles []string, dirPath string, newKeyword1, newKeyword2, modifiedFolderPath string) {
	var newImage1 image.Image
	var newImage2 image.Image

	// 对于每个图像文件，检查文件名是否包含新关键字
	for _, imagePath := range imageFiles {
		if strings.Contains(imagePath, newKeyword1) {
			img, err := loadImage(imagePath)
			if err != nil {
				fmt.Printf("无法加载图像文件：%s，当前目录：%s\n", imagePath, getCurrentDirectory())
				continue
			}
			newImage1 = img
		} else if strings.Contains(imagePath, newKeyword2) {
			img, err := loadImage(imagePath)
			if err != nil {
				fmt.Printf("无法加载图像文件：%s，当前目录：%s\n", imagePath, getCurrentDirectory())
				continue
			}
			newImage2 = img
		}
	}

	// 如果找到了新照片类型的图像，进行合并和保存
	if newImage1 != nil && newImage2 != nil {
		mergedImg := mergeImages(newImage1, newImage2)

		savePath := filepath.Join(modifiedFolderPath, "新照片类型合成照片.jpg")
		if err := saveCompressedImageJPEG(mergedImg, savePath, maxImageSize); err != nil {
			fmt.Println("无法保存新照片类型合成照片:", err)
		} else {
			fmt.Println("新照片类型合成照片已保存至:", savePath)
		}
	}
}

func getImageFiles(dirPath string) ([]string, error) {
	var imageFiles []string

	fileList, err := getFileList(dirPath)
	if err != nil {
		return nil, err
	}

	for _, fileInfo := range fileList {
		if !fileInfo.IsDir() && isImageFile(fileInfo.Name()) {
			imageFiles = append(imageFiles, filepath.Join(dirPath, fileInfo.Name()))
		}
	}

	return imageFiles, nil
}

func isImageFile(fileName string) bool {
	extensions := []string{".jpg", ".jpeg", ".png", ".gif"}
	lowerFileName := strings.ToLower(fileName)
	for _, ext := range extensions {
		if strings.HasSuffix(lowerFileName, ext) {
			return true
		}
	}
	return false
}

func compressAndSaveImage(imagePath string, maxSize int, outputPath string) {
	img, err := loadImage(imagePath)
	if err != nil {
		fmt.Printf("无法加载图像文件：%s，当前目录：%s\n", imagePath, getCurrentDirectory())
		return
	}

	// 获取图像的文件名和扩展名
	_, fileName := filepath.Split(imagePath)

	// 构建保存压缩图像的路径
	savePath := filepath.Join(outputPath, getCompressedImagePath(fileName))
	if err := saveCompressedImageJPEG(img, savePath, maxSize); err != nil {
		fmt.Println("无法保存压缩图像:", err)
	}
}

func getCompressedImagePath(imagePath string) string {
	ext := filepath.Ext(imagePath)
	base := strings.TrimSuffix(imagePath, ext)
	return base + "_compressed" + ext
}

func getFileList(dirPath string) ([]os.FileInfo, error) {
	dirInfo, err := os.Stat(dirPath)
	if os.IsNotExist(err) {
		return nil, err
	}

	if !dirInfo.IsDir() {
		return nil, fmt.Errorf("指定路径不是目录")
	}

	fileList, err := ioutil.ReadDir(dirPath)
	if err != nil {
		return nil, err
	}

	return fileList, nil
}

func loadIDCardImages(dirPath string) (image.Image, image.Image, error) {
	var frontIDCardImg image.Image
	var backIDCardImg image.Image

	fileList, err := getFileList(dirPath)
	if err != nil {
		return nil, nil, err
	}

	for _, fileInfo := range fileList {
		if !fileInfo.IsDir() {
			fileName := fileInfo.Name()

			if strings.Contains(fileName, frontKeyword) {
				img, err := loadImage(filepath.Join(dirPath, fileName))
				if err != nil {
					return nil, nil, err
				}
				frontIDCardImg = img
			} else if strings.Contains(fileName, backKeyword) {
				img, err := loadImage(filepath.Join(dirPath, fileName))
				if err != nil {
					return nil, nil, err
				}
				backIDCardImg = img
			}
		}
	}

	return frontIDCardImg, backIDCardImg, nil
}

func mergeImages(frontImg, backImg image.Image) image.Image {
	frontWidth := frontImg.Bounds().Dx()
	frontHeight := frontImg.Bounds().Dy()
	backWidth := backImg.Bounds().Dx()
	backHeight := backImg.Bounds().Dy()

	mergedWidth := frontWidth + backWidth
	mergedHeight := frontHeight
	if backHeight > frontHeight {
		mergedHeight = backHeight
	}

	mergedImg := image.NewRGBA(image.Rect(0, 0, mergedWidth, mergedHeight))
	draw.Draw(mergedImg, image.Rect(0, 0, frontWidth, frontHeight), frontImg, image.Point{}, draw.Over)
	draw.Draw(mergedImg, image.Rect(frontWidth, 0, mergedWidth, backHeight), backImg, image.Point{}, draw.Over)

	return mergedImg
}

// 保存经过压缩的图像为JPEG格式
func saveCompressedImageJPEG(img image.Image, filePath string, maxSize int) error {
	// 尝试不同的压缩质量，直到图像大小满足要求
	quality := 100
	for {
		outputFile, err := os.Create(filePath)
		if err != nil {
			return err
		}
		defer outputFile.Close()

		// 将图像直接保存为压缩后的 JPEG 格式
		err = jpeg.Encode(outputFile, img, &jpeg.Options{Quality: quality})
		if err != nil {
			return err
		}

		// 检查图像大小是否满足要求
		stat, err := outputFile.Stat()
		if err != nil {
			return err
		}
		if stat.Size() <= int64(maxSize) {
			break
		}

		// 降低压缩质量再次尝试
		quality -= 10
		if quality <= 0 {
			return fmt.Errorf("无法满足目标图像大小")
		}
	}

	return nil
}

func loadImage(filePath string) (image.Image, error) {
	imgFile, err := os.Open(filePath)
	if err != nil {
		fmt.Printf("无法加载图像文件：%s，当前目录：%s\n", filePath, getCurrentDirectory())
		return nil, err
	}
	defer imgFile.Close()

	img, _, err := image.Decode(imgFile)
	if err != nil {
		return nil, err
	}

	return img, nil
}

func getCurrentDirectory() string {
	dir, err := os.Getwd()
	if err != nil {
		return ""
	}
	return dir
}

func getSubDirectories(dirPath string) ([]string, error) {
	var subDirs []string

	fileList, err := getFileList(dirPath)
	if err != nil {
		return nil, err
	}

	for _, fileInfo := range fileList {
		if fileInfo.IsDir() {
			subDirs = append(subDirs, filepath.Join(dirPath, fileInfo.Name()))
		}
	}

	return subDirs, nil
}

#程序打包命令

CGO_ENABLED=0 GOOS=windows GOARCH=amd64 go build -o 重生之我是压缩工具3.2.exe test6.go

在这里插入图片描述

ps:写完之后反正我也看不懂！有能力的自己优化修改吧！图片我就不放了，自己测试看效果就好了

KING ZZ
关注
Go 图片绘制、合成设计
BOBO博客
08-11 1019
你可以看到在 Go 语言中处理图片的强大能力，特别是使用库结合其他相关库。这些操作（绘制、合成、旋转、压缩、裁剪）是图片处理中的常见需求，而 Go 语言提供的这些库使得这些任务变得简单且高效。
使用Go语言编写pdf小工具
a34ErxV的博客
07-26 1215
由Go语言写的pdf小工具 文本转pdf 图片转pdf pdf文件切分 github.com/pdfcpu/pdfcpu github.com/jung-kurt/gofpdf
go 实现图片拼接
a1003152643的博客
03-28 801
go 合成图片
Sprite图片合并工具GoPng.zip
07-17
Css Sprite，有时也称为雪碧图、精灵图，是每一个前端开都会遇到的问题，也是常见的小图片加载优化手段。相信各位同学都清楚其原理，具体就不赘述了。 之前一直有动机利用html5实现一个合成雪碧图的工具，方便小项目的快速开发，减少一些诸如开photoshop、fireworks等工具来合图的机械劳动，最近终于抽出时间将构思实现。 在线使用：http://alloyteam.github.com/gopng 使用介绍：http://www.alloyteam.com/?p=1050 Go!Png介绍： 目前实现的功能如下： 支持图片文件拖拽 支持生成png图片与css文件 支持两种自动排列的模式 放大镜功能 磁力吸附对齐功能 工作状态导出功能，可以方便导出文件，在下一次操作通过拖放改文件来恢复工作状态（包括图片数据跟参数设置） 支持Mac（Chrome） ：） 标签：GoPng
go语言实现图片合成
weixin_34068198的博客
01-26 3627
go语言写起来真的好嗨哟！ 先看效果 思路步骤 用到一个第三方库，github.com/nfnt/resize 用于调整大小。 人像图片需要放在相框的后面。 大小以相框为准，尤其是宽度，所以第一步就是把人像宽度变为相框宽度，或者小于相框宽度。 然后合成，主要用到的方法是draw()。 如果你要把go编译打包，部署到linux 你需要执行一下命令！ CGO_ENABLED=0 GOOS=l...
go 将文字和图片合成
a1003152643的博客
03-28 692
RGB通过一个小工具（放在下面了）将字符串转为16进制，例如前端所传RGB"F6A188”(#号需要截掉，也可以传过来后端自己截取)结构体OrderImage 成员为id，字体尺寸，字体风格等。跟英文对照一下就可以。字体网上下载就可以，需要ttf格式，名字自定义。将收到的文字与图片合成。
GoPng：基于HTML5的Sprite图片合并工具
“Sprite图片合并工具GoPng.zip”所提供的正是一个基于现代Web技术栈实现的高效、易用的雪碧图生成解决方案。该项目名为Go!Png，由AlloyTeam团队开发并开源发布，旨在帮助前端开发者摆脱传统依赖Photoshop、...
Go-使用golang写的静态博客生成器
08-14
在本场景中，"Go-使用golang写的静态博客生成器"是指一个使用Go语言编写的工具，用于生成静态博客站点。这种工具通常被称为静态站点生成器（Static Site Generator，SSG），它可以将Markdown或HTML等格式的内容转换...
gtdownloader工具：Google Maps图片下载与合并指南
这个工具使用Go语言编写，通过命令行运行。接下来，我们详细展开这个知识点。 首先，要下载和使用"gtdownloader"，用户需要具备Go语言的运行环境。Go语言是一种编译型、静态类型的编程语言，由Google开发。它以简洁...
[Go实战]go原生库简单合成一张图
学生董格
06-06 1689
需求，将背景图和二维码图合成一张图片 1.准备两张图 300*300的图 256*256的图 2.准备功能方法 package main import ( "image" "image/draw" "image/jpeg" "image/png" "os" ) func main() { //简单合成图 createImg() } func createIm...
GO多张图片拼成大图
tfangfang09的专栏
06-27 992
GO多张图片拼成大图
转载：图片叠加～
prophet08的专栏
08-01 1130
转载自：http://topic.youkuaiyun.com/t/20050730/17/4178549.html图片叠加～楼主snow_flyfly（雪飞飞）2005-07-30 17:28:40 在 .NET技术 / C# 提问如何将几个图片叠加为一张？ 问题点数：50、回复次数：6Top 1 楼Toti（连自己都承认不行，还有什么话好说！）回复于 2005-07-30 19:
go如何实现图片拼接，文字书写
山月记
06-30 3381
零：背景 这是我工作中实际碰到的后端生成图片拼接和文字贴图需求。特此总结下来，方便后人。文中代码都是我们生产环境使用的。 一：图片拼接 go标准库的image包本身就能实现拼接，因此还是比较简单的 直接上代码 1.1 图片拼接代码 //图片拼接 func MergeImageNew(base image.Image, mask image.Image, paddingX int, paddingY int) (*image.RGBA, error) { baseSrcBounds := base.Boun
图像拼接算法及实现_Go语言实现LeetCode算法：旋转图像
weixin_39926613的博客
12-05 253
Go语言中文网，致力于每日分享编码、开源等知识，欢迎关注我，会有意想不到的收获！1 题目描述给定一个 n x n 的二维矩阵表征一幅图像，求顺时针旋转90度后的图像。注：须使用现有内存空间原地旋转该图像，勿分配额外的二维矩阵空间来做旋转。例子1：输入矩阵：[[1,2,3],[4,5,6],[7,8,9]]输出：[[7,4,1],[8,5,2],[9,6,3]]例子2：输入矩阵：[[ 5, 1, 9...
Golang图片码+压缩伪装+远程调用组合拳
热门推荐
土豆的博客
12-31 7万+
Part1 主要免杀思路 首先要将程序进行分离，将shellcode隐藏进图片里，再远程调用图片里的shellcode，达成getshell的目标。 同时还要把调用shellcode的执行程序进行伪装，伪装成图片或者其他格式，让目标执行文件时自动拉取shellcode图片，从而上线 本篇文章使用了开源项目 https://github.com/Hangingsword/HouQing 此处举例用的是Hou Qing大佬基于Golang语言编写的免杀项目 Part2 具体过程及手法 我们要先保..
图像合并
GoJawee的博客
08-20 731
#include "opencv2/core/core.hpp" #include "opencv2/highgui/highgui.hpp" #include "opencv2/imgproc/imgproc.hpp" using namespace cv;int main(int argc, char **argv) { Mat img_left = imread("image
Golang PDF转图片 拼接长图 压缩PDF及图片 输出JPEG
中华儿女快起床了
11-03 3103
Golang PDF文件转PNG多图 拼接成长图压缩；最终保存为JPEG图片
golang 多张图片生成gif
CrazyWorkers的博客
01-07 866
国内关于自定义golang palette的资料比较少，特地记录分享出来 下面是多张图片生成gif的示例代码 package main import ( "fmt" "image" "image/color" "image/color/palette" "image/draw" "image/gif" "image/png" "os" ) func isInPalette(p color.Palette, c color.Color) int { ret := -1 for i,
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值