#!/usr/bin/perl
#======================================================================
#                    S P L I T R D I 
#                    doc: Sat Aug 21 22:20:27 2010
#                    dlm: Sat Aug 21 23:18:41 2010
#                    (c) 2010 A.M. Thurnherr
#                    uE-Info: 51 44 NIL 0 0 72 2 2 4 NIL ofnI
#======================================================================

# split RDI files based on list of ensemble numbers (e.g. from yoyo -t)

# HISTORY:
#	Aug 21, 2010: - created

# NOTES:
#	- it is assumed that the input file begins with ensemble #1
#	- input file extension 000 is assumed
#	- turning-point ensembles are written to preceding profile,
#	  for compatibility with [yoyo]

$0 =~ m{(.*/)[^/]+};
require "$1RDI_BB_Read.pl";
use Getopt::Std;

die("Usage: $0 " .
	"[out-file -b)asename <bn>] " .
	"[out-file -n) <digits>] " .
	"<RDI file> <ens> <ens[...]>\n")
		unless (&getopts('b:n:') && @ARGV>=3);

chomp($opt_b = `basename $ARGV[0] .000`)		# basename
	unless defined($opt_b);
$opt_n = 2										# number of digits in profile #
	unless defined($opt_n);
		
readHeader($ARGV[0],\%hdr); shift;				# get length of ensembles
$ens_len = $hdr{ENSEMBLE_BYTES} + 2;

$first_ens = $ARGV[0]+1; shift;					# initialize loop
$last_ens  = $ARGV[0]; shift;
$cnr = 0;

do {											# split data
	sysseek(WBRF,($first_ens-2)*$ens_len,0) ||
		die("$WBRcfn: $!");
	$last_ens++ unless defined($ARGV[0]);
	$nBytes = ($last_ens-$first_ens+1) * $ens_len;
	sysread(WBRF,$buf,$nBytes) == $nBytes ||
		die("$WBRcfn: file truncated");

	$fn = $opt_b . sprintf("_%0${opt_n}d.000",$cnr++);
	open(F,">$fn") || die("$fn: $!\n");
	syswrite(F,$buf,$nBytes) == $nBytes ||
		die("$fn: $!\n");
	close(F);
	printf(STDERR "$fn: %d ensembles ($nBytes bytes)\n",$last_ens-$first_ens+1);
	
	$first_ens = $last_ens+1;
	$last_ens  = $ARGV[0]; shift;
} while defined($last_ens);

exit(0);