org.jpmml.model.visitors.TreePathFinder 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.7.2

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

import java.util.ArrayList;
import java.util.Collections;
import java.util.Deque;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.dmg.pmml.Node;
import org.dmg.pmml.PMMLObject;
import org.dmg.pmml.Visitable;
import org.dmg.pmml.VisitorAction;

/**
 * 
 * A Visitor that determines paths from the root Node element of the tree model to all its leaf Node elements.
 * 
 */
public class TreePathFinder extends AbstractVisitor {

	private Map> paths = new HashMap<>();


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

		super.applyTo(visitable);
	}

	@Override
	public VisitorAction visit(Node node){

		if(!node.hasNodes()){
			process(node);
		}

		return super.visit(node);
	}

	private void process(Node node){
		List path = new ArrayList<>();
		path.add(node);

		Deque parents = getParents();
		for(PMMLObject parent : parents){

			if(!(parent instanceof Node)){
				break;
			}

			path.add((Node)parent);
		}

		Collections.reverse(path);

		this.paths.put(node, path);
	}

	/**
	 * @return A map of all paths.
	 * Map keys are leaf Node elements.
	 * Map values are paths leading from the root Node element to the specified leaf Node element (inclusive).
	 */
	public Map> getPaths(){
		return this.paths;
	}
}