[rigol] / scope.c

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <stdint.h>
#include <string.h>
#include <strings.h>

#include "scope.h"
#include "usbtmc.h"

/* Just USB for now... */
int sendscpi(struct scope* sc, char* cmd, unsigned char *resp, int resplen)
{
	return usbtmc_sendscpi(sc, cmd, resp, resplen);
}

void closescope(struct scope* sc)
{
	return usbtmc_close(sc);
}

void claimscope(struct scope* sc)
{
	return usbtmc_claim(sc);
}

void releasescope(struct scope* sc)
{
	/* Disable keylock, so the user doesn't have to press the 'force'-button */
	sendscpi(sc, ":KEY:LOCK DISABLE",NULL,0); 
	return usbtmc_release(sc);
}

void resetscope(struct scope* sc)
{
	return usbtmc_reset(sc);
}

struct scope* initscope(void)
{
	struct scope *sc;

	sc = usbtmc_initscope();

	if (!sc) {
		printf("No scope found.\n");
		exit(EXIT_FAILURE);
	}

	claimscope(sc);
	sendscpi(sc, "*IDN?", (unsigned char*)sc->idn, sizeof(sc->idn)); 
	releasescope(sc);

	printf("Scope found (%s)\n", sc->idn);

	return sc;
}

char *scope_idn(struct scope *sc)
{
	return sc->idn;
}

#define COPY_SCOPE_STRING(sc, cmd, dst) { \
						char *buf; \
						buf = scope_get_string(sc, cmd, sizeof(dst)); \
						if (buf) { \
							strcpy(dst, buf); \
							free(buf); \
						}\
					}

char *scope_get_string(struct scope *sc, char *cmd, int maxlen)
{
	unsigned char *buf;
	int res;

	buf = malloc(maxlen+1);
	if (buf == NULL) {
		perror("malloc(scope_get_strings)");
		exit(EXIT_FAILURE);
	}

	res = sendscpi(sc, cmd, buf, maxlen);
	if (res < 0) {
		fprintf(stderr, "Command %s failed with %d\n", cmd, res);
		free(buf);
		return NULL;
	}

	buf[res] = 0;

	return (char*)buf;
}

int scope_get_truth_value(struct scope *sc, char *cmd)
{
	char buf[128];
	int res;

	bzero(buf, sizeof(buf));
	res = sendscpi(sc, cmd, (unsigned char*)buf, sizeof(buf)-1);
	if (res < 0) {
		fprintf(stderr, "Command %s failed with %d\n", cmd, res);
		return 0;
	}

	printf("%s %s\n", cmd, buf);

	if (strcasecmp(buf, "on") == 0) {
		return 1;
	} else if (strcasecmp(buf, "enable") == 0) {
		return 1;
	}

	return 0;
}

int scope_get_int(struct scope *sc, char *cmd)
{
	char buf[128];
	int res;

	bzero(buf, sizeof(buf));
	res = sendscpi(sc, cmd, (unsigned char*)buf, sizeof(buf)-1);
	if (res < 0) {
		fprintf(stderr, "Command %s failed with %d\n", cmd, res);
		return 0;
	}

	return atoi(buf);
}

double scope_get_double(struct scope *sc, char*cmd)
{
	char buf[128];
	int res;
	double ret;

	bzero(buf, sizeof(buf));
	res = sendscpi(sc, cmd, (unsigned char*)buf, sizeof(buf)-1);
	if (res < 0) {
		fprintf(stderr, "Command %s failed with %d\n", cmd, res);
		return 0.0;
	}

	ret = strtod(buf, NULL);

	return ret;
}

