yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/rfc6020/effective/AbstractEffectiveDocumentedDataNodeContainer.java

   1 /*
   2  * Copyright (c) 2014 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.stmt.rfc6020.effective;
   9
  10 import java.util.Collection;
  11 import java.util.HashMap;
  12 import java.util.HashSet;
  13 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
  14
  15 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
  16 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
  17 import java.util.Map;
  18 import java.util.Set;
  19 import org.opendaylight.yangtools.yang.common.QName;
  20 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
  21 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
  22 import org.opendaylight.yangtools.yang.model.api.GroupingDefinition;
  23 import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
  24 import org.opendaylight.yangtools.yang.model.api.UsesNode;
  25
  26 public abstract class AbstractEffectiveDocumentedDataNodeContainer<A, D extends DeclaredStatement<A>>
  27         extends AbstractEffectiveDocumentedNode<A, D> implements
  28         DataNodeContainer {
  29
  30     private final Map<QName, DataSchemaNode> childNodes;
  31     private final Set<GroupingDefinition> groupings;
  32     private final Set<UsesNode> uses;
  33     private final Set<TypeDefinition<?>> typeDefinitions;
  34     private final Set<DataSchemaNode> publicChildNodes;
  35
  36     protected AbstractEffectiveDocumentedDataNodeContainer(
  37             final StmtContext<A, D, ?> ctx) {
  38         super(ctx);
  39
  40         Collection<? extends EffectiveStatement<?, ?>> effectiveSubstatements = effectiveSubstatements();
  41
  42         childNodes = new HashMap<QName, DataSchemaNode>();
  43         groupings = new HashSet<GroupingDefinition>();
  44         uses = new HashSet<UsesNode>();
  45         typeDefinitions = new HashSet<TypeDefinition<?>>();
  46         publicChildNodes = new HashSet<DataSchemaNode>();
  47
  48         for (EffectiveStatement<?, ?> effectiveStatement : effectiveSubstatements) {
  49             if (effectiveStatement instanceof DataSchemaNode) {
  50                 DataSchemaNode dataSchemaNode = (DataSchemaNode) effectiveStatement;
  51
  52                 childNodes.put(dataSchemaNode.getQName(), dataSchemaNode);
  53                 publicChildNodes.add(dataSchemaNode);
  54             }
  55             if (effectiveStatement instanceof UsesNode) {
  56                 UsesNode usesNode = (UsesNode) effectiveStatement;
  57                 uses.add(usesNode);
  58             }
  59             if (effectiveStatement instanceof TypeDefinition) {
  60                 TypeDefinition<?> typeDef = (TypeDefinition<?>) effectiveStatement;
  61                 typeDefinitions.add(typeDef);
  62             }
  63             if (effectiveStatement instanceof GroupingDefinition) {
  64                 GroupingDefinition grp = (GroupingDefinition) effectiveStatement;
  65                 groupings.add(grp);
  66             }
  67         }
  68     }
  69
  70     @Override
  71     public final Set<TypeDefinition<?>> getTypeDefinitions() {
  72         return typeDefinitions;
  73     }
  74
  75     @Override
  76     public final Set<DataSchemaNode> getChildNodes() {
  77         return publicChildNodes;
  78     }
  79
  80     @Override
  81     public final Set<GroupingDefinition> getGroupings() {
  82         return groupings;
  83     }
  84
  85     @Override
  86     public final DataSchemaNode getDataChildByName(final QName name) {
  87         // Child nodes are keyed by their container name, so we can do a direct
  88         // lookup
  89         return childNodes.get(name);
  90     }
  91
  92     @Override
  93     public final DataSchemaNode getDataChildByName(final String name) {
  94         for (DataSchemaNode node : childNodes.values()) {
  95             if (node.getQName().getLocalName().equals(name)) {
  96                 return node;
  97             }
  98         }
  99         return null;
 100     }
 101
 102     @Override
 103     public Set<UsesNode> getUses() {
 104         return uses;
 105     }
 106
 107 }