Java tutorial
/** * License Agreement for OpenSearchServer * * Copyright (C) 2013-2014 Emmanuel Keller / Jaeksoft * * http://www.open-search-server.com * * This file is part of OpenSearchServer. * * OpenSearchServer 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 3 of the License, or * (at your option) any later version. * * OpenSearchServer 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 OpenSearchServer. * If not, see <http://www.gnu.org/licenses/>. **/ package com.jaeksoft.searchlib.web.controller; import java.awt.Dimension; import java.awt.Image; import java.awt.Rectangle; import java.awt.geom.Rectangle2D; import java.awt.image.BufferedImage; import java.io.File; import java.io.FileInputStream; import java.io.FileNotFoundException; import java.io.IOException; import java.net.URI; import java.net.URISyntaxException; import java.net.URLDecoder; import java.util.ArrayList; import java.util.List; import javax.imageio.ImageIO; import javax.naming.NamingException; import jcifs.smb.NtlmPasswordAuthentication; import jcifs.smb.SmbFile; import org.apache.commons.io.FilenameUtils; import org.apache.commons.lang3.StringUtils; import org.apache.pdfbox.exceptions.CryptographyException; import org.apache.pdfbox.pdmodel.PDDocument; import org.icepdf.core.exceptions.PDFException; import org.icepdf.core.exceptions.PDFSecurityException; import org.icepdf.core.pobjects.Document; import org.icepdf.core.pobjects.PDimension; import org.icepdf.core.pobjects.Page; import org.icepdf.core.pobjects.graphics.text.LineText; import org.icepdf.core.pobjects.graphics.text.PageText; import org.icepdf.core.pobjects.graphics.text.WordText; import org.icepdf.core.util.GraphicsRenderingHints; import org.zkoss.bind.annotation.AfterCompose; import org.zkoss.bind.annotation.Command; import org.zkoss.bind.annotation.NotifyChange; import org.zkoss.zul.Filedownload; import com.jaeksoft.searchlib.Client; import com.jaeksoft.searchlib.ClientCatalog; import com.jaeksoft.searchlib.Logging; import com.jaeksoft.searchlib.SearchLibException; import com.jaeksoft.searchlib.crawler.web.spider.HttpDownloadThread; import com.jaeksoft.searchlib.ocr.HocrPdf; import com.jaeksoft.searchlib.ocr.HocrPdf.HocrPage; import com.jaeksoft.searchlib.renderer.RendererResult; import com.jaeksoft.searchlib.renderer.plugin.AuthPluginInterface; import com.jaeksoft.searchlib.util.ExecuteUtils.ExecutionException; import com.jaeksoft.searchlib.util.GhostScript; import com.jaeksoft.searchlib.util.IOUtils; import com.jaeksoft.searchlib.util.ImageUtils; import com.jaeksoft.searchlib.util.pdfbox.PDFBoxHighlighter; @AfterCompose(superclass = true) public class ViewerController extends CommonController { private int page; private int numberOfPages; private float zoom; private final static int[] zoomScale = { 10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 150, 200, 250, 300, 400 }; private HttpDownloadThread downloadThread; private File tempFile; private BufferedImage currentImage; private URI uri; private String search; private String[] keywords; private HocrPdf hocrPdf; private GhostScript ghostScript = null; public ViewerController() throws SearchLibException, IOException, NamingException, URISyntaxException { super(); hocrPdf = null; downloadThread = null; currentImage = null; Client client = null; numberOfPages = 0; page = 1; zoom = 100; String h = getRequestParameter("h"); RendererResult result = null; if (h != null) { String p = getRequestParameter("p"); Integer hashCode = Integer.parseInt(h); Integer pos = Integer.parseInt(p); result = ClientCatalog.getRendererResults().find(hashCode); if (result == null) return; client = result.getClient(); RendererResult.Item item = result.getItem(pos); uri = new URI(item.getUrl()); setSearch(result.getKeywords()); hocrPdf = item.getHocrPdf(); } else { String index = getRequestParameter("index"); String u = getRequestParameter("uri"); setSearch(getRequestParameter("search")); uri = new URI(u); client = ClientCatalog.getClient(index); } if ("file".equalsIgnoreCase(uri.getScheme())) { try { tempFile = new File(uri); } catch (Exception e) { Logging.warn(e); tempFile = new File(uri.getPath()); } } else if ("smb".equalsIgnoreCase(uri.getScheme())) { tempFile = File.createTempFile("oss_pdf_viewer", ".pdf"); NtlmPasswordAuthentication auth = null; SmbFile smbFile; String url = URLDecoder.decode(uri.toString(), "UTF-8"); if (result != null) { AuthPluginInterface.User user = result.getLoggedUser(); if (user != null && user.password != null) auth = new NtlmPasswordAuthentication(result.getAuthDomain(), user.username, user.password); else if (result.isAuthCredential()) auth = new NtlmPasswordAuthentication(result.getAuthDomain(), result.getAuthUsername(), result.getAuthPassword()); } smbFile = auth != null ? new SmbFile(url, auth) : new SmbFile(url); IOUtils.copy(smbFile.getInputStream(), tempFile, true); } else { tempFile = File.createTempFile("oss_pdf_viewer", ".pdf"); downloadThread = new HttpDownloadThread(client, uri, tempFile, true); downloadThread.execute(180); } } @Override protected void reset() throws SearchLibException { } /** * @return the page */ public int getPage() { return page; } /** * @param page * the page to set * @throws SearchLibException */ @NotifyChange({ "currentImage", "page" }) public void setPage(int page) { if (page < 1) page = 1; if (page > numberOfPages) page = numberOfPages; if (page == this.page) return; this.page = page; currentImage = null; } /** * @return the zoom */ public int getZoom() { return (int) zoom; } /** * @param zoom * the zoom to set */ @NotifyChange({ "currentImage", "zoom" }) public void setZoom(int zoom) { this.zoom = zoom; currentImage = null; } @Command @NotifyChange({ "currentImage", "page" }) public void onPageUp() { setPage(page + +1); } @Command @NotifyChange({ "currentImage", "page" }) public void onPageDown() { setPage(page - 1); } @Command @NotifyChange({ "currentImage", "zoom" }) public void onZoomUp() { for (int zc : zoomScale) { if (zc > zoom) { setZoom(zc); return; } } } @Command @NotifyChange({ "currentImage", "zoom" }) public void onZoomDown() { int lastzc = zoomScale[0]; for (int zc : zoomScale) { if (zc >= zoom) { setZoom(lastzc); return; } lastzc = zc; } } @Command @NotifyChange("*") public void onTimer() { } @Command public void onDownload() throws FileNotFoundException { Filedownload.save(new FileInputStream(tempFile), null, FilenameUtils.getName(uri.getPath())); } public boolean isDownloading() { if (downloadThread == null) return false; return downloadThread.isRunning(); } public boolean isDownloaded() { return !isDownloading(); } public String getMessage() { if (downloadThread == null) return null; Exception e = downloadThread.getException(); if (e != null) return e.getMessage(); if (downloadThread.isRunning()) return "Downloading"; return uri.toString(); } public int getDownloadPercent() { if (downloadThread == null) return 0; return downloadThread.getPercent(); } private void checkHocrHighlight(float pageWidth, float pageHeight, List<Rectangle> boxList) { if (hocrPdf == null || keywords == null) return; HocrPage page = hocrPdf.getPage(this.page - 1); float xFactor = pageWidth / page.getPageWidth(); float yFactor = pageHeight / page.getPageHeight(); if (page != null) for (String keyword : keywords) page.addBoxes(keyword, boxList, xFactor, yFactor); } private void checkIcePdfHighlight(Document pdf, float pageHeight, float zoomFactor, List<Rectangle> boxList) { if (keywords == null) return; PageText pageText = pdf.getPageViewText(this.page - 1); if (pageText != null) { List<LineText> lines = pageText.getPageLines(); if (lines != null) { for (LineText lineText : lines) { for (WordText wordText : lineText.getWords()) { for (String keyword : keywords) if (keyword.equalsIgnoreCase(wordText.getText())) { Rectangle2D.Float rectf = wordText.getBounds(); Rectangle rect = new Rectangle(); rect.x = (int) (rectf.x * zoomFactor); rect.y = (int) ((pageHeight - rectf.y - rectf.height) * zoomFactor); rect.width = (int) (rectf.width * zoomFactor); rect.height = (int) (rectf.height * zoomFactor); boxList.add(rect); break; } } } } } } private void loadIcePdf() throws PDFException, PDFSecurityException, IOException { Document pdf = null; try { int pdfPage = page - 1; pdf = new Document(); pdf.setFile(tempFile.getAbsolutePath()); PDimension pd = pdf.getPageDimension(pdfPage, 0.0f); float zoomFactor = zoom / 100; float pageWidth = pd.getWidth(); float pageHeight = pd.getHeight(); List<Rectangle> boxList = new ArrayList<Rectangle>(0); checkIcePdfHighlight(pdf, pageHeight, zoomFactor, boxList); currentImage = (BufferedImage) pdf.getPageImage(pdfPage, GraphicsRenderingHints.SCREEN, Page.BOUNDARY_CROPBOX, 0.0f, zoom / 100); checkHocrHighlight(pageWidth, pageHeight, boxList); ImageUtils.yellowHighlight(currentImage, boxList, 0.1F); numberOfPages = pdf.getNumberOfPages(); } finally { if (pdf != null) pdf.dispose(); } } /** * gs -q -dFirstPage=3 -dLastPage=3 -dNOPAUSE -dBATCH -sDEVICE=jpeg -r72 * -dEPSCrop -sOutputFile=out.jpg "document.pdf" * * @throws IOException * @throws InterruptedException */ private void loadGS(String pdfPassword) throws IOException, InterruptedException { if (ghostScript == null) ghostScript = new GhostScript(null); File imageFile = null; try { imageFile = File.createTempFile("oss-viewer", ".png"); float zoomFactor = zoom / 100; int factor = (int) ((zoomFactor * zoom)); if (factor > 2400) factor = 2400; ghostScript.generateImage(pdfPassword, page, tempFile, factor, imageFile); currentImage = ImageIO.read(imageFile); } finally { if (imageFile != null) if (imageFile.exists()) imageFile.delete(); } } private void checkPdfBoxHighlight(PDDocument document, List<Rectangle> boxList) throws IOException { PDFBoxHighlighter pdfstripper = new PDFBoxHighlighter(keywords, boxList, new Dimension(currentImage.getWidth(), currentImage.getHeight())); pdfstripper.setStartPage(page); pdfstripper.setEndPage(page); pdfstripper.getText(document); } private void loadPdfBox() throws IOException, CryptographyException, SearchLibException, InterruptedException { PDDocument document = null; try { document = PDDocument.loadNonSeq(tempFile, null); // Trying to open with empty password boolean isEncrypted = document.isEncrypted(); if (isEncrypted) document.decrypt(""); loadGS(isEncrypted ? "" : null); List<Rectangle> boxList = new ArrayList<Rectangle>(0); checkPdfBoxHighlight(document, boxList); checkHocrHighlight(currentImage.getWidth(), currentImage.getHeight(), boxList); ImageUtils.yellowHighlight(currentImage, boxList, 0.1F); numberOfPages = document.getNumberOfPages(); } finally { if (document != null) IOUtils.close(document); } } public Image getCurrentImage() throws SearchLibException { if (currentImage != null) return currentImage; if (downloadThread != null) { if (downloadThread.isRunning()) return null; if (downloadThread.getException() != null) return null; if (!downloadThread.isDownloadSuccess()) return null; } if (tempFile == null) return null; try { loadPdfBox(); return currentImage; } catch (ExecutionException e) { Logging.warn(e.getCommandLine()); Logging.warn(e.getReturnedText()); } catch (Exception e) { Logging.warn(e); } try { loadIcePdf(); return currentImage; } catch (Exception e) { Logging.error(e); throw new SearchLibException("Unable to generate the image.", e); } } public int getImageWidth() { if (currentImage == null) return 0; return currentImage.getWidth(null); } public int getImageHeight() { if (currentImage == null) return 0; return currentImage.getHeight(null); } /** * @return the search */ public String getSearch() { return search; } /** * @param search * the search to set */ @NotifyChange("currentImage") public void setSearch(String search) { this.search = search; keywords = StringUtils.split(search); currentImage = null; } }