optimize.power.ipf

#pragma rtGlobals=3		// Use modern global access method and strict wave access.


//*******************************************************************************************************************************

Function PopupChoice ()
	Variable Choose = 0
	Prompt Choose, "Is this fit good?", popup, "Yes: Save fit; No response: Save zero"
	DoPrompt "Goodness of Fit", Choose
  if (V_Flag)
    killvariables v_flag
    abort("User Canceled")								// User canceled
  endif
	return choose
End

function do_variable_prompt(string_list)
string string_list
variable /g target_response
variable /g max_power_0, min_power_0
  variable  return_var, temp_var, tmp_max_power, tmp_min_power
  wave power_0
  temp_var = target_response
  tmp_max_power = power_0[1]
  tmp_min_power = power_0[0]
  prompt tmp_max_power, "Max power"
  prompt tmp_min_power, "Min power"
  prompt temp_var, "Target response amplitude (pA)"
  Prompt return_var, "Protocol directory", popup, string_list
  doprompt "Set variables",return_var, temp_var, tmp_max_power, tmp_min_power
  target_response = temp_var
  max_power_0 = tmp_max_power
  min_power_0 = tmp_min_power
	return(return_var)
end

Function continue_prompt()
	Variable out = 0
	Prompt out, "Continue?", popup, "Continue; Write out last power; write out next power"
	DoPrompt "Continue", out
  if (V_Flag)
    killvariables v_flag
		out = 2								// User canceled
	endif
	return out
End

function do_fit(traceunc, xmin, xmax, w_coef)
	wave w_coef
	string traceunc
	variable xmin, xmax
	variable v_abortcode = 0
	try
		FuncFit/N/Q/H="00000" /NTHR=0 DiffTwoExp2 W_coef  $traceunc(xmin,xmax) /D
	catch
		W_coef = NaN
		Duplicate/O/R=(xmin,xmax) $traceunc, $("fit_"+traceunc)
	endtry

end

macro Optimize_Power()
	string/g response_data_file_name, response_number_string, new_response_number_string, new_response_data_file_name
	variable response_number, new_response_number
	String traceunc = "ach_1"
 	string  protocol_dir = s_path
	String tracepower
	tracepower = "ach_3"
	Variable i=0, V_FitMaxIters = 100, v_levelx = 0,  fit_action, fitrange=0.04, this_response
	variable Td = 0.004,  Tr = 0.002,
	variable /g max_power_0, min_power_0
  variable /g target_response = 12
	Variable uncgpnt, V_FitError, last_power
	variable /g max_power, min_power, next_power
	Make/O/D W_coef = NaN					//For holding DiffTwoExp coefficients
  string prm_file_name
  variable temp_var, peak_time
  string path_list =   ("C:Documents and Settings:shane:My Documents:;"+s_path)

  variable converge_indicator
// min_power_0 = power_0[0]
// max_power_0 = power_0[1]
	temp_var = do_variable_prompt(path_list)
	protocol_dir = stringfromlist((temp_var-1), path_list)
  make /n= 0 response_wave
  make /n= 0 power_wave

	//---------------------------------------------First check / create all waves

	if ( !WaveExists(root:uncage_power_wave))
		response_data_file_name = s_path+s_filename
		response_data_file_name = response_data_file_name[0,(strsearch(response_data_file_name,".itx",0)-1)]
		response_number_string = response_data_file_name[(strsearch(response_data_file_name,"Line",0)+4),Inf]
		response_number = str2num(response_number_string)
    prm_file_name = response_data_file_name+".prm"
    last_power = read_power_from_prm(prm_file_name)
    read_write_prm(last_power,response_data_file_name+".prm",protocol_dir+"sm.updated.protocol.prm")
	endif

do //do1
  if ( WaveExists(root:uncage_power_wave))
		new_response_number = response_number + 1
		new_response_number_string = PadString(new_response_number_string, (5-floor(log(response_number))), char2num("0") ) + num2str(new_response_number)
		new_response_data_file_name = response_data_file_name[0,(strsearch(response_data_file_name,"Line",0)+3)] + new_response_number_string +".itx"
		LoadWave/T new_response_data_file_name
		new_response_number_string =""
		response_data_file_name = new_response_data_file_name
		response_data_file_name = response_data_file_name[0,(strsearch(response_data_file_name,".itx",0)-1)]
    prm_file_name = response_data_file_name+".prm"
		response_number_string = response_data_file_name[(strsearch(response_data_file_name,"Line",0)+4),Inf]
		response_number = str2num(response_number_string)
	endif

	// i = 0;
  v_levelx = 0
	findlevel /Q/EDGE=1 /R= (v_levelx,) $tracepower, (0.8*wavemax($tracepower))
	uncgpnt = v_levelx
	findlevel /Q/EDGE=2 /R= (v_levelx,) $tracepower, (0.5*wavemax($tracepower))
  K0 =  mean(ach_1,0,uncgpnt)
  Loess/pass=1 /N=(2^7) /DEST=temp srcWave=$traceunc
  // duplicate /O $traceunc temp
  temp = (abs(temp - K0))
  wavestats /q temp
  peak_time = v_maxloc
  K1 = $traceunc(v_maxloc)-k0
    K2 = td

  wavestats /q /R = (0,uncgpnt) $traceunc

