<?php
/**
 * Jingga
 *
 * PHP Version 8.2
 *
 * @package   phpOMS\Math\Stochastic
 * @copyright Dennis Eichhorn
 * @license   OMS License 2.0
 * @version   1.0.0
 * @link      https://jingga.app
 */
declare(strict_types=1);

namespace phpOMS\Math\Stochastic;

use phpOMS\Math\Statistic\Average;
use phpOMS\Math\Statistic\MeasureOfDispersion;

/**
 * Naive bayes matching.
 *
 * @package phpOMS\Math\Stochastic
 * @license OMS License 2.0
 * @link    https://jingga.app
 * @since   1.0.0
 */
final class NaiveBayesClassifier
{
    /**
     * Dictionary of different criteria.
     *
     * @var array
     * @since 1.0.0
     */
    private array $dict = [];

    /**
     * Cached probabilities.
     *
     * @var array
     * @since 1.0.0
     */
    private array $probabilities = [
        'count'    => 0,
        'criteria' => [],
        'attr'     => [],
    ];

    /**
     * Train matches.
     *
     * @param string $criteria Criteria to match against
     * @param array  $matched  Matches
     *
     * @return void
     *
     * @since 1.0.0
     */
    public function train(string $criteria, array $matched) : void
    {
        if (!isset($this->probabilities['criteria'][$criteria])) {
            $this->probabilities['criteria'][$criteria] = [
                'count' => 0,
                'attr'  => [],
            ];
        }

        foreach ($matched as $dataset) {
            foreach ($dataset as $attr => $value) {
                if (!isset($this->dict[$criteria][$attr])) {
                    $this->dict[$criteria][$attr] = [
                        'type' => \is_array($value) ? 1 : 2,
                        'data' => [],
                    ];
                }

                if (!isset($this->probabilities['attr'][$attr])) {
                    $this->probabilities['attr'][$attr] = [
                        'count' => 0,
                        'data'  => [],
                    ];
                }

                if (!isset($this->probabilities['criteria'][$criteria]['attr'][$attr])) {
                    $this->probabilities['criteria'][$criteria]['attr'][$attr] = [
                        'count'    => 0,
                        'mean'     => 0,
                        'variance' => 0,
                    ];
                }

                if (\is_array($value)) {
                    foreach ($value as $word) {
                        if (!isset($this->dict[$criteria][$attr]['data'][$word])) {
                            $this->dict[$criteria][$attr]['data'][$word] = 0;
                        }

                        ++$this->dict[$criteria][$attr]['data'][$word];
                        ++$this->probabilities['attr'][$attr]['count'];
                    }
                } else {
                    $this->dict[$criteria][$attr]['data'][] = $value;

                    ++$this->probabilities['attr'][$attr]['count'];
                    ++$this->probabilities['criteria'][$criteria]['attr'][$attr]['count'];
                }
            }

            ++$this->probabilities['criteria'][$criteria]['count'];
            ++$this->probabilities['count'];
        }
    }

    /**
     * Check against matches.
     *
     * @param string $criteria Criteria to match against
     * @param array  $toMatch  Values to match
     * @param int    $minimum  Minimum amount of occurrences for consideration
     *
     * @return float
     *
     * @since 1.0.0
     */
    public function matchCriteria(string $criteria, array $toMatch, int $minimum = 3) : float
    {
        $this->preCalculateProbabilities($toMatch);

        $n = 0.0;
        foreach ($toMatch as $attr => $value) {
            if (!isset($this->dict[$criteria], $this->dict[$criteria][$attr])) {
                continue;
            }

            if (\is_array($value)) {
                /**
                 * @var string $word
                 */
                foreach ($value as $word) {
                    if (isset($this->dict[$criteria][$attr]['data'][$word])
                        && $this->dict[$criteria][$attr]['data'][$word] >= $minimum
                    ) {
                        $p = ($this->dict[$criteria][$attr]['data'][$word] / \array_sum($this->dict[$criteria][$attr]['data']))
                            * ($this->probabilities['criteria'][$criteria]['count'] / $this->probabilities['count'])
                            / $this->probabilities['attr'][$attr]['data'][$word];

                        $n += \log(1 - $p) - \log($p);
                    }
                }
            } else {
                $p = (1 / \sqrt(2 * \M_PI * $this->probabilities['criteria'][$criteria]['attr'][$attr]['variance'])
                        * \exp(-($value - $this->probabilities['criteria'][$criteria]['attr'][$attr]['mean']) ** 2
                            / (2 * $this->probabilities['criteria'][$criteria]['attr'][$attr]['variance'])
                        )
                    )
                    * ($this->probabilities['criteria'][$criteria]['count'] / $this->probabilities['count'])
                    / $this->probabilities['attr'][$attr]['data'];

                $n += \log(1 - $p) - \log($p);
            }
        }

        return 1 / (1 + \exp($n));
    }

    /**
     * Pre-calculate some probabilities used for the matching process
     *
     * @param array $toMatch Data to match. Some probabilities depend on the passed values.
     *
     * @return void
     *
     * @since 1.0.0
     */
    private function preCalculateProbabilities(array $toMatch) : void
    {
        $this->probabilities['attr'] = [];

        foreach ($this->dict as $criteria => $subDict) {
            foreach ($subDict as $attr => $valueArray) {
                if ($valueArray['type'] === 2) {
                    $this->probabilities['criteria'][$criteria]['attr'][$attr]['mean']     = Average::arithmeticMean($this->dict[$criteria][$attr]['data']);
                    $this->probabilities['criteria'][$criteria]['attr'][$attr]['variance'] = MeasureOfDispersion::sampleVariance(
                        $this->dict[$criteria][$attr]['data'],
                        $this->probabilities['criteria'][$criteria]['attr'][$attr]['mean']
                    );

                    if (!isset($this->probabilities['attr'][$attr])) {
                        $this->probabilities['attr'][$attr] = ['data' => 0.0];
                    }

                    $this->probabilities['attr'][$attr]['data'] += (1 / \sqrt(2 * \M_PI * $this->probabilities['criteria'][$criteria]['attr'][$attr]['variance'])
                            * \exp(-($toMatch[$attr] - $this->probabilities['criteria'][$criteria]['attr'][$attr]['mean']) ** 2
                                / (2 * $this->probabilities['criteria'][$criteria]['attr'][$attr]['variance'])
                            )
                        )
                        * ($this->probabilities['criteria'][$criteria]['count'] / $this->probabilities['count']);
                } else {
                    if (!isset($this->probabilities['attr'][$attr])) {
                        $this->probabilities['attr'][$attr] = ['data' => []];
                    }

                    foreach ($valueArray['data'] as $word => $_) {
                        if (!isset($this->dict[$criteria][$attr]['data'][$word])) {
                            continue;
                        }

                        if (!isset($this->probabilities['attr'][$attr]['data'][$word])) {
                            $this->probabilities['attr'][$attr]['data'][$word] = 0.0;
                        }

                        $this->probabilities['attr'][$attr]['data'][$word] += ($this->dict[$criteria][$attr]['data'][$word] / \array_sum($this->dict[$criteria][$attr]['data']))
                            * ($this->probabilities['criteria'][$criteria]['count'] / $this->probabilities['count']);
                    }
                }
            }
        }
    }
}