1:  2:  3:  4:  5:  6:  7:  8:  9: 10: 11: 12: 13: 14: 15: 16: 17: 18: 19: 20: 21: 22: 23: 24: 25: 26: 27: 28: 29: 30: 31: 32: 33: 34: 35: 36: 37: 
<?php
namespace Omeka\Db\Event\Listener;

use Doctrine\Common\Persistence\Event\LoadClassMetadataEventArgs;

/**
 * Load the resource discriminator map dynamically.
 */
class ResourceDiscriminatorMap
{
    /**
     * @var array
     */
    protected $discriminatorMap;

    /**
     * Set the resource discriminator map.
     */
    public function __construct(array $discriminatorMap)
    {
        $this->discriminatorMap = $discriminatorMap;
    }

    /**
     * Attach the discriminator map to the Resource entity.
     *
     * @param LoadClassMetadataEventArgs $event
     */
    public function loadClassMetadata(LoadClassMetadataEventArgs $event)
    {
        $classMetadata = $event->getClassMetadata();
        if ('Omeka\Entity\Resource' == $classMetadata->name) {
            $classMetadata->discriminatorMap = $this->discriminatorMap;
        }
    }
}