sucountkey.c

su 的源代码库

/* Copyright (c) Colorado School of Mines, 2006.*/
/* All rights reserved.                       */

/* SUCOUNTKEY: $Revision: 1.3 $ ; $Date: 2003/06/09 16:17:07 $	*/

#include "su.h"
#include "segy.h"
#include "header.h"
#include <signal.h>


/*********************** self documentation ******************************/
char *sdoc[] = {
"										",
" SUCOUNTKEY - COUNT the number of unique values for a given KEYword.		",
"										",
" sucountkey < input.su key=[sx,gx,cdp,...]					",
" Required parameter:								",
" key=			array of SU header keywords being counted		",
" Optional parameters:								",
" verbose=1		chatty, =0 just print keyword number			",
" Example:									",
"	  suplane | sucountkey key=tracl,tracr,offset				",
"										",
NULL};

/*
 * Credits: Baoniu Han, [e-mail], Nov, 2000
 */

/* Globals variables*/
segy tr;

/* internal structure */
typedef struct {
	int num;
	Value *val;
} my_Count;

int main (int argc, char **argv)
{
	cwp_String key[SU_NKEYS];		/* array of keywords		*/
	cwp_String type[SU_NKEYS];		/* array of keywords		*/
	int index[SU_NKEYS];			/* name of type of getparred key*/ 
	my_Count count_table[SU_NKEYS];		/* count table for keywords	*/

	int verbose;				/* verbose flag			*/

	int ikey;				/* key counter			*/
	int nkeys;				/* number of header fields set	*/
	int i;					/* count index			*/
	Value val0;				/* value of key field		*/
	long buf_size;
	
	/* Initialize */
	initargs(argc, argv); 
	requestdoc(1); 
	
	/* Get "key" values */ 
	if ((nkeys=countparval("key"))!=0) {
		getparstringarray("key",key); 
	} else {
		key[0]="cdp"; 
	}
	if (!getparint("verbose", &verbose))	verbose = 1;

	/* you can increase this number to change the buffer size */
	/* for each key word */
	buf_size=100000L;
	
	/* get types and indexes corresponding to the keys */
	for (ikey=0; ikey<nkeys; ++ikey) {
		type[ikey]=hdtype(key[ikey]); 
		index[ikey]=getindex(key[ikey]);
		count_table[ikey].num=0;
		count_table[ikey].val=(Value *) malloc(sizeof(Value)*buf_size);
	}


	/* get info from first trace */
	if (!gettr(&tr))  err("can't get first trace"); 
	for(ikey=0;ikey<nkeys;++ikey){
		count_table[ikey].num=1;
		gethval(&tr, index[ikey], &count_table[ikey].val[0]);
	}



	/* loop over traces */
	while (gettr(&tr)) {
		
	  for (ikey=0; ikey<nkeys; ++ikey) {
		gethval(&tr,index[ikey],&val0); 
		for(i=0;i<count_table[ikey].num;i++)
	 	    if (valcmp(type[ikey], val0, count_table[ikey].val[i])==0) break;

	  if (i==count_table[ikey].num){				
		gethval(&tr, index[ikey], &count_table[ikey].val[count_table[ikey].num]);		
		count_table[ikey].num++;
	  } 

	}

	}

	if (verbose==1)
		warn("The counting of key word is finished, the total number for specified header is as follow:");

	for (ikey=0;ikey<nkeys;++ikey){
		if (verbose==1)
			warn(" key %s has %d unique entries in this data", key[ikey], count_table[ikey].num);
		else
			fprintf(stdout,"%s %d ", key[ikey], count_table[ikey].num);

		
	}	

	return(CWP_Exit());
}