✨ Aspect Ratio, with dimension extrapolation

Author: MrGraversen

spring-boot-starter-replicate/src/main/java/io/graversen/replicate/common/AspectRatios.java

@@ -0,0 +1,37 @@
+package io.graversen.replicate.common;
+
+import lombok.AccessLevel;
+import lombok.Getter;
+import lombok.RequiredArgsConstructor;
+
+import java.util.Arrays;
+import java.util.Optional;
+
+@Getter
+@RequiredArgsConstructor(access = AccessLevel.PACKAGE)
+public enum AspectRatios {
+    RATIO_1_BY_1(1, 1),
+    RATIO_16_BY_9(16, 9),
+    RATIO_21_BY_9(21, 9),
+    RATIO_3_BY_2(3, 2),
+    RATIO_2_BY_3(2, 3),
+    RATIO_4_BY_5(4, 5),
+    RATIO_5_BY_4(5, 4),
+    RATIO_3_BY_4(3, 4),
+    RATIO_4_BY_3(4, 3),
+    RATIO_9_BY_16(9, 16),
+    RATIO_9_BY_21(9, 21);
+
+    private final int widthRatio;
+    private final int heightRatio;
+
+    public static Optional<AspectRatios> fromValue(String value) {
+        return Arrays.stream(AspectRatios.values())
+                .filter(ratio -> ratio.getAspectRatio().equals(value))
+                .findFirst();
+    }
+
+    public String getAspectRatio() {
+        return widthRatio + ":" + heightRatio;
+    }
+}

spring-boot-starter-replicate/src/main/java/io/graversen/replicate/common/TextToImageAspectRatio.java

@@ -0,0 +1,18 @@
+package io.graversen.replicate.common;
+
+import lombok.NonNull;
+import lombok.Value;
+
+@Value
+public class TextToImageAspectRatio {
+    @NonNull Integer height;
+    @NonNull AspectRatios aspectRatio;
+
+    public Integer getWidth() {
+        return (int) Math.round(height * ((double) aspectRatio.getWidthRatio() / aspectRatio.getHeightRatio()));
+    }
+
+    public String getAspectRatioAsString() {
+        return aspectRatio.getAspectRatio();
+    }
+}

spring-boot-starter-replicate/src/test/java/io/graversen/replicate/common/TextToImageAspectRatioTest.java

@@ -0,0 +1,90 @@
+package io.graversen.replicate.common;
+
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+class TextToImageAspectRatioTest {
+    @Test
+    public void testAspectRatio_1By1() {
+        TextToImageAspectRatio aspectRatio = new TextToImageAspectRatio(100, AspectRatios.RATIO_1_BY_1);
+        assertEquals(100, aspectRatio.getWidth());
+        assertEquals(100, aspectRatio.getHeight());
+    }
+
+    @Test
+    public void testAspectRatio_16By9() {
+        TextToImageAspectRatio aspectRatio = new TextToImageAspectRatio(180, AspectRatios.RATIO_16_BY_9);
+        assertEquals(320, aspectRatio.getWidth());
+        assertEquals(180, aspectRatio.getHeight());
+    }
+
+    @Test
+    public void testAspectRatio_21By9() {
+        TextToImageAspectRatio aspectRatio = new TextToImageAspectRatio(180, AspectRatios.RATIO_21_BY_9);
+        assertEquals(420, aspectRatio.getWidth());
+        assertEquals(180, aspectRatio.getHeight());
+    }
+
+    @Test
+    public void testAspectRatio_3By2() {
+        TextToImageAspectRatio aspectRatio = new TextToImageAspectRatio(200, AspectRatios.RATIO_3_BY_2);
+        assertEquals(300, aspectRatio.getWidth());
+        assertEquals(200, aspectRatio.getHeight());
+    }
+
+    @Test
+    public void testAspectRatio_2By3() {
+        TextToImageAspectRatio aspectRatio = new TextToImageAspectRatio(300, AspectRatios.RATIO_2_BY_3);
+        assertEquals(200, aspectRatio.getWidth());
+        assertEquals(300, aspectRatio.getHeight());
+    }
+
+    @Test
+    public void testAspectRatio_4By5() {
+        TextToImageAspectRatio aspectRatio = new TextToImageAspectRatio(1350, AspectRatios.RATIO_4_BY_5);
+        assertEquals(1080, aspectRatio.getWidth());
+        assertEquals(1350, aspectRatio.getHeight());
+    }
+
+    @Test
+    public void testAspectRatio_5By4() {
+        TextToImageAspectRatio aspectRatio = new TextToImageAspectRatio(1080, AspectRatios.RATIO_5_BY_4);
+        assertEquals(1350, aspectRatio.getWidth());
+        assertEquals(1080, aspectRatio.getHeight());
+    }
+
+    @Test
+    public void testAspectRatio_3By4() {
+        TextToImageAspectRatio aspectRatio = new TextToImageAspectRatio(300, AspectRatios.RATIO_3_BY_4);
+        assertEquals(225, aspectRatio.getWidth());
+        assertEquals(300, aspectRatio.getHeight());
+    }
+
+    @Test
+    public void testAspectRatio_4By3() {
+        TextToImageAspectRatio aspectRatio = new TextToImageAspectRatio(120, AspectRatios.RATIO_4_BY_3);
+        assertEquals(160, aspectRatio.getWidth());
+        assertEquals(120, aspectRatio.getHeight());
+    }
+
+    @Test
+    public void testAspectRatio_9By16() {
+        TextToImageAspectRatio aspectRatio = new TextToImageAspectRatio(320, AspectRatios.RATIO_9_BY_16);
+        assertEquals(180, aspectRatio.getWidth());
+        assertEquals(320, aspectRatio.getHeight());
+    }
+
+    @Test
+    public void testAspectRatio_9By21() {
+        TextToImageAspectRatio aspectRatio = new TextToImageAspectRatio(420, AspectRatios.RATIO_9_BY_21);
+        assertEquals(180, aspectRatio.getWidth());
+        assertEquals(420, aspectRatio.getHeight());
+    }
+
+    @Test
+    public void testGetAspectRatioAsString() {
+        TextToImageAspectRatio aspectRatio = new TextToImageAspectRatio(100, AspectRatios.RATIO_4_BY_3);
+        assertEquals("4:3", aspectRatio.getAspectRatioAsString());
+    }
+}