add drawMarks to android BitMapImageFactory

deepjavalibrary · Sep 9, 2024 · d86c567 · d86c567
1 parent f91a696
commit d86c567
Showing 1 changed file with 34 additions and 0 deletions.
diff --git a/android/core/src/main/java/ai/djl/android/core/BitmapImageFactory.java b/android/core/src/main/java/ai/djl/android/core/BitmapImageFactory.java
@@ -18,6 +18,7 @@
 import android.graphics.Color;
 import android.graphics.Paint;
 import android.graphics.Rect;
+import android.graphics.Path;
 
 import ai.djl.modality.cv.Image;
 import ai.djl.modality.cv.ImageFactory;
@@ -282,6 +283,29 @@ public void drawBoundingBoxes(DetectedObjects detections) {
             oldBitmap.recycle();
         }
 
+        /** {@inheritDoc} */
+        @Override
+        public void drawMarks(List<Point> points, int radius) {
+            Bitmap mutableBitmap = bitmap.copy(Bitmap.Config.ARGB_8888, true);
+            Canvas canvas = new Canvas(mutableBitmap);
+
+            // set the paint configure
+            Paint paint = new Paint();
+            paint.setStrokeWidth(2);
+            paint.setStyle(Paint.Style.FILL);
+            paint.setAntiAlias(true);
+
+            paint.setColor(darker(randomColor()));
+            for (Point point : points) {
+                int[][] star = createStar(point, radius);
+                drawPolygon(canvas, paint, star);
+            }
+
+            Bitmap oldBitmap = bitmap;
+            bitmap = mutableBitmap;
+            oldBitmap.recycle();
+        }
+
         /** {@inheritDoc} */
         @Override
         public void drawJoints(Joints joints) {
@@ -395,5 +419,15 @@ private void drawMask(Bitmap image, Mask mask) {
             Canvas canvas = new Canvas(image);
             canvas.drawBitmap(maskedImage, x, y, null);
         }
+
+        private void drawPolygon(Canvas canvas, Paint paint, int[][] polygon) {
+            Path polygonPath = new Path();
+            polygonPath.moveTo(polygon[0][0], polygonPath[1][0]);
+            for (int i = 1; i < polygon[0].length; i++) {
+                polygonPath.lineTo(polygon[0][i], polygon[1][i]);
+            }
+            polygonPath.close();
+            canvas.drawPath(polygonPath, paint);
+        }
     }
 }