void update_scope_measurements(struct scope *sc)
{
	sc->status.measure.ch1.vpp = scope_get_double(sc, ":MEAS:VPP? CHAN1");
	sc->status.measure.ch1.vmax = scope_get_double(sc, ":MEAS:VMAX? CHAN1");
	sc->status.measure.ch1.vmin = scope_get_double(sc, ":MEAS:VMIN? CHAN1");
	sc->status.measure.ch1.vamplitude = scope_get_double(sc, ":MEAS:VAMP? CHAN1");
	sc->status.measure.ch1.vtop = scope_get_double(sc, ":MEAS:VTOP? CHAN1");
	sc->status.measure.ch1.vbase = scope_get_double(sc, ":MEAS:VBAS? CHAN1");
	sc->status.measure.ch1.vaverage = scope_get_double(sc, ":MEAS:VAV? CHAN1");
	sc->status.measure.ch1.vrms = scope_get_double(sc, ":MEAS:VRMS? CHAN1");
	sc->status.measure.ch1.overshoot = scope_get_double(sc, ":MEAS:OVER? CHAN1");
	sc->status.measure.ch1.preshoot = scope_get_double(sc, ":MEAS:PRES? CHAN1");
	sc->status.measure.ch1.frequency = scope_get_double(sc, ":MEAS:FREQ? CHAN1");
	sc->status.measure.ch1.risetime = scope_get_double(sc, ":MEAS:RIS? CHAN1");
	sc->status.measure.ch1.falltime = scope_get_double(sc, ":MEAS:FALL? CHAN1");
	sc->status.measure.ch1.period = scope_get_double(sc, ":MEAS:PER? CHAN1");
	sc->status.measure.ch1.pwidth = scope_get_double(sc, ":MEAS:PWID? CHAN1");
	sc->status.measure.ch1.nwidth = scope_get_double(sc, ":MEAS:NWID? CHAN1");
	sc->status.measure.ch1.pdutycycle = scope_get_double(sc, ":MEAS:PDUT? CHAN1");
	sc->status.measure.ch1.ndutycycle = scope_get_double(sc, ":MEAS:NDUT? CHAN1");
	sc->status.measure.ch1.pdelay = scope_get_double(sc, ":MEAS:PDEL? CHAN1");
	sc->status.measure.ch1.ndelay = scope_get_double(sc, ":MEAS:NDEL? CHAN1");

	sc->status.measure.ch2.vpp = scope_get_double(sc, ":MEAS:VPP? CHAN2");
	sc->status.measure.ch2.vmax = scope_get_double(sc, ":MEAS:VMAX? CHAN2");
	sc->status.measure.ch2.vmin = scope_get_double(sc, ":MEAS:VMIN? CHAN2");
	sc->status.measure.ch2.vamplitude = scope_get_double(sc, ":MEAS:VAMP? CHAN2");
	sc->status.measure.ch2.vtop = scope_get_double(sc, ":MEAS:VTOP? CHAN2");
	sc->status.measure.ch2.vbase = scope_get_double(sc, ":MEAS:VBAS? CHAN2");
	sc->status.measure.ch2.vaverage = scope_get_double(sc, ":MEAS:VAV? CHAN2");
	sc->status.measure.ch2.vrms = scope_get_double(sc, ":MEAS:VRMS? CHAN2");
	sc->status.measure.ch2.overshoot = scope_get_double(sc, ":MEAS:OVER? CHAN2");
	sc->status.measure.ch2.preshoot = scope_get_double(sc, ":MEAS:PRES? CHAN2");
	sc->status.measure.ch2.frequency = scope_get_double(sc, ":MEAS:FREQ? CHAN2");
	sc->status.measure.ch2.risetime = scope_get_double(sc, ":MEAS:RIS? CHAN2");
	sc->status.measure.ch2.falltime = scope_get_double(sc, ":MEAS:FALL? CHAN2");
	sc->status.measure.ch2.period = scope_get_double(sc, ":MEAS:PER? CHAN2");
	sc->status.measure.ch2.pwidth = scope_get_double(sc, ":MEAS:PWID? CHAN2");
	sc->status.measure.ch2.nwidth = scope_get_double(sc, ":MEAS:NWID? CHAN2");
	sc->status.measure.ch2.pdutycycle = scope_get_double(sc, ":MEAS:PDUT? CHAN2");
	sc->status.measure.ch2.ndutycycle = scope_get_double(sc, ":MEAS:NDUT? CHAN2");
	sc->status.measure.ch2.pdelay = scope_get_double(sc, ":MEAS:PDEL? CHAN2");
	sc->status.measure.ch2.ndelay = scope_get_double(sc, ":MEAS:NDEL? CHAN2");

	sc->status.measure.total = scope_get_truth_value(sc, ":MEAS:TOT?");
	COPY_SCOPE_STRING(sc, ":MEAS:SOUR?", sc->status.measure.source);

}

