<?php

namespace Symfony\Bundle\DoctrineFixturesBundle\Common\DataFixtures;

use Doctrine\Common\DataFixtures\FixtureInterface;

use Doctrine\Common\DataFixtures\Loader as BaseLoader;

use Symfony\Component\DependencyInjection\ContainerInterface;

use Symfony\Component\DependencyInjection\ContainerAwareInterface;

class Loader extends BaseLoader

{

    private $container;

    public function __construct(ContainerInterface $container)

    {

        $this->container = $container;

    }

    public function addFixture(FixtureInterface $fixture)

    {

        if ($fixture instanceof ContainerAwareInterface) {

            $fixture->setContainer($this->container);

        }

        parent::addFixture($fixture);

    }

}