/******************************************************************************

  * PicReader.java - Process Pics into useful BufferedImages

  * 

  * PicMan - The BuckoSoft Picture Manager in Java

  * Copyright(c) 2005 - Dick Balaska

  * 

  */

 package com.buckosoft.PicMan.image;

 import java.awt.BasicStroke;

 import java.awt.Color;

 import java.awt.Dimension;

 import java.awt.Graphics2D;

 import java.awt.Image;

 import java.awt.geom.AffineTransform;

 import java.awt.image.AffineTransformOp;

 import java.awt.image.BufferedImage;

 import java.io.File;

 import java.util.Iterator;

 import java.util.concurrent.locks.ReentrantLock;

 import javax.imageio.ImageIO;

 import javax.imageio.ImageReader;

 import javax.imageio.stream.ImageInputStream;

 import org.apache.commons.logging.Log;

 import org.apache.commons.logging.LogFactory;

 import com.buckosoft.PicMan.business.PicManFacade;

 import com.buckosoft.PicMan.domain.Pic;

 import com.buckosoft.PicMan.domain.Thumbnail;

 import com.drew.imaging.ImageMetadataReader;

 import com.drew.metadata.Metadata;

 import com.drew.metadata.exif.ExifIFD0Directory;

 /** Process Pics into useful BufferedImages

  * @author Dick Balaska

  * @since 2005/08/01

  * @see <a href="http://cvs.buckosoft.com/Projects/PicMan/PicMan/src/main/java/com/buckosoft/PicMan/image/PicReader.java">PicReader.java</a>

  */

 public class PicReader {

         private        PicManFacade        pmf;

         private        ThumbCache                thumbCache;

         private QueueDepth queueDepth = new QueueDepth();

         private        ReentrantLock        lock = new ReentrantLock();

         private        static final boolean DEBUG = true;

         protected final Log log = LogFactory.getLog(getClass());

         public        void setPicMan(PicManFacade pmf) {

                 this.pmf = pmf;

         }

         /** Set the reference to the ThumbCache manager

          * @param thumbCache The thumbcache

          */

         public        void        setThumbCache(ThumbCache thumbCache) {

                 this.thumbCache = thumbCache;

         }

         public        PicManFacade getPicMan() { return(pmf); }

         /** Return how many people are waiting to read a pic 

          * @return The depth

          */

         public        int        getQueueDepth() { return(queueDepth.getDepth()); }

         public         Thumbnail        getThumbNail(Pic pic, int height, String label) {

                 Thumbnail tn = null;

                 if (pic == null)

                         return(getXThumb(height));

 //                if (label == null)

                 tn = thumbCache.getThumbNail(pic, height);

                 if (tn == null) {

                         enterQueue();

                         try {

                                 tn = readThumbNail(pic, height, label);

                         } catch (Throwable t) {

                                 exitQueue();

                                 RuntimeException e1 = new RuntimeException("Failed to read ThumbNail for \"" + pic.getName() + "\"", t);

                                 throw(e1);

                         }

                         exitQueue();

                 }

                 if (label != null) {

                         int xp = 0;

                         int yp = 0+tn.getImage().getHeight() - 1;

                         Graphics2D g = tn.getImage().createGraphics();

                         g.setColor(Color.BLACK);

                         g.drawString(label, xp+1, yp+1);

                         g.setColor(Color.WHITE);

                         g.drawString(label, xp,   yp);

                 }

                 return(tn);

         }

         /** Get the number of pictures files in a directory.

          * This method doesn't really belong here, but this is the only module 

          * that deals with files/pics in directories.

          * @param rid The Root id to query

          * @param dir The directory to read. i.e. "2003/20030428"

          * @return the count

          */

         public int getFilesInDirCount(int rid, String dir) {

                 String fullPath = pmf.getDB().getRoot(rid).getPath() + "/" + dir;

                 File f = new File(fullPath);

                 String list[] = f.list();

                 int count = 0;

                 int i;

                 if (list == null)

                         return(0);

                 for (i=0; i<list.length; i++) {

                         if (list[i].endsWith(".jpg"))

                                 count++;

                 }

                 return(count);

         }

         /** Read a file and create a thumbnail

          * @param pic

          * @param thumbHeight

          * @return

          */

         private Thumbnail readThumbNail(Pic pic, int thumbHeight, String label) {

                 String fullPath;

                 fullPath = getFullPath(pmf, pic);

                 if (DEBUG)

                         log.info("Read: " + pic.getPid() + ": " + fullPath);

                 Thumbnail tn;

                 // XXX: Need to determine file extension, not whack a ".jpg" on the end.

                 File f = null;

                 try {

                         f = new File(fullPath);

                 } catch (Exception e) {

                         return(getXThumb(thumbHeight));

                 }

                 Iterator<ImageReader> readers = ImageIO.getImageReadersByFormatName("jpg");

                 ImageReader reader = (ImageReader)readers.next();

                 BufferedImage bi;

                 Graphics2D        g;

                 if (DEBUG)

                         log.debug("Trying to read f=" + f);

                 ImageInputStream iis = null;

                 try {

                         iis = ImageIO.createImageInputStream(f);

                         if (DEBUG)

                                 log.debug("iis = " + iis);

                         if (iis == null) {

                                 return(getXThumb(thumbHeight));

                         }

                         reader.setInput(iis, true);

                         bi = reader.read(0);

                         iis.close();

                 } catch (IllegalStateException ise) {

                         if (DEBUG)

                                 log.info("caught IllegalStateException");

                         reader.dispose();

                         bi = null;

                         if (iis != null) {

                                 try {

                                         iis.close();

                                 } catch (Exception e) {}

                         }

                         return(getXThumb(thumbHeight));

                 } catch (Exception e) {

                         //if (DEBUG)

                                 log.info("caught Exception");

                         reader.dispose();

                         bi = null;

                         if (iis != null) {

                                 try {

                                         iis.close();

                                 } catch (Exception ex) {}

                         }

                         return(getXThumb(thumbHeight));

                 }

                 reader.dispose();

                 Metadata metadata = null;

                 try {

                         metadata = ImageMetadataReader.readMetadata(f);

 //                } catch (ImageProcessingException | IOException e) {

                 } catch (Exception e) {

                         // TODO Auto-generated catch block

                         e.printStackTrace();

                 }

                 log.debug("readThumbNail: File " + fullPath);

                 ExifIFD0Directory directory = metadata.getDirectory(ExifIFD0Directory.class);

                 String s = "--";

                 try {

                         s = directory.getString(ExifIFD0Directory.TAG_ORIENTATION);

                 } catch (Exception e) {}

                 int rot = 0;

                 try {

                         rot = directory.getInt(ExifIFD0Directory.TAG_ORIENTATION);

                 } catch (Exception e) {

                         //e.printStackTrace();

                 }

                 log.debug("   Orientation: '" + s + "' rot: " + rot);

 /*                for (Directory directory : metadata.getDirectories()) {

                     for (Tag tag : directory.getTags()) {

                         logger.info(tag);

                     }

                 }

 */        

 //                int        thumbHeight = pmf.getThumbHeight();

                 int        width = bi.getWidth();

                 int height = bi.getHeight();

                 double        dW = ((double)thumbHeight/(double)height) * width;

                 double        dH = ((double)thumbHeight/(double)width) * height;

                 int newW = (int)dW;

                 int newH = thumbHeight;

                 // logger.info("newW/H = " + newW + " / " + newH );

                 BufferedImage small;

                 if (rot == 6) {

                         log.debug("******* rotating 90");

                         small = new BufferedImage((int)dH, thumbHeight, BufferedImage.TYPE_INT_BGR);

                         BufferedImage bix;

                         bix = new BufferedImage(bi.getHeight(), bi.getWidth(), BufferedImage.TYPE_INT_BGR);

                         g = bix.createGraphics();

                         AffineTransform trans = new AffineTransform(0.0, 1.0, -1.0, 0.0, bi.getHeight(), 0.0);

                         g.transform(trans);

                         g.drawImage(bi, 0, 0, bi.getWidth(), bi.getHeight(), null);

                         bi = bix;                

                         g = small.createGraphics();

                         g.drawImage(bix.getScaledInstance(-1, thumbHeight, Image.SCALE_SMOOTH), 0, 0, small.getWidth(), small.getHeight(), null);

                 } else if (rot == 3) {

                         log.debug("******* rotating 180");

                         small = new BufferedImage((int)dW, thumbHeight, BufferedImage.TYPE_INT_BGR);

                         BufferedImage bix;

                         bix = new BufferedImage(bi.getWidth(), bi.getHeight(), BufferedImage.TYPE_INT_BGR);

                         //g = bix.createGraphics();

                         AffineTransform trans = AffineTransform.getScaleInstance(-1, -1);

                         trans.translate(-bix.getWidth(), -bix.getHeight());

                         AffineTransformOp op = new AffineTransformOp(trans, AffineTransformOp.TYPE_NEAREST_NEIGHBOR);

                         bix = op.filter(bi, null);

                         g = small.createGraphics();

                         g.drawImage(bix.getScaledInstance(-1, thumbHeight, Image.SCALE_SMOOTH), 0, 0, small.getWidth(), small.getHeight(), null);

                 } else if (rot == 8) {

                         log.debug("******* rotating -90");

                         small = new BufferedImage((int)dH, thumbHeight, BufferedImage.TYPE_INT_BGR);

                         BufferedImage bix;

                         bix = new BufferedImage(bi.getHeight(), bi.getWidth(), BufferedImage.TYPE_INT_BGR);

                         g = bix.createGraphics();

                         AffineTransform trans = new AffineTransform(0.0, 1.0, -1.0, 0.0, bi.getHeight(), 0.0);

                         g.transform(trans);

                         g.drawImage(bi, 0, 0, bi.getWidth(), bi.getHeight(), null);

                         bi = bix;                

                         bix = new BufferedImage(bi.getWidth(), bi.getHeight(), BufferedImage.TYPE_INT_BGR);

                         //g = bix.createGraphics();

                         trans = AffineTransform.getScaleInstance(-1, -1);

                         trans.translate(-bix.getWidth(), -bix.getHeight());

                         AffineTransformOp op = new AffineTransformOp(trans, AffineTransformOp.TYPE_NEAREST_NEIGHBOR);

                         bix = op.filter(bi, null);

                         g = small.createGraphics();

                         g.drawImage(bix.getScaledInstance(-1, thumbHeight, Image.SCALE_SMOOTH), 0, 0, small.getWidth(), small.getHeight(), null);

                 } else {

                         small = new BufferedImage(newW, newH, BufferedImage.TYPE_INT_BGR);

                         g = small.createGraphics();

                         g.drawImage(bi.getScaledInstance(-1, newH, Image.SCALE_SMOOTH), null, null);

                 }

                 tn = new Thumbnail();

                 tn.setName(pic.getName());

                 tn.setImage(small);

                 g = null;

                 bi = null;

                 if (!pic.isImportPic())

                         thumbCache.addToCache(pic, tn);

                 return(tn);

         }

         public        Thumbnail getXThumb(int thumbHeight) {

                 int z = thumbHeight;

                 Thumbnail tn = new Thumbnail();

                 BufferedImage small = new BufferedImage(z, z, BufferedImage.TYPE_INT_BGR);

                 Graphics2D g = small.createGraphics();

                 g.setBackground(Color.BLACK);

                 g.setColor(Color.RED);

 //                java.awt.geom.Rectangle2D.Double rd = new java.awt.geom.Rectangle2D.Double(-0.5, -0.5, 1, 1);

                 g.setStroke(new BasicStroke((float)5.0));

                 g.drawLine(0, 0, z, z);

                 g.drawLine(0, z, z, 0);

                 tn.setImage(small);

                 tn.setXThumb(true);

                 return(tn);

         }

         public static String getFullPath(PicManFacade pmf, Pic pic) {

                 String fullPath;

                 if (pic.isImportPic())

                         fullPath = pmf.getDB().getSystem().getImportDirectory() + "/" + pic.getName();

                 else

                         fullPath = pmf.getDB().getRoot(pic.getRid()).getPath() + "/" + pic.getLocation()

                                                 + "/" + pic.getName() + ".jpg";

                 return(fullPath);

         }

         private        void enterQueue() {

                 queueDepth.incDepth();

                 log.info("Enter New queue depth " + queueDepth.getDepth());

                 lock.lock();

         }

         private        void exitQueue() {

                 lock.unlock();

                 queueDepth.decDepth();

                 log.info("Exit  New queue depth " + queueDepth.getDepth());                

         }

         public Dimension determinePicSize(Pic pic) {

                 BufferedImage bi = readPic(pic);

                 if (bi == null)

                         return(new Dimension(-1, -1));

                 return(new Dimension(bi.getWidth(), bi.getHeight()));

         }

         public BufferedImage readPic(Pic pic) {

 //                return(readPic(pic, null, 1.0));

                 try {

                         return(readPic(getFullPath(pmf, pic)));

                 } catch (Exception e) {

                         e.printStackTrace();

                         return(null);

                 }

         }

         public BufferedImage readPic(String path) {

                 File f = null;

                 try {

                         f = new File(path);

                 } catch (Exception e) {

                         return(null);

                 }

                 Iterator<ImageReader> readers = ImageIO.getImageReadersByFormatName("jpg");

                 ImageReader reader = (ImageReader)readers.next();

                 BufferedImage bi;

                 if (DEBUG)

                         log.info("readPic: " + f);

                 ImageInputStream iis = null;

                 try {

                         iis = ImageIO.createImageInputStream(f);

                         //if (DEBUG)

                         //        log.info("iis = " + iis);

                         if (iis == null) {

                                 return(null);

                         }

                         reader.setInput(iis, true);

                         bi = reader.read(0);

                         iis.close();

                 } catch (IllegalStateException ise) {

                         if (DEBUG)

                                 log.info("caught IllegalStateException");

                         reader.dispose();

                         bi = null;

                         if (iis != null) {

                                 try {

                                         iis.close();

                                 } catch (Exception ex) {}

                         }

                         return(null);

                 } catch (Exception e) {

                         if (DEBUG)

                                 log.info("caught Exception");

                         reader.dispose();

                         bi = null;

                         if (iis != null) {

                                 try {

                                         iis.close();

                                 } catch (Exception ex) {}

                         }

                         return(null);

                 }

                 reader.dispose();

                 Metadata metadata = null;

                 try {

                         metadata = ImageMetadataReader.readMetadata(f);

 //                } catch (ImageProcessingException | IOException e) {

                 } catch (Throwable t) {

                         t.printStackTrace();

                 }

                 log.debug("readPic: File " + path);

                 int rot = 0;

                 String s = "--";

                 if (metadata != null) {

                         ExifIFD0Directory directory = metadata.getDirectory(ExifIFD0Directory.class);

                         try {

                                 s = directory.getString(ExifIFD0Directory.TAG_ORIENTATION);

                         } catch (Exception e) {}

                         try {

                                 rot = directory.getInt(ExifIFD0Directory.TAG_ORIENTATION);

                         } catch (Exception e) {

                                 log.debug("failed directory.getInt(ExifIFD0Directory.TAG_ORIENTATION) " + e.getLocalizedMessage());

                         }

                 }

                 log.debug("   Orientation: '" + s + "' rot: " + rot);

                 if (rot == 6 || rot == 8) {

                         log.info("******* rotating 90 ********");

                         BufferedImage bix;

                         bix = new BufferedImage(bi.getHeight(), bi.getWidth(), BufferedImage.TYPE_INT_BGR);

                         Graphics2D g = (Graphics2D)bix.createGraphics();

                         AffineTransform trans = new AffineTransform(0.0, 1.0, -1.0, 0.0, bi.getHeight(), 0.0);

                         g.transform(trans);

                         g.drawImage(bi, 0, 0, bi.getWidth(), bi.getHeight(), null);

                         bi = bix;                

 /*                } else if (rot == 8) {

                         log.info("******* rotating -90 ********");

                         BufferedImage bix;

                         bix = new BufferedImage(bi.getHeight(), bi.getWidth(), BufferedImage.TYPE_INT_BGR);

                         Graphics2D g = (Graphics2D)bix.createGraphics();

                         AffineTransform trans = new AffineTransform(0.0, 1.0, -1.0, 0.0, bi.getHeight(), 0.0);

                         g.transform(trans);

                         g.drawImage(bi, 0, 0, bi.getWidth(), bi.getHeight(), null);

                         bi = bix;                

 */                }

                  if (rot == 3 || rot == 8) {

                         log.debug("******* rotating 180 ");

                         BufferedImage bix;

                         bix = new BufferedImage(bi.getWidth(), bi.getHeight(), BufferedImage.TYPE_INT_BGR);

                         AffineTransform trans = AffineTransform.getScaleInstance(-1, -1);

                         trans.translate(-bix.getWidth(), -bix.getHeight());

                         AffineTransformOp op = new AffineTransformOp(trans, AffineTransformOp.TYPE_NEAREST_NEIGHBOR);

                         bix = op.filter(bi, null);

                         bi = bix;                

                 }

                 return(bi);

         }

         private class QueueDepth {

                 private int queueDepth = 0;

                 public void incDepth() {

                         synchronized(this) {

                                 queueDepth++;

                         }

                 }

                 public void decDepth() {

                         synchronized(this) {

                                 queueDepth--;

                         }

                 }

                 public int getDepth() {

                         synchronized(this) {

                                 return(queueDepth);

                         }

                 }

         }

 }