add components for cropping tags

Author: cccraig

.gitignore

@@ -3,6 +3,7 @@
 defined('MUGSHOT_PATH') or die('Hacking attempt!');
 
 include_once(PHPWG_ROOT_PATH . 'admin/include/functions.php');
+include_once(MUGSHOT_PATH . 'include/nn.training.thumbnails.php');
 
 function add_mugshot_methods($arr) {
   $service = &$arr[0];
@@ -41,6 +42,22 @@ function book_mugshots($data, &$service) {
     $rm = pwg_db_real_escape_string($value['removeThis']);
     $tagName = ($tag == -1 && $name != '') ? tag_id_from_tag_name($name) : $tag;
 
+    // Create or remove the training thumbnails, depending on webmaster settings.
+    if ($plugin_config['autotag']) {
+      $sql = "SELECT * FROM `". IMAGES_TABLE . "` WHERE `id`=".$imageId.";";
+
+      $imgData = pwg_db_fetch_assoc(pwg_query($sql));
+      
+      // Remove or add cropped faces in the images to a directory.
+      if($rm == 0 && $width >= 40 && $height >= 40 && extension_loaded('imagick') === true) {
+        crop_image_faces($imgData['path'], $imgData['md5sum'], $name, $imgW, $imgH, $width, $height, $left, $top);
+      }
+    
+      if($rm == 1) {
+        delete_image_faces($name, $imgData['md5sum']);
+      }
+    }
+
     // Remove a mugshot
     if ($rm == 1) {
       $dString .= ($tag != '') ? $tag . ',' : '';

LICENSE

@@ -0,0 +1,89 @@
+<?php
+
+defined('MUGSHOT_PATH') or die('Hacking attempt!');
+
+// Check if the directory is empty
+function dir_is_empty($dir) {
+  if (!file_exists($dir)) {
+    return;
+  }
+  $handle = opendir($dir);
+  while (false !== ($entry = readdir($handle))) {
+    if ($entry != "." && $entry != "..") {
+      return FALSE;
+    }
+  }
+  return TRUE;
+}
+
+// Automatically rotate the image before cropping
+function auto_rotate_image($image) {
+  $o = $image->getImageOrientation();
+
+  switch ($o) {
+    case 3:
+      $image->rotateImage('#000', 180);
+      break;
+
+    case 6:
+      $image->rotateImage('#000', 90);
+      break;
+
+    case 8:
+      $image->rotateImage('#000', -90);
+      break;
+  }
+
+  $image->setImageOrientation(imagick::ORIENTATION_TOPLEFT);
+}
+
+// Crop faces in the image
+function crop_image_faces($p, $imgFileName, $name, $iw, $ih, $width, $height, $left, $top) {
+  try {
+    $image = new Imagick($p);
+    auto_rotate_image($image);
+    $image -> resizeImage($iw, $ih, Imagick::FILTER_LANCZOS,1);
+    $image -> cropImage($width, $height, $left, $top);
+    $struc = str_replace(' ', '_', strtolower($name));
+    $structure = getcwd()."/plugins/MugShot/training/".$struc;
+
+    if (!file_exists($structure)) {
+      mkdir($structure, 0775, true);
+    }
+
+    // If the file exists that means you have,
+    // 1) tagged someone twice in the same photo,
+    // 2) tagged someone in the same photo that is asociated with another album.
+    // In Either case, we don't want duplicate photos that are almost exactly the same cluttering up our training data.
+    if (!file_exists($structure.'/'.$imgFileName)) {
+      $image -> writeImage($structure.'/'.$imgFileName.'.jpg');
+      chmod($structure, 0760);
+    }
+
+  } catch (Exception $e) {
+    error_log($e->getMessage(), 0);
+  }
+}
+
+// Delete image face if the tag is deleted
+function delete_image_faces($name, $imgFileName) {
+  try {
+    $struc = str_replace(' ', '_', strtolower($name));
+    $structure = getcwd()."/plugins/MugShot/training/".$struc;
+    $remTest = false;
+    $dirTest = false;
+
+    if (file_exists($structure.'/'.$imgFileName)) {
+      $remTest = unlink($structure.'/'.$imgFileName.'.jpg');
+    }
+
+    if (dir_is_empty($structure)) {
+      $dirTest = rmdir($structure);
+    }
+
+  } catch (Exception $e) {
+    error_log($e->getMessage(), 0);
+  }
+}
+
+?>

README.md

@@ -12,6 +12,9 @@
 .labels {width: 200px;float: left;}
 .inputs {width: 200px;margin-left: 200px;}
 .ta {border-radius:5px 5px 5px 5px;}
+input[type="radio"] {
+  margin-right: 10px;
+}
 {/literal}{/html_style}
 
 {combine_script id='LocalStorageCache' load='footer' path='admin/themes/default/js/LocalStorageCache.js'}
@@ -45,7 +48,7 @@ jQuery(document).ready(function() {
       <p>
     {if count($groups) > 0}
         <strong>{'The following groups can use MugShot to tag people'|@translate}</strong>
-        <br></br>
+        <br><br>
         <select data-selectize="groups" data-value="{$groups_selected|@json_encode|escape:html}"
           placeholder="{'Type in a search term'|translate}"
           name="groups[]" multiple style="width:600px;"></select>
@@ -54,10 +57,25 @@ jQuery(document).ready(function() {
     {/if}
       </p>
 
+    <!-- Enable automatic cropping of tagged faces -->
+			<p><br>
+        <label for="autotag">
+        <strong>{'Enable cropping of faces for automated tagging'|@translate}</strong><br>
+        <i>This requires ImageMagick to be installed, including the PHP plugin</i><br><br>
+    			{if $autotag }
+    				<input type="radio" id="autotag1" name="autotag" value="1" checked /><b><i>ALLOW</i></b> MugShot to crop tagged faces from photos<br>
+    				<input type="radio" id="autotag2" name="autotag" value="0"/><b><i>DO NOT ALLOW</i></b> MugShot to crop tagged faces from photos<br>
+    			{else}
+    				<input type="radio" id="autotag1" name="autotag" value="1"/><b><i>ALLOW</i></b> MugShot to crop tagged faces from photos<br>
+    				<input type="radio" id="autotag2" name="autotag" value="0" checked /><b><i>DO NOT ALLOW</i></b> MugShot to crop tagged faces from photos<br>
+    			{/if}
+        </label>
+			</p><br>
+		</div>
+
     {if count($groups) > 0}
     	<p class="formButtons">
     		<input type="submit" value="{'Save'|@translate}" name="save" />
     	</p>
     {/if}
-		</div>
 </form>