opencog-generate / opencog / generate / RandomCallback.h

/*
 * opencog/generate/RandomCallback.h
 *
 * Copyright (C) 2020 Linas Vepstas <linasvepstas@gmail.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License v3 as
 * published by the Free Software Foundation and including the exceptions
 * at http://opencog.org/wiki/Licenses
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program; if not, write to:
 * Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#ifndef _OPENCOG_RANDOM_CALLBACK_H
#define _OPENCOG_RANDOM_CALLBACK_H

#include <opencog/generate/CollectStyle.h>
#include <opencog/generate/Dictionary.h>
#include <opencog/generate/GenerateCallback.h>
#include <opencog/generate/LinkStyle.h>
#include <opencog/generate/RandomParameters.h>

namespace opencog
{
/** \addtogroup grp_generate
 *  @{
 */

/// Callback to provide sections (aka "puzzle pieces") to extend the
/// current assembly. Under construction. Meant to draw pieces using
/// a stochastic, random selection process.
///

class RandomCallback :
	public GenerateCallback,
	private LinkStyle,
	private CollectStyle
{
private:
	Dictionary _dict;
	RandomParameters* _parms;
	Handle _weight_key;
	size_t _steps_taken;

	// -------------------------------------------
	// Nucleation points.
	HandleSeqSeq _root_sections;
	std::vector<std::discrete_distribution<size_t>> _root_dist;

	// -------------------------------------------
	// Lexical selection
	Handle select_from_lexis(const OdoFrame&,
	                         const Handle&, size_t,
	                         const Handle&);

	// Cached normalized weighted chooser. Map of all
	// sections in the dictionary that contain this to-connector.
	// Used by `select()` to return the next attachable section.
	std::map<Handle, std::discrete_distribution<size_t>> _distmap;

	// -------------------------------------------
	Handle select_from_open(const OdoFrame&,
	                        const Handle&, size_t,
	                        const Handle&);
	struct OpenSelections
	{
		HandleSeqMap _opensect;

		// Chooser, selecting a one of the open sections the current frame.
		std::map<Handle, std::discrete_distribution<size_t>> _opendi;
	};

	OpenSelections _opensel;
	std::stack<OpenSelections> _opensel_stack;
	// -------------------------------------------

public:
	RandomCallback(AtomSpace*, const Dictionary&, RandomParameters&);
	virtual ~RandomCallback();

	virtual void clear(AtomSpace*);
	void set_weight_key(const Handle& pred) { _weight_key = pred; }

	virtual void root_set(const HandleSet&);
	virtual HandleSet next_root(void);

	virtual HandleSeq joints(const Handle& con) {
		return _dict.joints(con);
	}
	virtual Handle select(const OdoFrame&,
	                      const Handle&, size_t,
	                      const Handle&);

	virtual Handle make_link(const Handle&, const Handle&,
	                         const Handle&, const Handle&);
	virtual size_t num_links(const Handle&, const Handle&,
	                         const Handle&);
	virtual void push_frame(const OdoFrame&);
	virtual void pop_frame(const OdoFrame&);

	virtual bool step(const OdoFrame&);
	virtual void solution(const OdoFrame&);
	virtual Handle get_solutions(void);
};


/** @}*/
}  // namespace opencog

#endif // _OPENCOG_RANDOM_CALLBACK_H