Skip to content
GitLab
Select Git revision
  • 3.x
  • 1.x
  • 2.x default
  • 0.1.x
  • renderer
  • symfony3
  • 2.0.0
  • 1.0.0
  • 0.1.2
  • 0.1.1
  • 0.1
Find file BlameHistoryPermalink
ClassVisitor.php 2.90 KiB
1 
<?php
2
3 
/*
4 
 * © 2016 IRSTEA
5 
 * Guillaume Perréal <guillaume.perreal@irstea.fr>
6 
 * Tous droits réservés.
7 
 */
8
9 
namespace Irstea\PlantUmlBundle\Model;
10
11 
use Irstea\PlantUmlBundle\Model\Filter\AcceptAllFilter;
12 
use Irstea\PlantUmlBundle\Model\Namespace_\RootNamespace;
13 
use Irstea\PlantUmlBundle\Model\Node\Class_;
14 
use Irstea\PlantUmlBundle\Model\Node\Interface_;
15 
use Irstea\PlantUmlBundle\Model\Node\Trait_;
16 
use Irstea\PlantUmlBundle\Writer\WritableInterface;
17 
use Irstea\PlantUmlBundle\Writer\WritableNodeInterface;
18 
use Irstea\PlantUmlBundle\Writer\WriterInterface;
19 
use ReflectionClass;
20
21 
/**
22 
 * Description of Visitor
23 
 *
24 
 * @author Guillaume Perréal <guillaume.perreal@irstea.fr>
25 
 */
26 
class ClassVisitor implements ClassVisitorInterface, WritableInterface
27 
{
28 
    /**
29 
     * @var RootNamespace
30 
     */
31 
    protected $rootNamespace;
32
33 
    /**
34 
     * @var ClassFilterInterface
35 
     */
36 
    protected $filter;
37
38 
    /**
39 
     * @var DecoratorInterface
40 
     */
41 
    protected $decorator;
42
43 
    public function __construct(DecoratorInterface $decorator = null, ClassFilterInterface $filter = null)
44 
    {
45 
        $this->rootNamespace = new RootNamespace();
46 
        $this->filter = $filter ?: AcceptAllFilter::instance();
47 
        $this->decorator = $decorator ?: NullDecorator::instance();
48 
    }
49
50 
    /**
51 
     * @param ClassFilterInterface $filter
52 
     * @return self
53 
     */
54 
    public function setClassFilter(ClassFilterInterface $filter)
55 
    {
56 
        $this->filter = $filter;
57 
        return $this;
58 
    }
59
60 
    /**
61 
     * @param DecoratorInterface $decorator
62 
     * @return self
63 
     */
64 
    public function setDecorator(DecoratorInterface $decorator)
65 
    {
66 
        $this->decorator = $decorator;
67 
        return $this;
68 
    }
69
70 
    public function visitClass($className)
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
{ assert('is_string($className)', $className); if (isset($this->nodes[$className])) { return $this->nodes[$className]; } if (!$this->filter->accept($className)) { return $this->nodes[$className] = false; } return $this->createNode($className); } /** * * @param string $className * @return NodeInterface */ protected function createNode($className) { $class = new ReflectionClass($className); $namespace = $this->rootNamespace->getNamespace($class->getNamespaceName()); if ($class->isTrait()) { $node = new Trait_($namespace, $className); } elseif ($class->isInterface()) { $node = new Interface_($namespace, $className); } else { $node = new Class_($namespace, $className, $class->isAbstract(), $class->isFinal()); } $this->nodes[$class->getName()] = $node; $namespace->addNode($node); $this->decorator->decorate($className, $node, $this); return $node; } public function outputTo(WriterInterface $writer) { return $this->rootNamespace->outputTo($writer); } }