<?php

namespace WOE\Matrix\Operators;

use WOE\Matrix\Matrix;
use WOE\Matrix\Exception;
class DirectSum extends Operator
{
    /**
     * Execute the addition
     *
     * @param mixed $value The matrix or numeric value to add to the current base value
     * @return $this The operation object, allowing multiple additions to be chained
     * @throws Exception If the provided argument is not appropriate for the operation
     */
    public function execute($value): Operator
    {
        if (is_array($value)) {
            $value = new Matrix($value);
        }
        if ($value instanceof Matrix) {
            return $this->directSumMatrix($value);
        }
        throw new Exception('Invalid argument for addition');
    }
    /**
     * Execute the direct sum for a matrix
     *
     * @param \Matrix $value The numeric value to concatenate/direct sum with the current base value
     * @return $this The operation object, allowing multiple additions to be chained
     **/
    private function directSumMatrix($value): Operator
    {
        $originalColumnCount = count($this->matrix[0]);
        $originalRowCount = count($this->matrix);
        $valColumnCount = $value->columns;
        $valRowCount = $value->rows;
        $value = $value->toArray();
        for ($row = 0; $row < $this->rows; ++$row) {
            $this->matrix[$row] = array_merge($this->matrix[$row], array_fill(0, $valColumnCount, 0));
        }
        $this->matrix = array_merge($this->matrix, array_fill(0, $valRowCount, array_fill(0, $originalColumnCount, 0)));
        for ($row = $originalRowCount; $row < $originalRowCount + $valRowCount; ++$row) {
            array_splice($this->matrix[$row], $originalColumnCount, $valColumnCount, $value[$row - $originalRowCount]);
        }
        return $this;
    }
}