int update_scope_status(struct scope *sc)
{
	bzero(&(sc->status), sizeof(sc->status));

	COPY_SCOPE_STRING(sc, ":INFO:LANG?", sc->status.system.lang);

	sc->status.system.counter_enabled = scope_get_truth_value(sc, ":COUN:ENAB?");
	sc->status.system.beep_enabled = scope_get_truth_value(sc, ":BEEP:ENAB?");

	sc->status.keyboard.key_lock = scope_get_truth_value(sc, ":KEY:LOCK?");

	update_scope_measurements(sc);

	COPY_SCOPE_STRING(sc, ":DISP:TYPE?", sc->status.display.type);
	COPY_SCOPE_STRING(sc, ":DISP:GRID?", sc->status.display.grid);
	sc->status.display.persist = scope_get_truth_value(sc, ":DISP:PERS?");
	COPY_SCOPE_STRING(sc, ":DISP:MNUD?", sc->status.display.mnudisplay);
	sc->status.display.mnustatus = scope_get_truth_value(sc, ":DISP:MNUS?");
	COPY_SCOPE_STRING(sc, ":DISP:SCR?", sc->status.display.screen);
	sc->status.display.brightness = scope_get_int(sc, ":DISP:BRIG?");
	sc->status.display.intensity = scope_get_int(sc, ":DISP:INT?");

	COPY_SCOPE_STRING(sc, ":ACQ:TYPE?", sc->status.acquire.type);
	COPY_SCOPE_STRING(sc, ":ACQ:MODE?", sc->status.acquire.mode);

	sc->status.acquire.averages = scope_get_int(sc, ":ACQ:AVER?");
	sc->status.acquire.srate_ch1 = scope_get_double(sc, ":ACQ:SAMP? CHAN1");
	sc->status.acquire.srate_ch2 = scope_get_double(sc, ":ACQ:SAMP? CHAN2");
	sc->status.acquire.srate_digital = scope_get_double(sc, ":ACQ:SAMP? DIGITAL");

	COPY_SCOPE_STRING(sc, ":TIM:MODE?", sc->status.timebase.mode);
	sc->status.timebase.offset = scope_get_double(sc, ":TIM:OFFS?");
	sc->status.timebase.delayed_offset = scope_get_double(sc, ":TIM:DEL:OFFS?");
	sc->status.timebase.scale = scope_get_double(sc, ":TIM:SCAL?");
	COPY_SCOPE_STRING(sc, ":TIM:FORM?", sc->status.timebase.format);

	return 0;
}
Commit	Line	Data
713be7a4 MG	1	#include <stdio.h>
	2	#include <stdlib.h>
	3	#include <unistd.h>
7df76f49	4	#include <stdint.h>
78fb0984 MG	5	#include <string.h>
78fb0984 MG	6	#include <strings.h>
713be7a4	7
713be7a4	8	#include "scope.h"
7906b395	9	#include "usbtmc.h"
713be7a4 MG	10
	11	/* Just USB for now... */
	12	int sendscpi(struct scope* sc, char* cmd, unsigned char *resp, int resplen)
	13	{
7906b395	14	return usbtmc_sendscpi(sc, cmd, resp, resplen);
713be7a4 MG	15	}
	16
	17	void closescope(struct scope* sc)
	18	{
7906b395	19	return usbtmc_close(sc);
713be7a4 MG	20	}
	21
	22	void claimscope(struct scope* sc)
	23	{
7906b395	24	return usbtmc_claim(sc);
713be7a4 MG	25	}
	26
	27	void releasescope(struct scope* sc)
	28	{
10ca7ea8	29	/* Disable keylock, so the user doesn't have to press the 'force'-button */
713be7a4	30	sendscpi(sc, ":KEY:LOCK DISABLE",NULL,0);
7906b395	31	return usbtmc_release(sc);
713be7a4 MG	32	}
713be7a4 MG	33
10ca7ea8 MG	34	void resetscope(struct scope* sc)
	35	{
	36	return usbtmc_reset(sc);
	37	}
	38
