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

  * GetThumbNailController.java - The Spring controller that returns an image of a ThumbNail

  * 

  * PicMan - The BuckoSoft Picture Manager in Java

  * Copyright(c) 2005 - Dick Balaska

  * 

  */

 package com.buckosoft.PicMan.web;

 import java.awt.Dimension;

 import java.awt.Graphics2D;

 import java.awt.Image;

 import java.awt.image.BufferedImage;

 import java.io.IOException;

 import java.io.PrintWriter;

 import java.util.Iterator;

 import javax.imageio.ImageIO;

 import javax.imageio.ImageWriter;

 import javax.imageio.stream.ImageOutputStream;

 import javax.servlet.ServletException;

 import javax.servlet.ServletOutputStream;

 import javax.servlet.http.HttpServletRequest;

 import javax.servlet.http.HttpServletResponse;

 import org.apache.commons.logging.Log;

 import org.apache.commons.logging.LogFactory;

 import org.springframework.web.servlet.ModelAndView;

 import com.buckosoft.BSAccount.web.BSAccountPageController;

 import com.buckosoft.BSAccount.web.BSAccountUserWebSession;

 import com.buckosoft.PicMan.business.PicManFacade;

 import com.buckosoft.PicMan.db.DatabaseFacade;

 import com.buckosoft.PicMan.domain.Mosaic;

 import com.buckosoft.PicMan.domain.Pic;

 import com.buckosoft.PicMan.domain.Thumbnail;

 import com.buckosoft.PicMan.domain.User;

 /** The Spring controller that returns an image of a ThumbNail. <br>

  * @author Dick Balaska

  * @since 2005/08/01

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

  */

 public class GetThumbNailController extends BSAccountPageController {

         private static boolean DEBUG = false;

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

         private DatabaseFacade dbf;

         private PicManFacade pmf;

         /** Set the reference to the PicMan Database

          * @param dbf The DatabaseFacade

          */

         public void setDatabase(DatabaseFacade dbf) { this.dbf = dbf; }

         /** Set the reference to the PicMan API.

          * @param pmf The PicManFacade

          */

         public void setPicMan(PicManFacade pmf) { this.pmf = pmf; }

         /** Enable logger output on this module

          * @param debugFlag true == turn on debugging.

          */

         public void setDEBUG(boolean debugFlag) {

                 DEBUG = debugFlag;

         }

         /** Spring standard http request handler, returns an image/jpeg

          * @param request The http request

          * @param response The http response

          * @return null

          */ 

         public ModelAndView handleRequest(HttpServletRequest request, HttpServletResponse response)

                         throws ServletException, IOException {

                 String s;

                 Pic                pic = null;

                 String        label = null;

 //                boolean        importPic = false;        // special rules for reading import thumbs (don't cache because of temp filename)

                 Thumbnail tn = null;

                 BSAccountUserWebSession userWebSession = this.bsAccountMan.getUserWebSession(request);

                 User user;

 /*                if (!userWebSession.getUser().isRegisteredUser()) {

                         response.setContentType("text/plain");

                         response.sendError(401, "Not logged in");

                         return(null);

                 }

 */

                 user = (User)userWebSession.getUser();

                 s = request.getParameter("label");

                 if (s != null) {

                         label = s;

                 }

                 s = request.getParameter("moswork");

                 if (s != null) {

                         outMosaicPic(request, response);

                         return(null);

                 }

                 s = request.getParameter("posterMid");

                 if (s != null) {

                         outPosterMid(request, response);

                         return(null);

                 }

                 s = request.getParameter("importPic");

                 if (s != null) {

                         pic = new Pic(true, s);

                 }

                 if (pic == null) {

                         s = request.getParameter("pic");

                         if (s == null) {

                                 PrintWriter out = response.getWriter();

                                 out.print("No pic in request");

                                 return(null);

                         }

                 }

                 if (pic == null)

                         pic = dbf.getPic(s);

                 if (DEBUG) {

                         if (pic != null)

                                 logger.info("Getting thumb '" + pic.getName() + "'");

                         else

                                 logger.info("Getting Xthumb for '" + s + "'");

                 }

                 int height = dbf.getSystem().getThumbHeight();

                 s = request.getParameter("mosthumb");

                 if (s != null)

                         height = dbf.getSystem().getMosaicThumbHeight();

                 s = request.getParameter("home");

                 if (s != null) {

                         int width = pic.getWidth();

                         if (width == 0) {

                                 Dimension d = pmf.determinePicSize(pic);

                                 if (d.height == -1)

                                         width = height;

                                 else {

                                         logger.warn("Updating size for pic " + pic.getName());

                                         pic.setHeight(d.height);

                                         pic.setWidth(d.width);

                                         pmf.getDB().updatePic(pic);                                        

                                 }

                         } else {

                                 height = user.getHomePageThumbWidth() * pic.getHeight() / width;

                         }

                 }

                 s = request.getParameter("mos");        // mosaic testing

                 if (s != null)

                         tn = pmf.getTestMosaicThumbNail(pic, height, Integer.parseInt(s));

                 else

                         tn = pmf.getThumbNail(pic, height, label);

                 response.setContentType("image/jpeg");

                 ServletOutputStream sos = response.getOutputStream();

                 ImageOutputStream ios = ImageIO.createImageOutputStream(sos);

                 Iterator<ImageWriter> writers = ImageIO.getImageWritersByFormatName("jpg");

                 ImageWriter writer = (ImageWriter)writers.next();

                 writer.setOutput(ios);

                 writer.write(tn.getImage());

                 return(null);

         }

         private        void outMosaicPic(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

             BSAccountUserWebSession userWebSession = this.bsAccountMan.getUserWebSession(request);

             User user = (User)userWebSession.getUser();

                 response.setContentType("image/jpeg");

                 response.addHeader("Cache-Control", "max-age=0");

                 response.addHeader("Cache-Control", "no-cache");

                 response.addHeader("expires", "0");

                 response.addHeader("Expires", "Tue, 01 Jan 1980 1:00:00 GMT");

                 response.addHeader("Pragma", "no-cache");

                 ServletOutputStream sos = response.getOutputStream();

                 ImageOutputStream ios = ImageIO.createImageOutputStream(sos);

                 Iterator<ImageWriter> writers = ImageIO.getImageWritersByFormatName("jpg");

                 ImageWriter writer = (ImageWriter)writers.next();

                 writer.setOutput(ios);

                 BufferedImage small;

                 if (pmf.getMosaicMan() == null || pmf.getMosaicMan().getMosaicEngine() == null) {

                         small = pmf.getPicReader().getXThumb(dbf.getSystem().getThumbHeight()).getImage();

                 } else {

                         BufferedImage bi;

                         bi = pmf.getMosaicMan().getMosaicEngine().getImage(); 

                         int        width = bi.getWidth();

                         int height = bi.getHeight();

                         int        picHeight = user.getMosaicPreviewHeight();

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

                         int newW = (int)dW;

                         int newH = picHeight;

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

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

                         Graphics2D        g;

                         g = small.createGraphics();

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

                 }

                 writer.write(small);

         }

         private        void outPosterMid(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

 //           BSAccountUserWebSession userWebSession = this.bsAccountMan.getUserWebSession(request);

 //            User user = (User)userWebSession.getUser();

                 response.setContentType("image/jpeg");

                 response.addHeader("Cache-Control", "max-age=0");

                 response.addHeader("Cache-Control", "no-cache");

                 response.addHeader("expires", "0");

                 response.addHeader("Expires", "Tue, 01 Jan 1980 1:00:00 GMT");

                 response.addHeader("Pragma", "no-cache");

                 ServletOutputStream sos = response.getOutputStream();

                 ImageOutputStream ios = ImageIO.createImageOutputStream(sos);

                 Iterator<ImageWriter> writers = ImageIO.getImageWritersByFormatName("jpg");

                 ImageWriter writer = (ImageWriter)writers.next();

                 writer.setOutput(ios);

                 BufferedImage small;

                 String s = request.getParameter("posterMid");

                 Integer mid = Integer.parseInt(s);

                 Mosaic mosaic = pmf.getDB().getMosaic(mid);

                 String fname = mosaic.getOutPic();

                 BufferedImage bi = pmf.getPicReader().readPic(fname);

                 if (bi == null) {

                         small = pmf.getPicReader().getXThumb(dbf.getSystem().getThumbHeight()).getImage();

                 } else {

                         int        width = bi.getWidth();

                         int height = bi.getHeight();

                         int        picHeight = pmf.getDB().getThumbHeight();

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

                         int newW = (int)dW;

                         int newH = picHeight;

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

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

                         Graphics2D        g;

                         g = small.createGraphics();

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

                 }

                 writer.write(small);

         }

 }