#======================================================================

#                    C A L C _ T I M E L A G S . P L 

#                    doc: Fri Dec 17 21:59:07 2010

#                    dlm: Tue Dec 21 00:40:33 2010

#                    (c) 2010 A.M. Thurnherr

#                    uE-Info: 117 20 NIL 0 0 72 2 2 4 NIL ofnI

#======================================================================

# HISTORY:

#	Dec 17, 2010: - created

#	Dec 18, 2010: - adapted for multi-pass lagging

#	Dec 20: 2010: - added code to adjust start and end of profile ens

#				    based on extent of CTD profile and guestimated time

#				    ofset

# TODO:

#	- better seabed code (from LADCPproc)

#	- intermediate-step timelagging guess

#	- flip aliased ensembles

sub mad_w($$$)									# mean absolute deviation

{

	my($fe,$le,$so) = @_;						# first/last LADCP ens, CTD scan offset

	my($sad) = my($n) = 0;

	for (my($e)=$fe; $e<=$le; $e++) {

		my($s) = int(($LADCP{ENSEMBLE}[$e]->{ELAPSED} + $CTD{TIME_LAG} - $CTD{ELAPSED}[0]) / $CTD{DT} + 0.5);

		die("assertion failed\n" .

			"\ttest: abs($LADCP{ENSEMBLE}[$e]->{ELAPSED} + $CTD{TIME_LAG} - $CTD{ELAPSED}[$s]) <= $CTD{DT}/2\n" .

			"\te = $e, s = $s, ensemble = $LADCP{ENSEMBLE}[$e]->{NUMBER}"

		) unless (abs($LADCP{ENSEMBLE}[$e]->{ELAPSED} + $CTD{TIME_LAG} - $CTD{ELAPSED}[$s]) <= $CTD{DT}/2);

		my($dw) = $LADCP{ENSEMBLE}[$e]->{REFLR_W} - $CTD{W}[$s+$so];

		next unless (abs($dw) <= $opt_m);

		$sad += abs($dw);

		$n++;

	}

	return ($n>0) ? $sad/$n : 9e99;				# n == 0, e.g. in bottom gap

}

sub bestLag($$$$)								# find best lag in window

{

	my($fe,$le,$ww,$soi) = @_;					# first/last LADCP ens, window width, scan-offset increment

	my($bestso) = 0;							# error at first-guess offset

	my($bestmad) = mad_w($fe,$le,0);

	for (my($dso) = 1; $dso <= int($ww/2/$CTD{DT} + 0.5); $dso+=$soi) {

		my($mad) = mad_w($fe,$le,-$dso);

		$bestmad=$mad,$bestso=-$dso if ($mad < $bestmad);

		$mad = mad_w($fe,$le,$dso);

		$bestmad=$mad,$bestso=$dso if ($mad < $bestmad);

	}

	return ($bestso,$bestmad);

}

#----------------------------------------------------------------------

# carry out lag correlations and keep tally of the results

#	- fist and last 10% of LADCP profile ignored

#----------------------------------------------------------------------

sub calc_lag($$$)

