add method to find area to encode

segment-anything-models-java · May 28, 2024 · e5d0c61 · e5d0c61
1 parent 2deed03
commit e5d0c61
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/src/main/java/ai/nets/samj/ij/ui/IJ1PromptsProvider.java b/src/main/java/ai/nets/samj/ij/ui/IJ1PromptsProvider.java
@@ -53,8 +53,10 @@
 import ai.nets.samj.ui.PromptsResultsDisplay;
 
 import java.awt.Color;
+import java.awt.Dimension;
 import java.awt.Polygon;
 import java.awt.Rectangle;
+import java.awt.Toolkit;
 import java.awt.event.KeyEvent;
 import java.awt.event.KeyListener;
 import java.awt.event.MouseEvent;
@@ -225,7 +227,8 @@ public void notifyException(SAMJException type, Exception ex) {
 		} else if (SAMJException.ENCODING == type) {
 			IJ.error("Error providing or processing the prompts", nMessage);
 		} else {
-			IJ.error(nMessage);
+			IJ.error(ex.toString());
+			ex.printStackTrace();
 		}
 
 	}
@@ -524,7 +527,7 @@ private void submitAndClearPoints() {
 		//TODO log.info("Image window: Processing now points, this count: "+collectedPoints.size());
 		isCollectingPoints = false;
 		activeImage.deleteRoi();
-		Rectangle zoomedRectangle = this.activeCanvas.getSrcRect();
+		Rectangle zoomedRectangle = findEncodingArea();
 		try {
 			if (activeImage.getWidth() * activeImage.getHeight() > Math.pow(AbstractSamJ.MAX_ENCODED_AREA_RS, 2)
 					|| activeImage.getWidth() > AbstractSamJ.MAX_ENCODED_SIDE || activeImage.getHeight() > AbstractSamJ.MAX_ENCODED_SIDE)
@@ -542,6 +545,18 @@ private void submitAndClearPoints() {
 		temporalNegROIs = new ArrayList<Roi>();
 	}
 
+	private Rectangle findEncodingArea() {
+		Rectangle zoomedRectangle = this.activeCanvas.getSrcRect();
+		Dimension screenSize = Toolkit.getDefaultToolkit().getScreenSize();
+		double width = screenSize.getWidth();
+		double height = screenSize.getHeight();
+		Rectangle bounds = this.activeCanvas.getBounds();
+
+		double xRatio = zoomedRectangle.x / width;
+		double yRatio = zoomedRectangle.y / height;
+		return this.activeCanvas.getSrcRect();
+	}
+
 	private void submitRectPrompt(Interval rectInterval) {
 		try {
 			addToRoiManager(promptsToNet.fetch2dSegmentation(rectInterval), "rect");