1 files changed, 232 insertions, 0 deletions

diff --git a/src/io.c b/src/io.c
new file mode 100644
index 0000000..ab0f658
--- /dev/null
+++ b/src/io.c
@@ -0,0 +1,232 @@
+#include <stdio.h>
+#include <stdint.h>
+#include <string.h>
+
+#include "coordinates.h"
+#include "moves.h"
+#include "utils.h"
+
+/* Functions for nice output */
+char *edge_string(int i) {
+  return (i > -1 && i < 12) ? edge_string_list[i] : "-";
+}
+
+char *corner_string(int i) {
+  return (i > -1 && i < 8) ? corner_string_list[i] : "-";
+}
+
+char *move_string(int i) {
+  return (i > -1 && i < 19) ? move_string_list[i] : "err";
+}
+
+void print_ep_array(int ep[12]) {
+  for (int i = 0; i < 12; i++)
+    printf(" %s ", edge_string(ep[i]));
+}
+
+void print_ep_int(int ep) {
+  int aux[12];
+  ep_int_to_array(ep, aux);
+  print_ep_array(aux);
+}
+
+void print_cp_array(int cp[8]) {
+  for (int i = 0; i < 8; i++)
+    printf(" %s ", corner_string(cp[i]));
+}
+
+void print_cp_int(int cp) {
+  int aux[8];
+  cp_int_to_array(cp, aux);
+  print_cp_array(aux);
+}
+
+void print_eo_array(int eo[12]) {
+  for (int i = 0; i < 12; i++) {
+    if (eo[i])
+      printf("  x ");
+    else
+      printf("    ");
+  }
+}
+
+void print_eo_int(int eo) {
+  int aux[12];
+  eo_11bits_to_array(eo, aux);
+  print_eo_array(aux);
+}
+
+void print_co_array(int co[8]) {
+  for (int i = 0; i < 8; i++) {
+    if (co[i] == 0)
+      printf("     ");
+    if (co[i] == 1)
+      printf("  cw ");
+    if (co[i] == 2)
+      printf(" ccw ");
+  }
+}
+
+void print_co_int(int co) {
+  int aux[8];
+  co_7trits_to_array(co, aux);
+  print_co_array(aux);
+}
+
+void print_cube_scram(int *scram) {
+  int ep = 0, cp = 0, eofb = 0, coud = 0;
+  for (int i = 0; scram[i]; i++) {
+    ep = apply_move_ep_int(scram[i], ep);
+    cp = cp_transition_table[cp][scram[i]];
+    eofb = eofb_transition_table[eofb][scram[i]];
+    coud = coud_transition_table[coud][scram[i]];
+  }
+  printf("\t\t");         print_ep_int(0);    printf("\n");
+  printf("EP:\t\t");      print_ep_int(ep);   printf("\n");
+  printf("EO(F/B):\t"); print_eo_int(eofb); printf("\n");
+                                            printf("\n");
+  printf("\t\t");         print_cp_int(0);    printf("\n");
+  printf("CP:\t\t");      print_cp_int(cp);   printf("\n");
+  printf("CO(U/D):\t"); print_co_int(coud); printf("\n");
+}
+
+
+void copy_moves(int *src, int *dst) {
+  for (int i = 0; (dst[i] = src[i]); i++);
+}
+
+void append_moves(int *src, int *dst) {
+  int n = 0;
+  for (; dst[n]; n++);
+  copy_moves(src, dst+n);
+}
+
+/* Parse a string and saves the move in a. Supports NISS notation.
+ * Returns the number of moves, or -1 in case of error. */
+int read_moves(char *str, int *a) {
+  int count = 0;
+  int niss = 0;
+  for (int i = 0; str[i] && str[i] != '\n'; i++) {
+    while (str[i] == ' ' || str[i] == '\t') i++;
+    switch (str[i]) {
+      case 'U':
+        a[count++] = niss ? -U : U;
+        break;
+      case 'D':
+        a[count++] = niss ? -D : D;
+        break;
+      case 'R':
+        a[count++] = niss ? -R : R;
+        break;
+      case 'L':
+        a[count++] = niss ? -L : L;
+        break;
+      case 'F':
+        a[count++] = niss ? -F : F;
+        break;
+      case 'B':
+        a[count++] = niss ? -B : B;
+        break;
+      case '(':
+        if (niss)
+          return -1;
+        else
+          niss = 1;
+        break;
+      case ')':
+        if (!niss)
+          return -1;
+        else
+          niss = 0;
+        break;
+      default:
+        return -1;
+    }
+    switch (str[++i]) {
+      case '2':
+        a[count-1] += niss ? -1 : 1;
+        break;
+      case '\'':
+      case '3':
+        a[count-1] += niss ? -2 : 2;
+        break;
+      case '1':
+      default:
+        --i;
+    }
+  }
+  a[count] = 0;
+  return count;
+}
+
+/* Read moves from standard input, after a prompt. */
+int read_moves_from_prompt(int *a) {
+  char str[1000];
+  printf("Enter moves: ");
+  if (fgets(str, 1000, stdin) == NULL)
+    return -1;
+  return read_moves(str, a);
+}
+
+/* Read moves from a list of token, each containing one or more moves. */
+int read_moves_from_tok(int n, char tok[][100], int *a) {
+  char str[1000] = "";
+  for (int i = 0; i < n; i++)
+    strcat(str, tok[i]);
+  return read_moves(str, a);
+}
+
+/* Checks if a sequence of moves uses NISS */
+int uses_niss(int *str) {
+  for (int i = 0; str[i]; i++)
+    if (str[i] < 0)
+      return 1;
+  return 0;
+}
+
+/* A (B) -> B' A */
+int unniss(int *src, int *dst) {
+  int n = 0;
+  for (int i = 0; src[i]; i++)
+    if (src[i] < 0)
+      n++;
+
+  int norm_count = n, inv_count = n-1;
+  for (int i = 0; src[i]; i++)
+    if (src[i] > 0)
+      dst[norm_count++] = src[i];
+    else
+      dst[inv_count--] = inverse_move[-src[i]];
+
+  dst[norm_count] = 0;
+
+  return n;
+}
+
+int invert(int *src, int *dst) {
+  int aux[255];
+  for (int i = 0; (aux[i] = -src[i]); i++);
+  return unniss(aux, dst);
+}
+
+int len(int *scram) {
+  int m;
+  for (m = 0; scram[m]; m++);
+  return m;
+}
+
+void print_moves(int moves_list[]) {
+  int niss = 0;
+  for (int i = 0; moves_list[i]; i++) {
+    if (!niss && moves_list[i] < 0) {
+      printf("(");
+      niss = 1;
+    }
+    printf("%s", move_string_list[abs(moves_list[i])]);
+    if (niss && moves_list[i+1] >= 0) {
+      niss = 0;
+      printf(")");
+    }
+    printf(" ");
+  }
+}