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
  68     private void copyDeclaredStmts(SubstatementContext<A, D, E> original,
  69             QNameModule newQNameModule, TypeOfCopy typeOfCopy)
  70             throws SourceException {
  71         Collection<? extends StatementContextBase<?, ?, ?>> originalDeclaredSubstatements = original
  72                 .declaredSubstatements();
  73         for (StatementContextBase<?, ?, ?> stmtContext : originalDeclaredSubstatements) {
  74             if (GroupingUtils.needToCopyByUses(stmtContext)) {
  75                 StatementContextBase<?, ?, ?> copy = stmtContext.createCopy(
  76                         newQNameModule, this, typeOfCopy);
  77                 this.addEffectiveSubstatement(copy);
  78             } else if (GroupingUtils.isReusedByUses(stmtContext)) {
  79                 this.addEffectiveSubstatement(stmtContext);
  80             }
  81         }
  82     }
  83
  84     private void copyEffectiveStmts(SubstatementContext<A, D, E> original,
  85             QNameModule newQNameModule, TypeOfCopy typeOfCopy)
  86             throws SourceException {
  87         Collection<? extends StatementContextBase<?, ?, ?>> originalEffectiveSubstatements = original
  88                 .effectiveSubstatements();
  89         for (StatementContextBase<?, ?, ?> stmtContext : originalEffectiveSubstatements) {
  90             if (GroupingUtils.needToCopyByUses(stmtContext)) {
  91                 StatementContextBase<?, ?, ?> copy = stmtContext.createCopy(
  92                         newQNameModule, this, typeOfCopy);
  93                 this.addEffectiveSubstatement(copy);
  94             } else if (GroupingUtils.isReusedByUses(stmtContext)) {
  95                 this.addEffectiveSubstatement(stmtContext);
  96             }
  97         }
  98     }
  99
 100     @Override
 101     public StatementContextBase<?, ?, ?> getParentContext() {
 102         return parent;
 103     }
 104
 105     @Override
 106     public NamespaceStorageNode getParentNamespaceStorage() {
 107         return parent;
 108     }
 109
 110     @Override
 111     public Registry getBehaviourRegistry() {
 112         return parent.getBehaviourRegistry();
 113     }
 114
 115     @Override
 116     public RootStatementContext<?, ?, ?> getRoot() {
 117         return parent.getRoot();
 118     }
 119
 120     @Override
 121     public A getStatementArgument() {
 122         return argument;
 123     }
 124
 125     @Override
 126     public StatementContextBase<?, ?, ?> createCopy(
 127             StatementContextBase<?, ?, ?> newParent, TypeOfCopy typeOfCopy)
 128             throws SourceException {
 129         return createCopy(null, newParent, typeOfCopy);
 130     }
 131
 132     @Override
 133     public StatementContextBase<A, D, E> createCopy(QNameModule newQNameModule,
 134             StatementContextBase<?, ?, ?> newParent, TypeOfCopy typeOfCopy)
 135             throws SourceException {
 136         SubstatementContext<A, D, E> copy = new SubstatementContext<>(this,
 137                 newQNameModule, newParent, typeOfCopy);
 138
 139         copy.addAllToCopyHistory(this.getCopyHistory());
 140         copy.addToCopyHistory(typeOfCopy);
 141
 142         if(this.getOriginalCtx() != null) {
 143             copy.setOriginalCtx(this.getOriginalCtx());
 144         } else {
 145             copy.setOriginalCtx(this);
 146         }
 147
 148         definition().onStatementAdded(copy);
 149
 150         copy.copyDeclaredStmts(this, newQNameModule, typeOfCopy);
 151         copy.copyEffectiveStmts(this, newQNameModule, typeOfCopy);
 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 }