classes/Avatar.php

   1 <?php
   2 /**
   3  * Table Definition for avatar
   4  */
   5 require_once 'DB/DataObject.php';
   6
   7 class Avatar extends DB_DataObject
   8 {
   9     ###START_AUTOCODE
  10     /* the code below is auto generated do not remove the above tag */
  11
  12     public $__table = 'avatar';                          // table name
  13     public $profile_id;                      // int(4)  primary_key not_null
  14     public $original;                        // tinyint(1)
  15     public $width;                           // int(4)  primary_key not_null
  16     public $height;                          // int(4)  primary_key not_null
  17     public $mediatype;                       // varchar(32)   not_null
  18     public $filename;                        // varchar(255)
  19     public $url;                             // varchar(255)  unique_key
  20     public $created;                         // datetime()   not_null
  21     public $modified;                        // timestamp()   not_null default_CURRENT_TIMESTAMP
  22
  23     /* Static get */
  24     function staticGet($k,$v=NULL) { return DB_DataObject::staticGet('Avatar',$k,$v); }
  25
  26     /* the code above is auto generated do not remove the tag below */
  27     ###END_AUTOCODE
  28
  29         # We clean up the file, too
  30
  31         function delete() {
  32                 $filename = $this->filename;
  33                 if (parent::delete()) {
  34                         @unlink(common_avatar_path($filename));
  35                 }
  36         }
  37
  38         # Create and save scaled version of this avatar
  39         # XXX: maybe break into different methods
  40
  41         function scale($size) {
  42
  43                 $image_s = imagecreatetruecolor($size, $size);
  44                 $image_a = $this->to_image();
  45
  46                 $square = min($this->width, $this->height);
  47
  48                 imagecopyresampled($image_s, $image_a, 0, 0, 0, 0,
  49                                                    $size, $size, $square, $square);
  50
  51                 $ext = ($this->mediattype == 'image/jpeg') ? ".jpeg" : ".png";
  52
  53                 $filename = common_avatar_filename($this->profile_id, $ext, $size, common_timestamp());
  54
  55                 if ($this->mediatype == 'image/jpeg') {
  56                         imagejpeg($image_s, common_avatar_path($filename));
  57                 } else {
  58                         imagepng($image_s, common_avatar_path($filename));
  59                 }
  60
  61                 $scaled = DB_DataObject::factory('avatar');
  62                 $scaled->profile_id = $this->profile_id;
  63                 $scaled->width = $size;
  64                 $scaled->height = $size;
  65                 $scaled->original = false;
  66                 $scaled->mediatype = ($this->mediattype == 'image/jpeg') ? 'image/jpeg' : 'image/png';
  67                 $scaled->filename = $filename;
  68                 $scaled->url = common_avatar_url($filename);
  69                 $scaled->created = DB_DataObject_Cast::dateTime(); # current time
  70
  71                 if ($scaled->insert()) {
  72                         return $scaled;
  73                 } else {
  74                         return NULL;
  75                 }
  76         }
  77
  78         function to_image() {
  79                 $filepath = common_avatar_path($this->filename);
  80                 if ($this->mediatype == 'image/gif') {
  81                         return imagecreatefromgif($filepath);
  82                 } else if ($this->mediatype == 'image/jpeg') {
  83                         return imagecreatefromjpeg($filepath);
  84                 } else if ($this->mediatype == 'image/png') {
  85                         return imagecreatefrompng($filepath);
  86                 } else {
  87                         return NULL;
  88                 }
  89         }
  90 }