code-generator/binding-data-codec/src/main/java/org/opendaylight/yangtools/binding/data/codec/impl/ChoiceNodeCodecContext.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.binding.data.codec.impl;
   9
  10 import com.google.common.base.Preconditions;
  11 import com.google.common.collect.ImmutableMap;
  12 import com.google.common.collect.Iterables;
  13
  14 import java.util.HashMap;
  15 import java.util.Map;
  16
  17 import org.opendaylight.yangtools.yang.binding.DataObject;
  18 import org.opendaylight.yangtools.yang.binding.util.BindingReflections;
  19 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  20 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  21 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  22 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNodeContainer;
  23 import org.opendaylight.yangtools.yang.model.api.ChoiceCaseNode;
  24 import org.opendaylight.yangtools.yang.model.api.ChoiceNode;
  25 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
  26
  27 final class ChoiceNodeCodecContext extends DataContainerCodecContext<ChoiceNode> {
  28
  29     private final ImmutableMap<YangInstanceIdentifier.PathArgument, DataContainerCodecPrototype<?>> byYangCaseChild;
  30     private final ImmutableMap<Class<?>, DataContainerCodecPrototype<?>> byClass;
  31     private final ImmutableMap<Class<?>, DataContainerCodecPrototype<?>> byCaseChildClass;
  32
  33     public ChoiceNodeCodecContext(final DataContainerCodecPrototype<ChoiceNode> prototype) {
  34         super(prototype);
  35         Map<YangInstanceIdentifier.PathArgument, DataContainerCodecPrototype<?>> byYangCaseChildBuilder = new HashMap<>();
  36         Map<Class<?>, DataContainerCodecPrototype<?>> byClassBuilder = new HashMap<>();
  37         Map<Class<?>, DataContainerCodecPrototype<?>> byCaseChildClassBuilder = new HashMap<>();
  38
  39         for (Class<?> caze : factory().getRuntimeContext().getCases(bindingClass())) {
  40             DataContainerCodecPrototype<ChoiceCaseNode> cazeDef = loadCase(caze);
  41             byClassBuilder.put(cazeDef.getBindingClass(), cazeDef);
  42             for (Class<? extends DataObject> cazeChild : BindingReflections.getChildrenClasses((Class) caze)) {
  43                 byCaseChildClassBuilder.put(cazeChild, cazeDef);
  44             }
  45             for (DataSchemaNode cazeChild : cazeDef.getSchema().getChildNodes()) {
  46                 byYangCaseChildBuilder.put(new NodeIdentifier(cazeChild.getQName()), cazeDef);
  47             }
  48         }
  49
  50         byYangCaseChild = ImmutableMap.copyOf(byYangCaseChildBuilder);
  51         byClass = ImmutableMap.copyOf(byClassBuilder);
  52         byCaseChildClass = ImmutableMap.copyOf(byCaseChildClassBuilder);
  53     }
  54
  55     @Override
  56     protected DataContainerCodecContext<?> getStreamChild(final Class<?> childClass) {
  57         return byClass.get(childClass).get();
  58     }
  59
  60     Iterable<Class<?>> getCaseChildrenClasses() {
  61         return byCaseChildClass.keySet();
  62     }
  63
  64     protected DataContainerCodecPrototype<ChoiceCaseNode> loadCase(final Class<?> childClass) {
  65         ChoiceCaseNode childSchema = factory().getRuntimeContext().getCaseSchemaDefinition(schema(), childClass);
  66         return DataContainerCodecPrototype.from(childClass, childSchema, factory());
  67     }
  68
  69     @Override
  70     protected NodeCodecContext getYangIdentifierChild(final YangInstanceIdentifier.PathArgument arg) {
  71         DataContainerCodecPrototype<?> cazeProto = byYangCaseChild.get(arg);
  72         Preconditions.checkArgument(cazeProto != null, "Argument %s is not valid child of %s", arg, schema());
  73         return cazeProto.get().getYangIdentifierChild(arg);
  74     }
  75
  76     @Override
  77     protected Object dataFromNormalizedNode(final NormalizedNode<?, ?> data) {
  78         Preconditions
  79                 .checkArgument(data instanceof org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode);
  80         NormalizedNodeContainer<?, ?, NormalizedNode<?,?>> casted = (NormalizedNodeContainer<?, ?, NormalizedNode<?,?>>) data;
  81         NormalizedNode<?, ?> first = Iterables.getFirst(casted.getValue(), null);
  82
  83         if (first == null) {
  84             return null;
  85         }
  86         DataContainerCodecPrototype<?> caze = byYangCaseChild.get(first.getIdentifier());
  87         return caze.get().dataFromNormalizedNode(data);
  88     }
  89
  90     public DataContainerCodecContext<?> getCazeByChildClass(final Class<? extends DataObject> type) {
  91         return byCaseChildClass.get(type).get();
  92     }
  93
  94 }