tilt.handler.get.TiltImageHandler.java Source code

Introduction

Here is the source code for tilt.handler.get.TiltImageHandler.java

Source

/*
 * This file is part of TILT.
 *
 *  TILT is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  TILT is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with TILT.  If not, see <http://www.gnu.org/licenses/>.
 *  (c) copyright Desmond Schmidt 2014
 */

package tilt.handler.get;

import calliope.core.database.Connector;
import calliope.core.database.Connection;
import java.io.ByteArrayInputStream;
import java.net.InetAddress;
import java.net.URLConnection;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import tilt.constants.ImageType;
import tilt.constants.Params;
import tilt.constants.Database;
import tilt.exception.TiltException;
import tilt.exception.ImageException;
import tilt.image.Picture;
import tilt.image.PictureRegistry;
import tilt.handler.TiltGetHandler;
import tilt.Utils;
import tilt.handler.post.Options;
import tilt.handler.post.TextIndex;
import org.json.simple.JSONObject;

/**
 * Handles requests for various image types, including linked images
 * @author desmond
 */
public class TiltImageHandler extends TiltGetHandler {
    String docid;
    String pageid;
    String url;
    ImageType imageType;

    public void handle(HttpServletRequest request, HttpServletResponse response, String urn) throws TiltException {
        try {
            imageType = ImageType.read(request.getParameter(Params.PICTYPE));
            docid = request.getParameter(Params.DOCID);
            pageid = request.getParameter(Params.PAGEID);
            url = request.getParameter(Params.URL);
            if ((docid != null && pageid != null) || url != null) {
                if (url == null || url.length() == 0)
                    url = Utils.getUrl(request.getServerName(), docid, pageid);
                Picture p = PictureRegistry.get(url);
                if (p == null) {
                    TextIndex text = null;
                    PictureRegistry.prune();
                    Double[][] coords = getCropRect(request.getServerName(), docid, pageid);
                    Options opts = Options.get(request.getServerName(), docid);
                    String imageUrl = Utils.getUrl(request.getServerName(), docid, pageid);
                    String url = composeTextUrl(request, docid, pageid);
                    String textParam = Utils.getFromUrl(url);
                    if (textParam != null)
                        text = new TextIndex(textParam, "en_GB");
                    InetAddress poster = getIPAddress(request);
                    p = new Picture(opts, imageUrl, text, coords, poster);
                }
                byte[] pic = null;
                switch (imageType) {
                case load:
                    p.load();
                case original:
                    pic = p.getOrigData();
                    break;
                case preflight:
                    pic = p.getPreflightData();
                    break;
                case greyscale:
                    pic = p.getGreyscaleData();
                    break;
                case twotone:
                    pic = p.getTwoToneData();
                    break;
                case cleaned:
                    pic = p.getCleanedData();
                    break;
                case reconstructed:
                    pic = p.getReconstructedData();
                    break;
                case baselines:
                    pic = p.getBaselinesData();
                    break;
                case words:
                    pic = p.getWordsData();
                    break;
                case link:
                    pic = p.getGreyscaleData();
                    break;
                }
                if (pic != null) {
                    ByteArrayInputStream bis = new ByteArrayInputStream(pic);
                    String mimeType = URLConnection.guessContentTypeFromStream(bis);
                    response.setContentType(mimeType);
                    ServletOutputStream sos = response.getOutputStream();
                    sos.write(pic);
                    sos.close();
                } else {
                    response.getOutputStream().println("<p>image " + docid + " not found</p>");
                }
            } else
                response.getOutputStream().println("<p>please specify a docid and pageid, or a url</p>");
        } catch (Exception e) {
            throw new ImageException(e);
        }
    }
}