ClassVisitor.php

<?php declare(strict_types=1);
/*
 * Copyright (C) 2016-2018 IRSTEA
 * All rights reserved.
 */

namespace Irstea\PlantUmlBundle\Model;

use Irstea\PlantUmlBundle\Model\Decorator\NullDecorator;
use Irstea\PlantUmlBundle\Model\Filter\AcceptAllFilter;
use Irstea\PlantUmlBundle\Model\Namespace_\Php\RootNamespace;
use Irstea\PlantUmlBundle\Model\Node\Class_;
use Irstea\PlantUmlBundle\Model\Node\Interface_;
use Irstea\PlantUmlBundle\Model\Node\Trait_;
use Irstea\PlantUmlBundle\Writer\WriterInterface;
use ReflectionClass;

/**
 * Description of Visitor.
 */
class ClassVisitor implements ClassVisitorInterface
{
    /**
     * @var NamespaceInterface
     */
    protected $rootNamespace;

    /**
     * @var ClassFilterInterface
     */
    protected $filter;

    /**
     * @var DecoratorInterface
     */
    protected $decorator;

    public function __construct(
        DecoratorInterface $decorator = null,
        ClassFilterInterface $filter = null,
        NamespaceInterface $namespace = null
    ) {
        $this->filter = $filter ?: AcceptAllFilter::instance();
        $this->decorator = $decorator ?: NullDecorator::instance();
        $this->rootNamespace = $namespace ?: new RootNamespace();
    }

    /**
     * @param ClassFilterInterface $filter
     *
     * @return self
     */
    public function setClassFilter(ClassFilterInterface $filter)
    {
        $this->filter = $filter;

        return $this;
    }

    /**
     * @param DecoratorInterface $decorator
     *
     * @return self
     */
    public function setDecorator(DecoratorInterface $decorator)
    {
        $this->decorator = $decorator;

        return $this;
    }