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.NL;
  11 import static org.opendaylight.controller.sal.java.api.generator.Constants.RCB;
  12 import static org.opendaylight.controller.sal.java.api.generator.Constants.TAB;
  13
  14 import java.io.IOException;
  15 import java.io.StringWriter;
  16 import java.io.Writer;
  17 import java.util.List;
  18
  19 import org.opendaylight.controller.sal.binding.model.api.CodeGenerator;
  20 import org.opendaylight.controller.sal.binding.model.api.GeneratedProperty;
  21 import org.opendaylight.controller.sal.binding.model.api.GeneratedTransferObject;
  22 import org.opendaylight.controller.sal.binding.model.api.Type;
  23
  24 public class ClassCodeGenerator implements CodeGenerator {
  25
  26     @Override
  27     public Writer generate(Type type) throws IOException {
  28         final Writer writer = new StringWriter();
  29         if (type instanceof GeneratedTransferObject) {
  30             GeneratedTransferObject genTO = (GeneratedTransferObject) type;
  31             final List<GeneratedProperty> fields = genTO.getProperties();
  32
  33             writer.write(GeneratorUtil.createClassDeclarationWithPkgName(genTO,
  34                     ""));
  35             writer.write(NL);
  36             writer.write(NL);
  37
  38             if (fields != null) {
  39                 for (GeneratedProperty field : fields) {
  40                     writer.write(GeneratorUtil.createField(field, TAB) + NL);
  41                 }
  42                 writer.write(NL);
  43                 writer.write(GeneratorUtil.createConstructor(genTO, TAB) + NL);
  44                 writer.write(NL);
  45                 for (GeneratedProperty field : fields) {
  46                     writer.write(GeneratorUtil.createGetter(field, TAB) + NL);
  47                     if (!field.isReadOnly()) {
  48                         writer.write(GeneratorUtil.createSetter(field, TAB) + NL);
  49                     }
  50                 }
  51                 writer.write(NL);
  52
  53                 if (!genTO.getHashCodeIdentifiers().isEmpty()) {
  54                     writer.write(GeneratorUtil.createHashCode(
  55                             genTO.getHashCodeIdentifiers(), TAB)
  56                             + NL);
  57                 }
  58
  59                 if (!genTO.getEqualsIdentifiers().isEmpty()) {
  60                     writer.write(GeneratorUtil.createEquals(genTO,
  61                             genTO.getEqualsIdentifiers(), TAB)
  62                             + NL);
  63                 }
  64
  65                 if (!genTO.getToStringIdentifiers().isEmpty()) {
  66                     writer.write(GeneratorUtil.createToString(genTO,
  67                             genTO.getToStringIdentifiers(), TAB)
  68                             + NL);
  69
  70                 }
  71
  72                 writer.write(RCB);
  73             }
  74         }
  75         return writer;
  76     }
  77
  78 }