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