binding/mdsal-binding-java-api-generator/src/main/java/org/opendaylight/mdsal/binding/java/api/generator/AbstractBigRangeGenerator.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.mdsal.binding.java.api.generator;
   9
  10 import com.google.common.collect.Range;
  11 import java.lang.reflect.Array;
  12 import java.util.Set;
  13 import java.util.function.Function;
  14 import javax.annotation.Nonnull;
  15 import org.opendaylight.yangtools.yang.binding.CodeHelpers;
  16 import org.opendaylight.yangtools.yang.model.api.type.RangeConstraint;
  17
  18 /**
  19  * Abstract base for generators which require instantiation of boundary values
  20  * to check. These are implemented by generating an array constant within the
  21  * class, which contains {@link Range} instances, which hold pre-instantiated
  22  * boundary values.
  23  *
  24  * @param <T> type of the class
  25  */
  26 abstract class AbstractBigRangeGenerator<T extends Number & Comparable<T>> extends AbstractRangeGenerator<T> {
  27     protected AbstractBigRangeGenerator(final Class<T> typeClass) {
  28         super(typeClass);
  29     }
  30
  31     private static String range(final Function<Class<?>, String> classImporter) {
  32         return classImporter.apply(Range.class);
  33     }
  34
  35     private String itemType(final Function<Class<?>, String> classImporter) {
  36         return range(classImporter) + '<' + getTypeName() + '>';
  37     }
  38
  39     private String arrayType(final Function<Class<?>, String> classImporter) {
  40         return itemType(classImporter) + "[]";
  41     }
  42
  43     @Override
  44     protected final String generateRangeCheckerImplementation(final String checkerName,
  45             @Nonnull final RangeConstraint<?> constraint, final Function<Class<?>, String> classImporter) {
  46         final Set<? extends Range<? extends Number>> constraints = constraint.getAllowedRanges().asRanges();
  47         final String fieldName = checkerName.toUpperCase() + "_RANGES";
  48         final StringBuilder sb = new StringBuilder();
  49
  50         // Field to hold the Range objects in an array
  51         sb.append("private static final ").append(arrayType(classImporter)).append(' ').append(fieldName).append(";\n");
  52
  53         // Static initializer block for the array
  54         sb.append("static {\n");
  55         sb.append("    @SuppressWarnings(\"unchecked\")\n");
  56         sb.append("    final ").append(arrayType(classImporter)).append(" a = (").append(arrayType(classImporter)).append(") ")
  57         .append(classImporter.apply(Array.class)).append(".newInstance(").append(range(classImporter)).append(".class, ")
  58         .append(constraints.size()).append(");\n");
  59
  60         int i = 0;
  61         for (Range<? extends Number> r : constraints) {
  62             final String min = format(getValue(r.lowerEndpoint()));
  63             final String max = format(getValue(r.upperEndpoint()));
  64
  65             sb.append("    a[").append(i++).append("] = ").append(range(classImporter)).append(".closed(").append(min).append(", ")
  66             .append(max).append(");\n");
  67         }
  68
  69         sb.append("    ").append(fieldName).append(" = a;\n");
  70         sb.append("}\n");
  71
  72         // Static enforcement method
  73         sb.append("private static void ").append(checkerName).append("(final ").append(getTypeName()).append(" value) {\n");
  74         sb.append("    for (").append(itemType(classImporter)).append(" r : ").append(fieldName).append(") {\n");
  75         sb.append("        if (r.contains(value)) {\n");
  76         sb.append("            return;\n");
  77         sb.append("        }\n");
  78         sb.append("    }\n");
  79
  80         sb.append("    ").append(classImporter.apply(CodeHelpers.class)).append(".throwInvalidRange(").append(fieldName).append(", value);\n");
  81         sb.append("}\n");
  82
  83         return sb.toString();
  84     }
  85 }