CGII/framework/include/cgv/base/find_action.h

#pragma once

#include <cgv/base/action.h>

namespace cgv {
	namespace base {

/// simple action implementation that adds nodes implementing X to a results vector
template <class X>
class find_action : public action
{
	X* x;
protected:
	std::vector<X*>& result;
public:
	find_action(std::vector<X*>& _result) : action(true,true), result(_result) {}
	/// make the passed object current
	void select(base_ptr p) {
		x = p->get_interface<X>();
	}
	bool begin()
	{
		if (x) {
			result.push_back(x);
			return true;
		}
		else
			return false;
	}
	/// check if the current object implements the interface needed for this action
	bool implements_action() const {
		return x != 0;
	}
};

/// collect all nodes that implement interface X
template <class X>
void find_interface(base_ptr start, std::vector<X*>& result) {
	find_action<X> fa(result);
	traverser(fa).traverse(start);
}

	}
}