Merge "Fix for Bug 241."
[yangtools.git] / code-generator / binding-java-api-generator / src / main / java / org / opendaylight / yangtools / sal / java / api / generator / TOGenerator.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.GeneratedTransferObject;\r
12 import org.opendaylight.yangtools.sal.binding.model.api.Type;\r
13 \r
14 /**\r
15  *\r
16  * Transformator of the data from the virtual form to JAVA source code. The\r
17  * result source code represents JAVA class. For generating of the source code\r
18  * is used the template written in XTEND language.\r
19  *\r
20  */\r
21 public final class TOGenerator implements CodeGenerator {\r
22 \r
23     /**\r
24      * Generates JAVA source code for generated type <code>Type</code>. The code\r
25      * is generated according to the template source code template which is\r
26      * written in XTEND language.\r
27      */\r
28     @Override\r
29     public String generate(Type type) {\r
30         if (type instanceof GeneratedTransferObject) {\r
31             final GeneratedTransferObject genTO = (GeneratedTransferObject) type;\r
32             if(genTO.isUnionType()) {\r
33                 final UnionTemplate template = new UnionTemplate(genTO);\r
34                 return template.generate();\r
35             } else if (genTO.isUnionTypeBuilder()) {\r
36                 final UnionBuilderTemplate template = new UnionBuilderTemplate(genTO);\r
37                 return template.generate();\r
38             } else {\r
39                 final ClassTemplate template = new ClassTemplate(genTO);\r
40                 return template.generate();\r
41             }\r
42         }\r
43         return "";\r
44     }\r
45 \r
46     @Override\r
47     public boolean isAcceptable(Type type) {\r
48         return type instanceof GeneratedTransferObject;\r
49     }\r
50 \r
51     @Override\r
52     public String getUnitName(Type type) {\r
53         return type.getName();\r
54     }\r
55 \r
56 }\r