// Tek Query File Loader
// This code was written to load a Tektronix Query File which is a dump of a query to a Tek scope.
// NOTE: This is not the Tek WFM file format.
 
// The way to determine the header length is to look for the ":CURVE" keyword. It is formatted like this:
//		":CURVE <digit><number>"		.e.g., ":CURVE 41000"
// where <digit> represents the number of digits is the following number and <number> represents the number of
// samples of binary data that follow. In this case, "4" is the number of digits in "1000" and 1000 is the number of samples.
 
static Constant kHeaderLength = 282				// Length of header in bytes. Empirically determined.
static Constant kWaveDataType = 2				// Single precision floating point
 
//	LoadTekQueryFile(pathName, fileName, createTable, createGraph)
//
//	pathName is the name of an Igor symbolic path or "".
//	fileName is one of the following:
//		The name of a file in the folder designated by pathName.
//		A partial path from the folder designated by pathName to the file to be loaded.
//		A full path to the file to be loaded.
//		"" to get an Open File dialog.
 
// Add an item to Igor's Load Waves submenu (Data menu)
Menu "Load Waves"
	"Load Tek Query File...", LoadTekQueryFileMenuItem()
	"Load All Tek Query Files In Folder...", LoadAllTekQueryFiles("", "", 0, 0)
End
 
Function/S StripDoubleQuotes(str)
	String str					// A string starting and ending with double quotes
 
	if (CmpStr(str[0], "\"") != 0)
		return str				// String did not start with double quotes
	endif
 
	Variable len  = strlen(str)
	str = str[1,len-2]
	return str
End
 
Structure TekQueryHeaderInfo
	Variable encoding			// 0 = binary. -1 = unknown. Currently the only encoding I know about is binary. From ENCDG keyword.
	Variable byteOrder			// 0 = big-endian, 1 = little-endian.  From BYT_OR keyword.
	Variable dataType			// See WaveType for possible values. From BIT_NR and BN_FMT keywords.
	Variable numPoints			// Number of points in waveform. From NR_PT keyword.
	String xUnits				// From XUNIT keyword.
	Variable x0					// From XZERO keyword. 
	Variable dx					// From XINCR keyword. 
	String yUnits				// From YUNIT keyword.
	Variable yOffset				// From YOFF keyword.
	Variable yZero				// From YZERO keyword.
	Variable yMultiplier		// From YMULT keyword.
EndStructure
 
Function ExtractQueryHeaderInfo(refNum, headerInfo, quiet)
	Variable refNum			// File reference number
	STRUCT TekQueryHeaderInfo &headerInfo
	Variable quiet				// 0 - print information about errors
 
	Variable result = 0
 
	Variable originalPos
	FStatus refNum
	originalPos = V_filePos
 
	// Read header string
	String headerStr = ""
	Variable headerStrLength = kHeaderLength
	FSetPos refNum, 0
	headerStr = PadString(headerStr, kHeaderLength, 0x20)
	FBinRead refNum, headerStr
	FSetPos refNum, originalPos
 
	// Parse header string
	String str
	str = StringByKey("ENCDG", headerStr, " ")
	strswitch(str)
		case "BIN":
			headerInfo.encoding = 0
			break
		default:
			headerInfo.encoding = -1			// Unknown encoding
			result -= 1
			break
	endswitch
 
	str = StringByKey("BYT_OR", headerStr, " ")
	strswitch(str)
		case "MSB":
			headerInfo.byteOrder = 0
			break
		case "LSB":
			headerInfo.byteOrder = 1
			break
		default:
			headerInfo.byteOrder = -1			// Unknown byte order
			result -= 1
			break
	endswitch
 
	headerInfo.numPoints = NumberByKey("NR_PT", headerStr, " ")
 
	headerInfo.xUnits = StripDoubleQuotes(StringByKey("XUNIT", headerStr, " "))
	headerInfo.x0 = NumberByKey("XZERO", headerStr, " ")
	headerInfo.dx = NumberByKey("XINCR", headerStr, " ")
 
	headerInfo.yUnits = StripDoubleQuotes(StringByKey("YUNIT", headerStr, " "))
	headerInfo.yOffset = NumberByKey("YOFF", headerStr, " ")
	headerInfo.yZero = NumberByKey("YZERO", headerStr, " ")
	headerInfo.yMultiplier = NumberByKey("YMULT", headerStr, " ")
 
	Variable numBitsPerPoint = NumberByKey("BIT_NR", headerStr, " ")
	String binaryFormat = StringByKey("BN_FMT", headerStr, " ")
	strswitch(binaryFormat)
		case "RI":								// Presumably "real integer"
			switch (numBitsPerPoint)
				case 8:
					headerInfo.dataType = 8		// 8 bit signed integer
					break
 
				case 16:
					headerInfo.dataType = 16	// 16 bit signed integer
					break
 
				case 32:
					headerInfo.dataType = 32	// 32 bit signed integer
					break
 
				default:
					headerInfo.dataType = -1	// Unknown type
					result -= 1
					break
			endswitch		
			break
 
		default:
			headerInfo.dataType = -1			// Don't understand datatype
			result -= 1
			break
	endswitch
 
	return result
