parent
ab423e29b6
commit
22d7efb968
GPG Key ID:
947ADABD8533C476
@ -1,205 +0,0 @@ |
||||
package me.lensferno.dogename.ocr; |
||||
|
||||
import java.io.*; |
||||
import javax.swing.*; |
||||
import java.awt.*; |
||||
import java.awt.event.*; |
||||
import java.awt.image.*; |
||||
import javax.imageio.*; |
||||
|
||||
//edited code from https://blog.csdn.net/caoyuan10036/article/details/7199505
|
||||
|
||||
public class ScreenCapture { |
||||
// test main
|
||||
/* |
||||
public static void main(String[] args) { |
||||
File tempFile = new File( "files"+File.separator+"ocr.png"); |
||||
ScreenCapture capture = ScreenCapture.getInstance(); |
||||
capture.captureImage(); |
||||
JLabel imagebox = new JLabel(); |
||||
imagebox.setIcon(capture.getPickedIcon()); |
||||
try { |
||||
capture.saveToFile(tempFile); |
||||
System.out.println("Over"); |
||||
} catch (IOException e) { |
||||
e.printStackTrace(); |
||||
} |
||||
|
||||
} |
||||
*/ |
||||
|
||||
public static final String SCREEN_CAPTURE_LOCA="files"+File.separator+"ocrCache.png"; |
||||
|
||||
private static boolean canceled =false; |
||||
|
||||
|
||||
public static boolean getScreenCapture(){ |
||||
|
||||
try { |
||||
canceled=false; |
||||
File tempFile = new File(SCREEN_CAPTURE_LOCA); |
||||
ScreenCapture capture = ScreenCapture.getInstance(); |
||||
capture.captureImage(); |
||||
if(canceled){ |
||||
return false; |
||||
} |
||||
JLabel imagebox = new JLabel(); |
||||
imagebox.setIcon(capture.getPickedIcon()); |
||||
capture.saveToFile(tempFile); |
||||
System.out.println("Over"); |
||||
|
||||
return true; |
||||
} catch (Exception e) { |
||||
e.printStackTrace(); |
||||
return false; |
||||
} |
||||
} |
||||
|
||||
private ScreenCapture() { |
||||
|
||||
try { |
||||
robot = new Robot(); |
||||
} catch (AWTException e) { |
||||
System.err.println("Internal Error: " + e); |
||||
e.printStackTrace(); |
||||
} |
||||
JPanel cp = (JPanel) dialog.getContentPane(); |
||||
cp.setLayout(new BorderLayout()); |
||||
labFullScreenImage.addMouseListener(new MouseAdapter() { |
||||
public void mouseReleased(MouseEvent evn) { |
||||
if (evn.getButton()==MouseEvent.BUTTON3){ |
||||
canceled=true; |
||||
System.out.println("captureCancel"); |
||||
|
||||
dialog.setVisible(false); |
||||
return; |
||||
} |
||||
isFirstPoint = true; |
||||
pickedImage = fullScreenImage.getSubimage(recX, recY, recW, |
||||
recH); |
||||
dialog.setVisible(false); |
||||
|
||||
} |
||||
}); |
||||
|
||||
|
||||
labFullScreenImage.addMouseMotionListener(new MouseMotionAdapter() { |
||||
public void mouseDragged(MouseEvent evn) { |
||||
if (isFirstPoint) { |
||||
x1 = evn.getX(); |
||||
y1 = evn.getY(); |
||||
isFirstPoint = false; |
||||
} else { |
||||
x2 = evn.getX(); |
||||
y2 = evn.getY(); |
||||
int maxX = Math.max(x1, x2); |
||||
int maxY = Math.max(y1, y2); |
||||
int minX = Math.min(x1, x2); |
||||
int minY = Math.min(y1, y2); |
||||
recX = minX; |
||||
recY = minY; |
||||
recW = maxX - minX; |
||||
recH = maxY - minY; |
||||
labFullScreenImage.drawRectangle(recX, recY, recW, recH); |
||||
} |
||||
} |
||||
|
||||
public void mouseMoved(MouseEvent e) { |
||||
labFullScreenImage.drawCross(e.getX(), e.getY()); |
||||
//messageLabel.setLocation(e.getX(), e.getY());
|
||||
} |
||||
}); |
||||
|
||||
cp.add(BorderLayout.CENTER, labFullScreenImage); |
||||
//cp.add(messageLabel);
|
||||
dialog.setCursor(Cursor.getPredefinedCursor(Cursor.CROSSHAIR_CURSOR)); |
||||
dialog.setAlwaysOnTop(true); |
||||
dialog.setMaximumSize(Toolkit.getDefaultToolkit().getScreenSize()); |
||||
dialog.setUndecorated(true); |
||||
dialog.setSize(dialog.getMaximumSize()); |
||||
dialog.setModal(true); |
||||
} |
||||
|
||||
// Singleton Pattern
|
||||
public static ScreenCapture getInstance() { |
||||
return defaultCapturer; |
||||
} |
||||
|
||||
/** 捕捉屏幕的一个矫形区域 */ |
||||
public void captureImage() { |
||||
fullScreenImage = robot.createScreenCapture(new Rectangle(Toolkit |
||||
.getDefaultToolkit().getScreenSize())); |
||||
ImageIcon icon = new ImageIcon(fullScreenImage); |
||||
labFullScreenImage.setIcon(icon); |
||||
dialog.setVisible(true); |
||||
} |
||||
|
||||
/** 得到捕捉后的BufferedImage */ |
||||
public BufferedImage getPickedImage() { |
||||
return pickedImage; |
||||
} |
||||
|
||||
/** 得到捕捉后的Icon */ |
||||
public ImageIcon getPickedIcon() { |
||||
return new ImageIcon(getPickedImage()); |
||||
} |
||||
|
||||
/** |
||||
* 储存为一个文件,为PNG格式 |
||||
* |
||||
* @deprecated replaced by saveAsPNG(File file) |
||||
**/ |
||||
@Deprecated |
||||
public void saveToFile(File file) throws IOException { |
||||
ImageIO.write(getPickedImage(), defaultImageFormater, file); |
||||
} |
||||
|
||||
|
||||
/** 写入一个OutputStream */ |
||||
public void write(OutputStream out) throws IOException { |
||||
ImageIO.write(getPickedImage(), defaultImageFormater, out); |
||||
} |
||||
|
||||
// singleton design pattern
|
||||
private static ScreenCapture defaultCapturer = new ScreenCapture(); |
||||
private int x1, y1, x2, y2; |
||||
private int recX, recY, recH, recW; // 截取的图像
|
||||
private boolean isFirstPoint = true; |
||||
private BackgroundImage labFullScreenImage = new BackgroundImage(); |
||||
//private Label messageLabel = new Label("按任意键可退出。");
|
||||
private Robot robot; |
||||
private BufferedImage fullScreenImage; |
||||
private BufferedImage pickedImage; |
||||
private String defaultImageFormater = "png"; |
||||
private JDialog dialog = new JDialog(); |
||||
} |
||||
|
||||
/** 显示图片的Label */ |
||||
class BackgroundImage extends JLabel { |
||||
public void paintComponent(Graphics g) { |
||||
super.paintComponent(g); |
||||
g.drawRect(x, y, w, h); |
||||
String area = w + " * " + h; |
||||
g.drawString(area, x + w / 2 - 15, y + h / 2); |
||||
g.drawLine(lineX, 0, lineX, getHeight()); |
||||
|
||||
g.drawLine(0, lineY, getWidth(), lineY); |
||||
} |
||||
|
||||
public void drawRectangle(int x, int y, int width, int height) { |
||||
this.x = x; |
||||
this.y = y; |
||||
h = height; |
||||
w = width; |
||||
repaint(); |
||||
} |
||||
|
||||
public void drawCross(int x, int y) { |
||||
lineX = x; |
||||
lineY = y; |
||||
repaint(); |
||||
} |
||||
|
||||
int lineX, lineY; |
||||
int x, y, h, w; |
||||
} |
||||
@ -0,0 +1,245 @@ |
||||
package me.lensferno.dogename.utils.ocr; |
||||
|
||||
import javafx.application.Platform; |
||||
import javafx.beans.property.SimpleBooleanProperty; |
||||
import javafx.beans.property.SimpleStringProperty; |
||||
import javafx.scene.Scene; |
||||
import javafx.scene.image.Image; |
||||
import javafx.scene.image.ImageView; |
||||
import javafx.scene.input.KeyCode; |
||||
import javafx.scene.layout.Pane; |
||||
import javafx.scene.shape.Line; |
||||
import javafx.stage.Stage; |
||||
|
||||
import javax.imageio.ImageIO; |
||||
import java.awt.*; |
||||
import java.awt.image.BufferedImage; |
||||
import java.io.ByteArrayInputStream; |
||||
import java.io.ByteArrayOutputStream; |
||||
import java.io.File; |
||||
|
||||
public class ScreenCapture { |
||||
|
||||
private OcrTool ocrTool; |
||||
|
||||
private final String cacheImageFileLocation = "caches/image/ocrImageCache.jpg"; |
||||
|
||||
private final Runnable finalEvent = new Runnable() { |
||||
@Override |
||||
public void run() { |
||||
if (ocrTool == null) { |
||||
ocrTool = new OcrTool(); |
||||
} |
||||
ocrTool.requestOcrAPI(cacheImageFileLocation); |
||||
result.set(ocrTool.getResult()); |
||||
end.set(true); |
||||
} |
||||
}; |
||||
|
||||
private final SimpleStringProperty result = new SimpleStringProperty(); |
||||
private final SimpleBooleanProperty end = new SimpleBooleanProperty(true); |
||||
|
||||
private final ImageView screenImageView = new ImageView(); |
||||
private final Pane rootPane = new Pane(); |
||||
private final Stage stage = new Stage(); |
||||
|
||||
private final int screenWidth = Toolkit.getDefaultToolkit().getScreenSize().width; |
||||
private final int screenHeight = Toolkit.getDefaultToolkit().getScreenSize().height; |
||||
|
||||
ScreenshotTool screenshotTool = new ScreenshotTool(); |
||||
|
||||
public void startCapture() { |
||||
end.set(false); |
||||
Image fullscreenImage = screenshotTool.getFullScreenshotImageData(); |
||||
createScreenCaptureWindow(fullscreenImage); |
||||
} |
||||
|
||||
public void createScreenCaptureWindow(Image screenImage) { |
||||
this.screenImageView.setImage(screenImage); |
||||
stage.setFullScreen(true); |
||||
stage.setAlwaysOnTop(true); |
||||
stage.setFullScreenExitHint("拖动鼠标选择需要截取识别的区域，\n按ESC键可退出截屏选取"); |
||||
//stage.setFullScreenExitKeyCombination(KeyCombination.NO_MATCH);
|
||||
|
||||
rootPane.setPrefWidth(screenWidth); |
||||
rootPane.setPrefHeight(screenHeight); |
||||
rootPane.getChildren().add(screenImageView); |
||||
screenImageView.setFitWidth(screenWidth); |
||||
screenImageView.setFitHeight(screenHeight); |
||||
|
||||
Scene scene = new Scene(rootPane, screenWidth, screenHeight); |
||||
//rootPane.setStyle("-fx-background-color: black");
|
||||
scene.setOnKeyPressed(e -> { |
||||
if (e.getCode().equals(KeyCode.ESCAPE)){ |
||||
stage.close(); |
||||
end.set(true); |
||||
} |
||||
}); |
||||
|
||||
this.addMouseLine(); |
||||
scene.setOnMouseMoved(e -> this.drawMouseLine(e.getX(), e.getY())); |
||||
|
||||
this.addEventListeners(); |
||||
|
||||
stage.setScene(scene); |
||||
stage.show(); |
||||
} |
||||
|
||||
private double dragStartX = 0; |
||||
private double dragStartY = 0; |
||||
|
||||
private double dragEndX = 0; |
||||
private double dragEndY = 0; |
||||
|
||||
private double selectedAreaWidth = 0; |
||||
private double selectedAreaHeight = 0; |
||||
|
||||
private Line Width_Follow; |
||||
private Line Height_Follow; |
||||
private Line Width_Fixed; |
||||
private Line Height_Fixed; |
||||
|
||||
private void addEventListeners() { |
||||
rootPane.setOnMousePressed(e -> { |
||||
dragStartX = e.getX(); |
||||
dragStartY = e.getY(); |
||||
this.createNewSelectedAreaRectangle(); |
||||
this.setLinePosition(); |
||||
}); |
||||
rootPane.setOnMouseDragged(e -> { |
||||
this.drawMouseLine(e.getX(), e.getY()); |
||||
this.drawSelectedAreaRectangle(e.getX(), e.getY()); |
||||
}); |
||||
rootPane.setOnMouseReleased(e -> { |
||||
dragEndX = e.getX(); |
||||
dragEndY = e.getY(); |
||||
int clipPositionX = (int)dragStartX; |
||||
int clipPositionY = (int)dragStartY; |
||||
if (selectedAreaWidth < 0) { |
||||
clipPositionX = (int)dragEndX; |
||||
} |
||||
if (selectedAreaHeight < 0) { |
||||
clipPositionY = (int)dragEndY; |
||||
} |
||||
try { |
||||
this.saveImage(clipPositionX, clipPositionY, Math.abs((int)selectedAreaWidth), Math.abs((int)selectedAreaHeight)); |
||||
this.runFinalEvent(finalEvent); |
||||
} catch (Exception exc) { |
||||
exc.printStackTrace(); |
||||
} finally { |
||||
stage.close(); |
||||
} |
||||
}); |
||||
} |
||||
|
||||
private void runFinalEvent(Runnable finalEvent) { |
||||
Platform.runLater(finalEvent); |
||||
} |
||||
|
||||
private void saveImage(int x, int y, int width, int height){ |
||||
try { |
||||
File outputFile = new File(cacheImageFileLocation); |
||||
File outputDir = outputFile.getParentFile(); |
||||
|
||||
if (!outputDir.exists()) { |
||||
outputDir.mkdirs(); |
||||
outputFile.createNewFile(); |
||||
} |
||||
screenshotTool.saveClippedImage(x, y, width, height, "png", outputFile); |
||||
} catch (Exception e) { |
||||
e.printStackTrace(); |
||||
} |
||||
} |
||||
|
||||
private void createNewSelectedAreaRectangle() { |
||||
Width_Follow = new Line(); |
||||
Height_Follow = new Line(); |
||||
Width_Fixed = new Line(); |
||||
Height_Fixed = new Line(); |
||||
|
||||
Width_Follow.setStrokeWidth(5); |
||||
Height_Follow.setStrokeWidth(5); |
||||
Width_Fixed.setStrokeWidth(5); |
||||
Height_Fixed.setStrokeWidth(5); |
||||
|
||||
rootPane.getChildren().addAll(Width_Follow, Height_Follow, Width_Fixed, Height_Fixed); |
||||
} |
||||
|
||||
private void setLinePosition() { |
||||
Width_Fixed.layoutXProperty().set(dragStartX); |
||||
Width_Fixed.layoutYProperty().set(dragStartY); |
||||
|
||||
Height_Fixed.layoutYProperty().set(dragStartY); |
||||
Height_Fixed.layoutXProperty().set(dragStartX); |
||||
|
||||
Width_Follow.layoutXProperty().set(dragStartX); |
||||
Width_Follow.layoutYProperty().set(dragStartY); |
||||
|
||||
Height_Follow.layoutXProperty().set(dragStartX); |
||||
Height_Follow.layoutYProperty().set(dragStartY); |
||||
} |
||||
|
||||
private final Line MouseLine_X = new Line(); |
||||
private final Line MouseLine_Y = new Line(); |
||||
|
||||
private void addMouseLine() { |
||||
MouseLine_X.layoutXProperty().set(0); |
||||
MouseLine_X.endXProperty().set(screenWidth); |
||||
|
||||
MouseLine_Y.layoutYProperty().set(0); |
||||
MouseLine_Y.endYProperty().set(screenHeight); |
||||
|
||||
rootPane.getChildren().addAll(MouseLine_X, MouseLine_Y); |
||||
} |
||||
|
||||
private void drawMouseLine(double mouseX, double mouseY) { |
||||
MouseLine_X.setLayoutY(mouseY); |
||||
MouseLine_Y.setLayoutX(mouseX); |
||||
} |
||||
|
||||
private void drawSelectedAreaRectangle(double mouseX, double mouseY) { |
||||
|
||||
// 通过计算鼠标起始和当前坐标来确定四根线的长度和位置
|
||||
selectedAreaWidth = mouseX - dragStartX; |
||||
selectedAreaHeight = mouseY - dragStartY; |
||||
|
||||
Width_Fixed.endXProperty().set(selectedAreaWidth); |
||||
Height_Fixed.endYProperty().set(selectedAreaHeight); |
||||
|
||||
Width_Follow.layoutYProperty().set(mouseY); |
||||
Height_Follow.layoutXProperty().set(mouseX); |
||||
|
||||
Width_Follow.endXProperty().set(selectedAreaWidth); |
||||
Height_Follow.endYProperty().set(selectedAreaHeight); |
||||
} |
||||
|
||||
class ScreenshotTool { |
||||
BufferedImage fullscreenBufferedImage; |
||||
public Image getFullScreenshotImageData() { |
||||
try { |
||||
Robot robot = new Robot(); |
||||
ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream(); |
||||
|
||||
fullscreenBufferedImage = robot.createScreenCapture(new Rectangle(screenWidth, screenHeight)); |
||||
ImageIO.write(fullscreenBufferedImage, "png", byteArrayOutputStream); |
||||
|
||||
return new Image(new ByteArrayInputStream(byteArrayOutputStream.toByteArray())); |
||||
} catch (Exception e) { |
||||
e.printStackTrace(); |
||||
return null; |
||||
} |
||||
} |
||||
|
||||
public void saveClippedImage(int x, int y, int width, int height, String formateName, File output) throws Exception{ |
||||
ImageIO.write(fullscreenBufferedImage.getSubimage(x, y, width, height), formateName, output); |
||||
} |
||||
} |
||||
|
||||
public SimpleStringProperty resultProperty() { |
||||
return result; |
||||
} |
||||
|
||||
public SimpleBooleanProperty endProperty() { |
||||
return end; |
||||
} |
||||
} |
||||
Loading…
Reference in new issue