/**
  * __construct function.
  *
  * @access public
  * @param  object $objAlbum
  * @param  string $type
  * @return void
  */
 public function __construct($objAlbum, $type)
 {
     if (is_numeric($objAlbum)) {
         $objAlbum = \Photoalbums2AlbumModel::findByPk($objAlbum);
     }
     if (!is_object($objAlbum)) {
         return false;
     }
     $this->objAlbum = $objAlbum;
     $this->type = $type;
     $this->setPreviewImageId();
 }
 /**
  * generatePalette function.
  *
  * @access public
  * @return void
  */
 public function generatePalette()
 {
     // Get album
     $objAlbum = \Photoalbums2AlbumModel::findByPk($this->Input->get('id'));
     // Remove from palette
     if ($objAlbum->previewImageType != 'select_preview_image') {
         $this->removeFromPalette('tl_photoalbums2_album', 'default', 'previewImage');
     }
     // Remove from palette
     if ($objAlbum->imageSortType != 'custom') {
         $this->removeFromPalette('tl_photoalbums2_album', 'default', 'imageSort');
     }
 }