org.jpmml.model.visitors.ModelCleaner 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.11

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

import java.util.HashSet;
import java.util.Set;

import org.dmg.pmml.Field;
import org.dmg.pmml.Model;
import org.dmg.pmml.PMMLObject;
import org.dmg.pmml.Visitable;
import org.dmg.pmml.mining.MiningModel;

abstract
class ModelCleaner extends DeepFieldResolver {

	private Set activeFields = new HashSet<>();


	@Override
	public void applyTo(Visitable visitable){
		this.activeFields.clear();

		super.applyTo(visitable);
	}

	@Override
	public PMMLObject popParent(){
		PMMLObject parent = super.popParent();

		if(parent instanceof MiningModel){
			MiningModel miningModel = (MiningModel)parent;

			processMiningModel(miningModel);
		} else

		if(parent instanceof Model){
			Model model = (Model)parent;

			processModel(model);
		}

		return parent;
	}

	private void processMiningModel(MiningModel miningModel){
		Set activeFields = getActiveFields();

		activeFields.addAll(DeepFieldResolverUtil.getActiveFields(this, miningModel));
	}

	private void processModel(Model model){
		Set activeFields = getActiveFields();

		activeFields.addAll(DeepFieldResolverUtil.getActiveFields(this, model));
	}

	public Set getActiveFields(){
		return this.activeFields;
	}
}