w_coef = {k0,k1,k2}

	W_coef = {(W_coef[1]), uncgpnt, W_coef[2], ((peak_time-uncgpnt)*0.33), W_coef[0]}

	Display/N=Checking $traceunc;
	SetAxis bottom (uncgpnt-0.01),(uncgpnt+(2*fitrange))
	do_fit(traceunc, (uncgpnt-fitrange/3), (uncgpnt+fitrange), w_coef)
	AppendToGraph /c=(0,0,0) $("fit_"+traceunc)
  SetDrawEnv xcoord= bottom;SetDrawEnv dash= 3;DelayUpdate
  DrawLine uncgpnt,0,uncgpnt,1
	fit_action = PopupChoice ()
	if (fit_action == 2)		//Nothing there: save as zero for no response
		w_coef = 0
	endif
	DoWindow/K Checking
	if ( !WaveExists(root:uncage_power_wave))
		Make/O/N=0 uncage_power_wave
		max_power = max_power_0
		min_power = min_power_0
	endif
  last_power = read_power_from_prm(prm_file_name)
	this_response = abs(w_coef[0])
  insertpoints numpnts(response_wave),1,response_wave
  response_wave[(numpnts(response_wave)-1)] = this_response

  insertpoints numpnts(power_wave),1,power_wave
  power_wave[(numpnts(power_wave)-1)] = last_power
	print this_response
////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////
// calculate next uncage power
////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////
converge_indicator = next_power_geo(this_response, target_response, last_power)
if(interp(next_power, fit_ach_4_x, fit_ach_4) <= 10)
 next_power = interp(10, fit_ach_4, fit_ach_4_x)
endif
if(this_response < target_response)
if(next_power < last_power)
print "Next power estimate is less than last power, but last response is less than target. Default to next power <- last_power + 0.2"
  next_power = last_power + 0.2
endif
endif
print "Last power",last_power,interp(last_power, fit_ach_4_x, fit_ach_4)
  print "Next power",next_power,interp(next_power, fit_ach_4_x, fit_ach_4)
	if(converge_indicator)
		read_write_prm(next_power,protocol_dir+"sm.uncage.one.line.prm",protocol_dir+"sm.uncage.one.line.prm")
		break
	endif

	read_write_prm(next_power,prm_file_name,protocol_dir+"sm.updated.protocol.prm")

  temp_var = continue_prompt()
  if(temp_var == 2)
	  read_write_prm(last_power/1.41421,protocol_dir+"sm.uncage.one.line.prm",protocol_dir+"sm.uncage.line.0.prm")
  read_write_prm(last_power,protocol_dir+"sm.uncage.one.line.prm",protocol_dir+"sm.uncage.line.1.prm")
    read_write_prm(1.41421*last_power,protocol_dir+"sm.uncage.one.line.prm",protocol_dir+"sm.uncage.line.2.prm")
    break
  endif
  if(temp_var == 3)
	read_write_prm(next_power/1.41421,protocol_dir+"sm.uncage.one.line.prm",protocol_dir+"sm.uncage.line.0.prm")
  read_write_prm(next_power,protocol_dir+"sm.uncage.one.line.prm",protocol_dir+"sm.uncage.line.1.prm")
  read_write_prm(  1.41421*next_power,protocol_dir+"sm.uncage.one.line.prm",protocol_dir+"sm.uncage.line.1.prm")

    break
  endif
	killwaves ach_1, ach_3
  power_0[0] = min_power_0
  power_0[1] = max_power_0
  i = i + 1
while(1)//do1
		KillVariables/A;	KillStrings/A;	KillWaves /z/A
Endmacro


