suaddhead.c

su 的源代码库

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

/* SUADDHEAD: $Revision: 1.17 $ ; $Date: 2005/10/04 16:42:43 $		*/

#include "su.h"
#include "segy.h"
#include "header.h"

/*********************** self documentation ******************************/
char*  sdoc[] = {
" 									",
" SUADDHEAD - put headers on bare traces and set the tracl and ns fields",
" 									",
" suaddhead <stdin >stdout ns= ftn=0					",
" 									",
" Required parameter:							",
" 	ns=the number of samples per trace				",
" 									",
" Optional parameter:							",
#ifdef SU_LINE_HEADER
"	head=           file to read headers in				",
"                       not supplied --  will generate headers 		",
"                       given        --  will read in headers and attach",
"                                        floating point arrays to form 	",
"                                        traces 			", 
"                       (head can be created via sustrip program)	",
#endif
" 	ftn=0		Fortran flag					",
" 			0 = data written unformatted from C		",
" 			1 = data written unformatted from Fortran	",
"       tsort=3         trace sorting code:				",
"                                1 = as recorded (no sorting)		",
"                                2 = CDP ensemble			",
"                                3 = single fold continuous profile	",
"                                4 = horizontally stacked		", 
"       ntrpr=1         number of data traces per record		",
"                       if tsort=2, this is the number of traces per cdp", 
" 									",
" Trace header fields set: ns, tracl					",
" Use sushw/suchw to set other needed fields.				",
" 									",
" Caution: An incorrect ns field will munge subsequent processing.	",
" Note:    n1 and nt are acceptable aliases for ns.			",
" 									",
" Example:								",
" suaddhead ns=1024 <bare_traces | sushw key=dt a=4000 >segy_traces	",
" 									",
" This command line adds headers with ns=1024 samples.  The second part	",
" of the pipe sets the trace header field dt to 4 ms.			",
" 									",
NULL};
/**************** end self doc *******************************************/

/* Credits:
 *	SEP: Einar Kjartansson
 *	CWP: Jack K. Cohen
 *      UNOCAL: Zhiming Li	add ascii and binary headers
 */


extern unsigned char su_text_hdr[3200];
extern bhed su_binary_hdr;

segy tr;

int
main(int argc, char **argv)
{
	int ns;			/* number of samples			*/
	int ftn;		/* ftn=1 for Fortran			*/
	char junk[ISIZE];	/* to discard ftn junk  		*/
	cwp_Bool isreading=cwp_true;    /* true/false flag for while    */

	int ihead=0;		/* counter */
	int iread=0;		/* counter */
	int tsort, ntrpr;	/* Unocal header fields */
#ifdef SU_LINE_HEADER
	cwp_String head;	/* name of file holding headers         */
	FILE *infp=stdin, *outfp=stdout; /* input and output files 	*/
#endif
	FILE *headfp=NULL;	/* . file pointer for pointers		*/

	/* Initialize */
	initargs(argc, argv);
	requestdoc(1);


	/* Get parameters */
	if (!getparint("n1", &ns)
	 && !getparint("nt", &ns)
	 && !getparint("ns", &ns))  err("must specify ns=");
	if (!getparint("ftn", &ftn))	ftn = 0;
	if (ftn != 0 && ftn != 1)	err("ftn=%d must be 0 or 1", ftn);
	if (!getparint("ntrpr", &ntrpr)) ntrpr = 1;
	if (!getparint("tsort", &tsort)) tsort = 3;

#ifdef SU_LINE_HEADER

	if (!getparstring("head"  , &head)) {
		ihead = 0;

	} else {
		ihead = 1;
		if( !(headfp=efopen(head, "r")) ){

                   err( "unable to open header file " );
                }
              
	}


	/* create id headers */
	if(ihead==0) {
		su_binary_hdr.format = 1;
		su_binary_hdr.ntrpr = ntrpr;
		su_binary_hdr.tsort = tsort;
		su_binary_hdr.fold = ntrpr;
	} else {
		fgethdr(headfp,&su_text_hdr,&su_binary_hdr);
	}

	su_binary_hdr.hns = ns;

#endif
		
	while (isreading==cwp_true) {
		static int tracl = 0;	/* one-based trace number */

		/* If Fortran data, read past the record size bytes */
		if (ftn) efread(junk, ISIZE, 1, stdin);

		/* Do read of data for the segy */
		iread = fread((char *) tr.data, FSIZE, ns, stdin);
		if(iread!=ns) {
			return(CWP_Exit());

		} else {
			if(ihead==0) {
				tr.tracl = ++tracl;
			} else {
				efread(&tr, 1, HDRBYTES, headfp);
			}
			tr.ns = ns;
			tr.trid = TREAL;
			puttr(&tr);
		}

		/* If Fortran data, read past the record size bytes */
		if (ftn) efread(junk, ISIZE, 1, stdin);
	}

	return(CWP_Exit());
}