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