Retain DeclarationReference in DeclaredStatements
[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / meta / FractionDigitsStatementSupport.java
1 /*
2  * Copyright (c) 2017 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.meta;
9
10 import com.google.common.collect.ImmutableList;
11 import com.google.common.collect.ImmutableMap;
12 import com.google.common.collect.ImmutableMap.Builder;
13 import org.eclipse.jdt.annotation.NonNull;
14 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
15 import org.opendaylight.yangtools.yang.model.api.meta.DeclarationReference;
16 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
17 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
18 import org.opendaylight.yangtools.yang.model.api.stmt.FractionDigitsEffectiveStatement;
19 import org.opendaylight.yangtools.yang.model.api.stmt.FractionDigitsStatement;
20 import org.opendaylight.yangtools.yang.model.parser.api.YangParserConfiguration;
21 import org.opendaylight.yangtools.yang.model.ri.stmt.DeclaredStatementDecorators;
22 import org.opendaylight.yangtools.yang.model.ri.stmt.DeclaredStatements;
23 import org.opendaylight.yangtools.yang.model.ri.stmt.EffectiveStatements;
24 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
25 import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
26 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
27 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
28 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
29
30 public final class FractionDigitsStatementSupport
31         extends AbstractStatementSupport<Integer, FractionDigitsStatement, FractionDigitsEffectiveStatement> {
32     private static final SubstatementValidator SUBSTATEMENT_VALIDATOR =
33             SubstatementValidator.builder(YangStmtMapping.FRACTION_DIGITS).build();
34
35     // FIXME: move this to yang-model-spi
36     private static final ImmutableMap<FractionDigitsStatement, FractionDigitsEffectiveStatement> EMPTY_EFF;
37
38     static {
39         final Builder<FractionDigitsStatement, FractionDigitsEffectiveStatement> effBuilder = ImmutableMap.builder();
40         for (int i = 1; i <= 18; ++i) {
41             final FractionDigitsStatement decl = DeclaredStatements.createFractionDigits(i);
42             effBuilder.put(decl, EffectiveStatements.createFractionDigits(decl));
43         }
44         EMPTY_EFF = effBuilder.build();
45     }
46
47     public FractionDigitsStatementSupport(final YangParserConfiguration config) {
48         super(YangStmtMapping.FRACTION_DIGITS, StatementPolicy.contextIndependent(), config);
49     }
50
51     @Override
52     public Integer parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
53         final int fractionDigits;
54         try {
55             fractionDigits = Integer.parseInt(value);
56         } catch (NumberFormatException e) {
57             throw new SourceException(ctx, e, "%s is not valid fraction-digits integer argument", value);
58         }
59         if (fractionDigits < 1 || fractionDigits > 18) {
60             throw new SourceException("fraction-digits argument should be integer within [1..18]", ctx);
61         }
62         return fractionDigits;
63     }
64
65     @Override
66     protected SubstatementValidator getSubstatementValidator() {
67         return SUBSTATEMENT_VALIDATOR;
68     }
69
70     @Override
71     protected FractionDigitsStatement createDeclared(final StmtContext<Integer, FractionDigitsStatement, ?> ctx,
72             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
73         return DeclaredStatements.createFractionDigits(ctx.getArgument(), substatements);
74     }
75
76     @Override
77     protected FractionDigitsStatement attachDeclarationReference(final FractionDigitsStatement stmt,
78             final DeclarationReference reference) {
79         return DeclaredStatementDecorators.decorateFractionDigits(stmt, reference);
80     }
81
82     @Override
83     protected FractionDigitsEffectiveStatement createEffective(final Current<Integer, FractionDigitsStatement> stmt,
84             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
85         return substatements.isEmpty() ? createEmptyEffective(stmt.declared())
86             : EffectiveStatements.createFractionDigits(stmt.declared(), substatements);
87     }
88
89     private static @NonNull FractionDigitsEffectiveStatement createEmptyEffective(
90             final FractionDigitsStatement declared) {
91         final FractionDigitsEffectiveStatement shared = EMPTY_EFF.get(declared);
92         return shared != null ? shared : EffectiveStatements.createFractionDigits(declared);
93     }
94 }