Changeset 1428

Timestamp:

Apr 30, 2009 9:20:06 AM (15 years ago)

Author:

gah

Message:

fix image hardcopy transfer

File:

• trunk/packages/vizservers/pymolproxy/pymolproxy.c (modified) (11 diffs)

trunk/packages/vizservers/pymolproxy/pymolproxy.c

@@ -52 +52 @@
  */
 
+#include <assert.h>
+#include <ctype.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <getopt.h>
+#include <net/if.h>
+#include <netinet/in.h>
+#include <poll.h>
+#include <signal.h>
 #include <stdio.h>
-#include <time.h>
-#include <assert.h>
-#include <unistd.h>
 #include <stdlib.h>
 #include <string.h>
-#include <signal.h>
+#include <sys/ioctl.h>
+#include <sys/select.h>
+#include <sys/socket.h>
 #include <sys/stat.h>
-#include <sys/types.h>
-#include <sys/socket.h>
-#include <sys/wait.h>
-#include <sys/select.h>
-#include <poll.h>
 #include <sys/time.h>
 #include <sys/times.h>
-#include <fcntl.h>
-#include <netinet/in.h>
-#include <getopt.h>
-#include <errno.h>
-#include <sys/ioctl.h>
-#include <net/if.h>
+#include <sys/types.h>
+#include <sys/wait.h>
+#include <time.h>
+#include <unistd.h>
 #include <tcl.h>
 
@@ -354 +355 @@
         return proxyPtr->status;
     }
-    trace("Entering Expect(match=%s, maxSize=%d)\n", match, maxSize);
+    trace("Entering Expect(want=%s, maxSize=%d)\n", match, maxSize);
     c = match[0];
     length = strlen(match);
@@ -370 +371 @@
                 memcpy(out, line, nBytes);
                 clear_error(proxyPtr);
-                trace("Leaving Expect: match is (%.*s)\n", nBytes, out);
+                trace("Leaving Expect: got (%.*s)\n", nBytes, out);
                 return BUFFER_OK;
             }
@@ -636 +637 @@
 Pymol(PymolProxy *proxyPtr, char *format, ...)
 {
+    va_list ap;
+    char buffer[BUFSIZ];
+    char expect[BUFSIZ];
+    int result;
+    ssize_t nWritten;
+    size_t length;
+    char *p;
 
     if (proxyPtr->error) {
         return TCL_ERROR;
     }
-    {
-        va_list ap;
-        char cmd[BUFSIZ];
-        ssize_t nWritten;
-        size_t length;
-
-        va_start(ap, format);
-        vsnprintf(cmd, BUFSIZ-1, format, ap);
-        va_end(ap);
-        
-        trace("to-pymol>(%s)", cmd);
-        
-        /* Write the command out to the server. */
-        length = strlen(cmd);
-        nWritten = write(proxyPtr->serverInput, cmd, length);
-        if (nWritten != length) {
-            trace("short write to pymol (wrote=%d, should have been %d)",
-                  nWritten, length);
-        }
-    }
-    {
-        char out[BUFSIZ];
-        int result;
-
-        /* Now wait for the "PyMOL>" prompt. */
-        result = Expect(proxyPtr, "PyMOL>", out, BUFSIZ);
-        if (result == BUFFER_ERROR) {
-            trace("timeout reading data (buffer=%s)", out);
-            proxyPtr->error = 1;
-            proxyPtr->status = TCL_ERROR;
-            return proxyPtr->status;
-        }
-        return  proxyPtr->status;
-    }
+    
+    va_start(ap, format);
+    vsnprintf(buffer, BUFSIZ-1, format, ap);
+    va_end(ap);
+    
+    trace("to-pymol>(%s)", buffer);
+    
+    /* Write the command out to the server. */
+    length = strlen(buffer);
+    nWritten = write(proxyPtr->serverInput, buffer, length);
+    if (nWritten != length) {
+        trace("short write to pymol (wrote=%d, should have been %d)",
+              nWritten, length);
+    }
+
+    for (p = buffer; *p != '\0'; p++) {
+        if (isspace(*p)) {
+            *p = '\0';
+            break;
+        }
+    }
+    sprintf(expect, "PyMOL>%s", buffer);
+    /* Now wait for the "PyMOL>" prompt. */
+    result = Expect(proxyPtr, expect, buffer, BUFSIZ);
+    if (result == BUFFER_ERROR) {
+        trace("timeout reading data (buffer=%s)", buffer);
+        proxyPtr->error = 1;
+        proxyPtr->status = TCL_ERROR;
+        return proxyPtr->status;
+    }
+    return  proxyPtr->status;
 }
 
