test(getPatchIntensityCentroid): enhance tests

Author: opatiny

src/featureMatching/keypoints/__tests__/__image_snapshots__/get-patch-intensity-centroid-test-ts-better-triangle-90-keypoint-1-snap.png

@@ -54,6 +54,18 @@ test('check window is circular', () => {
   expect(result).toStrictEqual([{ column: 1, row: 0 }]);
 });
 
+test('triangle center of mass', () => {
+  const image = testUtils.createGreyImage([
+    [0, 0, 0, 1, 0],
+    [0, 0, 1, 1, 0],
+    [0, 1, 1, 1, 0],
+    [0, 0, 1, 1, 0],
+    [0, 0, 0, 1, 0],
+  ]);
+  const result = getPatchIntensityCentroid(image, { radius: 2 });
+  expect(result).toBeDeepCloseTo([{ column: 0.444, row: 0 }]);
+});
+
 test('patch, default options', () => {
   const image = testUtils.load('featureMatching/patch.png');
   image.invert({ out: image });
@@ -74,3 +86,75 @@ test('patch, default options', () => {
 
   expect(result).toMatchImageSnapshot();
 });
+
+test('better triangle keypoint', () => {
+  const origin = { row: 332, column: 253 };
+  const size = 7;
+  const radius = (size - 1) / 2;
+
+  const cropOrigin = {
+    row: origin.row - radius,
+    column: origin.column - radius,
+  };
+
+  const origialImage = testUtils
+    .load('featureMatching/polygons/betterScaleneTriangle.png')
+    .convertColor(ImageColorModel.GREY)
+    .invert();
+
+  const image = origialImage.crop({
+    origin: cropOrigin,
+    width: size,
+    height: size,
+  });
+
+  const centroid = getPatchIntensityCentroid(image, { radius })[0];
+  expect(centroid).toBeDeepCloseTo({ column: 1.281, row: 0.204 });
+
+  const center = image.getCoordinates(ImageCoordinates.CENTER);
+
+  const point = round(sum(center, centroid));
+
+  const result = image.convertColor(ImageColorModel.RGB);
+  result.drawCircle(center, radius + 1, { color: [255, 0, 0], out: result });
+
+  result.drawPoints([point], { color: [0, 255, 0], out: result });
+
+  expect(result).toMatchImageSnapshot();
+});
+
+test('better triangle 90 keypoint', () => {
+  const origin = { row: 730, column: 291 };
+  const size = 7;
+  const radius = (size - 1) / 2;
+
+  const cropOrigin = {
+    row: origin.row - radius,
+    column: origin.column - radius,
+  };
+
+  const origialImage = testUtils
+    .load('featureMatching/polygons/betterScaleneTriangle90.png')
+    .convertColor(ImageColorModel.GREY)
+    .invert();
+
+  const image = origialImage.crop({
+    origin: cropOrigin,
+    width: size,
+    height: size,
+  });
+
+  const centroid = getPatchIntensityCentroid(image, { radius })[0];
+  expect(centroid).toBeDeepCloseTo({ column: 0.634, row: -1.074 });
+
+  const center = image.getCoordinates(ImageCoordinates.CENTER);
+
+  const point = round(sum(center, centroid));
+
+  const result = image.convertColor(ImageColorModel.RGB);
+  result.drawCircle(center, radius + 1, { color: [255, 0, 0], out: result });
+
+  result.drawPoints([point], { color: [0, 255, 0], out: result });
+
+  expect(result).toMatchImageSnapshot();
+});