opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/ClassCodeGenerator.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.binding.generator.util.TypeConstants;
  20 import org.opendaylight.controller.sal.binding.model.api.CodeGenerator;
  21 import org.opendaylight.controller.sal.binding.model.api.Constant;
  22 import org.opendaylight.controller.sal.binding.model.api.Enumeration;
  23 import org.opendaylight.controller.sal.binding.model.api.GeneratedProperty;
  24 import org.opendaylight.controller.sal.binding.model.api.GeneratedTransferIdentityObject;
  25 import org.opendaylight.controller.sal.binding.model.api.GeneratedTransferObject;
  26 import org.opendaylight.controller.sal.binding.model.api.Type;
  27
  28 public final class ClassCodeGenerator implements CodeGenerator {
  29
  30     private Map<String, LinkedHashMap<String, Integer>> imports;
  31
  32     @Override
  33     public Writer generate(Type type) throws IOException {
  34         final Writer writer = new StringWriter();
  35
  36         if (type instanceof GeneratedTransferObject) {
  37             GeneratedTransferObject genTO = (GeneratedTransferObject) type;
  38             imports = GeneratorUtil.createImports(genTO);
  39
  40             final String currentPkg = genTO.getPackageName();
  41             final List<GeneratedProperty> fields = genTO.getProperties();
  42             final List<Enumeration> enums = genTO.getEnumerations();
  43             final List<Constant> consts = genTO.getConstantDefinitions();
  44
  45             writer.write(GeneratorUtil.createPackageDeclaration(currentPkg));
  46             writer.write(NL);
  47
  48             List<String> importLines = GeneratorUtil.createImportLines(imports);
  49             for (String line : importLines) {
  50                 writer.write(line + NL);
  51             }
  52             writer.write(NL);
  53
  54             writer.write(GeneratorUtil.createClassDeclaration(genTO, "", imports, genTO.isAbstract()));
  55             writer.write(NL);
  56             writer.write(NL);
  57
  58             if (consts != null) {
  59                 for (Constant con : consts) {
  60                     writer.write(GeneratorUtil.createConstant(con, TAB, imports, currentPkg));
  61                     writer.write(NL);
  62                 }
  63             }
  64
  65             if (enums != null) {
  66                 EnumGenerator enumGenerator = new EnumGenerator();
  67                 for (Enumeration e : enums) {
  68                     writer.write(enumGenerator.generateInnerEnumeration(e, TAB).toString());
  69                     writer.write(NL);
  70                 }
  71             }
  72
  73             boolean memberPatternListCodeRequired = false;
  74             memberPatternListCodeRequired = (GeneratorUtil.isConstantInTO(TypeConstants.PATTERN_CONSTANT_NAME, genTO));
  75             if (fields != null || memberPatternListCodeRequired) {
  76                 if (fields != null) {
  77                     for (GeneratedProperty field : fields) {
  78                         writer.write(GeneratorUtil.createField(field, TAB, imports, currentPkg) + NL);
  79                     }
  80                 }
  81                 if (memberPatternListCodeRequired) {
  82                     writer.write(TAB + PRIVATE + GAP + "List<Pattern>" + GAP + MEMBER_PATTERN_LIST + GAP + ASSIGN + GAP
  83                             + "new ArrayList<Pattern>()" + GAP + SC + NL);
  84
  85                 }
  86                 writer.write(NL);
  87                 writer.write(GeneratorUtil.createConstructor(genTO, TAB, imports, genTO.isAbstract()) + NL);
  88                 writer.write(NL);
  89                 if (fields != null) {
  90                     for (GeneratedProperty field : fields) {
  91                         writer.write(GeneratorUtil.createGetter(field, TAB, imports, currentPkg) + NL);
  92                         if (!field.isReadOnly()) {
  93                             writer.write(GeneratorUtil.createSetter(field, TAB, imports, currentPkg) + NL);
  94                         }
  95                     }
  96                 }
  97                 writer.write(NL);
  98
  99                 if (!genTO.getHashCodeIdentifiers().isEmpty()) {
 100                     writer.write(GeneratorUtil.createHashCode(genTO.getHashCodeIdentifiers(), TAB) + NL);
 101                 }
 102
 103                 if (!genTO.getEqualsIdentifiers().isEmpty()) {
 104                     writer.write(GeneratorUtil.createEquals(genTO, genTO.getEqualsIdentifiers(), TAB) + NL);
 105                 }
 106
 107                 if (!genTO.getToStringIdentifiers().isEmpty()) {
 108                     writer.write(GeneratorUtil.createToString(genTO, genTO.getToStringIdentifiers(), TAB) + NL);
 109
 110                 }
 111
 112                 writer.write(RCB);
 113             }
 114         }
 115         return writer;
 116     }
 117
 118 }