code-generator/binding-java-api-generator/src/main/java/org/opendaylight/yangtools/sal/java/api/generator/EnumGenerator.java

   1 /*\r
   2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.\r
   3  *\r
   4  * This program and the accompanying materials are made available under the\r
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,\r
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html\r
   7  */\r
   8 package org.opendaylight.yangtools.sal.java.api.generator;\r
   9 \r
  10 import org.opendaylight.yangtools.sal.binding.model.api.CodeGenerator;\r
  11 import org.opendaylight.yangtools.sal.binding.model.api.Enumeration;\r
  12 import org.opendaylight.yangtools.sal.binding.model.api.Type;\r
  13 \r
  14 public class EnumGenerator implements CodeGenerator {\r
  15 \r
  16     @Override\r
  17     public boolean isAcceptable(Type type) {\r
  18         return type instanceof Enumeration;\r
  19     }\r
  20 \r
  21     @Override\r
  22     public String generate(Type type) {\r
  23         if (type instanceof Enumeration) {\r
  24             final Enumeration enums = (Enumeration) type;\r
  25             final EnumTemplate enumTemplate = new EnumTemplate(enums);\r
  26             return enumTemplate.generate();\r
  27         }\r
  28         return "";\r
  29     }\r
  30 \r
  31     @Override\r
  32     public String getUnitName(Type type) {\r
  33         return type.getName();\r
  34     }\r
  35 \r
  36 }\r