update 2021年07月23日 18:29:34

Author: No-Github

README.md

@@ -0,0 +1,9 @@
+给文件夹内图片添加水印的工具
+
+源码来自 : https://blog.csdn.net/czy279470138/article/details/96993285
+
+仅作部分修改
+
+```
+./main.exe -name "1.png"
+```

logo.png

@@ -0,0 +1,166 @@
+package main
+import (
+	"flag"
+	"fmt"
+	"image"
+	"image/draw"
+	"image/jpeg"
+	"image/png"
+	"log"
+	"net/http"
+	"os"
+)
+
+var Input_name = flag.String("name", "null", "输入文件名")
+
+func main() {
+	flag.Parse()
+
+	for i := 0; i != flag.NArg(); i++ {
+		fmt.Printf("arg[%d]=%s\n", i, flag.Arg(i))
+	}
+
+	if *Input_name == ""{
+		fmt.Println("文件夹?")
+	}else {
+		fmt.Println("目标图片:",*Input_name)
+		imgwater()
+	}
+
+}
+
+func imgwater() {
+	srcImg := *Input_name             // 原始图片
+	imgWaterMarkPath := "./logo.png" // 水印图片
+
+	// 原始图片
+	originalImg, err := os.Open(srcImg)
+	if err != nil {
+		fmt.Println("打开原始图片出错")
+	}
+	//
+	defer originalImg.Close()
+
+	// 水印图片
+	waterMark, err := os.Open(imgWaterMarkPath)
+	if err != nil {
+		fmt.Println("打开水印图片出错")
+	}
+	defer waterMark.Close()
+
+	waterMarkImg, err := png.Decode(waterMark)
+	if err != nil {
+		fmt.Println("把水印图片解码为结构体时出错")
+	}
+
+	buff := make([]byte, 512)
+
+	file, err := os.Open(srcImg)
+	if err != nil {
+		log.Println(err)
+	}
+
+	_, err = file.Read(buff)
+	if err != nil {
+		log.Println(err)
+	}
+
+	imgType := http.DetectContentType(buff)
+
+	if imgType == "image/jpeg" {
+		fmt.Println("这是JPG文件")
+		imgJpeg, err := jpeg.Decode(originalImg)
+		if err != nil {
+			fmt.Println("把jpeg图片解码为结构体时出错")
+		}
+
+		b := imgJpeg.Bounds()
+		waterMarkWidth := b.Max.X
+		waterMarkHeight := b.Max.Y
+		fmt.Println("jpeg原始图片宽", waterMarkWidth, "jpeg原始图片高", waterMarkHeight)
+
+		m := image.NewRGBA(b)
+
+		// 原始图片
+		draw.Draw(m, b, imgJpeg, image.ZP, draw.Src)
+		// 水印图片
+		for i := 0; i < waterMarkWidth; i++ {
+			offsetWidth := 1000 * i
+			if offsetWidth < waterMarkWidth {
+				for j := 0; j < waterMarkHeight; j++ {
+					offsetHeight := 1000 * j
+					if offsetHeight < waterMarkHeight {
+						offset := image.Pt(offsetWidth, offsetHeight)
+						draw.Draw(m, waterMarkImg.Bounds().Add(offset), waterMarkImg, image.ZP, draw.Over)
+						fmt.Println("jpegOffset", offset)
+					}
+				}
+			}
+		}
+
+		// 生成新图片new.jpg,并设置图片质量 100
+		imgNew, err := os.Create("new.jpg") // 这里可以设置为上传图片 srcImg
+		if err != nil {
+			log.Println(err)
+		}
+
+		err = jpeg.Encode(imgNew, m, &jpeg.Options{100})
+		if err != nil {
+			log.Println(err)
+		}
+		// png.Encode(imgw, m)
+		defer imgNew.Close()
+
+		fmt.Println("添加JPG水印图片结束请查看")
+	}
+
+	if imgType == "image/png" {
+		fmt.Println("这是PNG文件")
+		imgPng, err := png.Decode(originalImg)
+		if err != nil {
+			fmt.Println("把PNG图片解码为结构体时出错")
+		}
+		b := imgPng.Bounds()
+		waterMarkWidth := b.Max.X
+		waterMarkHeight := b.Max.Y
+		fmt.Println("png原始图片宽", waterMarkWidth, "png原始图片高", waterMarkHeight)
+
+		m := image.NewRGBA(b)
+
+		// 原始图片
+		draw.Draw(m, b, imgPng, image.ZP, draw.Src)
+
+		// 水印图片
+		for i := 0; i < waterMarkWidth; i++ {
+			// resWidth := 0
+			offsetWidth := 1000 * i
+			// fmt.Println(offsetWidth)
+			if offsetWidth < waterMarkWidth {
+				for j := 0; j < waterMarkHeight; j++ {
+					offsetHeight := 1000 * j
+					if offsetHeight < waterMarkHeight {
+						offset := image.Pt(offsetWidth, offsetHeight)
+						draw.Draw(m, waterMarkImg.Bounds().Add(offset), waterMarkImg, image.ZP, draw.Over)
+						fmt.Println("pngOffset", offset)
+					}
+				}
+			}
+		}
+
+		// 生成新图片new.png
+		imgNew, err := os.Create("./new.png") // 这里可以设置为上传图片 srcImg
+		if err != nil {
+			log.Println(err)
+		}
+
+		err = png.Encode(imgNew, m)
+		if err != nil {
+			log.Println(err)
+		}
+		defer imgNew.Close()
+		fmt.Println("添加PNG水印图片结束请查看")
+	}
+	if imgType == "image/gif" {
+		fmt.Println("暂不支持 gif 格式。。。")
+	}
+}

main.go

@@ -0,0 +1,4 @@
+
+git add -A
+git commit -m "update $(date)"
+git push