macro fit_power()
DeletePoints (numpnts(ACH_3)-1),1, ACH_3
DeletePoints (numpnts(ACH_4)-1),1, ACH_4
DeletePoints 0,1, ACH_3
DeletePoints 0,1, ACH_4
//Display ACH_4 vs ACH_3
CurveFit/M=2/W=0 poly 3, ACH_4/X=ACH_3/D
duplicate fit_ach_4 fit_ach_4_x
fit_ach_4_x = x
duplicate ach_3 ach_4_x
//ach_4_x = x
display ach_4 vs ach_4_x
AppendToGraph fit_ACH_4 vs fit_ach_4_x
//display fit_ach_4 vs ach_4_x
ModifyGraph rgb(fit_ACH_4)=(0,0,0)
make /n=2 power_0
Edit/K=0 power_0;DelayUpdate
		KillVariables/A;	KillStrings/A;	KillWaves /z/A
endmacro


function next_power_geo(this_response, target_response, last_power)
variable this_response, target_response, last_power
wave ach_4_x, fit_ach_4, fit_ach_4_x
variable /g max_power, min_power, next_power, max_power_0
if(this_response < target_response)
next_power = 1.41421*interp(last_power, fit_ach_4_x, fit_ach_4)
next_power = interp(next_power, fit_ach_4,  fit_ach_4_x)
min_power = last_power
max_power = max(1.41421*next_power,max_power)
return 0
endif
if(this_response >= target_response)
next_power = interp(last_power, fit_ach_4_x, fit_ach_4)/1.41421
next_power = interp(next_power, fit_ach_4,  fit_ach_4_x)
min_power = min(next_power/1.41421,min_power)
max_power = last_power

return 0
endif
end

function next_power_bs(this_response, target_response, last_power)
variable this_response, target_response, last_power
wave ach_4_x, fit_ach_4, fit_ach_4_x
variable /g max_power, min_power, next_power, max_power_0
if(this_response > 1.2*target_response)
	max_power = last_power
	next_power = ((max_power+min_power)/2)
	return 0
endif
if(this_response <= 0.5*target_response)
	next_power = 1.41421*interp(last_power, fit_ach_4_x, fit_ach_4)
	next_power = interp(next_power, fit_ach_4,  fit_ach_4_x)
	min_power = last_power
  	max_power = max(interp(1.41421*interp(next_power, fit_ach_4_x, fit_ach_4), fit_ach_4,  fit_ach_4_x),max_power)
	return 0
endif

if(this_response < 0.8*target_response)
	min_power = last_power
	next_power = ((max_power+min_power)/2)
	return 0
endif

next_power = last_power
return 1
end

function next_power_fit(this_response, target_response, last_power)
variable this_response, target_response, last_power
wave ach_4_x, fit_ach_4, fit_ach_4_x
variable /g max_power, min_power, next_power, min_power_0


if(abs(this_response-target_response)<=0.2*target_response)
next_power = last_power
return 1
endif

if(this_response <= 0.5*target_response)
  next_power = 1.41421*interp(last_power, fit_ach_4_x, fit_ach_4)
  next_power = interp(next_power, fit_ach_4,  fit_ach_4_x)
  max_power = max(max_power, interp(1.41421*interp(next_power, fit_ach_4_x, fit_ach_4), fit_ach_4,  fit_ach_4_x))
  return 0
endif

next_power = interp(last_power, fit_ach_4_x, fit_ach_4) * ((target_response/this_response)^0.5)
	next_power = interp(next_power, fit_ach_4,  fit_ach_4_x)
  return 0
end

function next_power_nr(this_response, target_response, last_power)
variable this_response, target_response, last_power
wave ach_4_x, fit_ach_4
variable /g max_power, min_power, next_power, min_power_0
variable g, npts, sign_indicator
wave response_wave, fit_ach_4_x
sign_indicator = sign((target_response-this_response))

if(this_response <= 0.5*target_response)
	next_power = 1.41421*interp(last_power, fit_ach_4_x, fit_ach_4)
	next_power = interp(next_power, fit_ach_4,  fit_ach_4_x)
make /o/n= 0 response_wave
make /o/n= 0 power_wave
	return 0
endif

if(abs(this_response-target_response)<=0.2*target_response)
next_power = last_power
return 1
endif

if(numpnts(response_wave)<=1)
next_power = last_power + sign_indicator*0.2
return 0
endif
npts = numpnts(response_wave)
g = (response_wave[(npts-1)]-response_wave[(npts-2)])/(interp(power_wave[(npts-1)], fit_ach_4_x, fit_ach_4)-interp(power_wave[(npts-2)], fit_ach_4_x, fit_ach_4))
next_power = last_power + (target_response-this_response)/g
next_power = interp(next_power, fit_ach_4,  fit_ach_4_x)
if(numtype(next_power))
next_power = last_power + sign_indicator*0.2
endif
  return 0
