Linear_alternateWells_rewardFirstPoke_withITI.sc

% PROGRAM NAME: 	LINEAR TRACK ALTERNATION
% AUTHOR: 			MCZ
% DESCRIPTION:
%						This program delivers reward from each reward well (on a linear track) when the beam is broken, and the animal successfully alternates from one well to the other. Reward delivery ceases after 1sec.

%CONSTANTS

int deliverPeriod= 500   % how long to deliver the reward
int ITI = 1000

%VARIABLES

int count= 0                % blink count
int lastWell= 0             % last well
int currWell= 0            % current well
int rewardWell= 0       % reward well
int nowRewarding = 0 % keep track of reward being dispensed
int ITIOver = 1
int ITIStartTime = 0
int timeDiff = 0
int currentTime;


%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
% FUNCTIONS

% -----------------------
% Function Name: 	Reward
% Description:		This function administers reward to a marked well.
% -----------------------

function 1
	nowRewarding = 1 % nowRewarding
		portout[rewardWell]=1 % reward
		do in deliverPeriod % NOTE: USING A VARIABLE HERE WORKS
			portout[rewardWell]=0 % reset reward
			nowRewarding=0 % no longer rewarding
		end
end;


% -----------------------
% Function Name: 	Reward first poke
% Description:		This function adminsters reward to the first poke.
% -----------------------
function 2
	if lastWell==0 do
		rewardWell=currWell
		trigger(1)
	end
end;

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
% CALLBACKS -- EVENT-DRIVEN TRIGGERS

callback portin[1] up
	currentTime = clock()
	timeDiff = currentTime - ITIStartTime
	if timeDiff> ITI do
		disp('portin1 up')
		currWell=1 % well currently active
		trigger(2)
		if lastWell == 2 do
			disp('Poke 1 rewarded')
			rewardWell=1 % dispense reward from here
			trigger(1)% trigger reward
		else do
			disp('Poke 2 not rewarded')
		end
		ITIOver = 1
	else do
		disp('ITI lockout not over')
	end
end;

callback portin[1] down
	if ITIOver == 1 do
		ITIStartTime = clock()
		ITIOver = 0
	end 
	disp('portin1 down')
	lastWell=1 % well left, now last well
end;

callback portin[2] up
	currentTime = clock()
	if timeDiff > ITI do
		disp('portin2 up')
		currWell=2 % well currently active
		trigger(2)
		if lastWell == 1 do
			disp('Poke 2 rewarded')
			rewardWell=2 % dispense reward from here
			trigger(1) % trigger reward
		else do
			disp('Poke 2 not rewarded')
		end
		ITIOver = 1
	else do
		disp('ITI lockout not over')
	end
end;

callback portin[2] down
	if ITIOver == 1 do
		ITIStartTime = clock()
		ITIOver = 0
	end 
	disp('portin2 down')
	lastWell=2 % well left, now last well
end;