e3130e06849017690c6ead66796cbb07b6114b16
[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / type / BuiltinTypeStatement.java
1 /*
2  * Copyright (c) 2016 Pantheon Technologies s.r.o. 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.yangtools.yang.parser.rfc7950.stmt.type;
9
10 import static java.util.Objects.requireNonNull;
11
12 import com.google.common.collect.ImmutableMap;
13 import com.google.common.collect.ImmutableMap.Builder;
14 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
15 import org.opendaylight.yangtools.yang.model.api.meta.StatementSource;
16 import org.opendaylight.yangtools.yang.model.api.stmt.TypeStatement;
17 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.AbstractDeclaredStatement.WithRawStringArgument;
18
19 final class BuiltinTypeStatement extends WithRawStringArgument implements TypeStatement {
20     private static final ImmutableMap<String, BuiltinTypeStatement> BUILTINS;
21
22     static {
23         final Builder<String, BuiltinTypeStatement> builder = ImmutableMap.builder();
24         putBuiltin(builder, AbstractTypeStatementSupport.BINARY);
25         putBuiltin(builder, AbstractTypeStatementSupport.BOOLEAN);
26         putBuiltin(builder, AbstractTypeStatementSupport.EMPTY);
27         putBuiltin(builder, AbstractTypeStatementSupport.INSTANCE_IDENTIFIER);
28         putBuiltin(builder, AbstractTypeStatementSupport.INT8);
29         putBuiltin(builder, AbstractTypeStatementSupport.INT16);
30         putBuiltin(builder, AbstractTypeStatementSupport.INT32);
31         putBuiltin(builder, AbstractTypeStatementSupport.INT64);
32         putBuiltin(builder, AbstractTypeStatementSupport.STRING);
33         putBuiltin(builder, AbstractTypeStatementSupport.UINT8);
34         putBuiltin(builder, AbstractTypeStatementSupport.UINT16);
35         putBuiltin(builder, AbstractTypeStatementSupport.UINT32);
36         putBuiltin(builder, AbstractTypeStatementSupport.UINT64);
37         BUILTINS = builder.build();
38     }
39
40     private static void putBuiltin(final Builder<String, BuiltinTypeStatement> builder, final String argument) {
41         builder.put(argument, new BuiltinTypeStatement(argument));
42     }
43
44     private BuiltinTypeStatement(final String rawArgument) {
45         super(requireNonNull(rawArgument));
46     }
47
48     static TypeStatement maybeReplace(final TypeStatementImpl orig) {
49         if (orig.declaredSubstatements().isEmpty() && orig.getStatementSource() == StatementSource.DECLARATION
50                 && orig.statementDefinition() == YangStmtMapping.TYPE) {
51             final BuiltinTypeStatement builtin = BUILTINS.get(orig.argument());
52             if (builtin != null) {
53                 return builtin;
54             }
55         }
56
57         return orig;
58     }
59 }