End
 
//	LoadTekQueryFileDialog(createTable, createGraph)
//	Displays a dialog in which the user can enter parameters.
//	All of the parameters function both as inputs, to preset the dialog values,
//	and as output, to return the values to the calling routine.
//	The function result is 0 if OK or -1 if cancel.
Function LoadTekQueryFileDialog(createTable, createGraph)
	Variable &createTable				// Input and output.
	Variable &createGraph				// Input and output.
 
	if ( (createTable<1) || (createTable>2) )
		createTable = 1
	endif
	if ( (createGraph<1) || (createGraph>2) )
		createGraph = 1
	endif
 
	// This is necessary because you can't pass an & parameter to Prompt.
	Variable createTable2 = createTable
	Variable createGraph2 = createGraph
 
	Prompt createTable2, "Create table", popup "Yes;No"
	Prompt createGraph2, "Create graph", popup "Yes;No"
	DoPrompt "Load Tek Query File", createTable2, createGraph2
	if (V_Flag != 0)
		return -1			// User cancelled.
	endif
 
	createTable = createTable2
	createGraph = createGraph2
 
	return 0
End
 
Function LoadTekQueryFile(pathName, fileName, createTable, createGraph, quiet, newWaveName)
	String pathName			// Igor symbolic path name or "" for dialog.
	String fileName				// file name, partial path and file name, or full path or "" for dialog.
	Variable createTable		// 1 = Yes, 2 = No, 0 for dialog.
	Variable createGraph		// 1 = Yes, 2 = No, 0 for dialog.
	Variable quiet				// 0 = print diagnostic and error messages, 1 = print error messages only, 2 = do not print diagnostic or error messages
	String &newWaveName		// Output: Name of wave created.
 
	newWaveName  = ""
 
	Variable err
 
	if ( (createTable==0) || (createGraph==0) )
		err = LoadTekQueryFileDialog(createTable, createGraph)
		if (err != 0)
			return err
		endif
	endif
 
	// This puts up a dialog if the pathName and fileName do not specify a file.
	String message = "Select a Tek Query file"
	Variable refNum
	Open /R /Z=2 /P=$pathName /T="????" /M=message refNum as fileName	// /T flag presents "All files" in Open File dialog.
 
	// Save outputs from Open in a safe place.
	err = V_Flag
	String fullPath = S_fileName
 
	if (err != 0)
		return err			// -1 means user canceled.
	endif
 
	// Load the header information
	STRUCT TekQueryHeaderInfo headerInfo
	if (ExtractQueryHeaderInfo(refNum, headerInfo, quiet!=2) != 0)
		if (quiet != 2)
			Print "Error while loading header section of the file."
		endif
		Close refNum
		return -1
	endif
 
	Close refNum
 
	switch (headerInfo.encoding)
		case 0:								// Binary
			break
		default:
			if (quiet != 2)
				Print "The file's encoding is unknown. Only binary encoding is supported."
			endif
			return -1
	endswitch
 
	if (quiet == 0)
		Printf "Loading Tek Query data from \"%s\"\r", fullPath
	endif
 
	GBLoadWave /B=(headerInfo.byteOrder) /A=tekWave /T={headerInfo.dataType,kWaveDataType} /S=(kHeaderLength) /W=1 /U=(headerInfo.numPoints) /Q fullPath
	if (V_flag == 0)
		if (quiet != 2)
			Print "Error while loading binary data section of the file."
		endif
		return -1
	endif
	newWaveName = StringFromList(0, S_waveNames)
	Wave w = $newWaveName
 
	SetScale/P x, headerInfo.x0, headerInfo.dx, ""+headerInfo.xUnits, w
	SetScale d, 0, 0, ""+headerInfo.yUnits, w
 
	// I don't know what headerInfo.yZero is all about (YZERO keyword).
 
	w = (w - headerInfo.yOffset) * headerInfo.yMultiplier
 
	if (quiet == 0)
		Printf "Created wave %s, %d points.\r", newWaveName, numpnts(w)
	endif
 
	if (createTable == 1)
		Edit w.id
		ModifyTable format[1]=3,digits[1]=9		// Display time in suitable format.
	endif
	if (createGraph == 1)
		Display w
	endif
 
	return 0			// Zero signifies no error.	
