org.jpmml.model.visitors.VersionInspector Maven / Gradle / Ivy
/*
* Copyright (c) 2014 Villu Ruusmann
*/
package org.jpmml.model.visitors;
import java.lang.reflect.AnnotatedElement;
import java.lang.reflect.Field;
import java.util.Collection;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import org.dmg.pmml.Apply;
import org.dmg.pmml.PMMLFunctions;
import org.dmg.pmml.PMMLObject;
import org.dmg.pmml.Version;
import org.dmg.pmml.VisitorAction;
import org.jpmml.model.ReflectionUtil;
import org.jpmml.model.annotations.Added;
import org.jpmml.model.annotations.Optional;
import org.jpmml.model.annotations.Removed;
import org.jpmml.model.annotations.Required;
/**
*
* A Visitor that determines the range of valid PMML schema versions for a class model object.
*
*
* @see Added
* @see Optional
* @see Removed
* @see Required
*/
public class VersionInspector extends AbstractVisitor implements Resettable {
private Version minimum = Version.getMinimum();
private Version maximum = Version.getMaximum();
@Override
public void reset(){
this.minimum = Version.getMinimum();
this.maximum = Version.getMaximum();
}
@Override
public VisitorAction visit(PMMLObject object){
for(Class clazz = object.getClass(); clazz != null; clazz = clazz.getSuperclass()){
inspect(clazz);
}
List fields = ReflectionUtil.getFields(object.getClass());
for(Field field : fields){
Object value = ReflectionUtil.getFieldValue(field, object);
inspect(field, value);
// The field is set to an enum constant
if(value instanceof Enum){
Enum enumValue = (Enum)value;
Field enumField;
try {
Class enumClazz = enumValue.getClass();
enumField = enumClazz.getField(enumValue.name());
} catch(NoSuchFieldException nsfe){
throw new RuntimeException(nsfe);
}
inspect(enumField);
}
}
return super.visit(object);
}
@Override
public VisitorAction visit(Apply apply){
String function = apply.getFunction();
Version version = VersionInspector.functionVersions.get(function);
if(version != null){
updateMinimum(version);
}
return super.visit(apply);
}
private void inspect(Field field, Object value){
Class type = field.getType();
if(type.isPrimitive()){
if(ReflectionUtil.isDefaultValue(value)){
return;
}
} else
{
if(isNull(value)){
Optional optional = field.getAnnotation(Optional.class);
if(optional != null){
updateMinimum(optional.value());
}
Required required = field.getAnnotation(Required.class);
if(required != null){
updateMaximum(previous(required.value()));
}
return;
}
}
inspect(field);
}
private void inspect(AnnotatedElement element){
Added added = element.getAnnotation(Added.class);
if(added != null){
updateMinimum(added.value());
}
Removed removed = element.getAnnotation(Removed.class);
if(removed != null){
updateMaximum(removed.value());
}
}
/**
*
* The minimum (ie. earliest) PMML schema version that can fully represent this class model object.
*
*
* @see Version#getMinimum()
*/
public Version getMinimum(){
return this.minimum;
}
private void updateMinimum(Version minimum){
if(minimum != null && minimum.compareTo(this.minimum) > 0){
this.minimum = minimum;
}
}
/**
*
* The maximum (ie. latest) PMML schema version that can fully represent this class model object.
*
*
* @see Version#getMaximum()
*/
public Version getMaximum(){
return this.maximum;
}
private void updateMaximum(Version maximum){
if(maximum != null && maximum.compareTo(this.maximum) < 0){
this.maximum = maximum;
}
}
static
private boolean isNull(Object value){
if(value instanceof Collection){
Collection collection = (Collection)value;
return collection.isEmpty();
}
return (value == null);
}
static
private Version previous(Version version){
Version[] values = Version.values();
return values[version.ordinal() - 1];
}
private static Map functionVersions = new LinkedHashMap<>();
static {
declareFunctions(Version.PMML_3_0,
PMMLFunctions.ADD, PMMLFunctions.SUBTRACT, PMMLFunctions.MULTIPLY, PMMLFunctions.DIVIDE,
PMMLFunctions.MIN, PMMLFunctions.MAX, PMMLFunctions.SUM, PMMLFunctions.AVG,
PMMLFunctions.LOG10, PMMLFunctions.LN, PMMLFunctions.SQRT, PMMLFunctions.ABS,
PMMLFunctions.UPPERCASE, PMMLFunctions.SUBSTRING, PMMLFunctions.TRIMBLANKS,
PMMLFunctions.FORMATNUMBER, PMMLFunctions.FORMATDATETIME,
PMMLFunctions.DATEDAYSSINCEYEAR, PMMLFunctions.DATESECONDSSINCEYEAR, PMMLFunctions.DATESECONDSSINCEMIDNIGHT);
declareFunctions(Version.PMML_3_1,
PMMLFunctions.EXP, PMMLFunctions.POW, PMMLFunctions.THRESHOLD, PMMLFunctions.FLOOR, PMMLFunctions.CEIL, PMMLFunctions.ROUND);
declareFunctions(Version.PMML_4_0,
PMMLFunctions.ISMISSING, PMMLFunctions.ISNOTMISSING,
PMMLFunctions.EQUAL, PMMLFunctions.NOTEQUAL, PMMLFunctions.LESSTHAN, PMMLFunctions.LESSOREQUAL, PMMLFunctions.GREATERTHAN, PMMLFunctions.GREATEROREQUAL,
PMMLFunctions.AND, PMMLFunctions.OR,
PMMLFunctions.NOT,
PMMLFunctions.ISIN, PMMLFunctions.ISNOTIN,
PMMLFunctions.IF);
declareFunctions(Version.PMML_4_1,
PMMLFunctions.MEDIAN, PMMLFunctions.PRODUCT,
PMMLFunctions.LOWERCASE);
declareFunctions(Version.PMML_4_2,
PMMLFunctions.CONCAT, PMMLFunctions.REPLACE, PMMLFunctions.MATCHES);
declareFunctions(Version.PMML_4_3,
PMMLFunctions.NORMALCDF, PMMLFunctions.NORMALPDF, PMMLFunctions.STDNORMALCDF, PMMLFunctions.STDNORMALPDF, PMMLFunctions.ERF, PMMLFunctions.NORMALIDF, PMMLFunctions.STDNORMALIDF);
declareFunctions(Version.PMML_4_4,
PMMLFunctions.MODULO,
PMMLFunctions.ISVALID, PMMLFunctions.ISNOTVALID,
PMMLFunctions.EXPM1, PMMLFunctions.HYPOT, PMMLFunctions.LN1P, PMMLFunctions.RINT,
PMMLFunctions.SIN, PMMLFunctions.ASIN, PMMLFunctions.SINH, PMMLFunctions.COS, PMMLFunctions.ACOS, PMMLFunctions.COSH, PMMLFunctions.TAN, PMMLFunctions.ATAN, PMMLFunctions.TANH);
declareFunctions(Version.XPMML,
PMMLFunctions.ATAN2);
}
static
private void declareFunctions(Version version, String... functions){
for(String function : functions){
VersionInspector.functionVersions.put(function, version);
}
}
}