support ASCII escape code in Windows terminals

fangq · Sep 13, 2023 · ffc8ab0 · ffc8ab0
1 parent 9d2d35b
commit ffc8ab0
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 0 deletions.
diff --git a/src/mcx_tictoc.c b/src/mcx_tictoc.c
@@ -220,3 +220,35 @@ void sleep_ms(int milliseconds) {
     usleep(milliseconds * 1000);
 #endif
 }
+
+
+#if defined(_WIN32) && defined(USE_OS_TIMER) && !defined(MCX_CONTAINER)
+
+#ifndef ENABLE_VIRTUAL_TERMINAL_PROCESSING
+    #define ENABLE_VIRTUAL_TERMINAL_PROCESSING 0x0004
+#endif
+
+int EnableVTMode() {
+    // Set output mode to handle virtual terminal sequences
+    HANDLE hOut = GetStdHandle(STD_OUTPUT_HANDLE);
+
+    if (hOut == INVALID_HANDLE_VALUE) {
+        return 0;
+    }
+
+    DWORD dwMode = 0;
+
+    if (!GetConsoleMode(hOut, &dwMode)) {
+        return 0;
+    }
+
+    dwMode |= ENABLE_PROCESSED_OUTPUT | ENABLE_VIRTUAL_TERMINAL_PROCESSING;
+
+    if (!SetConsoleMode(hOut, dwMode)) {
+        return 0;
+    }
+
+    return 1;
+}
+
+#endif
diff --git a/src/mcx_tictoc.h b/src/mcx_tictoc.h
@@ -39,6 +39,10 @@ unsigned int StartTimer ();
 unsigned int GetTimeMillis ();
 void sleep_ms(int milliseconds);
 
+#if defined(_WIN32) && defined(USE_OS_TIMER) && !defined(MCX_CONTAINER)
+int EnableVTMode();
+#endif
+
 #ifdef  __cplusplus
 }
 #endif

diff --git a/src/mcx_utils.c b/src/mcx_utils.c
@@ -48,6 +48,10 @@
 #include "mcx_bench.h"
 #include "mcx_mie.h"
 
+#if defined(_WIN32) && defined(USE_OS_TIMER) && !defined(MCX_CONTAINER)
+    #include "mmc_tictoc.h"
+#endif
+
 #ifndef MCX_CONTAINER
     #include "zmat/zmatlib.h"
     #include "ubj/ubj.h"
@@ -4391,6 +4395,10 @@ void mcx_parsecmd(int argc, char* argv[], Config* cfg) {
     char logfile[MAX_PATH_LENGTH] = {0};
     float np = 0.f;
 
+#if defined(_WIN32) && defined(USE_OS_TIMER) && !defined(MCX_CONTAINER)
+    EnableVTMode();
+#endif
+
     if (argc <= 1) {
         mcx_usage(cfg, argv[0]);
         exit(0);