Revert ChoiceSchemaNode.getCases() to return a Collection
[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / choice / 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.parser.rfc7950.stmt.choice;
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.DerivableSchemaNode;
21 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
22 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
23 import org.opendaylight.yangtools.yang.model.api.stmt.ChoiceEffectiveStatement;
24 import org.opendaylight.yangtools.yang.model.api.stmt.ChoiceStatement;
25 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaTreeEffectiveStatement;
26 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.AbstractDeclaredEffectiveStatement.DefaultWithSchemaTree.WithSubstatements;
27 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStatementMixins.AugmentationTargetMixin;
28 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStatementMixins.DataSchemaNodeMixin;
29 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStatementMixins.MandatoryMixin;
30 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
31
32 final class ChoiceEffectiveStatementImpl extends WithSubstatements<QName, ChoiceStatement, ChoiceEffectiveStatement>
33         implements ChoiceEffectiveStatement, ChoiceSchemaNode, DerivableSchemaNode,
34                    DataSchemaNodeMixin<QName, ChoiceStatement>, AugmentationTargetMixin<QName, ChoiceStatement>,
35                    MandatoryMixin<QName, ChoiceStatement> {
36     private final CaseSchemaNode defaultCase;
37     private final ChoiceSchemaNode original;
38     private final @NonNull SchemaPath path;
39     private final int flags;
40
41     ChoiceEffectiveStatementImpl(final ChoiceStatement declared,
42             final StmtContext<QName, ChoiceStatement, ChoiceEffectiveStatement> ctx,
43             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements, final int flags,
44             final @Nullable CaseSchemaNode defaultCase, final @Nullable ChoiceSchemaNode original) {
45         super(declared, ctx, substatements);
46         this.flags = flags;
47         this.path = ctx.getSchemaPath().get();
48         this.defaultCase = defaultCase;
49         this.original = original;
50     }
51
52     @Override
53     public @NonNull QName argument() {
54         return getQName();
55     }
56
57     @Override
58     public SchemaPath getPath() {
59         return path;
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 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 String toString() {
90         return ChoiceEffectiveStatementImpl.class.getSimpleName() + "[" + "qname=" + getQName() + "]";
91     }
92 }