713be7a4 MG	39	struct scope* initscope(void)
713be7a4 MG	40	{
713be7a4 MG	41	struct scope *sc;
713be7a4 MG	42
7906b395	43	sc = usbtmc_initscope();
713be7a4	44
7906b395	45	if (!sc) {
2999345d MG	46	printf("No scope found.\n");
	47	exit(EXIT_FAILURE);
	48	}
713be7a4	49
713be7a4 MG	50	claimscope(sc);
	51	sendscpi(sc, "IDN?", (unsigned char)sc->idn, sizeof(sc->idn));
	52	releasescope(sc);
	53
	54	printf("Scope found (%s)\n", sc->idn);
	55
	56	return sc;
	57	}
	58
	59	char scope_idn(struct scope sc)
	60	{
	61	return sc->idn;
	62	}
78fb0984 MG	63
	64	#define COPY_SCOPE_STRING(sc, cmd, dst) { \
	65	char *buf; \
	66	buf = scope_get_string(sc, cmd, sizeof(dst)); \
	67	if (buf) { \
	68	strcpy(dst, buf); \
	69	free(buf); \
	70	}\
	71	}
	72
	73	char scope_get_string(struct scope sc, char *cmd, int maxlen)
	74	{
	75	unsigned char *buf;
	76	int res;
	77
759a1836	78	buf = malloc(maxlen+1);
78fb0984 MG	79	if (buf == NULL) {
	80	perror("malloc(scope_get_strings)");
	81	exit(EXIT_FAILURE);
	82	}
	83
	84	res = sendscpi(sc, cmd, buf, maxlen);
	85	if (res < 0) {
	86	fprintf(stderr, "Command %s failed with %d\n", cmd, res);
	87	free(buf);
	88	return NULL;
	89	}
	90
	91	buf[res] = 0;
	92
	93	return (char*)buf;
	94	}
	95
	96	int scope_get_truth_value(struct scope sc, char cmd)
	97	{
	98	char buf[128];
	99	int res;
	100
	101	bzero(buf, sizeof(buf));
	102	res = sendscpi(sc, cmd, (unsigned char*)buf, sizeof(buf)-1);
	103	if (res < 0) {
	104	fprintf(stderr, "Command %s failed with %d\n", cmd, res);
	105	return 0;
	106	}
	107
	108	printf("%s %s\n", cmd, buf);
	109
	110	if (strcasecmp(buf, "on") == 0) {
	111	return 1;
	112	} else if (strcasecmp(buf, "enable") == 0) {
	113	return 1;
	114	}
	115
	116	return 0;
	117	}
	118
	119	int scope_get_int(struct scope sc, char cmd)
	120	{
	121	char buf[128];
	122	int res;
	123
	124	bzero(buf, sizeof(buf));
	125	res = sendscpi(sc, cmd, (unsigned char*)buf, sizeof(buf)-1);
	126	if (res < 0) {
	127	fprintf(stderr, "Command %s failed with %d\n", cmd, res);
	128	return 0;
	129	}
	130
	131	return atoi(buf);
	132	}
	133
	134	double scope_get_double(struct scope sc, charcmd)
	135	{
	136	char buf[128];
	137	int res;
	138	double ret;
	139
	140	bzero(buf, sizeof(buf));
	141	res = sendscpi(sc, cmd, (unsigned char*)buf, sizeof(buf)-1);
	142	if (res < 0) {
143	fprintf(stderr, "Command %s failed with %d\n", cmd, res);
144	return 0.0;
145	}
146
147	ret = strtod(buf, NULL);
148
149	return ret;
150	}
151
adb5a679 MG	152	void update_scope_measurements(struct scope *sc)
	153	{
	154	sc->status.measure.ch1.vpp = scope_get_double(sc, ":MEAS:VPP? CHAN1");
	155	sc->status.measure.ch1.vmax = scope_get_double(sc, ":MEAS:VMAX? CHAN1");
	156	sc->status.measure.ch1.vmin = scope_get_double(sc, ":MEAS:VMIN? CHAN1");
	157	sc->status.measure.ch1.vamplitude = scope_get_double(sc, ":MEAS:VAMP? CHAN1");
	158	sc->status.measure.ch1.vtop = scope_get_double(sc, ":MEAS:VTOP? CHAN1");
	159	sc->status.measure.ch1.vbase = scope_get_double(sc, ":MEAS:VBAS? CHAN1");
	160	sc->status.measure.ch1.vaverage = scope_get_double(sc, ":MEAS:VAV? CHAN1");
	161	sc->status.measure.ch1.vrms = scope_get_double(sc, ":MEAS:VRMS? CHAN1");
	162	sc->status.measure.ch1.overshoot = scope_get_double(sc, ":MEAS:OVER? CHAN1");
	163	sc->status.measure.ch1.preshoot = scope_get_double(sc, ":MEAS:PRES? CHAN1");
	164	sc->status.measure.ch1.frequency = scope_get_double(sc, ":MEAS:FREQ? CHAN1");
	165	sc->status.measure.ch1.risetime = scope_get_double(sc, ":MEAS:RIS? CHAN1");
	166	sc->status.measure.ch1.falltime = scope_get_double(sc, ":MEAS:FALL? CHAN1");
	167	sc->status.measure.ch1.period = scope_get_double(sc, ":MEAS:PER? CHAN1");
	168	sc->status.measure.ch1.pwidth = scope_get_double(sc, ":MEAS:PWID? CHAN1");
	169	sc->status.measure.ch1.nwidth = scope_get_double(sc, ":MEAS:NWID? CHAN1");
	170	sc->status.measure.ch1.pdutycycle = scope_get_double(sc, ":MEAS:PDUT? CHAN1");
	171	sc->status.measure.ch1.ndutycycle = scope_get_double(sc, ":MEAS:NDUT? CHAN1");
	172	sc->status.measure.ch1.pdelay = scope_get_double(sc, ":MEAS:PDEL? CHAN1");
	173	sc->status.measure.ch1.ndelay = scope_get_double(sc, ":MEAS:NDEL? CHAN1");
	174
	175	sc->status.measure.ch2.vpp = scope_get_double(sc, ":MEAS:VPP? CHAN2");
	176	sc->status.measure.ch2.vmax = scope_get_double(sc, ":MEAS:VMAX? CHAN2");
	177	sc->status.measure.ch2.vmin = scope_get_double(sc, ":MEAS:VMIN? CHAN2");
	178	sc->status.measure.ch2.vamplitude = scope_get_double(sc, ":MEAS:VAMP? CHAN2");
	179	sc->status.measure.ch2.vtop = scope_get_double(sc, ":MEAS:VTOP? CHAN2");
	180	sc->status.measure.ch2.vbase = scope_get_double(sc, ":MEAS:VBAS? CHAN2");
	181	sc->status.measure.ch2.vaverage = scope_get_double(sc, ":MEAS:VAV? CHAN2");
	182	sc->status.measure.ch2.vrms = scope_get_double(sc, ":MEAS:VRMS? CHAN2");
	183	sc->status.measure.ch2.overshoot = scope_get_double(sc, ":MEAS:OVER? CHAN2");
	184	sc->status.measure.ch2.preshoot = scope_get_double(sc, ":MEAS:PRES? CHAN2");
	185	sc->status.measure.ch2.frequency = scope_get_double(sc, ":MEAS:FREQ? CHAN2");
	186	sc->status.measure.ch2.risetime = scope_get_double(sc, ":MEAS:RIS? CHAN2");
	187	sc->status.measure.ch2.falltime = scope_get_double(sc, ":MEAS:FALL? CHAN2");
	188	sc->status.measure.ch2.period = scope_get_double(sc, ":MEAS:PER? CHAN2");
	189	sc->status.measure.ch2.pwidth = scope_get_double(sc, ":MEAS:PWID? CHAN2");
	190	sc->status.measure.ch2.nwidth = scope_get_double(sc, ":MEAS:NWID? CHAN2");
	191	sc->status.measure.ch2.pdutycycle = scope_get_double(sc, ":MEAS:PDUT? CHAN2");
	192	sc->status.measure.ch2.ndutycycle = scope_get_double(sc, ":MEAS:NDUT? CHAN2");
	193	sc->status.measure.ch2.pdelay = scope_get_double(sc, ":MEAS:PDEL? CHAN2");
	194	sc->status.measure.ch2.ndelay = scope_get_double(sc, ":MEAS:NDEL? CHAN2");
	195
	196	sc->status.measure.total = scope_get_truth_value(sc, ":MEAS:TOT?");
	197	COPY_SCOPE_STRING(sc, ":MEAS:SOUR?", sc->status.measure.source);
	198
	199	}
	200
