iText之pdf书签编辑工具

 需要首先定义标签文件，格式如下：
序号     起始页码 书签名
1        1        第一章
1.1      1        第一章第一节
1.1.1    1        第一章第一节第一小节
1.1.2    3        第一章第一节第二小节
1.2      4        第一章第二节
2        5        第二章

现在的界面非常简单，只有四个按钮。但确实已经够用。
计划实现这样的功能 ：从目录页中自动解析出书签；以图形化的方式编辑书签，而不是写结构化文件。

 

1 package main; 2 3 import java.awt.Dimension; 4 import java.awt.Toolkit; 5 import java.awt.event.ActionEvent; 6 import java.awt.event.ActionListener; 7 8 import javax.swing.Box; 9 import javax.swing.BoxLayout; 10 import javax.swing.JButton; 11 import javax.swing.JFileChooser; 12 import javax.swing.JFrame; 13 import javax.swing.JPanel; 14 import javax.swing.SwingUtilities; 15 import javax.swing.border.EtchedBorder; 16 import javax.swing.filechooser.FileNameExtensionFilter; 17 18 /****************** 19 * bookmarks(list) item: 20 * 21 * HashMap = 22 * (key, value) 23 * "Title" "some title" 24 * "Action" "GoTo" 25 * "Page" "some page" 26 * 27 * others may be: 28 * "Style" "some style" 29 * "Color" "some color" 30 * "Open" "True | False" 31 * 32 ******************/ 33 34 public class PdfBookmarker extends JFrame implements ActionListener{ 35 private JButton btn_edit_file; 36 private JButton btn_save_path; 37 private JButton btn_script_file; 38 private JButton btn_run; 39 40 private JPanel leftPanel; 41 private JFileChooser chooser; 42 43 private String oldFile; 44 private String newFile; 45 private String scriptFile; 46 47 public PdfBookmarker() { 48 init(); 49 } 50 51 public static void main(String [] args) { 52 SwingUtilities.invokeLater(new Runnable() { 53 public void run() { 54 new PdfBookmarker(); 55 } 56 }); 57 } 58 59 public void actionPerformed(ActionEvent event) { 60 if (event.getSource() == btn_save_path) { 61 chooser.addChoosableFileFilter( new FileNameExtensionFilter("pdf files", "pdf")); 62 int returnValue = chooser.showSaveDialog(PdfBookmarker.this); 63 if (returnValue == 0) { 64 this.newFile = chooser.getSelectedFile().getAbsolutePath(); 65 this.btn_script_file.setEnabled(true); 66 } 67 } 68 else if (event.getSource() == btn_edit_file) { 69 chooser.addChoosableFileFilter( new FileNameExtensionFilter("pdf files", "pdf")); 70 int returnValue = chooser.showOpenDialog(PdfBookmarker.this); 71 if (returnValue == 0) { 72 this.oldFile = chooser.getSelectedFile().getAbsolutePath(); 73 this.btn_save_path.setEnabled(true); 74 } 75 } 76 else if (event.getSource() == btn_script_file) { 77 chooser.addChoosableFileFilter( new FileNameExtensionFilter("text files", "txt")); 78 int returnValue = chooser.showOpenDialog(PdfBookmarker.this); 79 if (returnValue == 0) { 80 this.scriptFile = chooser.getSelectedFile().getAbsolutePath(); 81 this.btn_run.setEnabled(true); 82 } 83 } 84 else if (event.getSource() == btn_run) { 85 new BookmarkHelper().setOutlines(oldFile, newFile, scriptFile); 86 87 this.btn_save_path.setEnabled(false); 88 this.btn_script_file.setEnabled(false); 89 this.btn_run.setEnabled(false); 90 } 91 } 92 93 private void init() { 94 leftPanel = new JPanel(); 95 leftPanel.setLayout(new BoxLayout(leftPanel,BoxLayout.Y_AXIS)); 96 leftPanel.setBorder(new EtchedBorder()); 97 98 btn_edit_file = new JButton("选择要添加书签的文件"); 99 btn_edit_file.setPreferredSize(new Dimension(180, btn_edit_file.getPreferredSize().height)); 100 btn_edit_file.setMaximumSize(new Dimension(180, btn_edit_file.getPreferredSize().height)); 101 btn_edit_file.addActionListener(this); 102 leftPanel.add(btn_edit_file); 103 leftPanel.add(Box.createRigidArea(new Dimension(0, 3))); 104 105 btn_save_path = new JButton("选择新文件的保存路径"); 106 btn_save_path.setPreferredSize(new Dimension(180, btn_save_path.getPreferredSize().height)); 107 btn_save_path.setMaximumSize(new Dimension(180, btn_save_path.getPreferredSize().height)); 108 btn_save_path.addActionListener(this); 109 btn_save_path.setEnabled(false); 110 leftPanel.add(btn_save_path); 111 leftPanel.add(Box.createRigidArea(new Dimension(0, 3))); 112 113 btn_script_file = new JButton("选择定义好的书签文件"); 114 btn_script_file.setPreferredSize(new Dimension(180, btn_script_file.getPreferredSize().height)); 115 btn_script_file.setMaximumSize(new Dimension(180, btn_script_file.getPreferredSize().height)); 116 btn_script_file.addActionListener(this); 117 btn_script_file.setEnabled(false); 118 leftPanel.add(btn_script_file); 119 leftPanel.add(Box.createRigidArea(new Dimension(0, 3))); 120 121 btn_run = new JButton("执行"); 122 btn_run.setPreferredSize(new Dimension(180, btn_run.getPreferredSize().height)); 123 btn_run.setMaximumSize(new Dimension(180, btn_run.getPreferredSize().height)); 124 btn_run.addActionListener(this); 125 btn_run.setEnabled(false); 126 leftPanel.add(btn_run); 127 leftPanel.add(Box.createRigidArea(new Dimension(0, 3))); 128 129 chooser = new JFileChooser(); 130 131 this.setSize(180, 160); 132 Dimension scrSize = Toolkit.getDefaultToolkit().getScreenSize(); 133 this.setLocation((scrSize.width - getSize().width)/2, 134 (scrSize.height - getSize().height)/2); 135 136 this.getContentPane().add(leftPanel); 137 this.setVisible(true); 138 this.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE); 139 } 140 } 141

 