@@ -843 +848 @@
     PymolProxy *proxyPtr = clientData;
     Image *imgPtr; 
-    int nScanned;
     size_t length;
-#ifdef notdef
-    float samples = 10.0;
-    ssize_t nWritten;
-#endif
     clear_error(proxyPtr);
 
@@ -859 +859 @@
     Pymol(proxyPtr,"refresh\n");
     Pymol(proxyPtr,"bmp -\n");
-    if (Expect(proxyPtr, "image follows: ", buffer, BUFSIZ) != BUFFER_OK) {
+    if (Expect(proxyPtr, "bmp image follows: ", buffer, BUFSIZ) != BUFFER_OK) {
         trace("can't find image follows line (%s)", buffer);
     }
-    nScanned = sscanf(buffer, "image follows: %d\n", &nBytes);
-    if (nScanned != 1) {
-        trace("can't scan image follows buffer (%s)", buffer);
+    if (sscanf(buffer, "bmp image follows: %d\n", &nBytes) != 1) {
+        Tcl_AppendResult(interp, "can't get # bytes from \"", buffer, "\"",
+                         (char *)NULL);
         return TCL_ERROR;
-
-    }
-#ifdef notdef
-    nWritten = write(3,&samples,sizeof(samples));
-#endif
+    } 
     sprintf(buffer, "nv>image %d %d %d %d\n", nBytes, proxyPtr->cacheId, 
             proxyPtr->frame, proxyPtr->rockOffset);
@@ -1278 +1274 @@
     Image *imgPtr;
 
-#ifdef notdef
-    float samples = 10.0;
-    ssize_t nWritten;
-#endif
     clear_error(proxyPtr);
 
@@ -1294 +1286 @@
     Pymol(proxyPtr,"png -\n");
 
-    Expect(proxyPtr, "image follows: ", buffer, 800);
-
-    sscanf(buffer, "image follows: %d\n", &nBytes);
- 
-#ifdef notdef
-    nWritten = write(3, &samples, sizeof(samples));
-#endif
-    if (nBytes == 0) {
-    }
+    Expect(proxyPtr, "png image follows: ", buffer, 800);
+
+    if (sscanf(buffer, "png image follows: %d\n", &nBytes) != 1) {
+        Tcl_AppendResult(interp, "can't get # bytes from \"", buffer, "\"",
+                         (char *)NULL);
+        return TCL_ERROR;
+    } 
     sprintf(buffer, "nv>image %d %d %d %d\n", nBytes, proxyPtr->cacheId, 
             proxyPtr->frame, proxyPtr->rockOffset);
@@ -1312 +1302 @@
         return  TCL_ERROR;
     }
-#ifdef notdef
-    Expect(proxyPtr, " ScenePNG", buffer,800);
-#endif
+    Expect(proxyPtr, " ScenePNG", buffer, 800);
+
     stats.nFrames++;
     stats.nBytes += nBytes;
@@ -1352 +1341 @@
         return TCL_ERROR;
     }
-
     /* Force pymol to update the current scene. */
-    Pymol(proxyPtr,"refresh\n");
-    Pymol(proxyPtr,"ray %d,%d\n", width, height);
-    Pymol(proxyPtr,"png -\n");
-
-    Expect(proxyPtr, "image follows: ", buffer, 800);
-
-    sscanf(buffer, "image follows: %d\n", &nBytes);
- 
-#ifdef notdef
-    nWritten = write(3, &samples, sizeof(samples));
-#endif
-    if (nBytes == 0) {
+    Pymol(proxyPtr,"refresh ; ray %d,%d ; png -, dpi=300\n", width, height);
+    Expect(proxyPtr, "png image follows: ", buffer, 800);
+
+    if (sscanf(buffer, "png image follows: %d\n", &nBytes) != 1) {
+        Tcl_AppendResult(interp, "can't get # bytes from \"", buffer, "\"",
+                         (char *)NULL);
+        return TCL_ERROR;
     }
     sprintf(buffer, "nv>image %d print \"%s\" %d\n", nBytes, token, 
@@ -1376 +1359 @@
         return  TCL_ERROR;
     }
-#ifdef notdef
-    Expect(proxyPtr, " ScenePNG", buffer,800);
-#endif
+    Expect(proxyPtr, " ScenePNG", buffer, 800);
+
     stats.nFrames++;
     stats.nBytes += nBytes;