78fb0984 MG	201	int update_scope_status(struct scope *sc)
	202	{
	203	bzero(&(sc->status), sizeof(sc->status));
	204
	205	COPY_SCOPE_STRING(sc, ":INFO:LANG?", sc->status.system.lang);
	206
	207	sc->status.system.counter_enabled = scope_get_truth_value(sc, ":COUN:ENAB?");
	208	sc->status.system.beep_enabled = scope_get_truth_value(sc, ":BEEP:ENAB?");
	209
	210	sc->status.keyboard.key_lock = scope_get_truth_value(sc, ":KEY:LOCK?");
	211
adb5a679 MG	212	update_scope_measurements(sc);
	213
	214	COPY_SCOPE_STRING(sc, ":DISP:TYPE?", sc->status.display.type);
	215	COPY_SCOPE_STRING(sc, ":DISP:GRID?", sc->status.display.grid);
	216	sc->status.display.persist = scope_get_truth_value(sc, ":DISP:PERS?");
	217	COPY_SCOPE_STRING(sc, ":DISP:MNUD?", sc->status.display.mnudisplay);
	218	sc->status.display.mnustatus = scope_get_truth_value(sc, ":DISP:MNUS?");
	219	COPY_SCOPE_STRING(sc, ":DISP:SCR?", sc->status.display.screen);
	220	sc->status.display.brightness = scope_get_int(sc, ":DISP:BRIG?");
	221	sc->status.display.intensity = scope_get_int(sc, ":DISP:INT?");
	222
78fb0984 MG	223	COPY_SCOPE_STRING(sc, ":ACQ:TYPE?", sc->status.acquire.type);
	224	COPY_SCOPE_STRING(sc, ":ACQ:MODE?", sc->status.acquire.mode);
	225
	226	sc->status.acquire.averages = scope_get_int(sc, ":ACQ:AVER?");
adb5a679 MG	227	sc->status.acquire.srate_ch1 = scope_get_double(sc, ":ACQ:SAMP? CHAN1");
adb5a679 MG	228	sc->status.acquire.srate_ch2 = scope_get_double(sc, ":ACQ:SAMP? CHAN2");
78fb0984 MG	229	sc->status.acquire.srate_digital = scope_get_double(sc, ":ACQ:SAMP? DIGITAL");
	230
	231	COPY_SCOPE_STRING(sc, ":TIM:MODE?", sc->status.timebase.mode);
	232	sc->status.timebase.offset = scope_get_double(sc, ":TIM:OFFS?");
	233	sc->status.timebase.delayed_offset = scope_get_double(sc, ":TIM:DEL:OFFS?");
	234	sc->status.timebase.scale = scope_get_double(sc, ":TIM:SCAL?");
	235	COPY_SCOPE_STRING(sc, ":TIM:FORM?", sc->status.timebase.format);
	236
	237	return 0;
	238	}