communication seems to work now reliably in canonical mode, with full
[sball] / src / sball.c
index 627fc1b..3273480 100644 (file)
@@ -10,6 +10,7 @@
 #include <termios.h>
 #include <sys/select.h>
 #include <sys/time.h>
+#include <sys/ioctl.h>
 
 #define INP_BUF_SZ     256
 
@@ -26,6 +27,9 @@ struct sball {
        char buf[INP_BUF_SZ];
        int len;
 
+       short mot[6];
+       unsigned int keystate;
+
        int (*parse)(struct sball*, int, char*, int);
 };
 
@@ -42,13 +46,16 @@ static int guess_num_buttons(const char *verstr);
 static void make_printable(char *buf, int len);
 static int read_timeout(int fd, char *buf, int bufsz, long tm_usec);
 
+static void print_state(struct sball *sb);
+
+
 struct sball *sball_open(const char *dev)
 {
        int fd, sz;
        char buf[128];
        struct sball *sb = 0;
 
-       if((fd = open(dev, O_RDWR | O_NOCTTY)) == -1) {
+       if((fd = open(dev, O_RDWR | O_NOCTTY | O_NONBLOCK)) == -1) {
                fprintf(stderr, "sball_open: failed to open device: %s: %s\n", dev, strerror(errno));
                return 0;
        }
@@ -64,8 +71,9 @@ struct sball *sball_open(const char *dev)
        if(stty_sball(fd) == -1) {
                goto err;
        }
+       write(fd, "\r@RESET\r", 8);
 
-       if((sz = read_timeout(fd, buf, sizeof buf - 1, 2000000)) > 0 && memcmp(buf, "\r@1", 3) == 0) {
+       if((sz = read_timeout(fd, buf, sizeof buf - 1, 2000000)) > 0 && strstr(buf, "\r@1")) {
                /* we got a response, so it's a spaceball */
                make_printable(buf, sz);
                printf("Spaceball detected: %s\n", buf);
@@ -74,7 +82,7 @@ struct sball *sball_open(const char *dev)
                printf("%d buttons\n", sb->nbuttons);
 
                /* set binary mode and enable automatic data packet sending */
-               write(fd, "\rCB\rMSS\r", 8);
+               write(fd, "\rCB\rMSSV\r", 9);
 
                sb->parse = sball_parsepkt;
                return sb;
@@ -88,7 +96,7 @@ struct sball *sball_open(const char *dev)
 
        if((sz = read_timeout(fd, buf, sizeof buf - 1, 250000)) > 0 && buf[0] == 'v') {
                make_printable(buf, sz);
-               printf("Magellan SpaceMouse detected: %s\n", buf);
+               printf("Magellan SpaceMouse detected:\n%s\n", buf);
 
                sb->nbuttons = guess_num_buttons(buf);
                printf("%d buttons\n", sb->nbuttons);
@@ -137,9 +145,25 @@ int sball_read(struct sball *sb)
        return 0;
 }
 
+int sball_axis(struct sball *sb, int axis)
+{
+       return sb->mot[axis];
+}
+
+unsigned int sball_buttons(struct sball *sb)
+{
+       return sb->keystate;
+}
+
+int sball_num_buttons(struct sball *sb)
+{
+       return sb->nbuttons;
+}
+
 /* Labtec spaceball: 9600 8n1 XON/XOFF */
 static int stty_sball(int fd)
 {
+       int status;
        struct termios term;
 
        if(tcgetattr(fd, &term) == -1) {
@@ -148,12 +172,17 @@ static int stty_sball(int fd)
        }
 
        term.c_oflag = 0;
-       term.c_lflag = 0;
+       term.c_lflag = ICANON;
        term.c_cc[VMIN] = 0;
-       term.c_cc[VTIME] = 1;
+       term.c_cc[VTIME] = 0;
+       term.c_cc[VEOF] = 0;
+       term.c_cc[VEOL] = '\r';
+       term.c_cc[VEOL2] = 0;
+       term.c_cc[VERASE] = 0;
+       term.c_cc[VKILL] = 0;
 
        term.c_cflag = CLOCAL | CREAD | CS8 | HUPCL;
-       term.c_iflag = IGNBRK | IGNPAR | IXON | IXOFF;
+       term.c_iflag = IGNBRK | IGNPAR;
 
        cfsetispeed(&term, B9600);
        cfsetospeed(&term, B9600);
@@ -163,12 +192,17 @@ static int stty_sball(int fd)
                return -1;
        }
 
+       if(ioctl(fd, TIOCMGET, &status) != -1) {
+               status |= TIOCM_DTR | TIOCM_RTS;
+               ioctl(fd, TIOCMSET, &status);
+       }
        return 0;
 }
 
 /* Logicad magellan spacemouse: 9600 8n2 CTS/RTS */
 static int stty_mag(int fd)
 {
+       int status;
        struct termios term;
 
        if(tcgetattr(fd, &term) == -1) {
@@ -177,9 +211,14 @@ static int stty_mag(int fd)
        }
 
        term.c_oflag = 0;
-       term.c_lflag = 0;
+       term.c_lflag = ICANON;
        term.c_cc[VMIN] = 0;
-       term.c_cc[VTIME] = 1;
+       term.c_cc[VTIME] = 0;
+       term.c_cc[VEOF] = 0;
+       term.c_cc[VEOL] = '\r';
+       term.c_cc[VEOL2] = 0;
+       term.c_cc[VERASE] = 0;
+       term.c_cc[VKILL] = 0;
 
        term.c_cflag = CLOCAL | CREAD | CS8 | CSTOPB | HUPCL | CRTSCTS;
        term.c_iflag = IGNBRK | IGNPAR;
@@ -187,11 +226,16 @@ static int stty_mag(int fd)
        cfsetispeed(&term, B9600);
        cfsetospeed(&term, B9600);
 
-       if(tcsetattr(fd, TCSANOW, &term) == -1) {
+       if(tcsetattr(fd, TCSAFLUSH, &term) == -1) {
                perror("sball_open: tcsetattr");
                return -1;
        }
+       tcflush(fd, TCIOFLUSH);
 
+       if(ioctl(fd, TIOCMGET, &status) != -1) {
+               status |= TIOCM_DTR | TIOCM_RTS;
+               ioctl(fd, TIOCMGET, &status);
+       }
        return 0;
 }
 
@@ -222,23 +266,9 @@ static int proc_input(struct sball *sb)
        return 0;
 }
 
-static void print_keystate(unsigned int keystate)
-{
-       int i;
-
-       printf("keystate: ");
-       for(i=0; i<12; i++) {
-               int b = 11 - i;
-               int hex = b < 10 ? b + '0' : b - 10 + 'a';
-               putchar(keystate & (1 << b) ? hex : '-');
-       }
-       putchar('\n');
-}
-
 static int mag_parsepkt(struct sball *sb, int id, char *data, int len)
 {
-       int i, mot[6];
-       unsigned int keystate;
+       int i;
 
        /*printf("magellan packet: %c - %s (%d bytes)\n", (char)id, data, len);*/
 
@@ -249,11 +279,16 @@ static int mag_parsepkt(struct sball *sb, int id, char *data, int len)
                        return -1;
                }
                for(i=0; i<6; i++) {
-                       mot[i] = ((((int)data[0] & 0xf) << 12) | (((int)data[1] & 0xf) << 8) |
+#ifdef SBALL_BIG_ENDIAN
+                       sb->mot[i] = ((((int)data[3] & 0xf) << 12) | (((int)data[2] & 0xf) << 8) |
+                                       (((int)data[1] & 0xf) << 4) | (data[0] & 0xf)) - 0x8000;
+#else
+                       sb->mot[i] = ((((int)data[0] & 0xf) << 12) | (((int)data[1] & 0xf) << 8) |
                                        (((int)data[2] & 0xf) << 4) | (data[3] & 0xf)) - 0x8000;
+#endif
                        data += 4;
                }
-               printf("motion: T %+4d %+4d %+4d  R %+4d %+4d %+4d\n", mot[0], mot[1], mot[2], mot[3], mot[4], mot[5]);
+               print_state(sb);
                break;
 
        case 'k':
@@ -261,8 +296,8 @@ static int mag_parsepkt(struct sball *sb, int id, char *data, int len)
                        fprintf(stderr, "magellan: invalid keyboard pakcet, expected 3 bytes, got: %d\n", len);
                        return -1;
                }
-               keystate = (data[0] & 0xf) | ((data[1] & 0xf) << 4) | (((unsigned int)data[2] & 0xf) << 8);
-               print_keystate(keystate);
+               sb->keystate = (data[0] & 0xf) | ((data[1] & 0xf) << 4) | (((unsigned int)data[2] & 0xf) << 8);
+               print_state(sb);
                break;
 
        case 'e':
@@ -284,9 +319,7 @@ static int mag_parsepkt(struct sball *sb, int id, char *data, int len)
 static int sball_parsepkt(struct sball *sb, int id, char *data, int len)
 {
        int i;
-       unsigned int keystate;
        char c, *rd, *wr;
-       short *mot;
 
        /* decode data packet, replacing escaped values with the correct ones */
        rd = wr = data;
@@ -304,6 +337,7 @@ static int sball_parsepkt(struct sball *sb, int id, char *data, int len)
                                break;
                        case '^':
                                *wr++ = '^';
+                               break;
                        default:
                                fprintf(stderr, "sball decode: ignoring invalid escape code: %xh\n", (unsigned int)c);
                        }
@@ -320,8 +354,18 @@ static int sball_parsepkt(struct sball *sb, int id, char *data, int len)
                        return -1;
                }
 
-               mot = (short*)(data + 2);       /* skip the period */
-               printf("motion: T %+6d %+6d %+6d  R %+6d %+6d %+6d\n", mot[0], mot[1], mot[2], mot[3], mot[4], mot[5]);
+#ifndef SBALL_BIG_ENDIAN
+               for(i=0; i<6; i++) {
+                       data += 2;
+                       c = data[0];
+                       data[0] = data[1];
+                       data[1] = c;
+                       sb->mot[i] = *(short*)data;
+               }
+#else
+               memcpy(sb->mot, data + 2, 12);
+#endif
+               print_state(sb);
                break;
 
        case 'K':
@@ -336,9 +380,9 @@ static int sball_parsepkt(struct sball *sb, int id, char *data, int len)
                 * data[0] bits 0-2 -> buttons 4,5,6
                 * data[0] bit 4 is (2003 pick) -> button 7
                 */
-               keystate = (data[1] & 0xf) | ((data[1] >> 4) & 3) | ((data[0] & 7) << 4) |
+               sb->keystate = (data[1] & 0xf) | ((data[1] >> 4) & 3) | ((data[0] & 7) << 4) |
                        ((data[0] & 0x10) >> 1);
-               print_keystate(keystate);
+               print_state(sb);
                break;
 
        case '.':
@@ -348,19 +392,21 @@ static int sball_parsepkt(struct sball *sb, int id, char *data, int len)
                }
                /* spaceball 4000 key packet */
                sb->flags |= SB4000;
-               /* update orientation flag */
+               /* update orientation flag (actually don't bother) */
+               /*
                if(data[0] & 0x20) {
                        sb->flags |= FLIPXY;
                } else {
                        sb->flags &= ~FLIPXY;
                }
+               */
 
                /* data[1] bits 0-5 -> buttons 0,1,2,3,4,5
                 * data[1] bit 7 -> button 6
                 * data[0] bits 0-4 -> buttons 7,8,9,10,11
                 */
-               keystate = (data[1] & 0x3f) | ((data[1] & 0x80) >> 1) | ((data[0] & 0x1f) << 7);
-               print_keystate(keystate);
+               sb->keystate = (data[1] & 0x3f) | ((data[1] & 0x80) >> 1) | ((data[0] & 0x1f) << 7);
+               print_state(sb);
                break;
 
        case 'E':
@@ -374,6 +420,9 @@ static int sball_parsepkt(struct sball *sb, int id, char *data, int len)
                }
                break;
 
