Phraseanet/lib/Alchemy/Phrasea/Order/OrderViewModel.php

<?php
/*
 * This file is part of Phraseanet
 *
 * (c) 2005-2016 Alchemy
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Alchemy\Phrasea\Order;

use Alchemy\Phrasea\Model\Entities\Order;
use Assert\Assertion;

class OrderViewModel
{
    /**
     * @var Order
     */
    private $order;
    /**
     * @var OrderElementViewModel[]
     */
    private $viewElements = [];

    /**
     * @param Order $order
     */
    public function __construct(Order $order)
    {
        $this->order = $order;
    }

    /**
     * @param $viewElements
     */
    public function setViewElements($viewElements)
    {
        Assertion::allIsInstanceOf($viewElements, OrderElementViewModel::class);

        $this->viewElements = $viewElements instanceof \Traversable ? iterator_to_array($viewElements) : $viewElements;
    }

    /**
     * @return Order
     */
    public function getOrder()
    {
        return $this->order;
    }

    /**
     * @return OrderElementViewModel[]
     */
    public function getElements()
    {
        return $this->viewElements;
    }
}