Example usage for com.itextpdf.text.pdf.parser PdfImageObject getFileType

List of usage examples for com.itextpdf.text.pdf.parser PdfImageObject getFileType

Introduction

In this page you can find the example usage for com.itextpdf.text.pdf.parser PdfImageObject getFileType.

Prototype

public String getFileType() 

Source Link

Usage

From source file:com.github.naofum.epubconverter.ReadPdf.java

License:Open Source License

public void renderImage(ImageRenderInfo renderInfo) {
    try {/*from w  w  w.j  a  v a 2 s  . c o m*/
        // Get image
        PdfImageObject image = renderInfo.getImage();
        if (image != null) {
            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            baos.write(image.getImageAsBytes());
            baos.flush();
            baos.close();

            // Check image type
            String imageType = image.getFileType();
            if (imageType == "png" || imageType == "gif" || imageType == "jpg") {
                if (imageType == "jpg") {
                    imageType = "jpeg";
                }

                // Save to ePUB
                String imageName = String.format("image%s.%s", renderInfo.getRef().getNumber(), imageType);
                if (!WriteZip.addImage("OEBPS/" + imageName, baos.toByteArray())) {
                    ReadPdf.imageList.add(imageName);
                }
            }

        }
    } catch (IOException e) {
        System.err.println("Failed to extract image (renderListener) " + e.getMessage());
    } catch (OutOfMemoryError e) {
        System.err.println("Out of memory in image extraction (renderListener) " + e.getMessage());
    }
}

From source file:cz.muni.pdfjbim.pdf.MyImageRenderListener.java

License:Apache License

/**
 * @see com.itextpdf.text.pdf.parser.RenderListener#renderImage(
 *     com.itextpdf.text.pdf.parser.ImageRenderInfo)
 *///w  w  w  . j  ava  2s .co  m
public void renderImage(ImageRenderInfo renderInfo) {
    try {
        String filename;

        PdfImageObject image = renderInfo.getImage();
        if (image == null) {
            return;
        }
        filename = String.format(path, renderInfo.getRef().getNumber(), image.getFileType());
        try (FileOutputStream os = new FileOutputStream(filename)) {
            os.write(image.getImageAsBytes());
            os.flush();
        } catch (IOException ex) {
            log.warn("IOException occurred when storing image object to file");
        }
    } catch (IOException e) {
        log.warn("IOException occurred when rendering pdf image object", e);
    }
}

From source file:org.gmdev.pdftrick.engine.ImageListenerShowThumb.java

License:Open Source License

/**
 * extracts the images contained in selected page and convert it in a BufferedImage to add and show to the center panel.
 * @param renderInfo//  w ww  .j  a v  a  2 s  .co m
 */
