drupol\phpartition\Partition

Complexity

Total Complexity

14

Size/Duplication

Total Lines	156
Duplicated Lines	0 %

Coupling/Cohesion

Dependencies

1

Test Coverage

Coverage

100%

Importance

Changes

0

10 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	1
A	addItem()	0	6	1
A	addItems()	0	8	2
A	getRawItems()	0	8	1
A	getWeight()	0	7	1
A	setAlgo()	0	4	1
A	getAlgo()	0	4	1
A	deleteItems()	0	6	2
A	delete()	0	4	1
A	sortByValue()	0	16	3

<?php

namespace drupol\phpartition;

use phootwork\collection\Map;

/**
 * Class Partition.
 *
 * @package drupol\phpartition
 */
class Partition extends Map
{

  /**
   * The algorithm in use.
   *
   * @var PartitionAlgorithmInterface
   */
    protected $algo;

  /**
   * Partition constructor.
   *
   * @param PartitionItem[] $elements
   *   The list of elements.
   */
    public function __construct(array $elements = array())
    {
        parent::__construct(array());
        $this->addItems($elements);
    }

  /**
   * Add an item to the partition.
   *
   * @param PartitionItem $item
   *   The item to add to the partition.
   *
   * @return $this
   *   Return itself.
   */
    public function addItem(PartitionItem $item)
    {
        $this->set(spl_object_hash($item), $item);

        return $this;
    }

  /**
   * Add items to the partition.
   *
   * @param PartitionItem[] $items
   *   The items to add to the partition.
   *
   * @return $this
   *   Return itself.
   */
    public function addItems(array $items = array())
    {
        foreach ($items as $item) {
            $this->addItem($item);
        }

        return $this;
    }

  /**
   * Get an array of original items.
   *
   * @return array
   *   The original items.
   */
    public function getRawItems()
    {
        return array_values(
            array_map(function ($item) {
                return $item->getItem();
            }, $this->toArray())
        );
    }

  /**
   * Get the weight of the partition.
   *
   * @return int
   *   The partition's weight.
   */
    public function getWeight()
    {
        return array_reduce($this->toArray(), function ($sum, $item) {
            $sum += $item->getValue();
            return $sum;
        });
    }

  /**
   * Set the algorithm to use.
   *
   * @param BasePartitionAlgorithm $algo
   *   The algorithm to use.
   */
    public function setAlgo(BasePartitionAlgorithm $algo)
    {
        $this->algo = $algo;
    }

  /**
   * Get the algorithm in use.
   *
   * @return PartitionAlgorithmInterface
   *   The algorithm in use.
   */
    public function getAlgo()
    {
        return $this->algo;
    }

  /**
   * Delete items from the partition.
   *
   * @param PartitionItem[] $items
   *   The items to delete.
   */
    public function deleteItems(array $items = array())
    {
        foreach ($items as $item) {
            $this->delete($item);
        }
    }

  /**
   * Delete an item from the partition.
   *
   * @param PartitionItem $item
   *   The item to delete.
   */
    public function delete(PartitionItem $item)
    {
        $this->remove(spl_object_hash($item));
    }

  /**
   * Sort the items of the partition in a particular order.
   *
   * @param string $order
   *   ASC for ascending, DESC for descending.
   *
   * @return $this
   */
    public function sortByValue($order = 'ASC')
    {
        if ('ASC' == $order) {
            $this->sort(function ($itemA, $itemB) {
                return $itemA->getValue() > $itemB->getValue();
            });
        }

        if ('DESC' == $order) {
            $this->sort(function ($itemA, $itemB) {
                return $itemA->getValue() < $itemB->getValue();
            });
        }

        return $this;
    }
}