org.jpmml.evaluator.ArrayUtil Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of pmml-evaluator Show documentation
JPMML class model evaluator
There is a newer version: 1.6.5
/*
 * Copyright (c) 2018 Villu Ruusmann
 *
 * This file is part of JPMML-Evaluator
 *
 * JPMML-Evaluator is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * JPMML-Evaluator is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with JPMML-Evaluator.  If not, see .
 */
package org.jpmml.evaluator;

import java.util.List;

import com.google.common.base.Function;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.LoadingCache;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Lists;
import org.dmg.pmml.Array;

public class ArrayUtil {

	private ArrayUtil(){
	}

	static
	public int getSize(Array array){
		Integer n = array.getN();
		if(n != null){
			return n;
		}

		List content = getContent(array);

		return content.size();
	}

	static
	public List getContent(Array array){
		return CacheUtil.getValue(array, ArrayUtil.contentCache);
	}

	static
	public List asNumberList(Array array){
		List content = getContent(array);

		Array.Type type = array.getType();
		if(type == null){
			throw new MissingAttributeException(array, PMMLAttributes.ARRAY_TYPE);
		}

		switch(type){
			case INT:
			case REAL:
				return (List)content;
			case STRING:
				throw new InvalidElementException(array);
			default:
				throw new UnsupportedAttributeException(array, type);
		}
	}

	static
	public List parse(Array array){
		Array.Type type = array.getType();
		if(type == null){
			throw new MissingAttributeException(array, PMMLAttributes.ARRAY_TYPE);
		}

		String value = array.getValue();

		List tokens;

		switch(type){
			case INT:
			case REAL:
				tokens = org.jpmml.model.ArrayUtil.parse(value, false);
				break;
			case STRING:
				tokens = org.jpmml.model.ArrayUtil.parse(value, true);
				break;
			default:
				throw new UnsupportedAttributeException(array, type);
		}

		Integer n = array.getN();
		if(n != null && n != tokens.size()){
			throw new InvalidElementException(array);
		}

		switch(type){
			case INT:
				return Lists.transform(tokens, INT_PARSER);
			case REAL:
				return Lists.transform(tokens, REAL_PARSER);
			case STRING:
				return Lists.transform(tokens, STRING_PARSER);
			default:
				throw new UnsupportedAttributeException(array, type);
		}
	}

	private static final Function INT_PARSER = new Function(){

		@Override
		public Integer apply(String string){
			return Numbers.INTEGER_INTERNER.intern(Integer.parseInt(string));
		}
	};

	private static final Function REAL_PARSER = new Function(){

		@Override
		public Double apply(String string){
			return Numbers.DOUBLE_INTERNER.intern(Double.parseDouble(string));
		}
	};

	private static final Function STRING_PARSER = new Function(){

		@Override
		public String apply(String string){
			return Strings.INTERNER.intern(string);
		}
	};

	private static final LoadingCache> contentCache = CacheUtil.buildLoadingCache(new CacheLoader>(){

		@Override
		public List load(Array array){
			return ImmutableList.copyOf(parse(array));
		}
	});
}