End
 
Function LoadTekQueryFileMenuItem()
	String newWaveName
	LoadTekQueryFile("", "", 1, 1, 0, newWaveName)
End
 
//	LoadAllTekQueryFilesDialog(extension, createTable, createGraph)
//	Displays a dialog in which the user can enter parameters.
//	All of the parameters function both as inputs, to preset the dialog values,
//	and as output, to return the values to the calling routine.
//	The function result is 0 if OK or -1 if cancel.
Function LoadAllTekQueryFilesDialog(extension, createTable, createGraph)
	String &extension					// Input and output
	Variable &createTable				// Input and output.
	Variable &createGraph				// Input and output.
 
	if (CmpStr(extension,".wfm")!=0 && CmpStr(extension,"????")!=0)
		extension = ".wfm"				// .wfm is really the wrong extension for a Tek Query file
	endif
	if ( (createTable<1) || (createTable>2) )
		createTable = 1
	endif
	if ( (createGraph<1) || (createGraph>2) )
		createGraph = 1
	endif
 
	// This is necessary because you can't pass an & parameter to Prompt.
	String extension2 = extension
	Variable createTable2 = createTable
	Variable createGraph2 = createGraph
 
	Prompt extension2, "Tek Query file name extension", popup ".wfm;Any extension"
	Prompt createTable2, "Create table", popup "Yes;No"
	Prompt createGraph2, "Create graph", popup "Yes;No"
	DoPrompt "Load All Tek Query Files", extension2, createTable2, createGraph2
	if (V_Flag != 0)
		return -1			// User cancelled.
	endif
 
	if (CmpStr(extension2, "Any extension") == 0)
		extension2 = "????"
	endif
 
	extension = extension2
	createTable = createTable2
	createGraph = createGraph2
 
	return 0
End
 
Function LoadAllTekQueryFiles(pathName, extension, createTable, createGraph)
	String pathName			// Name of an Igor symbolic folder created by Misc->NewPath
	String extension			// e.g., ".wfm". Pass "????" for any extension. Pass "" for dialog.
	Variable createTable		// 1 = Yes, 2 = No, 0 for dialog.
	Variable createGraph		// 1 = Yes, 2 = No, 0 for dialog.
 
	if (CmpStr(extension,"")==0 || createTable<1 || createTable>2 || createGraph<1 || createGraph>2)
		if (LoadAllTekQueryFilesDialog(extension, createTable, createGraph) != 0)
			return -1					// User cancelled.
		endif
	endif
 
	if (strlen(pathName) == 0)
		String message
 
		if (CmpStr(extension, "????") == 0)
			message = "Select a directory containing Tek Query files with any extension"
		else
			sprintf message, "Select a directory containing Tek Query files with %s extension", extension
		endif
 
		NewPath/O/M=message TekQueryDataPath		// This displays a dialog in which you can select a folder
		if (V_flag != 0)
			return V_flag								// -1 means user canceled
		endif
		pathName = "TekQueryDataPath"
	endif
 
	Variable err = 0
 
	String newWaveName
	String fileName
	Variable i = 0
	do
		fileName = IndexedFile($pathName, i, extension)
		if (strlen(fileName) == 0)
			break										// No more files
		endif
 
		err = LoadTekQueryFile(pathName, fileName, 2, 2, 1, newWaveName)
		if (err != 0)
			return err
		endif
 
		if (createTable == 1)
			if (i == 0)
				Edit $newWaveName
			else
				AppendToTable $newWaveName
			endif
		endif
 
		if (createGraph == 1)
			if (i == 0)
				Display $newWaveName
			else
				AppendToGraph $newWaveName
			endif
		endif
 
		i += 1
	while(1)
 
	return 0
End