opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/InterfaceGenerator.java

   1 /*
   2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8 package org.opendaylight.controller.sal.java.api.generator;
   9
  10 import static org.opendaylight.controller.sal.java.api.generator.Constants.*;
  11
  12 import java.io.IOException;
  13 import java.io.StringWriter;
  14 import java.io.Writer;
  15 import java.util.LinkedHashMap;
  16 import java.util.List;
  17 import java.util.Map;
  18
  19 import org.opendaylight.controller.sal.binding.model.api.CodeGenerator;
  20 import org.opendaylight.controller.sal.binding.model.api.Constant;
  21 import org.opendaylight.controller.sal.binding.model.api.Enumeration;
  22 import org.opendaylight.controller.sal.binding.model.api.GeneratedTransferObject;
  23 import org.opendaylight.controller.sal.binding.model.api.GeneratedType;
  24 import org.opendaylight.controller.sal.binding.model.api.MethodSignature;
  25 import org.opendaylight.controller.sal.binding.model.api.Type;
  26
  27 public final class InterfaceGenerator implements CodeGenerator {
  28
  29         private Map<String, LinkedHashMap<String, Integer>> imports;
  30
  31         private String generateEnums(List<Enumeration> enums) {
  32                 String result = "";
  33                 if (enums != null) {
  34                         EnumGenerator enumGenerator = new EnumGenerator();
  35                         for (Enumeration en : enums) {
  36                                 try {
  37                                         result = result
  38                                                         + (enumGenerator.generateInnerEnumeration(en, TAB).toString() + NL);
  39                                 } catch (IOException e) {
  40                                         e.printStackTrace();
  41                                 }
  42                         }
  43                 }
  44                 return result;
  45         }
  46
  47         private String generateConstants(List<Constant> constants, String pkgName) {
  48                 String result = "";
  49                 if (constants != null) {
  50                         for (Constant c : constants) {
  51                                 result = result
  52                                                 + GeneratorUtil
  53                                                                 .createConstant(c, TAB, imports, pkgName) + NL;
  54                         }
  55                         result.concat(NL);
  56                 }
  57                 return result;
  58         }
  59
  60         public String generateMethods(List<MethodSignature> methods, String pkgName) {
  61                 String result = "";
  62
  63                 if (methods != null) {
  64                         for (MethodSignature m : methods) {
  65                                 result = result
  66                                                 + GeneratorUtil.createMethodDeclaration(m, TAB,
  67                                                                 imports, pkgName) + NL;
  68                         }
  69                         result = result + NL;
  70                 }
  71                 return result;
  72         }
  73
  74         public Writer generate(Type type) throws IOException {
  75                 Writer writer = new StringWriter();
  76                 if (type instanceof GeneratedType
  77                                 && !(type instanceof GeneratedTransferObject)) {
  78                         GeneratedType genType = (GeneratedType) type;
  79                         imports = GeneratorUtil.createImports(genType);
  80
  81                         final String currentPkg = genType.getPackageName();
  82                         final List<Constant> constants = genType.getConstantDefinitions();
  83                         final List<MethodSignature> methods = genType
  84                                         .getMethodDefinitions();
  85                         final List<Enumeration> enums = genType.getEnumerations();
  86
  87                         writer.write(GeneratorUtil.createPackageDeclaration(genType
  88                                         .getPackageName()));
  89                         writer.write(NL);
  90
  91                         List<String> importLines = GeneratorUtil.createImportLines(imports);
  92                         for (String line : importLines) {
  93                                 writer.write(line + NL);
  94                         }
  95                         writer.write(NL);
  96                         writer.write(GeneratorUtil.createIfcDeclaration(genType, "",
  97                                         imports));
  98                         writer.write(NL);
  99
 100                         writer.write(generateEnums(enums));
 101                         writer.write(generateConstants(constants, currentPkg));
 102                         writer.write(generateMethods(methods, currentPkg));
 103
 104                         writer.write(RCB);
 105                 }
 106                 return writer;
 107         }
 108 }