end


Function DiffTwoExp2(w,t) : FitFunc
	Wave w
	Variable t

	//CurveFitDialog/ These comments were created by the Curve Fitting dialog. Altering them will
	//CurveFitDialog/ make the function less convenient to work with in the Curve Fitting dialog.
	//CurveFitDialog/ Equation:
	//CurveFitDialog/    IF    ((t-t0) < 0 )
	//CurveFitDialog/        f(t) = 0
	//CurveFitDialog/    ELSE
	//CurveFitDialog/        f(t) = (gsyn*(-exp(-(t-t0)/tr)+exp(-(t-t0)/td))/(-exp(-(t0+(td*tr/(td-tr))*ln(td/tr)-t0)/tr)+exp(-(t0+(td*tr/(td-tr))*ln(td/tr)-t0)/td)) -y0)
	//CurveFitDialog/    ENDIF
	//CurveFitDialog/ End of Equation
	//CurveFitDialog/ Independent Variables 1
	//CurveFitDialog/ t
	//CurveFitDialog/ Coefficients 5
	//CurveFitDialog/ w[0] = gsyn
	//CurveFitDialog/ w[1] = t0
	//CurveFitDialog/ w[2] = td
	//CurveFitDialog/ w[3] = tr
	//CurveFitDialog/ w[4] = y0
	IF    ((t-w[1]) < 0 )
		return w[4]
	ELSE
		return (w[4]+w[0]*(-exp(-(t-w[1])/w[3])+exp(-(t-w[1])/w[2]))/(-exp(-(w[1]+(w[2]*w[3]/(w[2]-w[3]))*ln(w[2]/w[3])-w[1])/w[3])+exp(-(w[1]+(w[2]*w[3]/(w[2]-w[3]))*ln(w[2]/w[3])-w[1])/w[2])))
	ENDIF
End

function read_write_prm(new_power,in_file_name,out_file_name)
	string in_file_name, out_file_name
	variable new_power
	variable file_ref_num
	variable out_file
	variable i_ = 0
	string string_in
	string write_buffer
	variable write_block_size = 2^9
	variable write_buffer_start
	variable write_buffer_stop
	variable power_line_start
	variable power_line_stop
	open /R  file_ref_num as in_file_name
	freadline /T="" file_ref_num, string_in
	close(file_ref_num)
	open out_file as out_file_name
	power_line_start = strsearch(string_in,"DAC6 Train Pulse Pot 1",0)
	power_line_stop = strsearch(string_in,num2char(13),power_line_start)
	string power_line_text = string_in[power_line_start,power_line_stop]
	power_line_text = power_line_text[0,strsearch(power_line_text,"=",0)] + num2str(new_power)+num2char(13)
	do
		write_buffer_start = i_*write_block_size
		write_buffer_stop =(i_+1)*write_block_size-1
		if(write_buffer_stop >= power_line_start)
			break
		endif
		write_buffer = string_in[write_buffer_start,write_buffer_stop]
		fprintf out_file, write_buffer
		i_ = i_+1
	while(strlen(write_buffer) > 0)
	write_buffer = string_in[write_buffer_start,power_line_start-1]
	fprintf out_file, write_buffer
	fprintf out_file, power_line_text
	string_in = string_in[power_line_stop+1, Inf]
	i_=0
	do
		write_buffer_start = i_*write_block_size
		write_buffer_stop =(i_+1)*write_block_size-1
		write_buffer = string_in[write_buffer_start,write_buffer_stop]
		fprintf out_file, write_buffer
		i_ = i_+1
	while(strlen(write_buffer) > 0)
	close(out_file)
end

function read_power_from_prm(prm_in)
  string prm_in
	variable file_ref_num, power_line_stop,power_line_start, last_power
  string string_in
	open /R  file_ref_num as prm_in
	freadline /T="" file_ref_num, string_in
	close(file_ref_num)
	power_line_start = strsearch(string_in,"DAC6 Train Pulse Pot 1",0)
	power_line_stop = strsearch(string_in,num2char(13),power_line_start)
	string power_line_text = string_in[power_line_start,power_line_stop]
	last_power = str2num(power_line_text[strsearch(power_line_text,"=",0)+1,inf])
	return last_power
end

macro uncage_pos_cor(x1,dx,y1,dy)
variable x1,dx,y1,dy
variable/G x2,y2
x2 = x1-dx
y2 = y1-dy
print x2,y2
endmacro