add methods to communication layer

segment-anything-models-java · Jul 17, 2024 · 49591c9 · 49591c9
1 parent 65ef7c4
commit 49591c9
Show file tree

Hide file tree

Showing 7 changed files with 187 additions and 0 deletions.
diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientSAM.java b/src/main/java/ai/nets/samj/communication/model/EfficientSAM.java
@@ -240,4 +240,34 @@ public String getInputImageAxes() {
 	public void setReturnOnlyBiggest(boolean onlyBiggest) {
 		this.onlyBiggest = onlyBiggest;
 	}
+
+	@Override
+	public String persistEncoding() throws IOException, InterruptedException {
+		try {
+			return efficientSamJ.persistEncoding();
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to persist the encoding: "+e.getMessage());
+			throw e;
+		}
+	}
+
+	@Override
+	public void selectEncoding(String encodingName) throws IOException, InterruptedException {
+		try {
+			efficientSamJ.selectEncoding(encodingName);
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to persist the encoding named '" + encodingName + "': "+e.getMessage());
+			throw e;
+		}
+	}
+
+	@Override
+	public void deleteEncoding(String encodingName) throws IOException, InterruptedException {
+		try {
+			efficientSamJ.deleteEncoding(encodingName);
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to delete the encoding named '" + encodingName + "': "+e.getMessage());
+			throw e;
+		}
+	}
 }
diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML0.java b/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML0.java
@@ -239,4 +239,34 @@ public String getInputImageAxes() {
 	public void setReturnOnlyBiggest(boolean onlyBiggest) {
 		this.onlyBiggest = onlyBiggest;
 	}
+
+	@Override
+	public String persistEncoding() throws IOException, InterruptedException {
+		try {
+			return efficientSamJ.persistEncoding();
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to persist the encoding: "+e.getMessage());
+			throw e;
+		}
+	}
+
+	@Override
+	public void selectEncoding(String encodingName) throws IOException, InterruptedException {
+		try {
+			efficientSamJ.selectEncoding(encodingName);
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to persist the encoding named '" + encodingName + "': "+e.getMessage());
+			throw e;
+		}
+	}
+
+	@Override
+	public void deleteEncoding(String encodingName) throws IOException, InterruptedException {
+		try {
+			efficientSamJ.deleteEncoding(encodingName);
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to delete the encoding named '" + encodingName + "': "+e.getMessage());
+			throw e;
+		}
+	}
 }
diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML1.java b/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML1.java
@@ -241,4 +241,34 @@ public String getInputImageAxes() {
 	public void setReturnOnlyBiggest(boolean onlyBiggest) {
 		this.onlyBiggest = onlyBiggest;
 	}
+
+	@Override
+	public String persistEncoding() throws IOException, InterruptedException {
+		try {
+			return efficientSamJ.persistEncoding();
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to persist the encoding: "+e.getMessage());
+			throw e;
+		}
+	}
+
+	@Override
+	public void selectEncoding(String encodingName) throws IOException, InterruptedException {
+		try {
+			efficientSamJ.selectEncoding(encodingName);
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to persist the encoding named '" + encodingName + "': "+e.getMessage());
+			throw e;
+		}
+	}
+
+	@Override
+	public void deleteEncoding(String encodingName) throws IOException, InterruptedException {
+		try {
+			efficientSamJ.deleteEncoding(encodingName);
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to delete the encoding named '" + encodingName + "': "+e.getMessage());
+			throw e;
+		}
+	}
 }
diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML2.java b/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML2.java
@@ -241,4 +241,34 @@ public String getInputImageAxes() {
 	public void setReturnOnlyBiggest(boolean onlyBiggest) {
 		this.onlyBiggest = onlyBiggest;
 	}
+
+	@Override
+	public String persistEncoding() throws IOException, InterruptedException {
+		try {
+			return efficientSamJ.persistEncoding();
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to persist the encoding: "+e.getMessage());
+			throw e;
+		}
+	}
+
+	@Override
+	public void selectEncoding(String encodingName) throws IOException, InterruptedException {
+		try {
+			efficientSamJ.selectEncoding(encodingName);
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to persist the encoding named '" + encodingName + "': "+e.getMessage());
+			throw e;
+		}
+	}
+
+	@Override
+	public void deleteEncoding(String encodingName) throws IOException, InterruptedException {
+		try {
+			efficientSamJ.deleteEncoding(encodingName);
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to delete the encoding named '" + encodingName + "': "+e.getMessage());
+			throw e;
+		}
+	}
 }
diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientViTSAMXL0.java b/src/main/java/ai/nets/samj/communication/model/EfficientViTSAMXL0.java
@@ -241,4 +241,34 @@ public String getInputImageAxes() {
 	public void setReturnOnlyBiggest(boolean onlyBiggest) {
 		this.onlyBiggest = onlyBiggest;
 	}
+
+	@Override
+	public String persistEncoding() throws IOException, InterruptedException {
+		try {
+			return efficientSamJ.persistEncoding();
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to persist the encoding: "+e.getMessage());
+			throw e;
+		}
+	}
+
+	@Override
+	public void selectEncoding(String encodingName) throws IOException, InterruptedException {
+		try {
+			efficientSamJ.selectEncoding(encodingName);
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to persist the encoding named '" + encodingName + "': "+e.getMessage());
+			throw e;
+		}
+	}
+
+	@Override
+	public void deleteEncoding(String encodingName) throws IOException, InterruptedException {
+		try {
+			efficientSamJ.deleteEncoding(encodingName);
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to delete the encoding named '" + encodingName + "': "+e.getMessage());
+			throw e;
+		}
+	}
 }
diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientViTSAMXL1.java b/src/main/java/ai/nets/samj/communication/model/EfficientViTSAMXL1.java
@@ -242,4 +242,34 @@ public String getInputImageAxes() {
 	public void setReturnOnlyBiggest(boolean onlyBiggest) {
 		this.onlyBiggest = onlyBiggest;
 	}
+
+	@Override
+	public String persistEncoding() throws IOException, InterruptedException {
+		try {
+			return efficientSamJ.persistEncoding();
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to persist the encoding: "+e.getMessage());
+			throw e;
+		}
+	}
+
+	@Override
+	public void selectEncoding(String encodingName) throws IOException, InterruptedException {
+		try {
+			efficientSamJ.selectEncoding(encodingName);
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to persist the encoding named '" + encodingName + "': "+e.getMessage());
+			throw e;
+		}
+	}
+
+	@Override
+	public void deleteEncoding(String encodingName) throws IOException, InterruptedException {
+		try {
+			efficientSamJ.deleteEncoding(encodingName);
+		} catch (IOException | InterruptedException | RuntimeException e) {
+			log.error(FULL_NAME+", unable to delete the encoding named '" + encodingName + "': "+e.getMessage());
+			throw e;
+		}
+	}
 }
diff --git a/src/main/java/ai/nets/samj/communication/model/SAMModel.java b/src/main/java/ai/nets/samj/communication/model/SAMModel.java
@@ -146,6 +146,13 @@ public interface SAMModel {
 	 */
 	public <T extends RealType<T> & NativeType<T>> List<Polygon> fetch2dSegmentationFromMask(RandomAccessibleInterval<T> rai) throws IOException, RuntimeException, InterruptedException;
 
+
+	public String persistEncoding() throws IOException, InterruptedException;
+
+	public void selectEncoding(String encodingName) throws IOException, InterruptedException;
+
+	public void deleteEncoding(String encodingName) throws IOException, InterruptedException;
+
 	/**
 	 * Close the Python process where the model is being executed
 	 */