#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <math.h>
#include <string.h>

#include "coords.h"
#include "tr.h"
#include "db.h"
#include "stock.h"

#define ARGC 8
#define BUFSIZE 256

typedef struct CTable {
	double	calib;
	char	chan[16];
} CTable;


// Filter function

int rerange( Dbptr *tr, float *data , void *private );

int rerange( Dbptr *tr, float *data, void *private ) {

	int		found;
	long	i, nsamp, nrecords;
	double	time, endtime, samprate, calib;
	char	sta[7], chan[9], chanb[9];

	Dbptr	*db = private;	

	dbgetv( *tr, 0, "sta", sta, "chan", chan, "time", &time, "endtime", &endtime, "samprate", &samprate, "nsamp", &nsamp, 0 );
	//fprintf( stderr, "\n%-5s %-3s %10.3f sps %5ld %f\n", sta, chan, samprate, nsamp, time ); 

	if ( !dbquery( *db, dbRECORD_COUNT, &nrecords ) ) {
		register_error( 1, "Can't get record_count\n" );
	}

	// Find the original calib:

	found = -1;
	for ( (*db).record=0; (*db).record<nrecords; (*db).record++ ) {
		dbgetv( *db, 0, "chan", chanb, "calib", &calib, 0 );
		//fprintf( stderr, "chan:%s  calib:%f\n", chanb, calib );

		if ((chan[2] == 'Z') && (chanb[2] == 'V')) {
			found = 0; break;
		}
		if ((chan[2] == 'N') && (chanb[2] == 'L')) {
			found = 0; break;
		} 
		if ((chan[2] == 'E') && (chanb[2] == 'T')) {
			found = 0; break;
		}
	}	
	
	if (found == -1) {
		register_error(1, "No channel match: %s\n", chanb);

	}

	//fprintf( stderr, "chan:%s  chanb:%s  calib:%f\n", chan, chanb, calib );

	// Invert the calib, i.e., de-calibrate the data

	calib = 1.0 / calib;

    for ( i=0 ; i<nsamp ; i++ ) { 
		//fprintf(stderr, " %10.0f", data[i] ) ; 
		data[i] = (float) round(calib * data[i]) ; 
    }
    //fprintf(stderr, "\n" ) ;
	
	return(0);
}


// main 

int main (int argc, char **argv) {

	char	*db_in, *db_out, *select, *chan_e, *chan_n, *chan_z, *file_name;
	char	*newchan[3];
	char	fchan[9], sta[9], loc[9];
	Dbptr	db, dbout, tr; 

	int		result = 0;

	elog_init( argc, argv );
	
	if( argc < ARGC ) {
		printf( "\n" );
		printf( "Error: missing parameters.\n" );
		printf( "Usage: %s db_in db_out select chan_e chan_n chan_z file_name\n", argv[0] );
		die( 1, "" );
	}

	db_in = calloc(BUFSIZE,sizeof(char));
	strncpy( db_in, argv[1], BUFSIZE-1 );

	db_out = calloc(BUFSIZE,sizeof(char));
	strncpy( db_out, argv[2], BUFSIZE-1 );

	select = calloc(BUFSIZE,sizeof(char));
	strncpy( select, argv[3], BUFSIZE-1 );
  
	chan_e = calloc(BUFSIZE,sizeof(char));
	strncpy( chan_e, argv[4], BUFSIZE-1 );

	chan_n = calloc(BUFSIZE,sizeof(char));
	strncpy( chan_n, argv[5], BUFSIZE-1 );

	chan_z = calloc(BUFSIZE,sizeof(char));
	strncpy( chan_z, argv[6], BUFSIZE-1 );

	file_name = calloc(BUFSIZE,sizeof(char));
	strncpy( file_name, argv[7], BUFSIZE-1 );

	//printf( "db_in:     %s\n", db_in );
	//printf( "db_out:    %s\n", db_out );
	//printf( "select:    %s\n", select );
	//printf( "chan_e:    %s\n", chan_e );
	//printf( "chan_n:    %s\n", chan_n );
	//printf( "chan_z:    %s\n", chan_z );
	//printf( "file_name: %s\n", file_name );
	
	newchan[0] = chan_e;
	newchan[1] = chan_n;
	newchan[2] = chan_z;	
	
	if ( dbopen( db_in, "r", &db) ) {
		die( 1, "Can't open database %s\n", db_in ) ;
	}

	db = dblookup( db, 0, "rotate", 0, 0 );
	db = dbsubset( db, select, 0 );		

	// Debug 
	//Tbl *list = strtbl( "sta", "chan", "time", "endtime", "calib", 0 );
	//dbselect( db, list, stderr );
	//free(list);

	db.record = 0;
	if ( dbgetv( db, 0, "sta", sta, 0 ) != 0 ) {
		printf( "Cannot get sta.\n" );
	}

	if ( dbopen( db_out, "r+", &dbout) < 0 ) {
		die( 1, "Can't open output database '%s'\n", db_out ) ;
	}

	dbout = dblookup( dbout, 0, "wfdisc", 0, 0 ) ;
	
	tr = dbinvalid() ;
	if ( trload_css( db, "time", "endtime", &tr, 0, 0 ) < 0 ) {
		die( 1, "ERROR: trload_css fails" ) ;
	}

	if ( trapply_calib( tr ) ) {
		die( 1, "ERROR: trapply_calib fails\n" ) ;
	}
	
	result = rotate_to_standard( tr, newchan ) ;
	if ( result < 0 ) {
		die( 1, "ERROR: rotate_to_standard fails" ) ;
	} else if ( result > 0 ) {
		complain( 0, "WARNING: some channels not rotated.\n" ) ;
	}

	// Only save the zne channels to the new db
	tr = dbsubset( tr, "chan =~ /^..[ZNE].*/", 0 );

	// Recalibrate the range so that the trace values are 
	// effectively returned to counts.  This reduces the peak 
	// Values so they are in range of STEIM2

	trapply( tr, rerange, (void*)&db );
	//trdisp( tr, "Test" );

	for( int i = 0;  i < 3; i++) {
		//printf( "newchan[%i]:%s\n", i, newchan[i] );
		strncpy( fchan, newchan[i], 3 );
		fchan[3] = '\0';
		strncpy( loc, newchan[i]+4, 2 );
		loc[2] = '\0';
		map_seed_chanloc( sta, fchan, loc,  newchan[i] );
		//printf("%s\t%s\t%s\t%s\n", sta, fchan, loc, newchan[i]);
	}

	// Specify a filename string
	char *fmt_str;
	fmt_str = calloc(BUFSIZE,sizeof(char));
	strcat( fmt_str, "%Y/%j/%{sta}_%{chan}_" );
	strcat( fmt_str, file_name );
	strcat( fmt_str, ".msd" );
	//printf( "fmt_str: |%s|\n", fmt_str );

	result = trsave_wf ( tr, dbout, "sd", fmt_str, trOVERWRITE ) ;	
	if ( result < 0 ) {
		die ( 1, "trsave_wf fails\n" ) ;
	}
	else if ( result > 0 ) {
		if ( result & TR_CLIPPED ) {
			complain ( 0, "TR_CLIPPED - waveforms  clipped." ) ; 
		}
		if ( result & TR_TRUNCATED ) {
			complain ( 0, "TR_TRUNCATED - waveforms truncated." ) ;
		}
	}

	dbclose(db);
	dbclose(dbout);
	dbclose(tr);
	//trfree(tr);
	//trfree(db);
	//trfree(dbout);
	
	//printf( "COMPLETE: %s\n", select );
	return result;
}