409 lines
16 KiB
Java
409 lines
16 KiB
Java
package com.point.strategy.common;
|
|
|
|
|
|
import javax.imageio.ImageIO;
|
|
import javax.swing.*;
|
|
import java.awt.*;
|
|
import java.awt.image.BufferedImage;
|
|
import java.io.*;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import org.apache.pdfbox.pdmodel.PDDocument;
|
|
import org.apache.pdfbox.pdmodel.PDPage;
|
|
import org.apache.pdfbox.pdmodel.PDPageContentStream;
|
|
import org.apache.pdfbox.pdmodel.font.PDFont;
|
|
import org.apache.pdfbox.pdmodel.font.PDType0Font;
|
|
import org.apache.pdfbox.pdmodel.graphics.color.PDColor;
|
|
import org.apache.pdfbox.pdmodel.graphics.color.PDDeviceRGB;
|
|
import org.apache.pdfbox.pdmodel.graphics.state.PDExtendedGraphicsState;
|
|
import java.io.ByteArrayOutputStream;
|
|
|
|
import java.awt.Color;
|
|
|
|
public class WatermarkImgUtils {
|
|
|
|
//水印颜色
|
|
private static Color markContentColor = Color.lightGray;
|
|
//水印字体，大小
|
|
private static Font font = new Font("宋体", Font.BOLD, 24);
|
|
//设置水印文字的旋转角度
|
|
private static Integer degree = 45;
|
|
//设置水印透明度
|
|
private static float alpha = 0.4f;
|
|
// 水印之间的间隔
|
|
private static final int XMOVE = 120;
|
|
// 水印之间的间隔
|
|
private static final int YMOVE = 160;
|
|
|
|
public static void main(String[] args) {
|
|
|
|
addWatermarkByFile("C:\\Users\\yironghui\\Desktop\\0001.png", "文字水印");
|
|
// addImageWatermarkByFile("D:\\text\\水印\\图片水印测试\\panoRes\\pano","D:\\text\\5.png");
|
|
|
|
}
|
|
|
|
/**
|
|
* 批量处理图片添加文字水印-平铺
|
|
* @param sourceImgPath 源图片路径 / 文件夹路径
|
|
* @param waterMarkContent 水印内容
|
|
*/
|
|
public static boolean addWatermarkByFile(String sourceImgPath, String waterMarkContent) {
|
|
//获取路径
|
|
File sourceImgFile = new File(sourceImgPath);
|
|
if (!sourceImgFile.exists()) {
|
|
System.out.println("文件路径错误：" + sourceImgFile.getPath());
|
|
return false;
|
|
} else {
|
|
if (sourceImgFile.isFile()){
|
|
addWatermark(sourceImgFile.getPath(), sourceImgFile.getPath(), waterMarkContent);
|
|
}else {
|
|
List<String> proList=getTotalSizeOfFilesInDir(sourceImgFile);
|
|
for (String s : proList) {
|
|
File f = new File(s);
|
|
if (!"thumb.jpg".equals(f.getName()) && !"preview.jpg".equals(f.getName())){
|
|
boolean b=addWatermark(s, s, waterMarkContent);
|
|
System.out.println(b+"\t"+s);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* 批量处理图片添加图片水印-平铺
|
|
* @param sourceImagePath
|
|
* @param waterImagePath
|
|
* @return
|
|
*/
|
|
public static boolean addImageWatermarkByFile(String sourceImagePath, String waterImagePath) {
|
|
//获取路径
|
|
File sourceImgFile = new File(sourceImagePath);
|
|
if (!sourceImgFile.exists()) {
|
|
System.out.println("文件路径错误：" + sourceImgFile.getPath());
|
|
return false;
|
|
} else {
|
|
if (sourceImgFile.isFile()){
|
|
addImageWatermark(new File(sourceImagePath),new File(waterImagePath));
|
|
}else {
|
|
List<String> proList=getTotalSizeOfFilesInDir(sourceImgFile);
|
|
for (String s : proList) {
|
|
File f = new File(s);
|
|
if (!"thumb.jpg".equals(f.getName()) && !"preview.jpg".equals(f.getName())){
|
|
boolean b=addImageWatermark(new File(s),new File(waterImagePath));
|
|
System.out.println(b+"\t"+s);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
|
|
/**
|
|
* 递归方式 查询全部文件
|
|
* @param file
|
|
* @return
|
|
*/
|
|
private static List<String> getTotalSizeOfFilesInDir(final File file) {
|
|
List<String> asList=new ArrayList<>();
|
|
if (file.isFile()) {
|
|
asList.add(file.getPath());
|
|
return asList;
|
|
}
|
|
final File[] children = file.listFiles();
|
|
if (children != null) {
|
|
for (final File child : children){
|
|
List<String> daList=getTotalSizeOfFilesInDir(child);
|
|
for (String s : daList) {
|
|
asList.add(s);
|
|
}
|
|
}
|
|
}
|
|
return asList;
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
* 添加文字水印平铺
|
|
* @description
|
|
* @param sourceImgPath 源图片路径
|
|
* @param tarImgPath 保存的图片路径
|
|
* @param waterMarkContent 水印内容
|
|
* @return void
|
|
*/
|
|
public static boolean addWatermark(String sourceImgPath, String tarImgPath, String waterMarkContent){
|
|
|
|
OutputStream outImgStream = null;
|
|
try {
|
|
File srcImgFile = new File(sourceImgPath);//得到文件
|
|
Image srcImg = ImageIO.read(srcImgFile);//文件转化为图片
|
|
int srcImgWidth = srcImg.getWidth(null);//获取图片的宽
|
|
int srcImgHeight = srcImg.getHeight(null);//获取图片的高
|
|
// 加水印
|
|
BufferedImage bufImg = new BufferedImage(srcImgWidth, srcImgHeight, BufferedImage.TYPE_INT_RGB);
|
|
Graphics2D g = bufImg.createGraphics();//得到画笔
|
|
g.drawImage(srcImg, 0, 0, srcImgWidth, srcImgHeight, null);
|
|
g.setColor(markContentColor); //设置水印颜色
|
|
g.setFont(font); //设置字体
|
|
g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, alpha));//设置水印文字透明度
|
|
if (null != degree) {
|
|
g.rotate(Math.toRadians(degree));//设置水印旋转
|
|
}
|
|
JLabel label = new JLabel(waterMarkContent);
|
|
FontMetrics metrics = label.getFontMetrics(font);
|
|
int width = metrics.stringWidth(label.getText())+XMOVE;//文字水印的宽
|
|
int rowsNumber = srcImgHeight/width;// 图片的高 除以 文字水印的宽 ——> 打印的行数(以文字水印的宽为间隔)
|
|
int columnsNumber = srcImgWidth/width;//图片的宽 除以 文字水印的宽 ——> 每行打印的列数(以文字水印的宽为间隔)
|
|
/*//防止图片太小而文字水印太长，所以至少打印一次
|
|
if(rowsNumber < 1){
|
|
rowsNumber = 1;
|
|
}
|
|
if(columnsNumber < 1){
|
|
columnsNumber = 1;
|
|
}*/
|
|
for(int j=0;j<rowsNumber;j++){
|
|
for(int i=0;i<columnsNumber;i++){
|
|
g.drawString(waterMarkContent, i*width + j*width, -i*width + j*width+YMOVE);//画出水印,并设置水印位置
|
|
}
|
|
}
|
|
g.dispose();// 释放资源
|
|
// 输出图片
|
|
outImgStream = new FileOutputStream(tarImgPath);
|
|
//获取到文件的后缀名
|
|
String fileName = srcImgFile.getName();
|
|
String formatName = fileName.substring(fileName.lastIndexOf(".") + 1);
|
|
//覆盖原图
|
|
ImageIO.write(bufImg, formatName, outImgStream);
|
|
return true;
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
e.getMessage();
|
|
return false;
|
|
} finally{
|
|
try {
|
|
if(outImgStream != null){
|
|
outImgStream.flush();
|
|
outImgStream.close();
|
|
}
|
|
return true;
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
e.getMessage();
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 添加图片水印平铺
|
|
* @param srcImageFile 目标图片
|
|
* @param waterImageFile 水印图片
|
|
* @return
|
|
*/
|
|
public static boolean addImageWatermark(File srcImageFile, File waterImageFile) {
|
|
try {
|
|
|
|
Image srcImg = ImageIO.read(srcImageFile);//文件转化为图片
|
|
int srcImgWidth = srcImg.getWidth(null);//获取图片的宽
|
|
int srcImgHeight = srcImg.getHeight(null);//获取图片的高
|
|
|
|
Image waterImage = ImageIO.read(waterImageFile);//文件转化为图片
|
|
int waterImageWidth = waterImage.getWidth(null);//获取水印图片的宽
|
|
|
|
//使用ImageIO的read方法读取图片
|
|
BufferedImage read = ImageIO.read(srcImageFile);
|
|
BufferedImage image = ImageIO.read(waterImageFile);
|
|
//获取画布
|
|
Graphics2D graphics = read.createGraphics();
|
|
//设置透明度为0.5
|
|
graphics.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP,alpha));
|
|
|
|
if (waterImageWidth<100){
|
|
waterImageWidth=200;
|
|
}
|
|
waterImageWidth+=XMOVE;
|
|
int rowsNumber = srcImgHeight/waterImageWidth;// 图片的高 除以 文字水印的宽 ——> 打印的行数(以文字水印的宽为间隔)
|
|
int columnsNumber = srcImgWidth/waterImageWidth;//图片的宽 除以 文字水印的宽 ——> 每行打印的列数(以文字水印的宽为间隔)
|
|
/*//防止图片太小而文字水印太长，所以至少打印一次
|
|
if(rowsNumber < 1){
|
|
rowsNumber = 1;
|
|
}
|
|
if(columnsNumber < 1){
|
|
columnsNumber = 1;
|
|
}*/
|
|
for(int j=0;j<rowsNumber;j++){
|
|
for(int i=0;i<columnsNumber;i++){
|
|
int x=i*waterImageWidth + j*waterImageWidth;
|
|
int y=-i*waterImageWidth + j*waterImageWidth+YMOVE;
|
|
//添加水印
|
|
graphics.drawImage(image,x,y,null);
|
|
}
|
|
}
|
|
//关闭透明度
|
|
//graphics.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER));
|
|
graphics.dispose();
|
|
//获取到文件的后缀名
|
|
String fileName = srcImageFile.getName();
|
|
String formatName = fileName.substring(fileName.lastIndexOf(".") + 1);
|
|
//使用ImageIO的write方法进行输出-覆盖原图
|
|
ImageIO.write(read,formatName,srcImageFile);
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
}
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* 批量处理图片添加文字水印-平铺
|
|
* @param sourceImgPath 源图片路径 / 文件夹路径
|
|
* @param waterMarkContent 水印内容
|
|
*/
|
|
public static ByteArrayOutputStream addWatermarkByFileIo(String sourceImgPath, String waterMarkContent) {
|
|
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
|
//获取路径
|
|
File sourceImgFile = new File(sourceImgPath);
|
|
if (!sourceImgFile.exists()) {
|
|
System.out.println("文件路径错误：" + sourceImgFile.getPath());
|
|
return outputStream;
|
|
} else {
|
|
if (sourceImgFile.isFile()){
|
|
outputStream = addWatermarkIo(sourceImgFile.getPath(), waterMarkContent);
|
|
}else {
|
|
List<String> proList=getTotalSizeOfFilesInDir(sourceImgFile);
|
|
for (String s : proList) {
|
|
File f = new File(s);
|
|
if (!"thumb.jpg".equals(f.getName()) && !"preview.jpg".equals(f.getName())){
|
|
outputStream = addWatermarkIo(s, waterMarkContent);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return outputStream;
|
|
}
|
|
|
|
/**
|
|
* 添加文字水印平铺
|
|
* @description
|
|
* @param sourceImgPath 源图片路径
|
|
* @param tarImgPath 保存的图片路径
|
|
* @param waterMarkContent 水印内容
|
|
* @return ByteArrayOutputStream 输出流
|
|
*/
|
|
public static ByteArrayOutputStream addWatermarkIo(String sourceImgPath, String waterMarkContent) {
|
|
|
|
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
|
OutputStream outImgStream = null;
|
|
|
|
try {
|
|
File srcImgFile = new File(sourceImgPath); // 获取文件
|
|
Image srcImg = ImageIO.read(srcImgFile); // 将文件转化为图片
|
|
int srcImgWidth = srcImg.getWidth(null); // 获取图片的宽
|
|
int srcImgHeight = srcImg.getHeight(null); // 获取图片的高
|
|
|
|
// 添加水印
|
|
BufferedImage bufImg = new BufferedImage(srcImgWidth, srcImgHeight, BufferedImage.TYPE_INT_RGB);
|
|
Graphics2D g = bufImg.createGraphics(); // 获取画笔
|
|
g.drawImage(srcImg, 0, 0, srcImgWidth, srcImgHeight, null);
|
|
g.setColor(markContentColor); // 设置水印颜色
|
|
g.setFont(font); // 设置字体
|
|
g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, alpha)); // 设置水印文字透明度
|
|
|
|
if (null != degree) {
|
|
g.rotate(Math.toRadians(degree)); // 设置水印旋转
|
|
}
|
|
|
|
JLabel label = new JLabel(waterMarkContent);
|
|
FontMetrics metrics = label.getFontMetrics(font);
|
|
int width = metrics.stringWidth(label.getText()) + XMOVE; // 文字水印的宽
|
|
int rowsNumber = srcImgHeight / width; // 图片的高 除以 文字水印的宽 ——> 打印的行数(以文字水印的宽为间隔)
|
|
int columnsNumber = srcImgWidth / width; // 图片的宽 除以 文字水印的宽 ——> 每行打印的列数(以文字水印的宽为间隔)
|
|
|
|
for (int j = 0; j < rowsNumber; j++) {
|
|
for (int i = 0; i < columnsNumber; i++) {
|
|
g.drawString(waterMarkContent, i * width + j * width, -i * width + j * width + YMOVE); // 画出水印,并设置水印位置
|
|
}
|
|
}
|
|
|
|
g.dispose(); // 释放资源
|
|
|
|
ImageIO.write(bufImg, "png", outputStream);
|
|
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
e.getMessage();
|
|
} finally {
|
|
try {
|
|
if (outImgStream != null) {
|
|
outImgStream.flush();
|
|
outImgStream.close();
|
|
}
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
e.getMessage();
|
|
}
|
|
}
|
|
|
|
return outputStream;
|
|
}
|
|
|
|
public static ByteArrayOutputStream addWatermarkIo1(String sourceImgPath, String waterMarkContent) {
|
|
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
|
OutputStream outImgStream = null;
|
|
|
|
try {
|
|
File srcImgFile = new File(sourceImgPath);
|
|
Image srcImg = ImageIO.read(srcImgFile);
|
|
int srcImgWidth = srcImg.getWidth(null);
|
|
int srcImgHeight = srcImg.getHeight(null);
|
|
|
|
BufferedImage bufImg = new BufferedImage(srcImgWidth, srcImgHeight, BufferedImage.TYPE_INT_RGB);
|
|
Graphics2D g = bufImg.createGraphics();
|
|
g.drawImage(srcImg, 0, 0, srcImgWidth, srcImgHeight, null);
|
|
// 设置水印颜色
|
|
g.setColor(Color.BLACK);
|
|
// 设置字体
|
|
g.setFont(new Font("Arial", Font.PLAIN, 20));
|
|
// 设置水印透明度
|
|
g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, 0.5f));
|
|
|
|
JLabel label = new JLabel(waterMarkContent);
|
|
FontMetrics metrics = label.getFontMetrics(g.getFont());
|
|
int width = metrics.stringWidth(label.getText());
|
|
int rowsNumber = srcImgHeight / width;
|
|
int columnsNumber = srcImgWidth / width;
|
|
|
|
for (int j = 0; j < rowsNumber; j++) {
|
|
for (int i = 0; i < columnsNumber; i++) {
|
|
g.drawString(waterMarkContent, i * width + j * width, -i * width + j * width + 20);
|
|
}
|
|
}
|
|
|
|
g.dispose();
|
|
|
|
ImageIO.write(bufImg, "png", outputStream);
|
|
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
} finally {
|
|
try {
|
|
if (outImgStream != null) {
|
|
outImgStream.flush();
|
|
outImgStream.close();
|
|
}
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
return outputStream;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|