+       case 'M':       /* ignore MSS responses */
+               break;
+
        default:
                /* DEBUG */
                fprintf(stderr, "sball: got '%c' packet:", (char)id);
@@ -427,15 +476,15 @@ static int guess_num_buttons(const char *verstr)
                }
        }
 
-       if(strstr(verstr, "MAGELLAN Version")) {
+       if(strstr(verstr, "MAGELLAN")) {
                return 9; /* magellan spacemouse */
        }
 
-       if(strstr(verstr, "SPACEBALL Version")) {
+       if(strstr(verstr, "SPACEBALL")) {
                return 12; /* spaceball 5000 */
        }
 
-       if(strstr(verstr, "CadMan Version")) {
+       if(strstr(verstr, "CadMan")) {
                return 2;
        }
 
@@ -495,3 +544,28 @@ static int read_timeout(int fd, char *buf, int bufsz, long tm_usec)
        return sz > 0 ? sz : -1;
 }
 
+static void print_motion(short *mot)
+{
+       printf(" T[%+6d %+6d %+6d]  R[%+6d %+6d %+6d]", mot[0], mot[1],
+                       mot[2], mot[3], mot[4], mot[5]);
+}
+
+static void print_keystate(unsigned int keystate)
+{
+       int i;
+
+       for(i=0; i<12; i++) {
+               int b = 11 - i;
+               int hex = b < 10 ? b + '0' : b - 10 + 'a';
+               putchar(keystate & (1 << b) ? hex : '-');
+       }
+}
+
+static void print_state(struct sball *sb)
+{
+       print_motion(sb->mot);
+       printf("  B[");
+       print_keystate(sb->keystate);
+       printf("]\r");
+       fflush(stdout);
+}