X-Git-Url: http://git.zerfleddert.de/cgi-bin/gitweb.cgi/rigol/blobdiff_plain/610d5b65ca67c740ee40fc044ce502ddf3cba27a..58a9e2764568da62bbb285f8892d68d028674d4a:/rigol.c diff --git a/rigol.c b/rigol.c index 24a4a10..ebf01fb 100644 --- a/rigol.c +++ b/rigol.c @@ -13,47 +13,19 @@ rmmod uhci_hcd; modprobe uhci_hcd #include #include #include +#include +#include #include #include +#include +#include #include #include +#include "usbtmc.h" +#include "png.h" -//This routine locates a scope by VID/PID and returns an opened handle to it. -usb_dev_handle *find_scope() { - struct usb_bus *bus; - struct usb_device *dev=NULL; - usb_find_busses(); - usb_find_devices(); - for (bus=usb_busses; bus; bus=bus->next) { - for (dev=bus->devices; dev; dev=dev->next) { - //fprintf(stderr,"Prod/dev: %04X:%04X\n",dev->descriptor.idVendor,dev->descriptor.idProduct); - if (dev->descriptor.idVendor==0x400 && dev->descriptor.idProduct==0x5dc) { - return usb_open(dev); - } - } - } - return NULL; -} - -//Helper-routine: Convert a little-endian 4-byte word to an int -void int2chars(unsigned char *buff,unsigned int a) { - buff[3]=(a>>24)&0xff; - buff[2]=(a>>16)&0xff; - buff[1]=(a>>8)&0xff; - buff[0]=(a)&0xff; -} - -//Helper-routine: Convert an int to little-endian 4-byte word -unsigned int chars2int(unsigned char *buff) { - unsigned int a; - a=buff[3]<<24; - a+=buff[2]<<16; - a+=buff[1]<<8; - a+=buff[0]; - return a; -} #define MIN(a,b) (((a)<(b))?(a):(b)) @@ -86,88 +58,6 @@ void printb (unsigned char *pkt, int len) } } -//Send a scpi-command to the scope. The response goes into the buffer -//called resp, with a size of resplen. If resp==NULL, no response -//is requested. -int sendscpi(usb_dev_handle *dev, char* cmd, - unsigned char *resp, int resplen) { - unsigned char *buff; - int len,r,i; - int cmdlen = strlen(cmd); - static unsigned char seq=0; - - - buff=malloc(0x40); - seq++; - buff[0]=1; //func - buff[1]=seq; buff[2]=~seq; //nseq - buff[3]=0; - int2chars(buff+4, cmdlen); - buff[8]=1; - buff[9]=0x37; - buff[10]=0x39; - buff[11]=0x39; - //fprintf(stderr,"Writing header len=%d\n", cmdlen); - //printb(buff,12); - r=usb_bulk_write(dev, 1, (char*)buff, 12, 1000); - //fprintf(stderr,"%i bytes written. Writing cmd\n",r); - //printb(cmd, cmdlen); - r=usb_bulk_write(dev, 1, cmd, cmdlen, 1000); - //fprintf(stderr,"%i bytes written.\n",r); - if (resp != NULL && resplen != 0) { - //send read command - buff[0]=2; //func - seq++; - buff[1]=seq; buff[2]=~seq; //nseq - int2chars(buff+4,0x40); - buff[8]=1; - buff[9]=0xA; - buff[10]=0; - buff[11]=0; - //fprintf(stderr,"Writing resp req header\n"); - //printb(buff,12); - r=usb_bulk_write(dev, 1, (char*)buff, 12, 1000); - //fprintf(stderr,"%i bytes written. Reading response hdr\n",r); - r=usb_bulk_read(dev, 2, (char*)buff, 0x40, 1000); - //printb(buff,r); - len=chars2int(buff+4); - //fprintf(stderr,"%i bytes read. Resplen=%i\n",r,len); - for (i=0; i<(r-12); i++) { - if (i 0x40-12) { - //fprintf(stderr," Reading response:\n"); - if (resplen 0) { sscanf ((char*)buf, "%lf", &temp); return temp; @@ -257,7 +147,7 @@ void do_get_buf (struct usb_dev_handle *sc) unsigned char data[512*1024]; double sampfreq; - sendscpi (sc, ":STOP", NULL, 0); + usbtmc_sendscpi (sc, ":STOP", NULL, 0); sampfreq = get_float_from_scope (sc, ":ACQ:SAMP?"); @@ -265,7 +155,7 @@ void do_get_buf (struct usb_dev_handle *sc) sprintf (buf, ":TIM:SCAL %.15f", 50 / sampfreq); printf ("sending scale cmd: %s\n", buf); - sendscpi (sc, buf, NULL, 0); + usbtmc_sendscpi (sc, buf, NULL, 0); sleep (1); @@ -273,9 +163,9 @@ void do_get_buf (struct usb_dev_handle *sc) for (i=-254*1024;i< 254*1024;i += 600) { sprintf (buf, ":TIM:OFFSET %.15f", i / sampfreq); printf ("Sending offset cmd: %s\n", buf); - sendscpi (sc, buf, NULL, 0); + usbtmc_sendscpi (sc, buf, NULL, 0); - l = sendscpi(sc, ":WAV:DATA? CHANEL1", ch1, 1024); + l = usbtmc_sendscpi(sc, ":WAV:DATA? CHANEL1", ch1, 1024); if (l != 1024) { printf ("hmm. didnt' get 1024 bytes. \n"); @@ -290,10 +180,77 @@ void do_get_buf (struct usb_dev_handle *sc) fwrite (data, bp, 1, fp); fclose (fp); - sendscpi (sc, ":TIM:OFFSET 0", NULL, 0); + usbtmc_sendscpi (sc, ":TIM:OFFSET 0", NULL, 0); } +void do_get_screen(struct usb_dev_handle *sc) +{ + unsigned char screen[320*234]; + time_t lt; + char filename[256]; + unsigned char *png; + int imglen; + int ret; + int l; + int fd; + pid_t display; + /* Hide "RMT" from screen */ + l = usbtmc_sendscpi(sc, ":KEY:LOCK DISABLE", NULL, 0); + usleep(20000); + + l = usbtmc_sendscpi(sc, ":LCD:DATA?", screen, sizeof(screen)); + + if (l != sizeof(screen)) { + printf ("hmm. didnt' get %d bytes, but %d\n\n", sizeof(screen), l); + } + + png = lcd2png(screen, &imglen); + + lt = time(NULL); + strftime(filename, sizeof(filename), "screen_%Y%m%d-%H%M%S.png", localtime(<)); + fd=open(filename, O_CREAT|O_WRONLY, 0644); + if (fd == -1) { + perror("open"); + exit(EXIT_FAILURE); + } + + while(imglen > 0) { + ret = write(fd, png, imglen); + if (ret == -1) { + perror("write"); + exit(EXIT_FAILURE); + } + imglen -= ret; + } + close(fd); + free(png); + + printf("Waveform saved as %s\n", filename); + + display = fork(); + switch(display) { + case 0: + execlp("display", "display", filename, NULL); + exit(0); + break; + case -1: + perror("fork"); + break; + default: + break; + } +} + +void child_reaper(int sig) +{ + pid_t child; + + do { + child = waitpid(-1, NULL, WNOHANG); + } while(child > 0); + +} int main(int argc, char **argv) { @@ -301,19 +258,24 @@ int main(int argc, char **argv) char *scpi; unsigned char *buff; int l; - //Init libusb - usb_init(); - //Locate and open the scope - sc=find_scope(); - if (!sc) { - printf("No scope found.\n"); - exit(1); - } else { - printf("Scope found.\n"); - } + struct sigaction act; + //Initialize scope - initscope(sc); + sc = usbtmc_initscope(); buff = malloc (1024*1024); + if (buff == NULL) { + perror("malloc"); + exit(EXIT_FAILURE); + } + + bzero(&act, sizeof(act)); + act.sa_handler = child_reaper; + act.sa_flags = SA_NOCLDSTOP|SA_RESTART; + if (sigaction(SIGCHLD, &act, NULL) == -1) { + perror("sigaction"); + exit(EXIT_FAILURE); + } + while (1) { scpi = readline ("> "); @@ -330,28 +292,32 @@ int main(int argc, char **argv) do_plot (sc); continue; } - if (strncmp (scpi, "databuf", 4) == 0) { + if (strncmp (scpi, "databuf", 7) == 0) { do_get_buf (sc); continue; } + if (strncmp (scpi, "screen", 6) == 0) { + do_get_screen (sc); + continue; + } l = strlen (scpi); //printf ("got buf(%d): ", l); //printb (scpi, l+2); if (strchr (scpi, '?')) { //printf ("Expecting reply\n"); - l = sendscpi(sc, scpi, buff, 1024*1024); + l = usbtmc_sendscpi(sc, scpi, buff, 1024*1024); //printf ("Got replylen = %d.\n", l); buff[l] = 0; //zero-terminate printb (buff, l); } else { //printf ("No reply expected\n"); - l=sendscpi(sc,scpi,NULL,0); + l=usbtmc_sendscpi(sc,scpi,NULL,0); } free (scpi); } //Disable keylock, so the user doesn't have to press the 'force'-button - l=sendscpi(sc, ":KEY:LOCK DISABLE",NULL,0); + l=usbtmc_sendscpi(sc, ":KEY:LOCK DISABLE",NULL,0); //Free up and exit usb_release_interface(sc,0);