X-Git-Url: http://git.zerfleddert.de/cgi-bin/gitweb.cgi/rigol/blobdiff_plain/0b7a29d92ef3656b4c740b313f901a9d62f593a3..659f6954a480994415f72580df61635fba3d877f:/rigol.c

diff --git a/rigol.c b/rigol.c
index 22a5cea..6ae9d73 100644
--- a/rigol.c
+++ b/rigol.c
@@ -135,11 +135,15 @@ int main(int argc, char **argv)
 			continue;
 		}
 		if (strncmp (scpi, "databuf", 7) == 0) {
-			do_get_buf (sc);
+			do_get_buf(sc);
 			continue;
 		}
 		if (strncmp (scpi, "screen", 6) == 0) {
-			do_get_screen (sc);
+			do_get_screen(sc);
+			continue;
+		}
+		if (strncmp (scpi, "display", 7) == 0) {
+			do_display_screen(sc);
 			continue;
 		}
 
@@ -161,8 +165,6 @@ int main(int argc, char **argv)
 	//Disable keylock, so the user doesn't have to press the 'force'-button
 	l=usbtmc_sendscpi(sc, ":KEY:LOCK DISABLE",NULL,0); 
 
-	//Free up and exit
-	usb_release_interface(sc,0);
-	usb_close(sc);
+	usbtmc_close(sc);
 	return 0;
 }