design_area_colourise_image.inc

<?php
require_once SQ_CORE_PACKAGE_PATH.'/designs/design_area/design_area.inc';

class Design_Area_Colourise_Image extends Design_Area
{


    function __construct($assetid=0)
    {
        parent::__construct($assetid);

    }//end constructor


    public function setAttrValue($name, $value)
    {
        // if this is one of the colour attributes then let's grab the value, add it the
        // $this->colours and then save that array
        if ($name == 'black_colour' || $name == 'white_colour'
            || stristr($name, 'from_colour') || stristr($name, 'to_colour')) {
            $new_value = $this->vars['colours']['value'];
            if (!is_array($new_value)) $new_value = Array();
            $new_value[$name] = $value;
            $name = 'colours';
            $value = $new_value;
        }

        if ($name == 'type' && $value != 'multiple_colours') {
            $value = 'greyscale';
        }

        return parent::setAttrValue($name, $value);

    }//end setAttrValue()


    public function attr($name)
    {
        if (empty($this->vars['colours']['value'][$name])) {
            return parent::attr($name);
        } else {
            return $this->vars['colours']['value'][$name];
        }

    }//end attr()


    protected function _printPaintingCode()
    {
        if (!isset($this->_tmp['image'])) {
            trigger_error('no image');
            return;
        }
        $image = 'assets/'.$this->type().'/'.$this->id.'/'.$this->_tmp['image'];

        if ($this->attr('filename_only')) {
            // they only want us to echo out the filename
            echo '<', '?php echo $DATA_PATH; ?>/', $image;
        } else {
            // print out a full tag
            echo '<img src="<', '?php echo $DATA_PATH; ?', '>/', $image, '" ', $this->attr('extra'), ' />';
        }

    }//end _printPaintingCode()


    protected function _canInit()
    {
        $image_asset = $GLOBALS['DESIGN_BEING_GENERATED']->getAssociatedFile($this->attr('image'));
        if (is_null($image_asset)) {
            trigger_localised_error('CORE0184', E_USER_WARNING, $this->attr('id_name'), $this->attr('image'));
            return FALSE;
        }

        // check the source file exists
        $source = $image_asset->data_path.'/'.$image_asset->attr('name');
        if (!file_exists($source)) {
            trigger_localised_error('CORE0183', E_USER_WARNING, $this->attr('id_name'), $source);
            return FALSE;
        }

        if ($this->attr('type') == 'greyscale') {
            $colours = $this->attr('colours');
            if (!isset($colours['black_colour']) || !isset($colours['white_colour'])) {
                trigger_localised_error('CORE0186', E_USER_WARNING, $this->attr('id_name'));
                return FALSE;
            }
        }

        return TRUE;

    }//end _canInit()


    protected function _printInitCode()
    {
        require_once SQ_FUDGE_PATH.'/general/file_system.inc';
        require_once SQ_FUDGE_PATH.'/image/image_manip.inc';

        $image_asset = $GLOBALS['DESIGN_BEING_GENERATED']->getAssociatedFile($this->attr('image'));
        $source = $image_asset->data_path.'/'.$image_asset->attr('name');

        // dest file has design id so that if we are used by customisations they each have their own file
        $dest = $this->data_path_public.'/colourised_'.$GLOBALS['DESIGN_BEING_GENERATED']->id.'_'.$this->attr('image');

        if (!create_directory($this->data_path_public)) {
            trigger_localised_error('CORE0185', E_USER_WARNING, $this->attr('id_name'), $this->data_path_public);
            return;
        }

        $colours = $this->attr('colours');

        switch ($this->attr('type')) {

            case 'greyscale' :
                if (Image_Manip::gradientPalette($source, $dest, $colours['black_colour'], $colours['white_colour'])) {
                    $this->_tmp['image'] = basename($dest);
                }
            break;

            case 'multiple_colours' :
                $colour_map = Array();
                $i = 1;
                while (isset($colours['from_colour_'.$i]) && isset($colours['to_colour_'.$i])) {
                    $colour_map[$colours['from_colour_'.$i]] = $colours['to_colour_'.$i];
                    $i++;
                }
                if (Image_Manip::remapColour($source, $dest, $colour_map, $this->attr('tolerance'))) {
                    $this->_tmp['image'] = basename($dest);
                }
            break;

            default :
                trigger_localised_error('CORE0182', E_USER_WARNING, $this->attr('id_name'), $this->get_val('type'));

        }//end switch

    }//end _printInitCode()


    public function getProtectedAttrs()
    {
        $res = parent::getProtectedAttrs();
        $res[] = 'colours';
        return $res;

    }//end getProtectedAttrs()


}//end class
?>