merge from master
[fanfix.git] / src / be / nikiroo / fanfix / output / LaTeX.java
1 package be.nikiroo.fanfix.output;
2
3 import java.io.BufferedWriter;
4 import java.io.File;
5 import java.io.FileOutputStream;
6 import java.io.IOException;
7 import java.io.OutputStreamWriter;
8
9 import be.nikiroo.fanfix.Instance;
10 import be.nikiroo.fanfix.bundles.Config;
11 import be.nikiroo.fanfix.bundles.StringId;
12 import be.nikiroo.fanfix.data.Chapter;
13 import be.nikiroo.fanfix.data.MetaData;
14 import be.nikiroo.fanfix.data.Paragraph.ParagraphType;
15 import be.nikiroo.fanfix.data.Story;
16
17 class LaTeX extends BasicOutput {
18 protected BufferedWriter writer;
19 private boolean lastWasQuote = false;
20
21 // quote chars
22 private char openQuote = Instance.getInstance().getTrans().getCharacter(StringId.OPEN_SINGLE_QUOTE);
23 private char closeQuote = Instance.getInstance().getTrans().getCharacter(StringId.CLOSE_SINGLE_QUOTE);
24 private char openDoubleQuote = Instance.getInstance().getTrans().getCharacter(StringId.OPEN_DOUBLE_QUOTE);
25 private char closeDoubleQuote = Instance.getInstance().getTrans().getCharacter(StringId.CLOSE_DOUBLE_QUOTE);
26
27 @Override
28 public File process(Story story, File targetDir, String targetName)
29 throws IOException {
30 String targetNameOrig = targetName;
31 targetName += getDefaultExtension(false);
32
33 File target = new File(targetDir, targetName);
34
35 writer = new BufferedWriter(new OutputStreamWriter(
36 new FileOutputStream(target), "UTF-8"));
37 try {
38 super.process(story, targetDir, targetNameOrig);
39 } finally {
40 writer.close();
41 writer = null;
42 }
43
44 return target;
45 }
46
47 @Override
48 public String getDefaultExtension(boolean readerTarget) {
49 return ".tex";
50 }
51
52 @Override
53 protected void writeStoryHeader(Story story) throws IOException {
54 String date = "";
55 String author = "";
56 String title = "\\title{}";
57 String lang = "";
58 if (story.getMeta() != null) {
59 MetaData meta = story.getMeta();
60 title = "\\title{" + latexEncode(meta.getTitle()) + "}";
61 date = "\\date{" + latexEncode(meta.getDate()) + "}";
62 author = "\\author{" + latexEncode(meta.getAuthor()) + "}";
63 lang = meta.getLang().toLowerCase();
64 if (lang != null && !lang.isEmpty()) {
65 lang = Instance.getInstance().getConfig().getStringX(Config.CONF_LATEX_LANG, lang);
66 if (lang == null) {
67 System.err.println(Instance.getInstance().getTrans().getString(StringId.LATEX_LANG_UNKNOWN, lang));
68 }
69 }
70 }
71
72 writer.append("%\n");
73 writer.append("% This LaTeX document was auto-generated by Fanfic Reader, created by Niki.\n");
74 writer.append("%\n\n");
75 writer.append("\\documentclass[a4paper]{book}\n");
76 if (lang != null && !lang.isEmpty()) {
77 writer.append("\\usepackage[" + lang + "]{babel}\n");
78 }
79 writer.append("\\usepackage[utf8]{inputenc}\n");
80 writer.append("\\usepackage[T1]{fontenc}\n");
81 writer.append("\\usepackage{lmodern}\n");
82 writer.append("\\newcommand{\\br}{\\vspace{10 mm}}\n");
83 writer.append("\\newcommand{\\say}{--- \\noindent\\emph}\n");
84 writer.append("\\hyphenpenalty=1000\n");
85 writer.append("\\tolerance=5000\n");
86 writer.append("\\begin{document}\n");
87 if (story.getMeta() != null && story.getMeta().getDate() != null)
88 writer.append(date + "\n");
89 writer.append(title + "\n");
90 writer.append(author + "\n");
91 writer.append("\\maketitle\n");
92 writer.append("\n");
93
94 // TODO: cover
95 }
96
97 @Override
98 protected void writeStoryFooter(Story story) throws IOException {
99 writer.append("\\end{document}\n");
100 }
101
102 @Override
103 protected void writeChapterHeader(Chapter chap) throws IOException {
104 writer.append("\n\n\\chapter{" + latexEncode(chap.getName()) + "}"
105 + "\n");
106 }
107
108 @Override
109 protected void writeChapterFooter(Chapter chap) throws IOException {
110 writer.write("\n");
111 }
112
113 @Override
114 protected String enbold(String word) {
115 return "\\textsc{" + word + "}";
116 }
117
118 @Override
119 protected String italize(String word) {
120 return "\\emph{" + word + "}";
121 }
122
123 @Override
124 protected void writeTextLine(ParagraphType type, String line)
125 throws IOException {
126
127 line = decorateText(latexEncode(line));
128
129 switch (type) {
130 case BLANK:
131 writer.write("\n");
132 lastWasQuote = false;
133 break;
134 case BREAK:
135 writer.write("\n\\br");
136 writer.write("\n");
137 lastWasQuote = false;
138 break;
139 case NORMAL:
140 writer.write(line);
141 writer.write("\n");
142 lastWasQuote = false;
143 break;
144 case QUOTE:
145 writer.write("\n\\say{" + line + "}\n");
146 if (lastWasQuote) {
147 writer.write("\n\\noindent{}");
148 }
149 lastWasQuote = true;
150 break;
151 case IMAGE:
152 // TODO what about images in LaTeX?
153 break;
154 }
155 }
156
157 private String latexEncode(String input) {
158 StringBuilder builder = new StringBuilder();
159 for (char car : input.toCharArray()) {
160 // TODO: check restricted chars?
161 if (car == '^' || car == '$' || car == '\\' || car == '#'
162 || car == '%') {
163 builder.append('\\');
164 builder.append(car);
165 } else if (car == openQuote) {
166 builder.append('`');
167 } else if (car == closeQuote) {
168 builder.append('\'');
169 } else if (car == openDoubleQuote) {
170 builder.append("``");
171 } else if (car == closeDoubleQuote) {
172 builder.append("''");
173 } else {
174 builder.append(car);
175 }
176 }
177
178 return builder.toString();
179 }
180 }