private void render(ImageRenderInfo renderInfo) {
    final HashMap<Integer, String> rotationFromPages = factory.getRotationFromPages();
    boolean isInline = false;

    PdfImageObject image = null;
    try {
        BufferedImage buffImg = null;
        if (renderInfo.getRef() == null) {
            isInline = true;
            inlineImageCounter += 1;
        }
        try {
            image = renderInfo.getImage();
        } catch (UnsupportedPdfException updfe) {
            try {
                if (isInline) {
                    buffImg = null;
                } else {
                    buffImg = CustomExtraImgReader.readIndexedPNG(renderInfo.getRef().getNumber(),
                            factory.getResultFile());
                }
            } catch (Exception e) {
                logger.error("Exception", e);
                PdfTrickMessages.append("ERROR", Consts.SENDLOG_MSG);
                unsupportedImage++;
                return;
            }
        }

        if (image != null) {
            BufferedImage buffPic = null;
            try {
                // if image is JBIG2 type i need a custom way and using jbig2 ImageIO plugin
                if (image.getFileType().equalsIgnoreCase("JBIG2")) {
                    buffPic = CustomExtraImgReader.readJBIG2(image);
                } else {
                    buffPic = image.getBufferedImage();
                }
            } catch (IIOException iioex) {
                byte[] imageByteArray = image.getImageAsBytes();
                try {
                    buffPic = CustomExtraImgReader.readCMYK_JPG(imageByteArray);
                } catch (Exception e) {
                    logger.error("Exception", e);
                    PdfTrickMessages.append("ERROR", Consts.SENDLOG_MSG);
                    unsupportedImage++;
                    return;
                }
            } catch (Exception e) {
                logger.error("Exception", e);
                PdfTrickMessages.append("ERROR", Consts.SENDLOG_MSG);
                unsupportedImage++;
                return;
            }

            // check if image contains a mask image
            BufferedImage buffMask = null;
            PdfDictionary imageDictionary = image.getDictionary();
            PRStream maskStream = (PRStream) imageDictionary.getAsStream(PdfName.SMASK);

            if (maskStream != null) {
                // if i have an smak object i check that is not a jpeg format, because this may cause some problem on offscreen rendering
                // usually all imges with mask are png ... and there aren't problems, if image is jpg i discard the mask :)
                if (!image.getFileType().equalsIgnoreCase("jpg") && buffPic != null) {
                    PdfImageObject maskImage = new PdfImageObject(maskStream);
                    buffMask = maskImage.getBufferedImage();
                    Image img = PdfTrickUtils.TransformGrayToTransparency(buffMask);
                    buffImg = PdfTrickUtils.ApplyTransparency(buffPic, img);
                } else {
                    buffImg = buffPic;
                }
            } else {
                buffImg = buffPic;
            }
        }

        String flip = "";
        String rotate = "";
        Matrix matrix = renderInfo.getImageCTM();
        String angle = "" + rotationFromPages.get(numPage);

        // experimental 
        float i11 = matrix.get(Matrix.I11); // if negative -> horizontal flip
        float i12 = matrix.get(Matrix.I12); // if negative -> 90 degree rotation
        float i21 = matrix.get(Matrix.I21); // if negative -> 270 degree rotation
        float i22 = matrix.get(Matrix.I22); // if negative -> vertical flip

        // flip and rotation ... from matrix if i11 or i22 is negative i have to flip image
        if (("" + i11).charAt(0) == '-') {
            flip = "fh";
        } else if (("" + i22).charAt(0) == '-') {
            flip = "fv";
        }

        if (angle.equalsIgnoreCase("270") || ("" + i21).charAt(0) == '-') {
            rotate = "270";
        } else if (angle.equalsIgnoreCase("180")) {
            rotate = "180";
        } else if (angle.equalsIgnoreCase("90") || ("" + i12).charAt(0) == '-') {
            rotate = "90";
        }

        if (buffImg != null) {
            buffImg = PdfTrickUtils.adjustImage(buffImg, flip, rotate);
            RenderedImageAttributes imageAttrs = null;
            if (isInline) {
                // set up inline image object attributes and store it in a hashmap
                InlineImage inImg = new InlineImage(buffImg, image != null ? image.getFileType() : "png");
                imageAttrs = new RenderedImageInline(inlineImageCounter, inImg, numPage, flip, rotate);
            } else {
                // set up image object for normal images
                imageAttrs = new RenderedImageNormal(numPage, renderInfo.getRef().getNumber(), flip, rotate);
            }

            // scaling image with original aspect ratio (if image exceded pic box)
            int w = buffImg.getWidth();
            int h = buffImg.getHeight();

            if (w > 170 || h > 170) {
                double faktor;
                if (w > h) {
                    faktor = 160 / (double) w;
                    int scaledW = (int) Math.round(faktor * w);
                    int scaledH = (int) Math.round(faktor * h);
                    buffImg = PdfTrickUtils.getScaledImagWithScalr(buffImg, scaledW, scaledH);

                } else {
                    faktor = 160 / (double) h;
                    int scaledW = (int) Math.round(faktor * w);
                    int scaledH = (int) Math.round(faktor * h);
                    buffImg = PdfTrickUtils.getScaledImagWithScalr(buffImg, scaledW, scaledH);
                }
            }
            numImg++;
            // dynamic update on the center panel (under EDT thread). UpdatePanelCenter upPcenter = this.upPcenter;
            upPcenter.setInlineImg(isInline);
            upPcenter.setBuffImg(buffImg);
            upPcenter.setImageAttrs(imageAttrs);
            try {
                SwingUtilities.invokeAndWait(upPcenter);
            } catch (InterruptedException e) {
                logger.error("Exception", e);
            } catch (InvocationTargetException e) {
                logger.error("Exception", e);
            }
            buffImg.flush();
        } else {
            unsupportedImage++;
        }
    } catch (IOException e) {
        logger.error("Exception", e);
        PdfTrickMessages.append("ERROR", Consts.SENDLOG_MSG);
        unsupportedImage++;
    } catch (Exception e) {
        logger.error("Exception", e);
        PdfTrickMessages.append("ERROR", Consts.SENDLOG_MSG);
        unsupportedImage++;
    }
}

From source file:org.gmdev.pdftrick.engine.ImagesExtractor.java

License:Open Source License

/**
 * /*from w  ww  .  ja va  2 s. c  o m*/
 * @param timeDirResult
 * @param resultFilePath
 * @param imageSelected
 * @param inlineImgSelected
 * @return TRUE if the extraction went good
 */
