|
|
package xyz.wbsite.jmacro.tool;
|
|
|
|
|
|
import cn.hutool.core.img.ImgUtil;
|
|
|
import cn.hutool.core.io.FileUtil;
|
|
|
import cn.hutool.core.util.StrUtil;
|
|
|
import cn.hutool.log.StaticLog;
|
|
|
import xyz.wbsite.jmacro.base.Legend;
|
|
|
import xyz.wbsite.jmacro.util.DialogUtil;
|
|
|
import xyz.wbsite.jmacro.util.Logger;
|
|
|
|
|
|
import javax.swing.*;
|
|
|
import javax.swing.filechooser.FileNameExtensionFilter;
|
|
|
import java.awt.*;
|
|
|
import java.awt.event.KeyAdapter;
|
|
|
import java.awt.event.KeyEvent;
|
|
|
import java.awt.event.MouseAdapter;
|
|
|
import java.awt.event.MouseEvent;
|
|
|
import java.awt.event.MouseMotionAdapter;
|
|
|
import java.awt.event.WindowAdapter;
|
|
|
import java.awt.event.WindowEvent;
|
|
|
import java.awt.image.BufferedImage;
|
|
|
import java.io.File;
|
|
|
import java.util.List;
|
|
|
import java.util.concurrent.atomic.AtomicBoolean;
|
|
|
import java.util.prefs.Preferences;
|
|
|
|
|
|
/**
|
|
|
* 辅助截屏工具
|
|
|
*
|
|
|
* @author wangbing
|
|
|
* @version 0.0.1
|
|
|
* @since 1.8
|
|
|
*/
|
|
|
public class PickLegend extends JFrame {
|
|
|
|
|
|
private static final String LAST_PATH = "capture.last.path";
|
|
|
|
|
|
/**
|
|
|
* 参照原点x坐标
|
|
|
*/
|
|
|
private int originX;
|
|
|
/**
|
|
|
* 参照原点y坐标
|
|
|
*/
|
|
|
private int originY;
|
|
|
|
|
|
/**
|
|
|
* 全屏幕截图
|
|
|
*/
|
|
|
private BufferedImage capture;
|
|
|
|
|
|
/**
|
|
|
* 背景蒙层
|
|
|
*/
|
|
|
private final Color mask = new Color(0, 0, 0, 0.2f);
|
|
|
|
|
|
/**
|
|
|
* 聚焦框线条
|
|
|
*/
|
|
|
private final Stroke focusWindow = new BasicStroke(1.0f);
|
|
|
|
|
|
/**
|
|
|
* 屏幕截图区域起始坐标
|
|
|
*/
|
|
|
private final Point start = new Point(0, 0);
|
|
|
|
|
|
/**
|
|
|
* 屏幕截图区域结束坐标
|
|
|
*/
|
|
|
private final Point end = new Point(0, 0);
|
|
|
|
|
|
/**
|
|
|
* 个人首选项配置工具
|
|
|
*/
|
|
|
private Preferences preferences;
|
|
|
|
|
|
/**
|
|
|
* 有参照原点的构造器
|
|
|
*
|
|
|
* @param originX 原点x
|
|
|
* @param originY 原点y
|
|
|
* @throws AWTException
|
|
|
*/
|
|
|
public PickLegend(int originX, int originY, File path) throws AWTException {
|
|
|
this();
|
|
|
this.originX = originX;
|
|
|
this.originY = originY;
|
|
|
this.preferences.put(LAST_PATH, path.isDirectory() ? path.getAbsolutePath() : path.getParent());
|
|
|
}
|
|
|
|
|
|
public PickLegend() throws AWTException {
|
|
|
init();
|
|
|
}
|
|
|
|
|
|
private void init() throws AWTException {
|
|
|
setExtendedState(Frame.MAXIMIZED_BOTH);
|
|
|
setLocation(0, 0);//位置
|
|
|
setUndecorated(true);
|
|
|
setAlwaysOnTop(true);
|
|
|
setBackground(mask);
|
|
|
setCursor(Cursor.getPredefinedCursor(Cursor.CROSSHAIR_CURSOR));
|
|
|
|
|
|
// 初始化偏好设置
|
|
|
preferences = Preferences.userRoot().node(this.getClass().getName());
|
|
|
|
|
|
// 获取屏幕截图
|
|
|
Robot robot = new Robot();
|
|
|
Toolkit tk = Toolkit.getDefaultToolkit();
|
|
|
capture = robot.createScreenCapture(new Rectangle(0, 0, tk.getScreenSize().width, tk.getScreenSize().height));
|
|
|
|
|
|
// 监听窗口关闭
|
|
|
addWindowListener(new WindowAdapter() {
|
|
|
@Override
|
|
|
public void windowClosing(WindowEvent e) {
|
|
|
// 关闭应用时要释放资源
|
|
|
dispose();
|
|
|
System.exit(0);//0正常退出，1非正常退出
|
|
|
}
|
|
|
});
|
|
|
|
|
|
addMouseListener(new MouseAdapter() {
|
|
|
@Override
|
|
|
public void mousePressed(MouseEvent e) {
|
|
|
super.mousePressed(e);
|
|
|
start.setLocation(e.getX(), e.getY());
|
|
|
end.setLocation(e.getX(), e.getY());
|
|
|
Logger.info("起始坐标：{}", start);
|
|
|
repaint();
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
public void mouseReleased(MouseEvent e) {
|
|
|
super.mouseReleased(e);
|
|
|
end.setLocation(e.getX(), e.getY());
|
|
|
Logger.info("结束坐标：{}", end);
|
|
|
save();
|
|
|
}
|
|
|
});
|
|
|
|
|
|
addMouseMotionListener(new MouseMotionAdapter() {
|
|
|
@Override
|
|
|
public void mouseDragged(MouseEvent e) {
|
|
|
super.mouseDragged(e);
|
|
|
end.setLocation(e.getX(), e.getY());
|
|
|
repaint();
|
|
|
}
|
|
|
});
|
|
|
|
|
|
addKeyListener(new KeyAdapter() {
|
|
|
|
|
|
@Override
|
|
|
public void keyReleased(KeyEvent e) {
|
|
|
switch (e.getKeyCode()) {
|
|
|
case KeyEvent.VK_ESCAPE: {
|
|
|
StaticLog.info("exit.");
|
|
|
close();
|
|
|
}
|
|
|
break;
|
|
|
}
|
|
|
|
|
|
}
|
|
|
});
|
|
|
|
|
|
setVisible(true);
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
public void doLayout() {
|
|
|
super.doLayout();
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
public void paint(Graphics g) {
|
|
|
super.paint(g);
|
|
|
Graphics2D g2d = (Graphics2D) g;
|
|
|
g2d.drawImage(capture, 0, 0, capture.getWidth(), capture.getHeight(), this);
|
|
|
|
|
|
// 在背景图上加个蒙层
|
|
|
g2d.setColor(mask);
|
|
|
g2d.fillRect(0, 0, capture.getWidth(), capture.getHeight());
|
|
|
|
|
|
int w = getCaptureWidth();
|
|
|
int h = getCaptureHeight();
|
|
|
if (w > 0 && h > 0) {
|
|
|
g2d.setStroke(focusWindow);
|
|
|
g2d.setColor(Color.green);
|
|
|
g2d.drawRect(getCaptureX(), getCaptureY(), w, h);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
public int getCaptureX() {
|
|
|
return (int) Math.min(start.getX(), end.getX());
|
|
|
}
|
|
|
|
|
|
public int getCaptureY() {
|
|
|
return (int) Math.min(start.getY(), end.getY());
|
|
|
}
|
|
|
|
|
|
public int getCaptureWidth() {
|
|
|
return (int) Math.abs(start.getX() - end.getX());
|
|
|
}
|
|
|
|
|
|
public int getCaptureHeight() {
|
|
|
return (int) Math.abs(start.getY() - end.getY());
|
|
|
}
|
|
|
|
|
|
public void save() {
|
|
|
// 上一次保存目录
|
|
|
String lastPath = preferences.get(LAST_PATH, "");
|
|
|
// 默认文件名
|
|
|
String filename = StrUtil.format("{}.png", System.currentTimeMillis());
|
|
|
|
|
|
JFileChooser jFileChooser = new JFileChooser();
|
|
|
jFileChooser.setFileFilter(new FileNameExtensionFilter("png", "png"));
|
|
|
jFileChooser.setFileSelectionMode(JFileChooser.FILES_ONLY);
|
|
|
jFileChooser.setCurrentDirectory(new File(lastPath));
|
|
|
jFileChooser.setSelectedFile(new File(filename));
|
|
|
int returnVal = jFileChooser.showSaveDialog(this);
|
|
|
if (returnVal == JFileChooser.APPROVE_OPTION) {
|
|
|
File selectedFile = jFileChooser.getSelectedFile();
|
|
|
String legendName = selectedFile.getName();
|
|
|
if (!legendName.endsWith(".png")) {
|
|
|
legendName += ".png";
|
|
|
}
|
|
|
|
|
|
File parentDir = selectedFile.getParentFile();
|
|
|
String finalLegendName = legendName;
|
|
|
|
|
|
// 提取公共的保存逻辑
|
|
|
Runnable saveAction = () -> {
|
|
|
File targetFile = new File(parentDir, finalLegendName);
|
|
|
doSave(targetFile);
|
|
|
close();
|
|
|
};
|
|
|
|
|
|
List<File> existingFiles = FileUtil.loopFiles(Legend.getDefaultBase(),
|
|
|
pathname -> pathname.getName().equals(finalLegendName));
|
|
|
|
|
|
if (existingFiles.isEmpty()) {
|
|
|
saveAction.run();
|
|
|
} else {
|
|
|
DialogUtil.confirm("提示", "存在相同图例，是否覆盖？", result -> {
|
|
|
if (result) {
|
|
|
saveAction.run();
|
|
|
} else {
|
|
|
close(); // 用户取消时也关闭窗口
|
|
|
}
|
|
|
});
|
|
|
}
|
|
|
} else {
|
|
|
close(); // 用户取消选择时关闭窗口
|
|
|
}
|
|
|
}
|
|
|
|
|
|
private void doSave(File targetFile) {
|
|
|
Logger.info("↓↓↓↓↓区域采集↓↓↓↓↓");
|
|
|
|
|
|
Logger.info("图例名称：{}", targetFile.getName().replaceAll("\\.png",""));
|
|
|
|
|
|
Logger.info("保存路径 {}", targetFile);
|
|
|
ImgUtil.cut(capture, targetFile, new Rectangle(
|
|
|
getCaptureX(),
|
|
|
getCaptureY(),
|
|
|
getCaptureWidth(),
|
|
|
getCaptureHeight()
|
|
|
));
|
|
|
|
|
|
preferences.put(LAST_PATH, targetFile.getParent());
|
|
|
Logger.info("保存目录 {}", targetFile.getParent());
|
|
|
Logger.info("保存地址 {}", targetFile);
|
|
|
Logger.info("用法示例：");
|
|
|
Logger.info("查找图例：findLegend(\"{}\", 0.9)",targetFile.getName().replaceAll("\\.png",""));
|
|
|
Logger.info("");
|
|
|
Logger.info("↑↑↑↑↑区域采集↑↑↑↑↑");
|
|
|
}
|
|
|
|
|
|
public void close() {
|
|
|
setVisible(false);
|
|
|
dispose();
|
|
|
}
|
|
|
}
|
