package cc.glsn.v15.neuralnet;

import java.util.LinkedList;

/**
 * Extends Brain to become an feedback neural network
 * rather than a simple feed forward.
 *
 * The specified numbers of feeback links are added as outputs
 * and those outputs are feed back in as inputs.
 *
 */
public class RecurrentBrain extends Brain
{
	private int feedbackLinks;
	public RecurrentBrain(NetFunction functg, LinkedList<Integer> layerSizes, int feedbackLinks)
	{
		super(functg,layerSizes);
		this.feedbackLinks=feedbackLinks;
		setupInputOutput();
	}
	public RecurrentBrain(NetFunction functg, LinkedList<Integer> layerSizes, String inputMappingStyle, int feedbackLinks)
	{
		super(functg,layerSizes,inputMappingStyle);
		this.feedbackLinks=feedbackLinks;
		setupInputOutput();
	}

	private void setupInputOutput()
	{
		for(int i=0; i<feedbackLinks; i++)
		{
			String name="feedback_" + i;
			addInput(name);
			setInput(name, 0.0);
			addOutput(name);
		}
	}

	public void resetFeeds(double val)
	{

		for(int i=0; i<feedbackLinks; i++)
		{
			String name="feedback_" + i;
			addInput(name);
			setInput(name, val);
		}	
	}

	/**
	 * Copies all output feedback values to the inputs
	 * Should be run before a new set of inputs are set.
	 *
	 * Order should be:
	 * updateFeeds()
	 * setInput()s
	 * getOutput()s
	 *
	 *
	 */
	public void updateFeeds()
	{
		double vals[]=new double[feedbackLinks];
		for(int i=0; i<feedbackLinks; i++)
		{
			String name="feedback_" + i;
			vals[i]=getOutput(name);
		}
		for(int i=0; i<feedbackLinks; i++)
		{
			String name="feedback_" + i;
			setInput(name,vals[i]);
		}
	}


}