Source and destination parameter support

cityssm · Dec 14, 2018 · 78631f6 · 78631f6
1 parent f45307c
commit 78631f6
Show file tree

Hide file tree

Showing 4 changed files with 86 additions and 24 deletions.
diff --git a/bin/ca/saultstemarie/pdfflattener/FlattenPDF$1.class b/bin/ca/saultstemarie/pdfflattener/FlattenPDF$1.class
diff --git a/bin/ca/saultstemarie/pdfflattener/FlattenPDF.class b/bin/ca/saultstemarie/pdfflattener/FlattenPDF.class
diff --git a/dist/PDFFlattener.jar b/dist/PDFFlattener.jar
diff --git a/src/ca/saultstemarie/pdfflattener/FlattenPDF.java b/src/ca/saultstemarie/pdfflattener/FlattenPDF.java
@@ -24,14 +24,32 @@
 
 public class FlattenPDF {
 
-	public static int IMAGE_DPI = 200;
+	/**
+	 * Whether or not debug output is printed to the console.
+	 */
 	public static boolean SHOW_DEBUG = true;
+
+	/**
+	 * The suggested prefix to add to output files. 
+	 */
+	public static String DESTINATION_FILENAME_PREFIX = "flat--";
+
+	/**
+	 * The DPI that should be used when generating images.
+	 * Higher DPI increases the memory requirements and output file sizes, but also produces sharper images.
+	 */
+	public static int IMAGE_DPI = 200;
+
 
 	public static void main(String[] args) {
 
 		// Settings
 
 		System.setProperty("sun.java2d.cmm", "sun.java2d.cmm.kcms.KcmsServiceProvider");
+
+		/*
+		 * Set up JFileChooser
+		 */
 
 		try {
 			UIManager.setLookAndFeel(UIManager.getSystemLookAndFeelClassName());
@@ -40,50 +58,94 @@ public static void main(String[] args) {
 			// ignore and use default
 		}
 
+		JFileChooser fileChooser = new JFileChooser();
+		fileChooser.addChoosableFileFilter(new FileNameExtensionFilter("PDF Documents", "pdf"));
+		fileChooser.setAcceptAllFileFilterUsed(false);
+
 		/*
 		 * Get the source file
 		 */
 
-		JFileChooser fileChooser = new JFileChooser();
-
-		fileChooser.setDialogTitle("Select the Source PDF File to Flatten");
-		fileChooser.setApproveButtonText("Set Source PDF");
-		fileChooser.addChoosableFileFilter(new FileNameExtensionFilter("PDF Documents", "pdf"));
-		fileChooser.setAcceptAllFileFilterUsed(false);
+		File sourceFile = null;
 
-		int returnValue = fileChooser.showOpenDialog(null);
+		boolean sourceFileFromArgument = false;
 
-		if (returnValue != JFileChooser.APPROVE_OPTION) {
-			JOptionPane.showMessageDialog(null, "No Source PDF file selected.", "Error", JOptionPane.ERROR_MESSAGE);
-			System.exit(1);
+		if (args.length >= 1) {
+			sourceFile = new File(args[0]);
+
+			if (!sourceFile.exists() || !sourceFile.isFile() || !sourceFile.canRead() || !sourceFile.getName().toLowerCase().endsWith(".pdf")) {
+				JOptionPane.showMessageDialog(null, "Invalid source file.\n" +
+						args[0], "Error", JOptionPane.ERROR_MESSAGE);
+				System.exit(1);
+			}
+			else {
+				sourceFileFromArgument = true;
+			}
 		}
 
-		File sourceFile = fileChooser.getSelectedFile();
+		if (!sourceFileFromArgument) {
+
+			fileChooser.setDialogTitle("Select the Source PDF File to Flatten");
+			fileChooser.setApproveButtonText("Set Source PDF");
+
+			int returnValue = fileChooser.showOpenDialog(null);
+
+			if (returnValue != JFileChooser.APPROVE_OPTION) {
+				JOptionPane.showMessageDialog(null, "No Source PDF file selected.", "Error", JOptionPane.ERROR_MESSAGE);
+				System.exit(1);
+			}
+
+			sourceFile = fileChooser.getSelectedFile();
+		}
 
 		/*
 		 * Get the destination file
 		 */
-
-		File destinationFile = new File(sourceFile.getAbsolutePath() + File.separator + "FLAT-" + sourceFile.getName());
-
-		fileChooser.setDialogTitle("Select the Destination PDF File");
-		fileChooser.setApproveButtonText("Set Destination PDF");
-		fileChooser.setSelectedFile(destinationFile);
 
-		returnValue = fileChooser.showSaveDialog(null);
+		File destinationFile = null;
+		boolean destinationFileFromArgument = false;
 
-		if (returnValue != JFileChooser.APPROVE_OPTION) {
-			JOptionPane.showMessageDialog(null, "No Destination PDF file selected.", "Error", JOptionPane.ERROR_MESSAGE);
-			System.exit(1);
+		if (args.length >= 2) {
+			destinationFile = new File(args[1]);
+
+			if (!sourceFile.getName().toLowerCase().endsWith(".pdf")) {
+				JOptionPane.showMessageDialog(null, "Invalid destination file.\n" +
+						args[0], "Error", JOptionPane.ERROR_MESSAGE);
+				System.exit(1);
+			}
+			else {
+				destinationFileFromArgument = true;
+			}
 		}
 
-		destinationFile = fileChooser.getSelectedFile();
+		if (!destinationFileFromArgument) {
+
+			destinationFile = new File(sourceFile.getAbsolutePath() + File.separator + DESTINATION_FILENAME_PREFIX + sourceFile.getName());
+
+			fileChooser.setDialogTitle("Select the Destination PDF File");
+			fileChooser.setApproveButtonText("Set Destination PDF");
+			fileChooser.setSelectedFile(destinationFile);
+
+			int returnValue = fileChooser.showSaveDialog(null);
+
+			if (returnValue != JFileChooser.APPROVE_OPTION) {
+				JOptionPane.showMessageDialog(null, "No Destination PDF file selected.", "Error", JOptionPane.ERROR_MESSAGE);
+				System.exit(1);
+			}
+
+			destinationFile = fileChooser.getSelectedFile();
+		}
 
 		if (sourceFile.getPath().equals(destinationFile.getPath()) && sourceFile.getName().equals(destinationFile.getName())) {
 			JOptionPane.showMessageDialog(null, "You cannot select the same PDF file as both the source and the destination.", "Error", JOptionPane.ERROR_MESSAGE);
 			System.exit(1);
 		}
 
+		if (!destinationFile.getName().toLowerCase().endsWith(".pdf")) {
+			destinationFile = new File(destinationFile.getAbsolutePath() + ".pdf");
+		}
+
+
 		/*
 		 * Do the flattening
 		 */
@@ -96,7 +158,7 @@ public static void main(String[] args) {
 				"Ready to Start", 
 				JOptionPane.INFORMATION_MESSAGE);
 
-		returnValue = flattenPDF (sourceFile, destinationFile);
+		int returnValue = flattenPDF (sourceFile, destinationFile);
 
 		if (returnValue == 0) {
 			JOptionPane.showMessageDialog(null, "The PDF file was flattened successfully.", "Success", JOptionPane.INFORMATION_MESSAGE);