]> git.mxchange.org Git - friendica.git/blobdiff - include/Photo.php
fix some whitespace issues
[friendica.git] / include / Photo.php
index 4d02b5c651206359d565598b193057924ba92a74..54db9278e4111bd3a8db4638231c9dedff15df08 100644 (file)
@@ -7,14 +7,34 @@ class Photo {
        private $width;
        private $height;
        private $valid;
+       private $type;
+       private $types;
+
+       /**
+        * supported mimetypes and corresponding file extensions
+        */
+       static function supportedTypes() {
+               $t = array();
+               $t['image/jpeg'] ='jpg';
+               if (imagetypes() & IMG_PNG) $t['image/png'] = 'png';
+               return $t;
+       }
+
+       public function __construct($data, $type="image/jpeg") {
 
-       public function __construct($data) {
+               $this->types = $this->supportedTypes();
+               if (!array_key_exists($type,$this->types)){
+                       $type='image/jpeg';
+               }
                $this->valid = false;
+               $this->type = $type;
                $this->image = @imagecreatefromstring($data);
                if($this->image !== FALSE) {
                        $this->width  = imagesx($this->image);
                        $this->height = imagesy($this->image);
                        $this->valid  = true;
+                       imagealphablending($this->image, false);
+                       imagesavealpha($this->image, true);
                }
        }
 
@@ -38,6 +58,13 @@ class Photo {
        public function getImage() {
                return $this->image;
        }
+       
+       public function getType() {
+               return $this->type;
+       }
+       public function getExt() {
+               return $this->types[$this->type];
+       }
 
        public function scaleImage($max) {
 
@@ -78,6 +105,9 @@ class Photo {
 
 
                $dest = imagecreatetruecolor( $dest_width, $dest_height );
+               imagealphablending($dest, false);
+               imagesavealpha($dest, true);
+               if ($this->type=='image/png') imagefill($dest, 0, 0, imagecolorallocatealpha($dest, 0, 0, 0, 127)); // fill with alpha
                imagecopyresampled($dest, $this->image, 0, 0, 0, 0, $dest_width, $dest_height, $width, $height);
                if($this->image)
                        imagedestroy($this->image);
@@ -87,6 +117,72 @@ class Photo {
 
        }
 
+       public function rotate($degrees) {
+               $this->image  = imagerotate($this->image,$degrees,0);
+               $this->width  = imagesx($this->image);
+               $this->height = imagesy($this->image);
+       }
+
+       public function flip($horiz = true, $vert = false) {
+               $w = imagesx($this->image);
+               $h = imagesy($this->image);
+               $flipped = imagecreate($w, $h);
+               if($horiz) {
+                       for ($x = 0; $x < $w; $x++) {
+                               imagecopy($flipped, $this->image, $x, 0, $w - $x - 1, 0, 1, $h);
+                       }
+               }
+               if($vert) {
+                       for ($y = 0; $y < $h; $y++) {
+                               imagecopy($flipped, $this->image, 0, $y, 0, $h - $y - 1, $w, 1);
+                       }
+               }
+               $this->image = $flipped;
+       }
+
+       public function orient($filename) {
+               // based off comment on http://php.net/manual/en/function.imagerotate.php
+
+               $exif = exif_read_data($filename);
+               $ort = $exif['Orientation'];
+
+               switch($ort)
+               {
+                       case 1: // nothing
+                       break;
+
+               case 2: // horizontal flip
+                   $this->flip();
+                   break;
+                               
+               case 3: // 180 rotate left
+               $this->rotate(180);
+                       break;
+                   
+               case 4: // vertical flip
+               $this->flip(false, true);
+                       break;
+               
+               case 5: // vertical flip + 90 rotate right
+               $this->flip(false, true);
+                $this->rotate(-90);
+                       break;
+               
+               case 6: // 90 rotate right
+                   $this->rotate(-90);
+                       break;
+               
+               case 7: // horizontal flip + 90 rotate right
+                   $this->flip();   
+                   $this->rotate(-90);
+                       break;
+               
+               case 8:    // 90 rotate left
+                   $this->rotate(90);
+                       break;
+           }
+       }
+
 
 
        public function scaleImageUp($min) {
@@ -128,6 +224,9 @@ class Photo {
 
 
                $dest = imagecreatetruecolor( $dest_width, $dest_height );
+               imagealphablending($dest, false);
+               imagesavealpha($dest, true);
+               if ($this->type=='image/png') imagefill($dest, 0, 0, imagecolorallocatealpha($dest, 0, 0, 0, 127)); // fill with alpha
                imagecopyresampled($dest, $this->image, 0, 0, 0, 0, $dest_width, $dest_height, $width, $height);
                if($this->image)
                        imagedestroy($this->image);
@@ -142,6 +241,9 @@ class Photo {
        public function scaleImageSquare($dim) {
 
                $dest = imagecreatetruecolor( $dim, $dim );
+               imagealphablending($dest, false);
+               imagesavealpha($dest, true);
+               if ($this->type=='image/png') imagefill($dest, 0, 0, imagecolorallocatealpha($dest, 0, 0, 0, 127)); // fill with alpha
                imagecopyresampled($dest, $this->image, 0, 0, 0, 0, $dim, $dim, $this->width, $this->height);
                if($this->image)
                        imagedestroy($this->image);
@@ -153,6 +255,9 @@ class Photo {
 
        public function cropImage($max,$x,$y,$w,$h) {
                $dest = imagecreatetruecolor( $max, $max );
+               imagealphablending($dest, false);
+               imagesavealpha($dest, true);
+               if ($this->type=='image/png') imagefill($dest, 0, 0, imagecolorallocatealpha($dest, 0, 0, 0, 127)); // fill with alpha
                imagecopyresampled($dest, $this->image, 0, 0, $x, $y, $max, $max, $w, $h);
                if($this->image)
                        imagedestroy($this->image);
@@ -162,20 +267,38 @@ class Photo {
        }
 
        public function saveImage($path) {
-               $quality = get_config('system','jpeg_quality');
-               if((! $quality) || ($quality > 100))
-                       $quality = JPEG_QUALITY;
-               imagejpeg($this->image,$path,$quality);
+               switch($this->type){
+                       case "image/png":
+                               $quality = get_config('system','png_quality');
+                               if((! $quality) || ($quality > 9))
+                                       $quality = PNG_QUALITY;
+                               imagepng($this->image, $path, $quality);
+                               break;
+                       default:
+                               $quality = get_config('system','jpeg_quality');
+                               if((! $quality) || ($quality > 100))
+                                       $quality = JPEG_QUALITY;
+                               imagejpeg($this->image,$path,$quality);
+               }
+               
        }
 
        public function imageString() {
                ob_start();
-
-               $quality = get_config('system','jpeg_quality');
-               if((! $quality) || ($quality > 100))
-                       $quality = JPEG_QUALITY;
-
-               imagejpeg($this->image,NULL,$quality);
+               switch($this->type){
+                       case "image/png":
+                               $quality = get_config('system','png_quality');
+                               if((! $quality) || ($quality > 9))
+                                       $quality = PNG_QUALITY;
+                               imagepng($this->image,NULL, $quality);
+                               break;
+                       default:
+                               $quality = get_config('system','jpeg_quality');
+                               if((! $quality) || ($quality > 100))
+                                       $quality = JPEG_QUALITY;
+
+                               imagejpeg($this->image,NULL,$quality);
+               }
                $s = ob_get_contents();
                ob_end_clean();
                return $s;
@@ -193,52 +316,147 @@ class Photo {
                else
                        $guid = get_guid();
 
-               $r = q("INSERT INTO `photo`
-                       ( `uid`, `contact-id`, `guid`, `resource-id`, `created`, `edited`, `filename`, `album`, `height`, `width`, `data`, `scale`, `profile`, `allow_cid`, `allow_gid`, `deny_cid`, `deny_gid` )
-                       VALUES ( %d, %d, '%s', '%s', '%s', '%s', '%s', '%s', %d, %d, '%s', %d, %d, '%s', '%s', '%s', '%s' )",
-                       intval($uid),
-                       intval($cid),
-                       dbesc($guid),
-                       dbesc($rid),
-                       dbesc(datetime_convert()),
-                       dbesc(datetime_convert()),
-                       dbesc(basename($filename)),
-                       dbesc($album),
-                       intval($this->height),
-                       intval($this->width),
-                       dbesc($this->imageString()),
-                       intval($scale),
-                       intval($profile),
-                       dbesc($allow_cid),
-                       dbesc($allow_gid),
-                       dbesc($deny_cid),
-                       dbesc($deny_gid)
+               $x = q("select id from photo where `resource-id` = '%s' and uid = %d and `contact-id` = %d and `scale` = %d limit 1",
+                               dbesc($rid),
+                               intval($uid),
+                               intval($cid),
+                               intval($scale)
                );
+               if(count($x)) {
+                       $r = q("UPDATE `photo`
+                               set `uid` = %d, 
+                               `contact-id` = %d, 
+                               `guid` = '%s', 
+                               `resource-id` = '%s', 
+                               `created` = '%s',
+                               `edited` = '%s', 
+                               `filename` = '%s', 
+                               `type` = '%s', 
+                               `album` = '%s', 
+                               `height` = %d, 
+                               `width` = %d, 
+                               `data` = '%s', 
+                               `scale` = %d, 
+                               `profile` = %d, 
+                               `allow_cid` = '%s', 
+                               `allow_gid` = '%s', 
+                               `deny_cid` = '%s',
+                               `deny_gid` = '%s' 
+                               where id = %d limit 1",
+
+                               intval($uid),
+                               intval($cid),
+                               dbesc($guid),
+                               dbesc($rid),
+                               dbesc(datetime_convert()),
+                               dbesc(datetime_convert()),
+                               dbesc(basename($filename)),
+                               dbesc($this->type),
+                               dbesc($album),
+                               intval($this->height),
+                               intval($this->width),
+                               dbesc($this->imageString()),
+                               intval($scale),
+                               intval($profile),
+                               dbesc($allow_cid),
+                               dbesc($allow_gid),
+                               dbesc($deny_cid),
+                               dbesc($deny_gid),
+                               intval($x[0]['id'])
+                       );
+               }
+               else {
+                       $r = q("INSERT INTO `photo`
+                               ( `uid`, `contact-id`, `guid`, `resource-id`, `created`, `edited`, `filename`, type, `album`, `height`, `width`, `data`, `scale`, `profile`, `allow_cid`, `allow_gid`, `deny_cid`, `deny_gid` )
+                               VALUES ( %d, %d, '%s', '%s', '%s', '%s', '%s', '%s', '%s', %d, %d, '%s', %d, %d, '%s', '%s', '%s', '%s' )",
+                               intval($uid),
+                               intval($cid),
+                               dbesc($guid),
+                               dbesc($rid),
+                               dbesc(datetime_convert()),
+                               dbesc(datetime_convert()),
+                               dbesc(basename($filename)),
+                               dbesc($this->type),
+                               dbesc($album),
+                               intval($this->height),
+                               intval($this->width),
+                               dbesc($this->imageString()),
+                               intval($scale),
+                               intval($profile),
+                               dbesc($allow_cid),
+                               dbesc($allow_gid),
+                               dbesc($deny_cid),
+                               dbesc($deny_gid)
+                       );
+               }
                return $r;
        }
+}}
 
 
+/**
+ * Guess image mimetype from filename or from Content-Type header
+ * 
+ * @arg $filename string Image filename
+ * @arg $fromcurl boolean Check Content-Type header from curl request
+ */
+function guess_image_type($filename, $fromcurl=false) {
+    logger('Photo: guess_image_type: '.$filename . ($fromcurl?' from curl headers':''), LOGGER_DEBUG);
+       $type = null;
+       if ($fromcurl) {
+               $a = get_app(); 
+               $headers=array();
+               $h = explode("\n",$a->get_curl_headers());
+               foreach ($h as $l) {
+                       list($k,$v) = array_map("trim", explode(":", trim($l), 2));
+                       $headers[$k] = $v;
+               }
+               if (array_key_exists('Content-Type', $headers))
+                       $type = $headers['Content-Type'];
+       }
+       if (is_null($type)){
+               $ext = pathinfo($filename, PATHINFO_EXTENSION);
+               $types = Photo::supportedTypes();
+               $type = "image/jpeg";
+               foreach ($types as $m=>$e){
+                       if ($ext==$e) $type = $m;
+               }
 
-
-
-}}
-
+       }
+    logger('Photo: guess_image_type: type='.$type, LOGGER_DEBUG);
+       return $type;
+       
+}
 
 function import_profile_photo($photo,$uid,$cid) {
 
        $a = get_app();
 
+       $r = q("select `resource-id` from photo where `uid` = %d and `contact-id` = %d and `scale` = 4 and `album` = 'Contact Photos' limit 1",
+               intval($uid),
+               intval($cid)
+       );
+       if(count($r)) {
+               $hash = $r[0]['resource-id'];
+       }
+       else {
+               $hash = photo_new_resource();
+       }
+               
        $photo_failure = false;
 
        $filename = basename($photo);
        $img_str = fetch_url($photo,true);
-       $img = new Photo($img_str);
+       
+       // guess mimetype from headers or filename
+       $type = guess_image_type($photo,true);
+
+       
+       $img = new Photo($img_str, $type);
        if($img->is_valid()) {
 
                $img->scaleImageSquare(175);
                                        
-               $hash = photo_new_resource();
-
                $r = $img->store($uid, $cid, $hash, $filename, 'Contact Photos', 4 );
 
                if($r === false)
@@ -258,11 +476,9 @@ function import_profile_photo($photo,$uid,$cid) {
                if($r === false)
                        $photo_failure = true;
 
-
-
-               $photo = $a->get_baseurl() . '/photo/' . $hash . '-4.jpg';
-               $thumb = $a->get_baseurl() . '/photo/' . $hash . '-5.jpg';
-               $micro = $a->get_baseurl() . '/photo/' . $hash . '-6.jpg';
+               $photo = $a->get_baseurl() . '/photo/' . $hash . '-4.' . $img->getExt();
+               $thumb = $a->get_baseurl() . '/photo/' . $hash . '-5.' . $img->getExt();
+               $micro = $a->get_baseurl() . '/photo/' . $hash . '-6.' . $img->getExt();
        }
        else
                $photo_failure = true;