/*
 * SymbolicFactorGraph.h
 *
 *  Created on: Oct 29, 2009
 *      Author: Frank Dellaert
 */

#pragma once

#include <string>
#include <list>
#include <gtsam/inference/FactorGraph.h>
#include <gtsam/inference/SymbolicFactor.h>
#include <gtsam/inference/SymbolicBayesNet.h>
#include <gtsam/inference/Key.h>

namespace gtsam {

	class SymbolicConditional;

	/** Symbolic Factor Graph */
	class SymbolicFactorGraph: public FactorGraph<SymbolicFactor> {
	public:

		/** Construct empty factor graph */
		SymbolicFactorGraph() {}

		/** Push back unary factor */
		void push_factor(const Symbol& key) {
			boost::shared_ptr<SymbolicFactor> factor(new SymbolicFactor(key));
			push_back(factor);
		}

		/** Push back binary factor */
		void push_factor(const Symbol& key1, const Symbol& key2) {
			boost::shared_ptr<SymbolicFactor> factor(new SymbolicFactor(key1,key2));
			push_back(factor);
		}

		/** Push back ternary factor */
		void push_factor(const Symbol& key1, const Symbol& key2, const Symbol& key3) {
			boost::shared_ptr<SymbolicFactor> factor(new SymbolicFactor(key1,key2,key3));
			push_back(factor);
		}

		/** Push back 4-way factor */
		void push_factor(const Symbol& key1, const Symbol& key2, const Symbol& key3, const Symbol& key4) {
			boost::shared_ptr<SymbolicFactor> factor(new SymbolicFactor(key1,key2,key3,key4));
			push_back(factor);
		}

		/**
		 * Construct from a factor graph of any type
		 */
		template<class Factor>
		SymbolicFactorGraph(const FactorGraph<Factor>& fg) {
			for (size_t i = 0; i < fg.size(); i++) {
				boost::shared_ptr<Factor> f = fg[i];
				std::list<Symbol> keys = f->keys();
				SymbolicFactor::shared_ptr factor(new SymbolicFactor(keys));
				push_back(factor);
			}
		}

  	/**
     * Eliminate a single node yielding a conditional Gaussian
     * Eliminates the factors from the factor graph through findAndRemoveFactors
     * and adds a new factor on the separator to the factor graph
     */
    boost::shared_ptr<SymbolicConditional> eliminateOne(const Symbol& key);

		/**
		 * eliminate factor graph in place(!) in the given order, yielding
		 * a chordal Bayes net
		 */
		SymbolicBayesNet eliminate(const Ordering& ordering);

		/**
		 * Same as eliminate in the SymbolicFactorGraph case
		 */
		SymbolicBayesNet eliminateFrontals(const Ordering& ordering);
	};

} // namespace gtsam