Added YANG typedef support in the YANG parser
[controller.git] / opendaylight / sal / yang-prototype / yang / yang-model-util / src / main / java / org / opendaylight / controller / model / util / YangTypesConverter.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.controller.model.util;\r
9 \r
10 import java.util.HashMap;\r
11 import java.util.HashSet;\r
12 import java.util.List;\r
13 import java.util.Map;\r
14 import java.util.Set;\r
15 \r
16 import org.opendaylight.controller.model.api.type.BinaryTypeDefinition;\r
17 import org.opendaylight.controller.model.api.type.BitsTypeDefinition;\r
18 import org.opendaylight.controller.model.api.type.BooleanTypeDefinition;\r
19 import org.opendaylight.controller.model.api.type.DecimalTypeDefinition;\r
20 import org.opendaylight.controller.model.api.type.IntegerTypeDefinition;\r
21 import org.opendaylight.controller.model.api.type.RangeConstraint;\r
22 import org.opendaylight.controller.model.api.type.StringTypeDefinition;\r
23 import org.opendaylight.controller.model.api.type.UnsignedIntegerTypeDefinition;\r
24 import org.opendaylight.controller.yang.common.QName;\r
25 import org.opendaylight.controller.yang.model.api.TypeDefinition;\r
26 \r
27 public class YangTypesConverter {\r
28 \r
29     private static final Map<String, TypeDefinition<? extends TypeDefinition<?>>> baseYangTypeMap = new HashMap<String, TypeDefinition<? extends TypeDefinition<?>>>();\r
30     private static final Set<String> baseYangTypes = new HashSet<String>();\r
31 \r
32     private static final TypeDefinition<BinaryTypeDefinition> BINARY = new BinaryType();\r
33     private static final TypeDefinition<BitsTypeDefinition> BITS = new BitsType();\r
34     private static final TypeDefinition<BooleanTypeDefinition> BOOLEAN_TYPE = new BooleanType();\r
35     private static final TypeDefinition<IntegerTypeDefinition> INT8_TYPE = new Int8();\r
36     private static final TypeDefinition<IntegerTypeDefinition> INT16_TYPE = new Int16();\r
37     private static final TypeDefinition<IntegerTypeDefinition> INT32_TYPE = new Int32();\r
38     private static final TypeDefinition<IntegerTypeDefinition> INT64_TYPE = new Int64();\r
39     private static final TypeDefinition<StringTypeDefinition> STRING_TYPE = new StringType();\r
40     private static final TypeDefinition<UnsignedIntegerTypeDefinition> UINT8_TYPE = new Uint8();\r
41     private static final TypeDefinition<UnsignedIntegerTypeDefinition> UINT16_TYPE = new Uint16();\r
42     private static final TypeDefinition<UnsignedIntegerTypeDefinition> UINT32_TYPE = new Uint32();\r
43     private static final TypeDefinition<UnsignedIntegerTypeDefinition> UINT64_TYPE = new Uint64();\r
44 \r
45     static {\r
46         baseYangTypeMap.put("binary", BINARY);\r
47         baseYangTypeMap.put("bits", BITS);\r
48         baseYangTypeMap.put("boolean", BOOLEAN_TYPE);\r
49         baseYangTypeMap.put("int8", INT8_TYPE);\r
50         baseYangTypeMap.put("int16", INT16_TYPE);\r
51         baseYangTypeMap.put("int32", INT32_TYPE);\r
52         baseYangTypeMap.put("int64", INT64_TYPE);\r
53         baseYangTypeMap.put("string", STRING_TYPE);\r
54         baseYangTypeMap.put("uint8", UINT8_TYPE);\r
55         baseYangTypeMap.put("uint16", UINT16_TYPE);\r
56         baseYangTypeMap.put("uint32", UINT32_TYPE);\r
57         baseYangTypeMap.put("uint64", UINT64_TYPE);\r
58 \r
59         baseYangTypes.add("binary");\r
60         baseYangTypes.add("bits");\r
61         baseYangTypes.add("boolean");\r
62         baseYangTypes.add("decimal64");\r
63         baseYangTypes.add("empty");\r
64         baseYangTypes.add("enumeration");\r
65         baseYangTypes.add("identityref");\r
66         baseYangTypes.add("instance-identifier");\r
67         baseYangTypes.add("int8");\r
68         baseYangTypes.add("int16");\r
69         baseYangTypes.add("int32");\r
70         baseYangTypes.add("int64");\r
71         baseYangTypes.add("leafref");\r
72         baseYangTypes.add("string");\r
73         baseYangTypes.add("uint8");\r
74         baseYangTypes.add("uint16");\r
75         baseYangTypes.add("uint32");\r
76         baseYangTypes.add("uint64");\r
77         baseYangTypes.add("union");\r
78     }\r
79 \r
80     public static boolean isBaseYangType(String type) {\r
81         return baseYangTypes.contains(type);\r
82     }\r
83 \r
84     public static TypeDefinition<?> javaTypeForBaseYangType(QName typeQName) {\r
85         TypeDefinition<?> type = baseYangTypeMap.get(typeQName.getLocalName());\r
86         return type;\r
87     }\r
88 \r
89     public static TypeDefinition<?> javaTypeForBaseYangType(String typeName) {\r
90         TypeDefinition<?> type = baseYangTypeMap.get(typeName);\r
91         return type;\r
92     }\r
93 \r
94     public static TypeDefinition<IntegerTypeDefinition> javaTypeForBaseYangSignedIntegerType(\r
95             String typeName, List<RangeConstraint> ranges) {\r
96         if (typeName.equals("int8")) {\r
97             return new Int8(ranges, null, null);\r
98         } else if (typeName.equals("int16")) {\r
99             return new Int16(ranges, null, null);\r
100         } else if (typeName.equals("int32")) {\r
101             return new Int32(ranges, null, null);\r
102         } else if (typeName.equals("int64")) {\r
103             return new Int64(ranges, null, null);\r
104         }\r
105         return null;\r
106     }\r
107 \r
108     public static TypeDefinition<UnsignedIntegerTypeDefinition> javaTypeForBaseYangUnsignedIntegerType(\r
109             final String typeName, List<RangeConstraint> ranges) {\r
110         if (typeName.equals("uint8")) {\r
111             return new Uint8(ranges, null, null);\r
112         } else if (typeName.equals("uint16")) {\r
113             return new Uint16(ranges, null, null);\r
114         } else if (typeName.equals("uint32")) {\r
115             return new Uint32(ranges, null, null);\r
116         } else if (typeName.equals("uint64")) {\r
117             return new Uint64(ranges, null, null);\r
118         }\r
119         return null;\r
120     }\r
121 \r
122     public static TypeDefinition<DecimalTypeDefinition> javaTypeForBaseYangDecimal64Type(\r
123             List<RangeConstraint> rangeStatements, int fractionDigits) {\r
124         return new Decimal64(rangeStatements, fractionDigits);\r
125     }\r
126 \r
127 }\r