/*******************************************************************************
 * DIANNE  - Framework for distributed artificial neural networks
 * Copyright (C) 2015  iMinds - IBCN - UGent
 *
 * This file is part of DIANNE.
 *
 * DIANNE is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * 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 Affero 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, see <http://www.gnu.org/licenses/>.
 *
 * Contributors:
 *     Tim Verbelen, Steven Bohez
 *******************************************************************************/
package be.iminds.iot.dianne.api.rl.dataset;

import be.iminds.iot.dianne.api.dataset.Sample;
import be.iminds.iot.dianne.tensor.Tensor;

/**
 * A helper class for representing one sample of an experience pool
 * 
 * @author tverbele
 *
 */
public class ExperiencePoolSample extends Sample {

	// store reward as a 1d tensor - tensor type required to support ExperiencePoolBatch
	public Tensor reward;
	
	public Tensor nextState;
	
	// terminal tensor is 1 in case of non terminal, 0 in case of terminal
	public Tensor terminal;
	
	public ExperiencePoolSample(){}
	
	public ExperiencePoolSample(Tensor state, Tensor action, float reward, Tensor nextState){
		super(state, action);
		
		this.reward = new Tensor(1);
		this.reward.set(reward, 0);
		
		this.nextState = nextState;
		
		this.terminal = new Tensor(1);
		this.terminal.set(nextState == null ? 0.0f : 1.0f, 0);
	}
	
	public ExperiencePoolSample(Tensor state, Tensor action, Tensor reward, Tensor nextState, Tensor terminal){
		super(state, action);
		
		this.reward = reward;
		this.nextState = nextState;
		this.terminal = terminal;
	}
	
	public Tensor getState(){
		return input;
	}
	
	public Tensor getAction(){
		return target;
	}
	
	public float getScalarReward(){
		return reward.get(0);
	}
	
	public Tensor getReward(){
		return reward;
	}
	
	public Tensor getNextState(){
		return nextState;
	}
	
	public Tensor getTerminal(){
		return terminal;
	}
	
	public boolean isTerminal(){
		return terminal.get(0) == 0.0f;
	}
	
	@Override
	public String toString(){
		StringBuilder b = new StringBuilder();
		b.append("State: ")
		.append(input)
		.append(" - Action: ")
		.append(target)
		.append(" - Reward: ")
		.append(reward)
		.append(" - Next state: ")
		.append(nextState)
		.append(" - Terminal: ").append(isTerminal());
		return b.toString();
	}
}

@@ -2704,22 +2704,33 @@
 .append(
-r
+getScalarR
 eward
+()
 )%0A%09%09.app

print scalar reward in ExperiencePoolSample