org.jpmml.model.visitors.Interner Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pmml-model Show documentation

JPMML class model

There is a newer version: 1.6.6

/*
 * Copyright (c) 2016 Villu Ruusmann
 */
package org.jpmml.model.visitors;

import java.lang.reflect.Field;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.List;
import java.util.ListIterator;
import java.util.Objects;

import org.dmg.pmml.PMMLObject;
import org.jpmml.model.ReflectionUtil;

abstract
public class Interner extends AbstractVisitor {

	private Class type = null;


	public Interner(Class type){
		setType(type);
	}

	abstract
	public V intern(V value);

	public Class getType(){
		return this.type;
	}

	private void setType(Class type){
		this.type = Objects.requireNonNull(type);
	}

	protected void apply(Field field, PMMLObject object){
		Class type = getType();

		Class fieldType = field.getType();

		if(Objects.equals(List.class, fieldType)){
			ParameterizedType listType = (ParameterizedType)field.getGenericType();

			Type[] typeArguments = listType.getActualTypeArguments();
			if(typeArguments.length != 1){
				throw new IllegalArgumentException();
			}

			Class listElementType = (Class)typeArguments[0];
			if(listElementType.isAssignableFrom(type)){
				List values = (List)ReflectionUtil.getFieldValue(field, object);

				if(values != null && !values.isEmpty()){

					for(ListIterator it = values.listIterator(); it.hasNext(); ){
						V value = it.next();

						if(type.isInstance(value)){
							V internedValue = intern(value);

							it.set(internedValue);
						}
					}
				}
			}

			return;
		} // End if

		if(fieldType.isAssignableFrom(type)){
			Object value = ReflectionUtil.getFieldValue(field, object);

			if(value != null){

				if(type.isInstance(value)){
					V internedValue = intern(type.cast(value));

					ReflectionUtil.setFieldValue(field, object, internedValue);
				}
			}
		}
	}
}