From 97964ddb1c854931562402a1a6129b2f3eca60ee Mon Sep 17 00:00:00 2001
From: Antonio32A <~@antonio32a.com>
Date: Thu, 28 Jul 2022 17:29:53 +0200
Subject: [PATCH] Port mosaic resizing logic to properly calculate the width
 and height of the result image.

---
 src/mosaic.ts | 104 +++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 102 insertions(+), 2 deletions(-)

diff --git a/src/mosaic.ts b/src/mosaic.ts
index 867e1aa..546513f 100644
--- a/src/mosaic.ts
+++ b/src/mosaic.ts
@@ -44,9 +44,10 @@ export const handleMosaic = async (
       path += `/${mosaicMedia[3]}`;
     }
 
+    const size = calcSize(mediaList.map(i => ({ width: i.width, height: i.height } as Size)));
     return {
-      height: mediaList.reduce((acc, media) => acc + media.height, 0),
-      width: mediaList.reduce((acc, media) => acc + media.width, 0),
+      height: size.height,
+      width: size.width,
       formats: {
         jpeg: `${baseUrl}jpeg/${id}${path}`,
         webp: `${baseUrl}webp/${id}${path}`
@@ -54,3 +55,102 @@ export const handleMosaic = async (
     } as APIMosaicPhoto;
   }
 };
+
+// Port of https://github.com/FixTweet/mosaic/blob/feature/size-endpoint/src/mosaic.rs#L236
+
+const SPACING_SIZE = 10;
+/*
+ * This number will be multiplied by the height and weight
+ * if all combined images are over 2000 pixels in height or width.
+ * In my tests setting this to 0.5 increased performance by 4x (at the cost of 50% resolution)
+ * NOTE: This only works for when there's 4 images supplied.
+ */
+const BIG_IMAGE_MULTIPLIER = 1;
+
+const calcSize = (images: Size[]): Size => {
+  if (images.length === 1) {
+    return images[0];
+  } else if (images.length === 2) {
+    const size = calcHorizontalSize(images[0], images[1]);
+    return size as Size;
+  } else if (images.length === 3) {
+    const size = calcHorizontalSize(images[0], images[1]);
+    const thirdSize = calcVerticalSize(size, images[2]);
+
+    if (thirdSize.secondHeight * 1.5 > size.height) {
+      return calcHorizontalSize(size, images[2]) as Size;
+    } else {
+      return thirdSize as Size;
+    }
+  } else if (images.length === 4) {
+    const top = calcHorizontalSize(images[0], images[1]);
+    const bottom = calcHorizontalSize(images[2], images[3]);
+    const all = calcVerticalSize(top, bottom);
+
+    const sizeMult = (all.width > 2000 || all.height > 2000) ? BIG_IMAGE_MULTIPLIER : 1;
+    return {
+      width: all.width * sizeMult,
+      height: all.height * sizeMult
+    };
+  } else {
+    throw new Error('Invalid number of images');
+  }
+};
+
+const calcHorizontalSize = (first: Size, second: Size): HorizontalSize => {
+  let small = second;
+  let big = first;
+  let swapped = false;
+  if (second.height > first.height) {
+    small = first;
+    big = second;
+    swapped = true;
+  }
+
+  const smallWidth = Math.round((big.height / small.height) * small.width);
+  return {
+    width: smallWidth + SPACING_SIZE + big.width,
+    height: big.height,
+    firstWidth: swapped ? smallWidth : big.width,
+    secondWidth: swapped ? big.width : smallWidth
+  } as HorizontalSize;
+};
+
+const calcVerticalSize = (first: Size, second: Size): VerticalSize => {
+  let small = second;
+  let big = first;
+  let swapped = false;
+  if (second.width > first.width) {
+    small = first;
+    big = second;
+    swapped = true;
+  }
+
+  const smallHeight = Math.round(big.width / small.width * small.height);
+  return {
+    width: big.width,
+    height: smallHeight + SPACING_SIZE + big.height,
+    firstHeight: swapped ? smallHeight : big.height,
+    secondHeight: swapped ? big.height : smallHeight
+  } as VerticalSize;
+};
+
+interface Size {
+  width: number;
+  height: number;
+}
+
+interface HorizontalSize {
+  width: number;
+  height: number;
+  firstWidth: number;
+  secondWidth: number;
+}
+
+interface VerticalSize {
+  width: number;
+  height: number;
+  firstHeight: number;
+  secondHeight: number;
+}
+