Add methods add() to cores

Intervention · Nov 26, 2023 · 21acb14 · 21acb14
1 parent c8bc9ca
commit 21acb14
Show file tree

Hide file tree

Showing 5 changed files with 57 additions and 0 deletions.
diff --git a/src/Drivers/Gd/Core.php b/src/Drivers/Gd/Core.php
@@ -10,6 +10,13 @@ class Core extends Collection implements CoreInterface
 {
     protected int $loops = 0;
 
+    public function add(FrameInterface $frame): CoreInterface
+    {
+        $this->push($frame);
+
+        return $this;
+    }
+
     public function native(): mixed
     {
         return $this->first()->native();

diff --git a/src/Drivers/Imagick/Core.php b/src/Drivers/Imagick/Core.php
@@ -17,6 +17,26 @@ public function __construct(protected Imagick $imagick)
     {
     }
 
+    public function add(FrameInterface $frame): CoreInterface
+    {
+        $imagick = $frame->native();
+
+        $imagick->setImageDelay($frame->delay());
+        $imagick->setImageDispose($frame->dispose());
+
+        $size = $frame->size();
+        $imagick->setImagePage(
+            $size->width(),
+            $size->height(),
+            $frame->offsetLeft(),
+            $frame->offsetTop()
+        );
+
+        $this->imagick->addImage($imagick);
+
+        return $this;
+    }
+
     public function count(): int
     {
         return $this->imagick->getNumberImages();

diff --git a/src/Interfaces/CoreInterface.php b/src/Interfaces/CoreInterface.php
@@ -36,6 +36,14 @@ public function count(): int;
      */
     public function frame(int $position): FrameInterface;
 
+    /**
+     * Add new frame to core
+     *
+     * @param FrameInterface $frame
+     * @return CoreInterface
+     */
+    public function add(FrameInterface $frame): CoreInterface;
+
     /**
      * Return number of repetitions of an animated image
      *

diff --git a/tests/Drivers/Gd/CoreTest.php b/tests/Drivers/Gd/CoreTest.php
@@ -17,6 +17,17 @@ public function testNative(): void
         $this->assertInstanceOf(GdImage::class, $core->native());
     }
 
+    public function testAdd(): void
+    {
+        $gd1 = imagecreatetruecolor(3, 2);
+        $gd2 = imagecreatetruecolor(3, 2);
+        $core = new Core([new Frame($gd1)]);
+        $this->assertEquals(1, $core->count());
+        $result = $core->add(new Frame($gd2));
+        $this->assertEquals(2, $core->count());
+        $this->assertInstanceOf(Core::class, $result);
+    }
+
     public function testSetNative(): void
     {
         $gd1 = imagecreatetruecolor(3, 2);

diff --git a/tests/Drivers/Imagick/CoreTest.php b/tests/Drivers/Imagick/CoreTest.php
@@ -17,6 +17,17 @@ public function testConstructor(): void
         $this->assertInstanceOf(Core::class, $core);
     }
 
+    public function testAdd(): void
+    {
+        $imagick = new Imagick();
+        $imagick->newImage(100, 100, new ImagickPixel('red'));
+        $core = new Core($imagick);
+        $this->assertEquals(1, $core->count());
+        $result = $core->add(new Frame(clone $imagick));
+        $this->assertEquals(2, $core->count());
+        $this->assertInstanceOf(Core::class, $result);
+    }
+
     public function testCount(): void
     {
         $imagick = new Imagick();