private boolean extractImgSel(String timeDirResult, String resultFilePath,
        HashMap<String, RenderedImageAttributes> imageSelected,
        HashMap<String, RenderedImageAttributes> inlineImgSelected) {

    final Properties messages = factory.getMessages();
    String result = timeDirResult + "/" + "Img_%s.%s";
    PdfReader reader = null;
    boolean retExtract = true;

    try {
        reader = new PdfReader(resultFilePath);
        Set<String> keys = imageSelected.keySet();
        Iterator<String> i = keys.iterator();
        int z = 1;
        while (i.hasNext()) {
            BufferedImage buff = null;
            String key = i.next();
            RenderedImageAttributes imgAttr = imageSelected.get(key);
            String flip = imgAttr.getFlip();
            String rotate = imgAttr.getRotate();
            int ref = imgAttr.getReference();

            PdfObject o = reader.getPdfObject(ref);
            PdfStream stream = (PdfStream) o;
            PdfImageObject io = null;
            try {
                io = new PdfImageObject((PRStream) stream);
            } catch (UnsupportedPdfException updfe) {
                try {
                    buff = CustomExtraImgReader.readIndexedPNG(ref, resultFilePath);
                    buff = PdfTrickUtils.adjustImage(buff, flip, rotate);
                    String type = "png";
                    String filename = String.format(result, z, type);
                    File outputfile = new File(filename);
                    ImageIO.write(buff, type, outputfile);
                } catch (Exception e) {
                    logger.error("Exception", e);
                    PdfTrickMessages.append("ERROR", Consts.SENDLOG_MSG);
                }
            }

            if (io != null) {
                String type = io.getFileType();
                String filename = String.format(result, z, type);
                BufferedImage buffPic = null;
                try {
                    if (type.equalsIgnoreCase("JBIG2")) {
                        buffPic = CustomExtraImgReader.readJBIG2(io);
                    } else {
                        buffPic = io.getBufferedImage();
                    }
                } catch (IIOException e) {
                    byte[] imageByteArray = io.getImageAsBytes();
                    try {
                        buffPic = CustomExtraImgReader.readCMYK_JPG(imageByteArray);
                    } catch (Exception ex) {
                        logger.error("Exception", e);
                        PdfTrickMessages.append("ERROR", Consts.SENDLOG_MSG);
                    }
                }

                // check if image contains a mask image
                BufferedImage buffMask = null;
                PdfDictionary imageDictionary = io.getDictionary();
                PRStream maskStream = (PRStream) imageDictionary.getAsStream(PdfName.SMASK);

                if (maskStream != null) {
                    // i have an smask object i check that is not a jpeg format, because this may cause some problem on offscreen rendering
                    // usually all imges with mask are png ... and there aren't problem, if image is jpg i discard the mask :)
                    if (!type.equalsIgnoreCase("jpg")) {
                        PdfImageObject maskImage = new PdfImageObject(maskStream);
                        buffMask = maskImage.getBufferedImage();

                        Image img = PdfTrickUtils.TransformGrayToTransparency(buffMask);
                        buff = PdfTrickUtils.ApplyTransparency(buffPic, img);
                    } else {
                        buff = buffPic;
                    }
                } else {
                    buff = buffPic;
                }

                if (buff != null) {
                    buff = PdfTrickUtils.adjustImage(buff, flip, rotate);

                    //particular cases of encoding
                    String encode = "";
                    if (type.equalsIgnoreCase("jp2")) {
                        encode = "jpeg 2000";
                    } else if (type.equalsIgnoreCase("jbig2")) {
                        encode = "jpeg";
                        filename = String.format(result, z, "jpg");
                    } else {
                        encode = type;
                    }

                    File outputfile = new File(filename);
                    ImageIO.write(buff, encode, outputfile);
                }
            }
            z++;
        }
        reader.close();
        if (inlineImgSelected.size() > 0) {
            extractInlineImgSel(timeDirResult, inlineImgSelected, z);
        }
        PdfTrickMessages.append("INFO", messages.getProperty("tmsg_19"));
    } catch (Exception e) {
        logger.error("Exception", e);
        PdfTrickMessages.append("ERROR", Consts.SENDLOG_MSG);
        retExtract = false;
    }
    return retExtract;
}

From source file:pdfextract.ImageRenderListener.java

public void renderImage(ImageRenderInfo renderInfo) {
    try {/*from w ww  . j  ava2 s . com*/
        PdfImageObject image = renderInfo.getImage();
        if (image == null)
            return;
        int number = renderInfo.getRef() != null ? renderInfo.getRef().getNumber() : counter++;
        String filename = String.format("%s-%s.%s", name, number, image.getFileType());
        FileOutputStream os = new FileOutputStream(filename);
        os.write(image.getImageAsBytes());
        os.flush();
        os.close();

        PdfDictionary imageDictionary = image.getDictionary();
        PRStream maskStream = (PRStream) imageDictionary.getAsStream(PdfName.SMASK);
        if (maskStream != null) {
            PdfImageObject maskImage = new PdfImageObject(maskStream);
            filename = String.format("%s-%s-mask.%s", name, number, maskImage.getFileType());
            os = new FileOutputStream(filename);
            os.write(maskImage.getImageAsBytes());
            os.flush();
            os.close();
        }
    } catch (IOException e) {
        e.printStackTrace();
    }
}