1 package main; 2 3 import java.io.BufferedReader; 4 import java.io.FileNotFoundException; 5 import java.io.FileOutputStream; 6 import java.io.FileReader; 7 import java.io.IOException; 8 import java.util.ArrayList; 9 import java.util.HashMap; 10 import java.util.List; 11 import java.util.Map; 12 13 import com.lowagie.text.DocumentException; 14 import com.lowagie.text.pdf.PdfReader; 15 import com.lowagie.text.pdf.PdfStamper; 16 17 public class BookmarkHelper { 18 private static int line = -1; 19 private List data = null; 20 21 public void setOutlines(String oldFile, String newFile, String outlinesFile) { 22 try { 23 //create a reader for a certain document 24 PdfReader reader = new PdfReader(oldFile); 25 26 // we create a stamper that will copy the document to a new file 27 PdfStamper stamp = new PdfStamper(reader, new FileOutputStream(newFile)); 28 29 //read the outlines structure file 30 readOutlinesFile(outlinesFile); 31 32 //set the outlines 33 stamp.setOutlines(createOutlines()); 34 35 // closing PdfStamper will generate the new PDF file 36 stamp.close(); 37 } catch (IOException e) { 38 e.printStackTrace(); 39 } catch (DocumentException e) { 40 e.printStackTrace(); 41 } 42 43 } 44 45 public List createOutlines() { 46 List list = new ArrayList(); 47 48 HashMap map = null; 49 int lastLine = line + 1; 50 51 do { 52 line++; 53 54 map = new HashMap(); 55 map.put("Title", ((Map)data.get(line)).get("Title") ); 56 map.put("Action", "GoTo"); 57 map.put("Page", ((Map)data.get(line)).get("Page") ); 58 59 if (hasChildren(line)) { 60 map.put("Kids", createOutlines()); 61 } 62 63 list.add(map); 64 } while (hasSubling(lastLine)); 65 66 return list; 67 } 68 69 /* 70 *List item: 71 *String[] = {INDEX, PAGE, TITLE} 72 */ 73 private void readOutlinesFile(String file) { 74 List lines = new ArrayList(); 75 Map map = null; 76 77 try { 78 BufferedReader reader = new BufferedReader(new FileReader(file)); 79 80 String line = null; 81 String subStr = null; 82 int tabIndex = 0; 83 int spaceIndex = 0; 84 85 while ( (line = reader.readLine()) != null) { 86 map = new HashMap(); 87 88 tabIndex = line.indexOf("/t"); 89 tabIndex = (tabIndex >= 0) ? tabIndex : Integer.MAX_VALUE; 90 spaceIndex = line.indexOf(" "); 91 spaceIndex = (spaceIndex >= 0) ? spaceIndex : Integer.MAX_VALUE; 92 subStr = line.substring(0, tabIndex < spaceIndex ? tabIndex : spaceIndex); 93 map.put("Index", subStr); 94 line = line.substring(subStr.length() + 1).trim(); 95 96 tabIndex = line.indexOf("/t"); 97 tabIndex = (tabIndex >= 0) ? tabIndex : Integer.MAX_VALUE; 98 spaceIndex = line.indexOf(" "); 99 spaceIndex = (spaceIndex >= 0) ? spaceIndex : Integer.MAX_VALUE; 100 subStr = line.substring(0, tabIndex < spaceIndex ? tabIndex : spaceIndex); 101 map.put("Page", subStr); 102 103 map.put("Title", line.substring(subStr.length() + 1).trim()); 104 105 lines.add(map); 106 } 107 } catch (FileNotFoundException e) { 108 e.printStackTrace(); 109 } catch (IOException e) { 110 e.printStackTrace(); 111 } 112 113 this.data = lines; 114 } 115 116 private boolean hasSubling(int lastLine) { 117 if (line >= data.size() -1) return false; 118 119 String lastIndexStr = (String)( (Map)data.get(lastLine)).get("Index"); 120 String currIndexStr = (String)( (Map)data.get(line + 1)).get("Index"); 121 122 return (lastIndexStr.length() == currIndexStr.length()); 123 } 124 125 private boolean hasChildren(int line) { 126 if (line >= data.size() -1) return false; 127 128 String currIndexStr = (String)( (Map)data.get(line)).get("Index"); 129 String nextIndexStr = (String)( (Map)data.get(line + 1)).get("Index"); 130 131 if (line + 1 < data.size() 132 && currIndexStr.length() < nextIndexStr.length()) 133 return true; 134 135 return false; 136 } 137 } 138 

 

Adding bookmark or outline tree to PDF

Outline tree can be construct with PdfOutline object. PdfAction can be trigger from outline tree, which executes actions such as goto local page, javascript and hyperlink.

 

package com.geek.tutorial.itext.bookmarks; import com.lowagie.text.Document; import com.lowagie.text.pdf.PdfWriter; import com.lowagie.text.Chunk; import com.lowagie.text.Paragraph; import com.lowagie.text.pdf.PdfOutline; import com.lowagie.text.pdf.PdfContentByte; import com.lowagie.text.pdf.PdfAction; import java.io.FileOutputStream; public class Outline { public Outline() throws Exception{ Document document = new Document(); PdfWriter writer = PdfWriter.getInstance(document, new FileOutputStream("outline.pdf")); document.open(); // Code 1 document.add(new Chunk("Chapter 1") .setLocalDestination("1")); document.newPage(); document.add(new Chunk("Chapter 2") .setLocalDestination("2")); document.add(new Paragraph(new Chunk("Sub 2.1") .setLocalDestination("2.1"))); document.add(new Paragraph(new Chunk("Sub 2.2") .setLocalDestination("2.2"))); document.newPage(); document.add(new Chunk("Chapter 3") .setLocalDestination("3")); // Code 2 PdfContentByte cb = writer.getDirectContent(); PdfOutline root = cb.getRootOutline(); // Code 3 PdfOutline oline1 = new PdfOutline(root, PdfAction.gotoLocalPage("1", false),"Chapter 1"); PdfOutline oline2 = new PdfOutline(root, PdfAction.gotoLocalPage("2", false),"Chapter 2"); oline2.setOpen(false); PdfOutline oline2_1 = new PdfOutline(oline2, PdfAction.gotoLocalPage("2.1", false),"Sub 2.1"); PdfOutline oline2_2 = new PdfOutline(oline2, PdfAction.gotoLocalPage("2.2", false),"Sub 2.2"); PdfOutline oline3 = new PdfOutline(root, PdfAction.gotoLocalPage("3", false),"Chapter 3"); document.close(); } public static void main(String[] args) { try{ Outline outline = new Outline(); }catch(Exception e){ System.out.println(e); } } }

 

Code 1
Create 3 pages and set destination to titles. Add 2 sub destination in page 2.

Code 2
Get the root of the outline tree.

Code 3
Add bookmarks for each pages and two sub section to page 2. Bookmarks expanded by default, you can collapse the tree with this code

 

oline2.setOpen(false);