Fix DerivableSchemaNode.getOriginal()
[yangtools.git] / model / yang-model-ri / src / main / java / org / opendaylight / yangtools / yang / model / ri / stmt / impl / eff / ChoiceEffectiveStatementImpl.java
1 /*
2  * Copyright (c) 2015 Cisco Systems, Inc. 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.model.ri.stmt.impl.eff;
9
10 import static java.util.Objects.requireNonNull;
11
12 import com.google.common.collect.ImmutableList;
13 import java.util.Collection;
14 import java.util.Optional;
15 import org.eclipse.jdt.annotation.NonNull;
16 import org.eclipse.jdt.annotation.Nullable;
17 import org.opendaylight.yangtools.yang.common.QName;
18 import org.opendaylight.yangtools.yang.model.api.CaseSchemaNode;
19 import org.opendaylight.yangtools.yang.model.api.ChoiceSchemaNode;
20 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
21 import org.opendaylight.yangtools.yang.model.api.stmt.ChoiceEffectiveStatement;
22 import org.opendaylight.yangtools.yang.model.api.stmt.ChoiceStatement;
23 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaTreeEffectiveStatement;
24 import org.opendaylight.yangtools.yang.model.spi.meta.AbstractDeclaredEffectiveStatement.DefaultWithDataTree;
25 import org.opendaylight.yangtools.yang.model.spi.meta.EffectiveStatementMixins.AugmentationTargetMixin;
26 import org.opendaylight.yangtools.yang.model.spi.meta.EffectiveStatementMixins.DataSchemaNodeMixin;
27 import org.opendaylight.yangtools.yang.model.spi.meta.EffectiveStatementMixins.MandatoryMixin;
28
29 public final class ChoiceEffectiveStatementImpl
30         extends DefaultWithDataTree<QName, ChoiceStatement, ChoiceEffectiveStatement>
31         implements ChoiceEffectiveStatement, ChoiceSchemaNode, DataSchemaNodeMixin<ChoiceStatement>,
32             AugmentationTargetMixin<QName, ChoiceStatement>, MandatoryMixin<QName, ChoiceStatement> {
33     private final CaseSchemaNode defaultCase;
34     private final ChoiceSchemaNode original;
35     private final @NonNull QName argument;
36     private final int flags;
37
38     public ChoiceEffectiveStatementImpl(final ChoiceStatement declared,
39             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements, final QName argument,
40             final int flags, final @Nullable CaseSchemaNode defaultCase, final @Nullable ChoiceSchemaNode original) {
41         super(declared, substatements);
42         this.argument = requireNonNull(argument);
43         this.flags = flags;
44         this.defaultCase = defaultCase;
45         this.original = original;
46     }
47
48     public ChoiceEffectiveStatementImpl(final ChoiceEffectiveStatementImpl origEffective, final QName argument,
49             final int flags, final ChoiceSchemaNode newOriginal) {
50         super(origEffective);
51         this.argument = requireNonNull(argument);
52         this.flags = flags;
53         this.defaultCase = origEffective.defaultCase;
54         this.original = newOriginal;
55     }
56
57     @Override
58     public QName argument() {
59         return argument;
60     }
61
62     @Override
63     public int flags() {
64         return flags;
65     }
66
67     @Override
68     public Optional<ChoiceSchemaNode> getOriginal() {
69         return Optional.ofNullable(original);
70     }
71
72     @Override
73     public Optional<? extends CaseSchemaNode> findCase(final QName qname) {
74         final SchemaTreeEffectiveStatement<?> child = schemaTreeNamespace().get(requireNonNull(qname));
75         return child instanceof CaseSchemaNode ? Optional.of((CaseSchemaNode) child) : Optional.empty();
76     }
77
78     @Override
79     public Collection<? extends @NonNull CaseSchemaNode> getCases() {
80         return filterEffectiveStatements(CaseSchemaNode.class);
81     }
82
83     @Override
84     public Optional<CaseSchemaNode> getDefaultCase() {
85         return Optional.ofNullable(defaultCase);
86     }
87
88     @Override
89     public ChoiceEffectiveStatement asEffectiveStatement() {
90         return this;
91     }
92
93     @Override
94     public String toString() {
95         return ChoiceEffectiveStatementImpl.class.getSimpleName() + "[" + "qname=" + argument + "]";
96     }
97 }