1 /*
2  * Copyright (c) 2002, 2013, Oracle and/or its affiliates. All rights reserved.
3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4  *
5  * This code is free software; you can redistribute it and/or modify it
6  * under the terms of the GNU General Public License version 2 only, as
7  * published by the Free Software Foundation.  Oracle designates this
8  * particular file as subject to the "Classpath" exception as provided
9  * by Oracle in the LICENSE file that accompanied this code.
10  *
11  * This code is distributed in the hope that it will be useful, but WITHOUT
12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
14  * version 2 for more details (a copy is included in the LICENSE file that
15  * accompanied this code).
16  *
17  * You should have received a copy of the GNU General Public License version
18  * 2 along with this work; if not, write to the Free Software Foundation,
19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20  *
21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22  * or visit www.oracle.com if you need additional information or have any
23  * questions.
24  */
25 package build.tools.generatenimbus;
26 
27 import javax.xml.stream.XMLInputFactory;
28 import javax.xml.stream.XMLStreamException;
29 import javax.xml.stream.XMLStreamReader;
30 import java.io.*;
31 import java.util.HashMap;
32 import java.util.Map;
33 
34 /**
35  * Generates the various Java artifacts based on a SynthModel.
36  * <p/>
37  * Generated source files are split up among two different locations. There are those source files that are meant to be
38  * edited (generally, only the LookAndFeel class itself) and those that are autogenerated (everything else).
39  * <p/>
40  * All autogenerated files are placed in "buildPackageRoot" and are package private. A LAF author (one who has access to
41  * the generated sources) will be able to access any of the generated classes. Those referencing the library, however,
42  * will only be able to access the main LookAndFeel class itself (since everything else is package private).
43  *
44  * @author  Richard Bair
45  * @author  Jasper Potts
46  */
47 public class Generator {
48     private static Generator instance;
49 
50     /** A map of variables that are used for variable substitution in the template files. */
51     private Map<String, String> variables;
52     private boolean full = false;
53     private File buildPackageRoot;
54     private String packageNamePrefix;
55     private String lafName;
56     private SynthModel model;
57 
58     /**
59      * MAIN APPLICATION
60      * <p/>
61      * This is for using the generator as part of the java build process
62      *
63      * @param args The commandline arguments
64      */
main(String[] args)65     public static void main(String[] args) throws Exception {
66         if (args.length == 0 || (args.length % 2) != 0) {
67             System.out.println("Usage: generator [-options]\n" +
68                     "    -full <true|false>     True if we should build the whole LAF or false for building just states and painters.\n" +
69                     "    -skinFile <value>      Path to the skin.laf file for the LAF to be generated from.\n" +
70                     "    -buildDir <value>      The directory beneath which the build-controlled artifacts (such as the Painters) should\n" +
71                     "                           be placed. This is the root directory beneath which the necessary packages and source\n" +
72                     "                           files will be created.\n" +
73                     "    -resourcesDir <value>  The resources directory containing templates and images.\n" +
74                     "    -packagePrefix <value> The package name associated with this synth look and feel. For example,\n" +
75                     "                           \"org.mypackage.mylaf\"\n" +
76                     "    -lafName <value>       The name of the laf, such as \"MyLAF\".\n");
77         } else {
78             boolean full = false;
79             File skinFile = new File(System.getProperty("user.dir"));
80             File buildDir = new File(System.getProperty("user.dir"));
81             File resourcesDir = new File(System.getProperty("user.dir"));
82             String packagePrefix = "org.mypackage.mylaf";
83             String lafName = "MyLAF";
84             for (int i = 0; i < args.length; i += 2) {
85                 String key = args[i].trim().toLowerCase();
86                 String value = args[i + 1].trim();
87                 if ("-full".equals(key)) {
88                     full = Boolean.parseBoolean(value);
89                 } else if ("-skinfile".equals(key)) {
90                     skinFile = new File(value);
91                 } else if ("-builddir".equals(key)) {
92                     buildDir = new File(value);
93                 } else if ("-resourcesdir".equals(key)) {
94                     resourcesDir = new File(value);
95                 } else if ("-packageprefix".equals(key)) {
96                     packagePrefix = value;
97                 } else if ("-lafname".equals(key)) {
98                     lafName = value;
99                 }
100             }
101             System.out.println("### GENERATING LAF CODE ################################");
102             System.out.println("   full          :" + full);
103             System.out.println("   skinFile      :" + skinFile.getAbsolutePath());
104             System.out.println("   buildDir      :" + buildDir.getAbsolutePath());
105             System.out.println("   resourcesDir  :" + resourcesDir.getAbsolutePath());
106             System.out.println("   packagePrefix :" +packagePrefix);
107             System.out.println("   lafName       :" +lafName);
108 
109             SynthModel model;
110             XMLInputFactory inputFactory = XMLInputFactory.newInstance();
111             XMLStreamReader reader;
112             try( InputStream fis = new FileInputStream(skinFile);
113                  InputStream is = new BufferedInputStream(fis)) {
114                 reader = inputFactory.createXMLStreamReader(is);
115                 model = new SynthModel(reader);
116             }
117             Generator.init(full, buildDir, packagePrefix, lafName, model);
118             Generator.getInstance().generate();
119         }
120     }
121 
122     /**
123      * Creates a new Generator, capable of outputting the source code artifacts related to a given SynthModel. It is
124      * capable of generating the one-time artifacts in addition to the regeneration of build-controlled artifacts.
125      *
126      * @param full              True if we should build the whole LAF or false for building just states and painters.
127      * @param buildDir          The directory beneath which the build-controlled artifacts (such as the Painters) should
128      *                          be placed. This is the root directory beneath which the necessary packages and source
129      *                          files will be created.
130      * @param srcDir            The directory beneath which the normal user-controlled artifacts (such as the core
131      *                          LookAndFeel file) should be placed. These are one-time generated files. This is the root
132      *                          directory beneath which the necessary packages and source files will be created.
133      * @param packageNamePrefix The package name associated with this synth look and feel. For example,
134      *                          org.mypackage.mylaf
135      * @param lafName           The name of the laf, such as MyLAF.
136      * @param model             The actual SynthModel to base these generated files on.
137      */
Generator(boolean full, File buildDir, String packageNamePrefix, String lafName, SynthModel model)138     private Generator(boolean full, File buildDir,
139             String packageNamePrefix, String lafName, SynthModel model) {
140         this.full = full;
141         //validate the input variables
142         if (packageNamePrefix == null) {
143             throw new IllegalArgumentException("You must specify a package name prefix");
144         }
145         if (buildDir == null) {
146             throw new IllegalArgumentException("You must specify the build directory");
147         }
148         if (model == null) {
149             throw new IllegalArgumentException("You must specify the SynthModel");
150         }
151         if (lafName == null) {
152             throw new IllegalArgumentException("You must specify the name of the look and feel");
153         }
154 
155         //construct the map which is used to do variable substitution of the template
156         //files
157         variables = new HashMap<String, String>();
158         variables.put("PACKAGE", packageNamePrefix);
159         variables.put("LAF_NAME", lafName);
160 
161         //generate and save references to the package-root directories.
162         //(That is, given the buildDir and srcDir, generate references to the
163         //org.mypackage.mylaf subdirectories)
164         buildPackageRoot = new File(buildDir, packageNamePrefix.replaceAll("\\.", "\\/"));
165         buildPackageRoot.mkdirs();
166 
167         //save the variables
168         this.packageNamePrefix = packageNamePrefix;
169         this.lafName = lafName;
170         this.model = model;
171     }
172 
init(boolean full, File buildDir, String packageNamePrefix, String lafName, SynthModel model)173     public static void init(boolean full, File buildDir,
174             String packageNamePrefix, String lafName, SynthModel model) {
175         instance = new Generator(full, buildDir, packageNamePrefix, lafName, model);
176         model.initStyles();
177     }
178 
getInstance()179     public static Generator getInstance() {
180         return instance;
181     }
182 
getVariables()183     public static Map<String, String> getVariables() {
184         return new HashMap<String, String>(instance.variables);
185     }
186 
generate()187     public void generate() {
188         if (full) {
189             //create the LookAndFeel file
190             writeSrcFileImpl("LookAndFeel", variables, lafName + "LookAndFeel");
191 
192             writeSrcFileImpl("AbstractRegionPainter", variables);
193             writeSrcFileImpl("BlendingMode", variables);
194             writeSrcFileImpl("SynthPainterImpl", variables);
195             writeSrcFileImpl("IconImpl", variables, lafName + "Icon.java");
196             writeSrcFileImpl("StyleImpl", variables, lafName + "Style.java");
197             writeSrcFileImpl("Effect", variables);
198             writeSrcFileImpl("EffectUtils", variables);
199             writeSrcFileImpl("ShadowEffect", variables);
200             writeSrcFileImpl("DropShadowEffect", variables);
201             writeSrcFileImpl("InnerShadowEffect", variables);
202             writeSrcFileImpl("InnerGlowEffect", variables);
203             writeSrcFileImpl("OuterGlowEffect", variables);
204             writeSrcFileImpl("State", variables);
205             writeSrcFileImpl("ImageCache", variables);
206             writeSrcFileImpl("ImageScalingHelper", variables);
207         }
208         //next, populate the first set of ui defaults based on what is in the
209         //various palettes of the synth model
210         StringBuilder defBuffer = new StringBuilder();
211         StringBuilder styleBuffer = new StringBuilder();
212         model.write(defBuffer, styleBuffer, packageNamePrefix);
213 
214         Map<String, String> vars = getVariables();
215         vars.put("UI_DEFAULT_INIT", defBuffer.toString());
216         vars.put("STYLE_INIT", styleBuffer.toString());
217         writeSrcFile("Defaults", vars, lafName + "Defaults");
218     }
219 
writeSrcFileImpl(String name, Map<String, String> variables)220     private void writeSrcFileImpl(String name, Map<String, String> variables) {
221         writeSrcFileImpl(name, variables, name);
222     }
223 
writeSrcFileImpl(String templateName, Map<String, String> variables, String outputName)224     private void writeSrcFileImpl(String templateName,
225             Map<String, String> variables, String outputName) {
226         PrintWriter out = null;
227         try {
228             InputStream stream = getClass().getResourceAsStream(
229                     "resources/" + templateName + ".template");
230             TemplateReader in = new TemplateReader(variables, stream);
231 
232             out = new PrintWriter(new File(buildPackageRoot, outputName + ".java"));
233             String line = in.readLine();
234             while (line != null) {
235                 out.println(line);
236                 line = in.readLine();
237             }
238         } catch (IOException e) {
239             throw new RuntimeException("IOException in writer", e);
240         } finally {
241             if (out != null) out.close();
242         }
243     }
244 
writeSrcFile(String templateName, Map<String, String> variables, String outputName)245     public static void writeSrcFile(String templateName,
246             Map<String, String> variables, String outputName) {
247         instance.writeSrcFileImpl(templateName, variables, outputName);
248     }
249 
250     /** A BufferedReader implementation that automatically performs
251      * string replacements as needed.
252      */
253     private static final class TemplateReader extends BufferedReader {
254         private Map<String, String> variables;
255 
TemplateReader(Map<String, String> variables, InputStream template)256         TemplateReader(Map<String, String> variables, InputStream template) {
257             super(new InputStreamReader(template));
258             this.variables = variables;
259         }
260 
readLine()261         @Override public String readLine() throws IOException {
262             return substituteVariables(super.readLine());
263         }
264 
substituteVariables(String input)265         private String substituteVariables(String input) {
266             if (input == null) return null;
267             for (Map.Entry<String, String> variable : variables.entrySet()) {
268                 input = input.replace("${" + variable.getKey() + "}", variable.getValue());
269             }
270             return input;
271         }
272     }
273 }
274