public void highlight(
final Pattern searchText, final Pattern markingPattern, Color color, int pageNr) {
if (textCache == null || document == null) {
throw new IllegalArgumentException("TextCache was not initialized");
}
final List<PDPage> pages = document.getDocumentCatalog().getAllPages();
try {
boolean found = false;
final PDPage page = pages.get(pageNr - 1);
PDPageContentStream contentStream = new PDPageContentStream(document, page, true, true);
PDExtendedGraphicsState graphicsState = new PDExtendedGraphicsState();
graphicsState.setNonStrokingAlphaConstant(0.5f);
PDResources resources = page.findResources();
Map graphicsStateDictionary = resources.getGraphicsStates();
if (graphicsStateDictionary == null) {
// There is no graphics state dictionary in the resources dictionary, create one.
graphicsStateDictionary = new TreeMap();
}
graphicsStateDictionary.put("highlights", graphicsState);
resources.setGraphicsStates(graphicsStateDictionary);
for (Match searchMatch : textCache.match(pageNr, searchText)) {
if (textCache.match(searchMatch.positions, markingPattern).size() > 0) {
for (Match markingMatch : textCache.match(searchMatch.positions, markingPattern)) {
if (markupMatch(color, contentStream, markingMatch)) {
found = true;
}
}
} else {
System.out.println(
"Cannot highlight: " + markingPattern.pattern() + " on page " + (pageNr - 1));
}
if (found) {
break;
}
}
contentStream.close();
} catch (Exception e) {
e.printStackTrace();
} catch (Error e1) {
e1.printStackTrace();
throw e1;
}
}
private boolean markupMatch(Color color, PDPageContentStream contentStream, Match markingMatch)
throws IOException {
final List<PDRectangle> textBoundingBoxes = getTextBoundingBoxes(markingMatch.positions);
if (textBoundingBoxes.size() > 0) {
contentStream.appendRawCommands("/highlights gs\n");
contentStream.setNonStrokingColor(color);
for (PDRectangle textBoundingBox : textBoundingBoxes) {
contentStream.fillRect(
textBoundingBox.getLowerLeftX(),
textBoundingBox.getLowerLeftY(),
Math.max(
Math.abs(textBoundingBox.getUpperRightX() - textBoundingBox.getLowerLeftX()), 10),
10);
}
return true;
}
return false;
}
@Test
public void testAppend() throws Exception {
final String originalText = "Test";
final String textToAppend = "Append";
PDDocument document = new PDDocument();
PDPage page = new PDPage(PDPage.PAGE_SIZE_A4);
document.addPage(page);
PDPageContentStream contentStream = new PDPageContentStream(document, page);
contentStream.setFont(PDType1Font.HELVETICA, 12);
contentStream.beginText();
contentStream.moveTextPositionByAmount(20, 400);
contentStream.drawString(originalText);
contentStream.endText();
contentStream.close();
template.sendBodyAndHeader(
"direct:start", textToAppend, PdfHeaderConstants.PDF_DOCUMENT_HEADER_NAME, document);
resultEndpoint.setExpectedMessageCount(1);
resultEndpoint.expectedMessagesMatches(
new Predicate() {
@Override
public boolean matches(Exchange exchange) {
Object body = exchange.getIn().getBody();
assertThat(body, instanceOf(ByteArrayOutputStream.class));
try {
PDDocument doc =
PDDocument.load(
new ByteArrayInputStream(((ByteArrayOutputStream) body).toByteArray()));
PDFTextStripper pdfTextStripper = new PDFTextStripper();
String text = pdfTextStripper.getText(doc);
assertEquals(2, doc.getNumberOfPages());
assertThat(text, containsString(originalText));
assertThat(text, containsString(textToAppend));
} catch (IOException e) {
throw new RuntimeException(e);
}
return true;
}
});
resultEndpoint.assertIsSatisfied();
}
/**
* @param args
* @throws Exception
*/
public static void main(String[] args) throws Exception {
// Create a document and add a page to it
PDDocument document = new PDDocument();
PDPage page1 = new PDPage(PDPage.PAGE_SIZE_A4);
// PDPage.PAGE_SIZE_LETTER is also possible
PDRectangle rect = page1.getMediaBox();
// rect can be used to get the page width and height
document.addPage(page1);
// Create a new font object selecting one of the PDF base fonts
PDFont fontPlain = PDType1Font.HELVETICA;
PDFont fontBold = PDType1Font.HELVETICA_BOLD;
PDFont fontItalic = PDType1Font.HELVETICA_OBLIQUE;
PDFont fontMono = PDType1Font.COURIER;
// Start a new content stream which will "hold" the to be created content
PDPageContentStream cos = new PDPageContentStream(document, page1);
int line = 1;
// Define a text content stream using the selected font, move the cursor and draw some text
cos.beginText();
cos.setFont(fontPlain, 12);
cos.setNonStrokingColor(Color.RED);
// width is 100; height is 20 lines down everytime
cos.moveTextPositionByAmount(100, rect.getHeight() - 20 * (++line));
cos.drawString("eeey boy");
cos.endText();
cos.beginText();
cos.setFont(fontItalic, 12);
cos.setNonStrokingColor(Color.GREEN);
cos.moveTextPositionByAmount(100, rect.getHeight() - 20 * (++line));
cos.drawString("How do you do?");
cos.endText();
cos.beginText();
cos.setFont(fontBold, 12);
cos.setNonStrokingColor(Color.BLACK);
cos.moveTextPositionByAmount(100, rect.getHeight() - 20 * (++line));
cos.drawString("Dit duurde veel langer dan nodig");
cos.endText();
cos.beginText();
cos.setFont(fontMono, 12);
cos.setNonStrokingColor(Color.BLACK);
cos.moveTextPositionByAmount(100, rect.getHeight() - 20 * (++line));
cos.drawString("maar ik was wat dingentjes aan t uitzoeken.");
cos.endText();
// Add some gucci into this place
try {
BufferedImage awtImage = ImageIO.read(new File("Gucci.jpg"));
PDXObjectImage ximage = new PDPixelMap(document, awtImage);
float scale = 0.5f; // alter this value to set the image size
cos.drawXObject(ximage, 100, 400, ximage.getWidth() * scale, ximage.getHeight() * scale);
} catch (IIOException Fnfex) {
System.out.println("No image for you");
}
// Make sure that the content stream is closed:
cos.close();
// New page
PDPage page2 = new PDPage(PDPage.PAGE_SIZE_A4);
document.addPage(page2);
cos = new PDPageContentStream(document, page2);
// draw a red box in the lower left hand corner
cos.setNonStrokingColor(Color.RED);
cos.fillRect(10, 10, 100, 100);
// add two lines of different widths
cos.setLineWidth(1);
cos.addLine(200, 250, 400, 250);
cos.closeAndStroke();
cos.setLineWidth(5);
cos.addLine(200, 300, 400, 300);
cos.closeAndStroke();
// close the content stream for page 2
cos.close();
// Save the results and ensure that the document is properly closed:
document.save("GUCCI.pdf");
document.close();
}
@Test
public void testAppendEncrypted() throws Exception {
final String originalText = "Test";
final String textToAppend = "Append";
PDDocument document = new PDDocument();
PDPage page = new PDPage(PDPage.PAGE_SIZE_A4);
document.addPage(page);
PDPageContentStream contentStream = new PDPageContentStream(document, page);
contentStream.setFont(PDType1Font.HELVETICA, 12);
contentStream.beginText();
contentStream.moveTextPositionByAmount(20, 400);
contentStream.drawString(originalText);
contentStream.endText();
contentStream.close();
final String ownerPass = "******";
final String userPass = "******";
AccessPermission accessPermission = new AccessPermission();
accessPermission.setCanExtractContent(false);
StandardProtectionPolicy protectionPolicy =
new StandardProtectionPolicy(ownerPass, userPass, accessPermission);
protectionPolicy.setEncryptionKeyLength(128);
document.protect(protectionPolicy);
ByteArrayOutputStream output = new ByteArrayOutputStream();
document.save(output);
// Encryption happens after saving.
PDDocument encryptedDocument = PDDocument.load(new ByteArrayInputStream(output.toByteArray()));
Map<String, Object> headers = new HashMap<String, Object>();
headers.put(PdfHeaderConstants.PDF_DOCUMENT_HEADER_NAME, encryptedDocument);
headers.put(
PdfHeaderConstants.DECRYPTION_MATERIAL_HEADER_NAME,
new StandardDecryptionMaterial(userPass));
template.sendBodyAndHeaders("direct:start", textToAppend, headers);
resultEndpoint.setExpectedMessageCount(1);
resultEndpoint.expectedMessagesMatches(
new Predicate() {
@Override
public boolean matches(Exchange exchange) {
Object body = exchange.getIn().getBody();
assertThat(body, instanceOf(ByteArrayOutputStream.class));
try {
PDDocument doc =
PDDocument.load(
new ByteArrayInputStream(((ByteArrayOutputStream) body).toByteArray()));
PDFTextStripper pdfTextStripper = new PDFTextStripper();
String text = pdfTextStripper.getText(doc);
assertEquals(2, doc.getNumberOfPages());
assertThat(text, containsString(originalText));
assertThat(text, containsString(textToAppend));
} catch (IOException e) {
throw new RuntimeException(e);
}
return true;
}
});
resultEndpoint.assertIsSatisfied();
}
