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