yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/reactor/SubstatementContext.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.stmt.reactor;
   9
  10 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier;
  11 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils;
  12 import org.opendaylight.yangtools.yang.model.api.stmt.KeyStatement;
  13 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
  14 import java.util.List;
  15 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.GroupingUtils;
  16 import org.opendaylight.yangtools.yang.common.QNameModule;
  17 import java.util.Collection;
  18 import org.opendaylight.yangtools.yang.common.QName;
  19 import com.google.common.base.Preconditions;
  20 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
  21 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
  22 import org.opendaylight.yangtools.yang.parser.spi.meta.NamespaceBehaviour.NamespaceStorageNode;
  23 import org.opendaylight.yangtools.yang.parser.spi.meta.NamespaceBehaviour.Registry;
  24 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
  25
  26 class SubstatementContext<A, D extends DeclaredStatement<A>, E extends EffectiveStatement<A, D>>
  27         extends StatementContextBase<A, D, E> {
  28
  29     private final StatementContextBase<?, ?, ?> parent;
  30     private final A argument;
  31
  32     SubstatementContext(StatementContextBase<?, ?, ?> parent,
  33             ContextBuilder<A, D, E> builder) throws SourceException {
  34         super(builder);
  35         this.parent = Preconditions.checkNotNull(parent,
  36                 "Parent must not be null");
  37         this.argument = builder.getDefinition().parseArgumentValue(this,
  38                 builder.getRawArgument());
  39     }
  40
  41     @SuppressWarnings("unchecked")
  42     SubstatementContext(SubstatementContext<A, D, E> original,
  43             QNameModule newQNameModule,
  44             StatementContextBase<?, ?, ?> newParent, TypeOfCopy typeOfCopy)
  45             throws SourceException {
  46         super(original);
  47         this.parent = newParent;
  48
  49         if (newQNameModule != null) {
  50             if (original.argument instanceof QName) {
  51                 QName originalQName = (QName) original.argument;
  52                 this.argument = (A) QName.create(newQNameModule,
  53                         originalQName.getLocalName());
  54             } else if (StmtContextUtils.producesDeclared(original,
  55                     KeyStatement.class)) {
  56                 this.argument = (A) StmtContextUtils
  57                         .replaceModuleQNameForKey(
  58                                 (StmtContext<Collection<SchemaNodeIdentifier>, KeyStatement, ?>) original,
  59                                 newQNameModule);
  60             } else {
  61                 this.argument = original.argument;
  62             }
  63         } else {
  64             this.argument = original.argument;
  65         }
  66
  67         copyDeclaredStmts(original, newQNameModule, typeOfCopy);
  68
  69         copyEffectiveStmts(original, newQNameModule, typeOfCopy);
  70     }
  71
  72     private void copyDeclaredStmts(SubstatementContext<A, D, E> original,
  73             QNameModule newQNameModule, TypeOfCopy typeOfCopy)
  74             throws SourceException {
  75         Collection<? extends StatementContextBase<?, ?, ?>> originalDeclaredSubstatements = original
  76                 .declaredSubstatements();
  77         for (StatementContextBase<?, ?, ?> stmtContext : originalDeclaredSubstatements) {
  78             if (GroupingUtils.needToCopyByUses(stmtContext)) {
  79                 StatementContextBase<?, ?, ?> copy = stmtContext.createCopy(
  80                         newQNameModule, this, typeOfCopy);
  81                 this.addEffectiveSubstatement(copy);
  82             } else if (GroupingUtils.isReusedByUses(stmtContext)) {
  83                 this.addEffectiveSubstatement(stmtContext);
  84             }
  85         }
  86     }
  87
  88     private void copyEffectiveStmts(SubstatementContext<A, D, E> original,
  89             QNameModule newQNameModule, TypeOfCopy typeOfCopy)
  90             throws SourceException {
  91         Collection<? extends StatementContextBase<?, ?, ?>> originalEffectiveSubstatements = original
  92                 .effectiveSubstatements();
  93         for (StatementContextBase<?, ?, ?> stmtContext : originalEffectiveSubstatements) {
  94             if (GroupingUtils.needToCopyByUses(stmtContext)) {
  95                 StatementContextBase<?, ?, ?> copy = stmtContext.createCopy(
  96                         newQNameModule, this, typeOfCopy);
  97                 this.addEffectiveSubstatement(copy);
  98             } else if (GroupingUtils.isReusedByUses(stmtContext)) {
  99                 this.addEffectiveSubstatement(stmtContext);
 100             }
 101         }
 102     }
 103
 104     @Override
 105     public StatementContextBase<?, ?, ?> getParentContext() {
 106         return parent;
 107     }
 108
 109     @Override
 110     public NamespaceStorageNode getParentNamespaceStorage() {
 111         return parent;
 112     }
 113
 114     @Override
 115     public Registry getBehaviourRegistry() {
 116         return parent.getBehaviourRegistry();
 117     }
 118
 119     @Override
 120     public RootStatementContext<?, ?, ?> getRoot() {
 121         return parent.getRoot();
 122     }
 123
 124     @Override
 125     public A getStatementArgument() {
 126         return argument;
 127     }
 128
 129     @Override
 130     public StatementContextBase<?, ?, ?> createCopy(
 131             StatementContextBase<?, ?, ?> newParent, TypeOfCopy typeOfCopy)
 132             throws SourceException {
 133         return createCopy(null, newParent, typeOfCopy);
 134     }
 135
 136     @Override
 137     public StatementContextBase<A, D, E> createCopy(QNameModule newQNameModule,
 138             StatementContextBase<?, ?, ?> newParent, TypeOfCopy typeOfCopy)
 139             throws SourceException {
 140         SubstatementContext<A, D, E> copy = new SubstatementContext<>(this,
 141                 newQNameModule, newParent, typeOfCopy);
 142
 143         copy.addAllToCopyHistory(this.getCopyHistory());
 144         copy.addToCopyHistory(typeOfCopy);
 145
 146         if(this.getOriginalCtx() != null) {
 147             copy.setOriginalCtx(this.getOriginalCtx());
 148         } else {
 149             copy.setOriginalCtx(this);
 150         }
 151
 152         return copy;
 153     }
 154
 155     @Override
 156     public List<Object> getArgumentsFromRoot() {
 157         List<Object> argumentsFromRoot = parent.getArgumentsFromRoot();
 158         argumentsFromRoot.add(argument);
 159         return argumentsFromRoot;
 160     }
 161
 162     @Override
 163     public List<StmtContext<?, ?, ?>> getStmtContextsFromRoot() {
 164         List<StmtContext<?, ?, ?>> stmtContextsList = parent
 165                 .getStmtContextsFromRoot();
 166         stmtContextsList.add(this);
 167         return stmtContextsList;
 168     }
 169
 170     @Override
 171     public boolean isRootContext() {
 172         return false;
 173     }
 174 }