2 * Copyright (c) 2013 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.controller.sal.java.api.generator;
10 import java.io.BufferedWriter;
12 import java.io.FileWriter;
13 import java.io.IOException;
14 import java.io.Writer;
15 import java.util.ArrayList;
16 import java.util.HashSet;
17 import java.util.List;
20 import org.opendaylight.controller.sal.binding.model.api.CodeGenerator;
21 import org.opendaylight.controller.sal.binding.model.api.GeneratedTransferObject;
22 import org.opendaylight.controller.sal.binding.model.api.GeneratedType;
23 import org.opendaylight.controller.sal.binding.model.api.Type;
24 import org.slf4j.Logger;
25 import org.slf4j.LoggerFactory;
27 public class GeneratorJavaFile {
29 private static final Logger log = LoggerFactory
30 .getLogger(GeneratorJavaFile.class);
31 private final CodeGenerator interfaceGenerator;
32 private final ClassCodeGenerator classGenerator;
33 private final Set<GeneratedType> genTypes;
34 private final Set<GeneratedTransferObject> genTransferObjects;
36 public GeneratorJavaFile(final CodeGenerator codeGenerator,
37 final Set<GeneratedType> types) {
38 this.interfaceGenerator = codeGenerator;
39 this.genTypes = types;
40 this.genTransferObjects = new HashSet<>();
41 classGenerator = new ClassCodeGenerator();
44 public GeneratorJavaFile(final Set<GeneratedType> types,
45 final Set<GeneratedTransferObject> genTransferObjects) {
46 this.interfaceGenerator = new InterfaceGenerator();
47 this.classGenerator = new ClassCodeGenerator();
48 this.genTypes = types;
49 this.genTransferObjects = genTransferObjects;
52 public List<File> generateToFile(final File parentDirectory) throws IOException {
53 final List<File> result = new ArrayList<>();
54 for (GeneratedType type : genTypes) {
55 final File genFile = generateTypeToJavaFile(parentDirectory, type,
58 if (genFile != null) {
62 for (GeneratedTransferObject transferObject : genTransferObjects) {
63 final File genFile = generateTypeToJavaFile(parentDirectory,
64 transferObject, classGenerator);
66 if (genFile != null) {
73 private File generateTypeToJavaFile(final File parentDir, final Type type,
74 final CodeGenerator generator) throws IOException {
75 if (parentDir == null) {
76 log.warn("Parent Directory not specified, files will be generated "
77 + "accordingly to generated Type package path.");
80 log.error("Cannot generate Type into Java File because " +
81 "Generated Type is NULL!");
82 throw new IllegalArgumentException("Generated Type Cannot be NULL!");
84 if (generator == null) {
85 log.error("Cannot generate Type into Java File because " +
86 "Code Generator instance is NULL!");
87 throw new IllegalArgumentException("Code Generator Cannot be NULL!");
89 final File packageDir = packageToDirectory(parentDir,
90 type.getPackageName());
92 if (!packageDir.exists()) {
95 final File file = new File(packageDir, type.getName() + ".java");
96 try (final FileWriter fw = new FileWriter(file)) {
99 try (final BufferedWriter bw = new BufferedWriter(fw)) {
100 Writer writer = generator.generate(type);
101 bw.write(writer.toString());
103 } catch (IOException e) {
104 log.error(e.getMessage());
105 throw new IOException(e.getMessage());
110 private File packageToDirectory(final File parentDirectory,
111 final String packageName) {
112 if (packageName == null) {
113 throw new IllegalArgumentException("Package Name cannot be NULL!");
116 final String[] subDirNames = packageName.split("\\.");
117 final StringBuilder dirPathBuilder = new StringBuilder();
118 dirPathBuilder.append(subDirNames[0]);
119 for (int i = 1; i < subDirNames.length; ++i) {
120 dirPathBuilder.append(File.separator);
121 dirPathBuilder.append(subDirNames[i]);
123 return new File(parentDirectory, dirPathBuilder.toString());