lib/imagefile.php

   1 <?php
   2 /**
   3  * StatusNet, the distributed open-source microblogging tool
   4  *
   5  * Abstraction for an image file
   6  *
   7  * PHP version 5
   8  *
   9  * LICENCE: This program is free software: you can redistribute it and/or modify
  10  * it under the terms of the GNU Affero General Public License as published by
  11  * the Free Software Foundation, either version 3 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU Affero General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Affero General Public License
  20  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21  *
  22  * @category  Image
  23  * @package   StatusNet
  24  * @author    Evan Prodromou <evan@status.net>
  25  * @author    Zach Copley <zach@status.net>
  26  * @copyright 2008-2009 StatusNet, Inc.
  27  * @license   http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
  28  * @link      http://status.net/
  29  */
  30
  31 if (!defined('GNUSOCIAL')) { exit(1); }
  32
  33 /**
  34  * A wrapper on uploaded files
  35  *
  36  * Makes it slightly easier to accept an image file from upload.
  37  *
  38  * @category Image
  39  * @package  StatusNet
  40  * @author   Evan Prodromou <evan@status.net>
  41  * @author   Zach Copley <zach@status.net>
  42  * @license  http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
  43  * @link     http://status.net/
  44  */
  45
  46 class ImageFile
  47 {
  48     var $id;
  49     var $filepath;
  50     var $barename;
  51     var $type;
  52     var $height;
  53     var $width;
  54     var $rotate=0;  // degrees to rotate for properly oriented image (extrapolated from EXIF etc.)
  55     var $animated = false;  // Animated image? (has more than 2 frames)
  56
  57     function __construct($id=null, $filepath=null, $type=null, $width=null, $height=null)
  58     {
  59         $this->id = $id;
  60         $this->filepath = $filepath;
  61
  62         $info = @getimagesize($this->filepath);
  63
  64         if (!(
  65             ($info[2] == IMAGETYPE_GIF && function_exists('imagecreatefromgif')) ||
  66             ($info[2] == IMAGETYPE_JPEG && function_exists('imagecreatefromjpeg')) ||
  67             $info[2] == IMAGETYPE_BMP ||
  68             ($info[2] == IMAGETYPE_WBMP && function_exists('imagecreatefromwbmp')) ||
  69             ($info[2] == IMAGETYPE_XBM && function_exists('imagecreatefromxbm')) ||
  70             ($info[2] == IMAGETYPE_PNG && function_exists('imagecreatefrompng')))) {
  71
  72             // TRANS: Exception thrown when trying to upload an unsupported image file format.
  73             throw new UnsupportedMediaException(_('Unsupported image format.'), $this->filepath);
  74         }
  75
  76         $this->type = ($info) ? $info[2]:$type;
  77         $this->width = ($info) ? $info[0]:$width;
  78         $this->height = ($info) ? $info[1]:$height;
  79
  80         if ($this->type == IMAGETYPE_JPEG && function_exists('exif_read_data')) {
  81             // Orientation value to rotate thumbnails properly
  82             $exif = exif_read_data($this->filepath);
  83             if (is_array($exif) && isset($exif['Orientation'])) {
  84                 switch ((int)$exif['Orientation']) {
  85                 case 1: // top is top
  86                     $this->rotate = 0;
  87                     break;
  88                 case 3: // top is bottom
  89                     $this->rotate = 180;
  90                     break;
  91                 case 6: // top is right
  92                     $this->rotate = -90;
  93                     break;
  94                 case 8: // top is left
  95                     $this->rotate = 90;
  96                     break;
  97                 }
  98                 // If we ever write this back, Orientation should be set to '1'
  99             }
 100         }
 101
 102         Event::handle('FillImageFileMetadata', array($this));
 103     }
 104
 105     public static function fromFileObject(File $file)
 106     {
 107         $imgPath = null;
 108         $media = common_get_mime_media($file->mimetype);
 109         if (Event::handle('CreateFileImageThumbnailSource', array($file, &$imgPath, $media))) {
 110             switch ($media) {
 111             case 'image':
 112                 $imgPath = $file->getPath();
 113                 break;
 114             default:
 115                 throw new UnsupportedMediaException(_('Unsupported media format.'), $file->getPath());
 116             }
 117         }
 118
 119         if (!file_exists($imgPath)) {
 120             throw new ServerException(sprintf('Image not available locally: %s', $imgPath));
 121         }
 122
 123         try {
 124             $image = new ImageFile($file->id, $imgPath);
 125         } catch (UnsupportedMediaException $e) {
 126             // Avoid deleting the original
 127             if ($imgPath != $file->getPath()) {
 128                 unlink($imgPath);
 129             }
 130             throw $e;
 131         }
 132         return $image;
 133     }
 134
 135     public function getPath()
 136     {
 137         if (!file_exists($this->filepath)) {
 138             throw new ServerException('No file in ImageFile filepath');
 139         }
 140
 141         return $this->filepath;
 142     }
 143
 144     static function fromUpload($param='upload')
 145     {
 146         switch ($_FILES[$param]['error']) {
 147          case UPLOAD_ERR_OK: // success, jump out
 148             break;
 149
 150          case UPLOAD_ERR_INI_SIZE:
 151          case UPLOAD_ERR_FORM_SIZE:
 152             // TRANS: Exception thrown when too large a file is uploaded.
 153             // TRANS: %s is the maximum file size, for example "500b", "10kB" or "2MB".
 154             throw new Exception(sprintf(_('That file is too big. The maximum file size is %s.'), ImageFile::maxFileSize()));
 155
 156          case UPLOAD_ERR_PARTIAL:
 157             @unlink($_FILES[$param]['tmp_name']);
 158             // TRANS: Exception thrown when uploading an image and that action could not be completed.
 159             throw new Exception(_('Partial upload.'));
 160
 161          case UPLOAD_ERR_NO_FILE:
 162             // No file; probably just a non-AJAX submission.
 163          default:
 164             common_log(LOG_ERR, __METHOD__ . ": Unknown upload error " . $_FILES[$param]['error']);
 165             // TRANS: Exception thrown when uploading an image fails for an unknown reason.
 166             throw new Exception(_('System error uploading file.'));
 167         }
 168
 169         $info = @getimagesize($_FILES[$param]['tmp_name']);
 170
 171         if (!$info) {
 172             @unlink($_FILES[$param]['tmp_name']);
 173             // TRANS: Exception thrown when uploading a file as image that is not an image or is a corrupt file.
 174             throw new UnsupportedMediaException(_('Not an image or corrupt file.'), '[deleted]');
 175         }
 176
 177         return new ImageFile(null, $_FILES[$param]['tmp_name']);
 178     }
 179
 180     /**
 181      * Compat interface for old code generating avatar thumbnails...
 182      * Saves the scaled file directly into the avatar area.
 183      *
 184      * @param int $size target width & height -- must be square
 185      * @param int $x (default 0) upper-left corner to crop from
 186      * @param int $y (default 0) upper-left corner to crop from
 187      * @param int $w (default full) width of image area to crop
 188      * @param int $h (default full) height of image area to crop
 189      * @return string filename
 190      */
 191     function resize($size, $x = 0, $y = 0, $w = null, $h = null)
 192     {
 193         $targetType = $this->preferredType();
 194         $outname = Avatar::filename($this->id,
 195                                     image_type_to_extension($targetType),
 196                                     $size,
 197                                     common_timestamp());
 198         $outpath = Avatar::path($outname);
 199         $this->resizeTo($outpath, array('width'=>$size, 'height'=>$size,
 200                                         'x'=>$x,        'y'=>$y,
 201                                         'w'=>$w,        'h'=>$h));
 202         return $outname;
 203     }
 204
 205     /**
 206      * Copy the image file to the given destination.
 207      * For obscure formats, this will automatically convert to PNG;
 208      * otherwise the original file will be copied as-is.
 209      *
 210      * @param string $outpath
 211      * @return string filename
 212      */
 213     function copyTo($outpath)
 214     {
 215         return $this->resizeTo($outpath, array('width' =>$this->width,
 216                                                'height'=>$this->height));
 217     }
 218
 219     /**
 220      * Create and save a thumbnail image.
 221      *
 222      * @param string $outpath
 223      * @param array $box    width, height, boundary box (x,y,w,h) defaults to full image
 224      * @return string full local filesystem filename
 225      */
 226     function resizeTo($outpath, array $box=array())
 227     {
 228         $box['width'] = isset($box['width']) ? intval($box['width']) : $this->width;
 229         $box['height'] = isset($box['height']) ? intval($box['height']) : $this->height;
 230         $box['x'] = isset($box['x']) ? intval($box['x']) : 0;
 231         $box['y'] = isset($box['y']) ? intval($box['y']) : 0;
 232         $box['w'] = isset($box['w']) ? intval($box['w']) : $this->width;
 233         $box['h'] = isset($box['h']) ? intval($box['h']) : $this->height;
 234
 235         // Doublecheck that parameters are sane and integers.
 236         if ($box['width'] < 1 || $box['width'] > common_config('thumbnail', 'maxsize')
 237                 || $box['height'] < 1 || $box['height'] > common_config('thumbnail', 'maxsize')
 238                 || $box['w'] < 1 || $box['x'] >= $this->width
 239                 || $box['h'] < 1 || $box['y'] >= $this->height) {
 240             // Fail on bad width parameter. If this occurs, it's due to algorithm in ImageFile->scaleToFit
 241             common_debug("Boundary box parameters for resize of {$this->filepath} : ".var_export($box,true));
 242             throw new ServerException('Bad thumbnail size parameters.');
 243         }
 244
 245         if (!file_exists($this->filepath)) {
 246             // TRANS: Exception thrown during resize when image has been registered as present, but is no longer there.
 247             throw new Exception(_('Lost our file.'));
 248         }
 249
 250         // Don't crop/scale if it isn't necessary
 251         if ($box['width'] === $this->width
 252             && $box['height'] === $this->height
 253             && $box['x'] === 0
 254             && $box['y'] === 0
 255             && $box['w'] === $this->width
 256             && $box['h'] === $this->height
 257             && $this->type == $this->preferredType()) {
 258
 259             @copy($this->filepath, $outpath);
 260             return $outpath;
 261         }
 262
 263         if (Event::handle('StartResizeImageFile', array($this, $outpath, $box))) {
 264             $this->resizeToFile($outpath, $box);
 265         }
 266
 267         if (!file_exists($outpath)) {
 268             throw new UseFileAsThumbnailException($this->id);
 269         }
 270
 271         return $outpath;
 272     }
 273
 274     protected function resizeToFile($outpath, array $box)
 275     {
 276         switch ($this->type) {
 277         case IMAGETYPE_GIF:
 278             $image_src = imagecreatefromgif($this->filepath);
 279             break;
 280         case IMAGETYPE_JPEG:
 281             $image_src = imagecreatefromjpeg($this->filepath);
 282             break;
 283         case IMAGETYPE_PNG:
 284             $image_src = imagecreatefrompng($this->filepath);
 285             break;
 286         case IMAGETYPE_BMP:
 287             $image_src = imagecreatefrombmp($this->filepath);
 288             break;
 289         case IMAGETYPE_WBMP:
 290             $image_src = imagecreatefromwbmp($this->filepath);
 291             break;
 292         case IMAGETYPE_XBM:
 293             $image_src = imagecreatefromxbm($this->filepath);
 294             break;
 295         default:
 296             // TRANS: Exception thrown when trying to resize an unknown file type.
 297             throw new Exception(_('Unknown file type'));
 298         }
 299
 300         if ($this->rotate != 0) {
 301             $image_src = imagerotate($image_src, $this->rotate, 0);
 302         }
 303
 304         $image_dest = imagecreatetruecolor($box['width'], $box['height']);
 305
 306         if ($this->type == IMAGETYPE_GIF || $this->type == IMAGETYPE_PNG || $this->type == IMAGETYPE_BMP) {
 307
 308             $transparent_idx = imagecolortransparent($image_src);
 309
 310             if ($transparent_idx >= 0) {
 311
 312                 $transparent_color = imagecolorsforindex($image_src, $transparent_idx);
 313                 $transparent_idx = imagecolorallocate($image_dest, $transparent_color['red'], $transparent_color['green'], $transparent_color['blue']);
 314                 imagefill($image_dest, 0, 0, $transparent_idx);
 315                 imagecolortransparent($image_dest, $transparent_idx);
 316
 317             } elseif ($this->type == IMAGETYPE_PNG) {
 318
 319                 imagealphablending($image_dest, false);
 320                 $transparent = imagecolorallocatealpha($image_dest, 0, 0, 0, 127);
 321                 imagefill($image_dest, 0, 0, $transparent);
 322                 imagesavealpha($image_dest, true);
 323
 324             }
 325         }
 326
 327         imagecopyresampled($image_dest, $image_src, 0, 0, $box['x'], $box['y'], $box['width'], $box['height'], $box['w'], $box['h']);
 328
 329         switch ($this->preferredType()) {
 330          case IMAGETYPE_GIF:
 331             imagegif($image_dest, $outpath);
 332             break;
 333          case IMAGETYPE_JPEG:
 334             imagejpeg($image_dest, $outpath, common_config('image', 'jpegquality'));
 335             break;
 336          case IMAGETYPE_PNG:
 337             imagepng($image_dest, $outpath);
 338             break;
 339          default:
 340             // TRANS: Exception thrown when trying resize an unknown file type.
 341             throw new Exception(_('Unknown file type'));
 342         }
 343
 344         imagedestroy($image_src);
 345         imagedestroy($image_dest);
 346     }
 347
 348
 349     /**
 350      * Several obscure file types should be normalized to PNG on resize.
 351      *
 352      * @fixme consider flattening anything not GIF or JPEG to PNG
 353      * @return int
 354      */
 355     function preferredType()
 356     {
 357         if($this->type == IMAGETYPE_BMP) {
 358             //we don't want to save BMP... it's an inefficient, rare, antiquated format
 359             //save png instead
 360             return IMAGETYPE_PNG;
 361         } else if($this->type == IMAGETYPE_WBMP) {
 362             //we don't want to save WBMP... it's a rare format that we can't guarantee clients will support
 363             //save png instead
 364             return IMAGETYPE_PNG;
 365         } else if($this->type == IMAGETYPE_XBM) {
 366             //we don't want to save XBM... it's a rare format that we can't guarantee clients will support
 367             //save png instead
 368             return IMAGETYPE_PNG;
 369         }
 370         return $this->type;
 371     }
 372
 373     function unlink()
 374     {
 375         @unlink($this->filepath);
 376     }
 377
 378     static function maxFileSize()
 379     {
 380         $value = ImageFile::maxFileSizeInt();
 381
 382         if ($value > 1024 * 1024) {
 383             $value = $value/(1024*1024);
 384             // TRANS: Number of megabytes. %d is the number.
 385             return sprintf(_m('%dMB','%dMB',$value),$value);
 386         } else if ($value > 1024) {
 387             $value = $value/1024;
 388             // TRANS: Number of kilobytes. %d is the number.
 389             return sprintf(_m('%dkB','%dkB',$value),$value);
 390         } else {
 391             // TRANS: Number of bytes. %d is the number.
 392             return sprintf(_m('%dB','%dB',$value),$value);
 393         }
 394     }
 395
 396     static function maxFileSizeInt()
 397     {
 398         return min(ImageFile::strToInt(ini_get('post_max_size')),
 399                    ImageFile::strToInt(ini_get('upload_max_filesize')),
 400                    ImageFile::strToInt(ini_get('memory_limit')));
 401     }
 402
 403     static function strToInt($str)
 404     {
 405         $unit = substr($str, -1);
 406         $num = substr($str, 0, -1);
 407
 408         switch(strtoupper($unit)){
 409          case 'G':
 410             $num *= 1024;
 411          case 'M':
 412             $num *= 1024;
 413          case 'K':
 414             $num *= 1024;
 415         }
 416
 417         return $num;
 418     }
 419
 420     public function scaleToFit($maxWidth=null, $maxHeight=null, $crop=null)
 421     {
 422         return self::getScalingValues($this->width, $this->height,
 423                                         $maxWidth, $maxHeight, $crop, $this->rotate);
 424     }
 425
 426     /*
 427      * Gets scaling values for images of various types. Cropping can be enabled.
 428      *
 429      * Values will scale _up_ to fit max values if cropping is enabled!
 430      * With cropping disabled, the max value of each axis will be respected.
 431      *
 432      * @param $width    int Original width
 433      * @param $height   int Original height
 434      * @param $maxW     int Resulting max width
 435      * @param $maxH     int Resulting max height
 436      * @param $crop     int Crop to the size (not preserving aspect ratio)
 437      */
 438     public static function getScalingValues($width, $height,
 439                                         $maxW=null, $maxH=null,
 440                                         $crop=null, $rotate=0)
 441     {
 442         $maxW = $maxW ?: common_config('thumbnail', 'width');
 443         $maxH = $maxH ?: common_config('thumbnail', 'height');
 444
 445         if ($maxW < 1 || ($maxH !== null && $maxH < 1)) {
 446             throw new ServerException('Bad parameters for ImageFile::getScalingValues');
 447         } elseif ($maxH === null) {
 448             // if maxH is null, we set maxH to equal maxW and enable crop
 449             $maxH = $maxW;
 450             $crop = true;
 451         }
 452
 453         // Because GD doesn't understand EXIF orientation etc.
 454         if (abs($rotate) == 90) {
 455             $tmp = $width;
 456             $width = $height;
 457             $height = $tmp;
 458         }
 459
 460         // Cropping data (for original image size). Default values, 0 and null,
 461         // imply no cropping and with preserved aspect ratio (per axis).
 462         $cx = 0;    // crop x
 463         $cy = 0;    // crop y
 464         $cw = null; // crop area width
 465         $ch = null; // crop area height
 466
 467         if ($crop) {
 468             $s_ar = $width / $height;
 469             $t_ar = $maxW / $maxH;
 470
 471             $rw = $maxW;
 472             $rh = $maxH;
 473
 474             // Source aspect ratio differs from target, recalculate crop points!
 475             if ($s_ar > $t_ar) {
 476                 $cx = floor($width / 2 - $height * $t_ar / 2);
 477                 $cw = ceil($height * $t_ar);
 478             } elseif ($s_ar < $t_ar) {
 479                 $cy = floor($height / 2 - $width / $t_ar / 2);
 480                 $ch = ceil($width / $t_ar);
 481             }
 482         } else {
 483             $rw = $maxW;
 484             $rh = ceil($height * $rw / $width);
 485
 486             // Scaling caused too large height, decrease to max accepted value
 487             if ($rh > $maxH) {
 488                 $rh = $maxH;
 489                 $rw = ceil($width * $rh / $height);
 490             }
 491         }
 492         return array(intval($rw), intval($rh),
 493                     intval($cx), intval($cy),
 494                     is_null($cw) ? $width : intval($cw),
 495                     is_null($ch) ? $height : intval($ch));
 496     }
 497 }
 498
 499 //PHP doesn't (as of 2/24/2010) have an imagecreatefrombmp so conditionally define one
 500 if(!function_exists('imagecreatefrombmp')){
 501     //taken shamelessly from http://www.php.net/manual/en/function.imagecreatefromwbmp.php#86214
 502     function imagecreatefrombmp($p_sFile)
 503     {
 504         //    Load the image into a string
 505         $file    =    fopen($p_sFile,"rb");
 506         $read    =    fread($file,10);
 507         while(!feof($file)&&($read<>""))
 508             $read    .=    fread($file,1024);
 509
 510         $temp    =    unpack("H*",$read);
 511         $hex    =    $temp[1];
 512         $header    =    substr($hex,0,108);
 513
 514         //    Process the header
 515         //    Structure: http://www.fastgraph.com/help/bmp_header_format.html
 516         if (substr($header,0,4)=="424d")
 517         {
 518             //    Cut it in parts of 2 bytes
 519             $header_parts    =    str_split($header,2);
 520
 521             //    Get the width        4 bytes
 522             $width            =    hexdec($header_parts[19].$header_parts[18]);
 523
 524             //    Get the height        4 bytes
 525             $height            =    hexdec($header_parts[23].$header_parts[22]);
 526
 527             //    Unset the header params
 528             unset($header_parts);
 529         }
 530
 531         //    Define starting X and Y
 532         $x                =    0;
 533         $y                =    1;
 534
 535         //    Create newimage
 536         $image            =    imagecreatetruecolor($width,$height);
 537
 538         //    Grab the body from the image
 539         $body            =    substr($hex,108);
 540
 541         //    Calculate if padding at the end-line is needed
 542         //    Divided by two to keep overview.
 543         //    1 byte = 2 HEX-chars
 544         $body_size        =    (strlen($body)/2);
 545         $header_size    =    ($width*$height);
 546
 547         //    Use end-line padding? Only when needed
 548         $usePadding        =    ($body_size>($header_size*3)+4);
 549
 550         //    Using a for-loop with index-calculation instaid of str_split to avoid large memory consumption
 551         //    Calculate the next DWORD-position in the body
 552         for ($i=0;$i<$body_size;$i+=3)
 553         {
 554             //    Calculate line-ending and padding
 555             if ($x>=$width)
 556             {
 557                 //    If padding needed, ignore image-padding
 558                 //    Shift i to the ending of the current 32-bit-block
 559                 if ($usePadding)
 560                     $i    +=    $width%4;
 561
 562                 //    Reset horizontal position
 563                 $x    =    0;
 564
 565                 //    Raise the height-position (bottom-up)
 566                 $y++;
 567
 568                 //    Reached the image-height? Break the for-loop
 569                 if ($y>$height)
 570                     break;
 571             }
 572
 573             //    Calculation of the RGB-pixel (defined as BGR in image-data)
 574             //    Define $i_pos as absolute position in the body
 575             $i_pos    =    $i*2;
 576             $r        =    hexdec($body[$i_pos+4].$body[$i_pos+5]);
 577             $g        =    hexdec($body[$i_pos+2].$body[$i_pos+3]);
 578             $b        =    hexdec($body[$i_pos].$body[$i_pos+1]);
 579
 580             //    Calculate and draw the pixel
 581             $color    =    imagecolorallocate($image,$r,$g,$b);
 582             imagesetpixel($image,$x,$height-$y,$color);
 583
 584             //    Raise the horizontal position
 585             $x++;
 586         }
 587
 588         //    Unset the body / free the memory
 589         unset($body);
 590
 591         //    Return image-object
 592         return $image;
 593     }
 594 }   // if(!function_exists('imagecreatefrombmp'))