Use correct variable to verify thread state access.

[cycript.git] / Display.cpp

diff --git a/Display.cpp b/Display.cpp
index fbca8df2faeb7485594854b3abddc883a560b02f..7e088cb03961b2c9cfd699b771838342c2ee783f 100644 (file)
--- a/Display.cpp
+++ b/Display.cpp
@@ -72,9 +72,6 @@ CYCursor CYDisplayOutput(int (*put)(int), int width, const char *data, ssize_t o
 
             case CYIgnoreStart:
                 CYDisplayOutput_(put, data);
-                ++offset;
-            break;
-
             case CYIgnoreEnd:
                 ++offset;
             break;
@@ -90,19 +87,25 @@ CYCursor CYDisplayOutput(int (*put)(int), int width, const char *data, ssize_t o
 
         }
     }
+}
 
-    return point;
+void CYDisplayMove_(char *negative, char *positive, int offset) {
+    if (offset < 0)
+        putp(tparm(negative, -offset));
+    else if (offset > 0)
+        putp(tparm(positive, offset));
 }
 
 void CYDisplayMove(CYCursor target) {
-    int offset(target.real() - current_.real());
+    CYCursor offset(target - current_);
 
-    if (offset < 0)
-        putp(tparm(parm_up_cursor, -offset));
-    else if (offset > 0)
-        putp(tparm(parm_down_cursor, offset));
+    CYDisplayMove_(parm_up_cursor, parm_down_cursor, offset.real());
+
+    if (char *parm = tparm(column_address, target.imag()))
+        putp(parm);
+    else
+        CYDisplayMove_(parm_left_cursor, parm_right_cursor, offset.imag());
 
-    putp(tparm(column_address, target.imag()));
     current_ = target;
 }
 
@@ -127,19 +130,18 @@ void CYDisplayUpdate() {
     if (width_ != width) {
         current_ = CYCursor();
         CYDisplayOutput(NULL, width, prompt);
-        CYDisplayOutput(NULL, width, buffer, point_);
+        current_ = CYDisplayOutput(NULL, width, buffer, point_);
     }
 
     CYDisplayMove(CYCursor());
     CYDisplayOutput(putchar, width, prompt);
-
     CYCursor target(CYDisplayOutput(putchar, width, stream.str().c_str(), rl_point));
-    if (target.imag() == 0)
-        putp(cursor_down);
 
+    if (current_.imag() == 0)
+        CYDisplayOutput(putchar, width, " ");
     putp(clr_eos);
-    CYDisplayMove(target);
 
+    CYDisplayMove(target);
     fflush(stdout);
 
     width_ = width;