binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/yangtools/binding/data/codec/gen/spi/AbstractSource.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.gen.spi;
   9
  10 import com.google.common.collect.Iterators;
  11 import com.google.common.collect.UnmodifiableIterator;
  12
  13 import java.util.Collections;
  14 import java.util.HashSet;
  15 import java.util.Set;
  16
  17 import org.opendaylight.yangtools.sal.binding.model.api.Type;
  18
  19 public abstract class AbstractSource {
  20
  21     private final Set<StaticConstantDefinition> staticConstants = new HashSet<>();
  22
  23     public final <T> void staticConstant(final String name, final Class<T> type, final T value) {
  24         staticConstants.add(new StaticConstantDefinition(name, type, value));
  25     }
  26
  27     public final Set<StaticConstantDefinition> getStaticConstants() {
  28         return Collections.unmodifiableSet(staticConstants);
  29     }
  30
  31     protected final CharSequence invoke(final CharSequence object, final String methodName, final Object... args) {
  32         StringBuilder builder = new StringBuilder();
  33         if (object != null) {
  34             builder.append(object);
  35             builder.append('.');
  36         }
  37         builder.append(methodName);
  38         builder.append('(');
  39
  40         UnmodifiableIterator<Object> iterator = Iterators.forArray(args);
  41         while (iterator.hasNext()) {
  42             builder.append(iterator.next());
  43             if (iterator.hasNext()) {
  44                 builder.append(',');
  45             }
  46         }
  47         builder.append(')');
  48         return builder;
  49     }
  50
  51     protected final CharSequence assign(final String var, final CharSequence value) {
  52         return assign((String) null, var, value);
  53     }
  54
  55     protected final CharSequence assign(final String type, final String var, final CharSequence value) {
  56         StringBuilder builder = new StringBuilder();
  57         if(type != null) {
  58             builder.append(type);
  59             builder.append(' ');
  60         }
  61         builder.append(var);
  62         builder.append(" = ");
  63         builder.append(value);
  64         return builder;
  65     }
  66
  67     protected final CharSequence assign(final Type type, final String var, final CharSequence value) {
  68         return assign(type.getFullyQualifiedName(), var, value);
  69     }
  70
  71     protected final CharSequence cast(final Type type, final CharSequence value) {
  72         return cast(type.getFullyQualifiedName(), value);
  73     }
  74
  75     protected final CharSequence forEach(final String iterable,final String iteratorName, final String valueType,final String valueName, final CharSequence body) {
  76         StringBuilder b = new StringBuilder();
  77         b.append(statement(assign(java.util.Iterator.class.getName(), iteratorName,invoke(iterable, "iterator"))));
  78         b.append("while (").append(invoke(iteratorName, "hasNext")).append(") {\n");
  79         b.append(statement(assign(valueType, valueName,cast(valueType, invoke(iteratorName, "next")))));
  80         b.append(body);
  81         b.append("\n}\n");
  82         return b;
  83     }
  84
  85     protected final CharSequence statement(final CharSequence statement) {
  86         return new StringBuilder().append(statement).append(";\n");
  87     }
  88
  89     protected final CharSequence cast(final String type, final CharSequence value) {
  90         StringBuilder builder = new StringBuilder();
  91         builder.append("((");
  92         builder.append(type);
  93         builder.append(") ");
  94         builder.append(value);
  95         builder.append(')');
  96         return builder;
  97     }
  98
  99 }