init new
This commit is contained in:
408
src/main/java/com/point/strategy/common/WatermarkImgUtils.java
Normal file
408
src/main/java/com/point/strategy/common/WatermarkImgUtils.java
Normal file
@@ -0,0 +1,408 @@
|
||||
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;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user