{

	my($n_windows,$w_size,$scan_increment) = @_;

RETRY:

	progress("Calculating $n_windows time-lags from ${w_size}s-long windows at %dHz resolution...\n",

		int(1/$scan_increment/$CTD{DT}+0.5));

	my($approx_CTD_profile_start_ens) =

		$firstGoodEns + int(($CTD{ELAPSED}[0] - $CTD{TIME_LAG}) / $LADCP{MEAN_DT});

	my($approx_CTD_profile_end_ens) =

		$firstGoodEns + int(($CTD{ELAPSED}[$#{$CTD{ELAPSED}}] + $CTD{ELAPSED}[0] - $CTD{TIME_LAG}) / $LADCP{MEAN_DT});

	my($approx_joint_profile_start_ens) = max($firstGoodEns,$approx_CTD_profile_start_ens);

	my($approx_joint_profile_end_ens) 	= min($lastGoodEns,$approx_CTD_profile_end_ens);

	debugmsg("profile start: $firstGoodEns -> $approx_joint_profile_start_ens\n");

	debugmsg("profile end  : $lastGoodEns -> $approx_joint_profile_end_ens\n");

	my($skip_ens) = int(($approx_joint_profile_end_ens - $approx_joint_profile_start_ens) / 10 + 0.5);

	my(%nBest);

	for (my($wi)=0; $wi<$n_windows; $wi++) {

		my($fe) = $approx_joint_profile_start_ens + $skip_ens + $wi*int(($approx_joint_profile_end_ens-$approx_joint_profile_start_ens-2*$skip_ens)/$n_windows+0.5);

		my($so,$mad) = bestLag($fe,$fe+int($w_size/$LADCP{MEAN_DT}+0.5),$w_size,$scan_increment);

		debugmsg("%.1f cm/s mad(w) at %3d scans offset\n",100*$mad,$so);

		$nBest{$so}++;

	}

	my(@best_lag);

	foreach my $i (keys(%nBest)) {

		$best_lag[0] = $i if ($nBest{$i} > $nBest{$best_lag[0]});

	}

	foreach my $i (keys(%nBest)) {

		next if ($i == $best_lag[0]);

		$best_lag[1] = $i if ($nBest{$i} > $nBest{$best_lag[1]});

	}

	foreach my $i (keys(%nBest)) {

		next if ($i == $best_lag[0] || $i == $best_lag[1]);

		$best_lag[2] = $i if ($nBest{$i} > $nBest{$best_lag[2]});

	}

	progress("\t3 most popular offsets: %d (%d%%), %d (%d%%), %d (%d%%)\n",

		$best_lag[0],int(($nBest{$best_lag[0]}/$n_windows)*100+0.5),

		$best_lag[1],int(($nBest{$best_lag[1]}/$n_windows)*100+0.5),

		$best_lag[2],int(($nBest{$best_lag[2]}/$n_windows)*100+0.5));

	if ($nBest{$best_lag[0]}+$nBest{$best_lag[1]}+$nBest{$best_lag[2]} <= 6) {

		warning(0,"cannot determine valid lag => trying again with doubled window size\n");

		undef(%nBest);

		$w_size *= 2;

		goto RETRY;

	}

	croak(sprintf("$0: cannot determine a valid lag; top 3 tags account for %d%% of total (use -3 to relax criterion)\n",

			int(100*($nBest{$best_lag[0]}+$nBest{$best_lag[1]}+$nBest{$best_lag[2]})/$n_windows+0.5)))

		unless ($nBest{$best_lag[0]}+$nBest{$best_lag[1]}+$nBest{$best_lag[2]} >= $opt_3*$n_windows);

	my($wmo) = ($nBest{$best_lag[0]}*$best_lag[0] +

				$nBest{$best_lag[1]}*$best_lag[1] +

				$nBest{$best_lag[2]}*$best_lag[2]) / ($nBest{$best_lag[0]} +

													  $nBest{$best_lag[1]} +

													  $nBest{$best_lag[2]});

	progress("\tweighted-mean offset = %.1f scans\n",$wmo);

	if ($wmo > 0.9*$w_size/2/$CTD{DT}) {

		warning(0,"lag too close to the edge of the window --- trying again after adjusting the guestimated offset\n");

		$CTD{TIME_LAG} += $w_size/2;

		undef(%nBest);

		goto RETRY;

	}

	if (-$wmo > 0.9*$w_size/2/$CTD{DT}) {

		warning(0,"lag too close to the edge of the window --- trying again after adjusting the guestimated offset\n");

		$CTD{TIME_LAG} -= $w_size/2;

		undef(%nBest);

		goto RETRY;

	}

	return $CTD{TIME_LAG}+$wmo*$CTD{DT};

}

1;