#include <stdlib.h>
#include <iostream>
#include "Trials.h"

using namespace std;


// This module sets up the number of trials and number of conditions
// each condition is defined by an integer.
Trials::Trials(int nc, int tp)
{
	int i,T;

	nCond = nc;	
	trialsPerCond = tp;

	trialVec = new int[nCond];
	nTrials = nCond*trialsPerCond;

	T = 0;

	for(i=0;i<nCond;++i)
	{
			trialVec[i] = i;
	}
	// scramble the trial order
	scramble();
	tpCount = 0;
	condCount  = 0;	
	trialCount = 0;

}

void Trials::scramble()
{
	int i,j,tmp;
	for(i=0;i<nCond;++i)
	{
		j = rand()%nCond;
		tmp = trialVec[j];
		trialVec[j] = trialVec[i];
		trialVec[i] = tmp;
	}
	tpCount = 0;
	condCount  = 0;	
	trialCount = 0;
}

int Trials::getNextTrial()
{
	int tv;
	if(trialCount < nTrials)
	{
		 tv = trialVec[condCount];
		 ++tpCount;
		 if(tpCount == trialsPerCond)
		 {
			 tpCount = 0;
			 ++ condCount;
		 }
		 ++trialCount;

	}
	else tv = -1;

	//cerr << "GET NEXT " << tpCount << " " << condCount << " " << tv << "\n";

	return tv;
}