Использование Flying Saucer для рендеринга изображений в PDF в памяти

Я использую Flying Saucer для преобразования XHTML в документ PDF. Я получил код для работы только с базовым HTML и встроенным CSS, однако теперь я пытаюсь добавить изображение в виде своего рода заголовка в PDF. Мне интересно, если есть какой-либо способ добавить изображение, прочитав в файле изображения как объект изображения Java, а затем добавив, что каким-то образом в PDF (или в XHTML), как будто он получает виртуальный "url", представляющий объект Image, который я могу использовать для рендеринга PDF). Кто-нибудь когда-либо делал что-нибудь подобное?

Заранее благодарим за любую помощь, которую вы можете предоставить!

Ответ 1

Я должен был сделать это на прошлой неделе, поэтому надеюсь, что смогу сразу ответить вам.

Летающая тарелка

Самый простой способ - добавить изображение, которое вы хотите в качестве разметки в свой HTML-шаблон, перед рендерингом с Flying Saucer. В Flying Saucer вам нужно будет реализовать ReplacedElementFactory, чтобы вы могли заменить любую разметку перед рендерингом с данными изображения.

/**
 * Replaced element in order to replace elements like 
 * <tt>&lt;div class="media" data-src="image.png" /></tt> with the real
 * media content.
 */
public class MediaReplacedElementFactory implements ReplacedElementFactory {
    private final ReplacedElementFactory superFactory;

    public MediaReplacedElementFactory(ReplacedElementFactory superFactory) {
        this.superFactory = superFactory;
    }

    @Override
    public ReplacedElement createReplacedElement(LayoutContext layoutContext, BlockBox blockBox, UserAgentCallback userAgentCallback, int cssWidth, int cssHeight) {
        Element element = blockBox.getElement();
        if (element == null) {
            return null;
        }
        String nodeName = element.getNodeName();
        String className = element.getAttribute("class");
        // Replace any <div class="media" data-src="image.png" /> with the
        // binary data of `image.png` into the PDF.
        if ("div".equals(nodeName) && "media".equals(className)) {
            if (!element.hasAttribute("data-src")) {
                throw new RuntimeException("An element with class `media` is missing a `data-src` attribute indicating the media file.");
            }
            InputStream input = null;
            try {
                input = new FileInputStream("/base/folder/" + element.getAttribute("data-src"));
                final byte[] bytes = IOUtils.toByteArray(input);
                final Image image = Image.getInstance(bytes);
                final FSImage fsImage = new ITextFSImage(image);
                if (fsImage != null) {
                    if ((cssWidth != -1) || (cssHeight != -1)) {
                        fsImage.scale(cssWidth, cssHeight);
                    }
                    return new ITextImageElement(fsImage);
                }
            } catch (Exception e) {
                throw new RuntimeException("There was a problem trying to read a template embedded graphic.", e);
            } finally {
                IOUtils.closeQuietly(input);
            }
        }
        return this.superFactory.createReplacedElement(layoutContext, blockBox, userAgentCallback, cssWidth, cssHeight);
    }

    @Override
    public void reset() {
        this.superFactory.reset();
    }

    @Override
    public void remove(Element e) {
        this.superFactory.remove(e);
    }

    @Override
    public void setFormSubmissionListener(FormSubmissionListener listener) {
        this.superFactory.setFormSubmissionListener(listener);
    }
}

Вы заметите, что я здесь hardcoded /base/folder, который является папкой, в которой находится файл HTML, так как он будет корневым URL-адресом для Flying Saucer для разрешения медиа. Вы можете изменить его на правильное местоположение, исходя из любого места, которое вы хотите (например, Свойства).

HTML

В вашей разметке HTML вы указываете где-то a <div class="media" data-src="somefile.png" /> так:

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" 
   "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
<html xmlns="http://www.w3.org/1999/xhtml">
    <head>
        <title>My document</title>
        <style type="text/css">
        #logo { /* something if needed */ }
        </style>
    </head>
    <body>
        <!-- Header -->
        <div id="logo" class="media" data-src="media/logo.png" style="width: 177px; height: 60px" />
        ...
    </body>
</html>

Rendering

И, наконец, вам нужно указать свой ReplacedElementFactory на Flying-Saucer при рендеринге:

String content = loadHtml();
ITextRenderer renderer = new ITextRenderer();
renderer.getSharedContext().setReplacedElementFactory(new MediaReplacedElementFactory(renderer.getSharedContext().getReplacedElementFactory()));
renderer.setDocumentFromString(content.toString());
renderer.layout();
final ByteArrayOutputStream baos = new ByteArrayOutputStream();
renderer.createPDF(baos);
// baos.toByteArray();

Я использую Freemarker для генерации HTML из шаблона, а затем с большим успехом подаю результат FlyingSaucer. Это довольно аккуратная библиотека.

Ответ 2

Спасибо Алекс за подробное решение. Я использую это решение и обнаружил, что для его работы есть еще одна строка.

public ReplacedElement createReplacedElement(LayoutContext layoutContext, BlockBox blockBox, UserAgentCallback userAgentCallback, int cssWidth, int cssHeight) {
  Element element = blockBox.getElement();
  ....
  ....
  final Image image = Image.getInstance(bytes);
  final int factor = ((ITextUserAgent)userAgentCallback).getSharedContext().getDotsPerPixel(); //Need to add this line
  image.scaleAbsolute(image.getPlainWidth() * factor, image.getPlainHeight() * factor) //Need to add this line
  final FSImage fsImage = new ITextFSImage(image);
  ....
  ....

Нам нужно прочитать DPP из SharedContext и масштабировать изображение, чтобы отобразить изображение в PDF.

Другое предложение: Мы можем напрямую расширить ITextReplacedElement вместо реализации ReplacedElementFactory. В этом случае мы можем установить ReplacedElementFactory в SharedContext следующим образом:

renderer.getSharedContext().setReplacedElementFactory(new MediaReplacedElementFactory(renderer.getOutputDevice()); 

Ответ 3

то, что сработало для меня, представляет собой встроенное изображение. Поэтому сначала конвертируйте изображение в base64, а затем вставьте его:

    byte[] image = ...
    ITextRenderer renderer = new ITextRenderer();
    renderer.setDocumentFromString("<html>\n" +
                                   "    <body>\n" +
                                   "        <h1>Image</h1>\n" +
                                   " <div><img src=\"data:image/png;base64," + Base64.getEncoder().encodeToString(image) + "\"></img></div>\n" +
                                   "    </body>\n" +
                                   "</html>");
    renderer.layout();
    renderer.createPDF(response.getOutputStream());