com.lowagie.toolbox.plugins.watermarker.WatermarkerTool Maven / Gradle / Ivy
/*
* $Id: Watermarker.java 3271 2008-04-18 20:39:42Z xlv $
* Copyright (c) 2005-2007 Bruno Lowagie, Carsten Hammer
*
* Permission is hereby granted, free of charge, to any person
* obtaining a copy of this software and associated documentation
* files (the "Software"), to deal in the Software without
* restriction, including without limitation the rights to use,
* copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following
* conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
* OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
* HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
* WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
* OTHER DEALINGS IN THE SOFTWARE.
*/
/*
* This class was originally published under the MPL by Bruno Lowagie
* and Carsten Hammer.
* It was a part of iText, a Java-PDF library. You can now use it under
* the MIT License; for backward compatibility you can also use it under
* the MPL version 1.1: http://www.mozilla.org/MPL/
* A copy of the MPL license is bundled with the source code FYI.
*/
package com.lowagie.toolbox.plugins.watermarker;
import static java.awt.Color.BLACK;
import static java.awt.Color.decode;
import static java.lang.Float.parseFloat;
import static java.lang.Integer.parseInt;
import java.awt.Color;
import java.io.File;
import java.io.FileOutputStream;
import javax.swing.JInternalFrame;
import javax.swing.JOptionPane;
import com.lowagie.text.pdf.PdfReader;
import com.lowagie.text.pdf.PdfStamper;
import com.lowagie.toolbox.AbstractTool;
import com.lowagie.toolbox.arguments.AbstractArgument;
import com.lowagie.toolbox.arguments.FileArgument;
import com.lowagie.toolbox.arguments.FloatArgument;
import com.lowagie.toolbox.arguments.IntegerArgument;
import com.lowagie.toolbox.arguments.StringArgument;
import com.lowagie.toolbox.arguments.filters.PdfFilter;
/**
* This tool lets you add a text watermark to all pages of a document.
*
* @since 2.1.1 (imported from itexttoolbox project)
*/
public class WatermarkerTool extends AbstractTool {
FileArgument destfile;
/**
* This tool lets you add a text watermark to all pages of a document.
*/
public WatermarkerTool() {
super();
FileArgument inputfile = new FileArgument(this, "srcfile",
"The file you want to watermark", false, new PdfFilter());
arguments.add(inputfile);
arguments.add(new StringArgument(this, "watermark",
"The text that can be used as watermark"));
arguments.add(new IntegerArgument(this, "fontsize",
"The fontsize of the watermark text"));
arguments.add(new FloatArgument(this, "opacity",
"The opacity of the watermark text"));
destfile = new FileArgument(this, "destfile",
"The file to which the watermarked PDF has to be written",
true, new PdfFilter());
arguments.add(destfile);
arguments.add(new StringArgument(this, "color",
"The color of the watermark text"));
inputfile.addPropertyChangeListener(destfile);
}
/**
* Creates the internal frame.
*/
@Override
protected void createFrame() {
internalFrame = new JInternalFrame("Watermark", true, false, true);
internalFrame.setSize(300, 80);
internalFrame.setJMenuBar(getMenubar());
System.out.println("=== Watermark OPENED ===");
}
/**
* Executes the tool (in most cases this generates a PDF file).
*/
@Override
public void execute() {
try {
if (getValue("srcfile") == null) {
throw new InstantiationException(
"You need to choose a sourcefile");
}
if (getValue("destfile") == null) {
throw new InstantiationException(
"You need to choose a destination file");
}
if (getValue("watermark") == null) {
throw new InstantiationException(
"You need to add a text for the watermark");
}
if (getValue("fontsize") == null) {
throw new InstantiationException(
"You need to add a fontsize for the watermark");
}
if (getValue("opacity") == null) {
throw new InstantiationException(
"You need to add a opacity for the watermark");
}
Color color = BLACK;
if (getValue("color") != null) {
color = decode((String) getValue("color"));
}
PdfReader reader = new PdfReader(((File) getValue("srcfile")).getAbsolutePath());
PdfStamper stamp = new PdfStamper(reader, new FileOutputStream((File) getValue("destfile")));
String text = (String) getValue("watermark");
int fontsize = parseInt((String) getValue("fontsize"));
float opacity = parseFloat((String) getValue("opacity"));
Writer writer = new Writer(reader, stamp, text, fontsize, opacity, color);
writer.write();
} catch (Exception e) {
JOptionPane.showMessageDialog(internalFrame, e.getMessage(), e
.getClass().getName(), JOptionPane.ERROR_MESSAGE);
System.err.println(e.getMessage());
}
}
/**
* Gets the PDF file that should be generated (or null if the output isn't a
* PDF file).
*
* @return the PDF file that should be generated
* @throws InstantiationException on error
*/
@Override
protected File getDestPathPDF() throws InstantiationException {
return (File) getValue("destfile");
}
/**
* Indicates that the value of an argument has changed.
*
* @param arg
* the argument that has changed
*/
@Override
public void valueHasChanged(AbstractArgument arg) {
if (internalFrame == null) {
// if the internal frame is null, the tool was called from the
// command line
return;
}
if (destfile.getValue() == null
&& arg.getName().equalsIgnoreCase("srcfile")) {
String filename = arg.getValue().toString();
String filenameout = filename.substring(0,
filename.indexOf(".", filename.length() - 4))
+ "_out.pdf";
destfile.setValue(filenameout);
}
}
/**
* This methods helps you running this tool as a standalone application.
*
*
* Call it like this from command line: java
* com.lowagie.tools.plugins.Watermarker input.pdf Draft 230 0.2 output.pdf #FF000000
*
*
* "input.pdf" is the input file name to be processed
*
* "Draft" is the text written as transparent "watermark" on top of each
* page
*
* "230" is the font size
*
* "0.2" is the opacity (1.0 completely opaque, 0.0 completely transparent)
*
* "output.pdf" is the output file name
*
* (Optional) "#FF0000" is the color (in hex format like #nnnnnn or 0xnnnnnn), #000000 (black) by default
*
*
* Call it from within other Java code:
*
*
* Watermarker.main(new
* String[]{"input.pdf","Draft","230","0.2","output.pdf","#FF000000"});
*
* @param args
* the srcfile, watermark text and destfile
*/
public static void main(String[] args) {
WatermarkerTool watermarkerTool = new WatermarkerTool();
if (args.length < 5 || args.length > 6) {
System.err.println(watermarkerTool.getUsage());
}
watermarkerTool.setMainArguments(args);
watermarkerTool.execute();
}
}