org.jpmml.translator.FieldInfo Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pmml-transpiler Show documentation
JPMML class model transpiler
There is a newer version: 1.3.8
/*
 * Copyright (c) 2019 Villu Ruusmann
 *
 * This file is part of JPMML-Transpiler
 *
 * JPMML-Transpiler 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-Transpiler 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-Transpiler.  If not, see .
 */
package org.jpmml.translator;

import java.util.Objects;

import org.dmg.pmml.Field;

public class FieldInfo {

	private Field field = null;

	private FieldInfo ref = null;

	private FunctionInvocation functionInvocation = null;

	private Integer count = null;

	private Encoder encoder = null;

	private String variableName = null;

	private String memberName = null;


	public FieldInfo(Field field){
		this(field, null);
	}

	public FieldInfo(Field field, Encoder encoder){
		setField(field);
		setEncoder(encoder);
	}

	public Field getField(){
		return this.field;
	}

	private void setField(Field field){
		this.field = Objects.requireNonNull(field);
	}

	public FieldInfo getRef(){
		return this.ref;
	}

	public void setRef(FieldInfo ref){
		this.ref = ref;
	}

	public FunctionInvocation getFunctionInvocation(){
		return this.functionInvocation;
	}

	public void setFunctionInvocation(FunctionInvocation functionInvocation){
		this.functionInvocation = functionInvocation;
	}

	public void updateCount(Integer count){

		if(count == null){
			return;
		} // End if

		if(this.count == null){
			this.count = count;
		} else

		{
			this.count += count;
		}
	}

	public Integer getCount(){
		return this.count;
	}

	public void setCount(Integer count){
		this.count = count;
	}

	public Encoder getEncoder(){
		return this.encoder;
	}

	public void setEncoder(Encoder encoder){
		this.encoder = encoder;
	}

	public String getVariableName(){

		if(this.variableName == null){
			this.variableName = createVariableName();
		}

		return this.variableName;
	}

	public void setVariableName(String varibaleName){
		this.variableName = varibaleName;
	}

	public String getMemberName(){

		if(this.memberName == null){
			this.memberName = createMemberName();
		}

		return this.memberName;
	}

	public void setMemberName(String memberName){
		this.memberName = memberName;
	}

	private String createVariableName(){
		Encoder encoder = getEncoder();

		if(encoder != null){
			return encoder.getVariableName(this);
		} else

		{
			Field field = getField();

			return IdentifierUtil.sanitize(field.requireName());
		}
	}

	private String createMemberName(){
		Encoder encoder = getEncoder();

		if(encoder != null){
			return encoder.getMemberName(this);
		} else

		{
			Field field = getField();

			return IdentifierUtil.sanitize(field.requireName());
		}
	}
}