color stable

Joseph Redmon

2016-03-14 3439232730b35ce392e163f8bf0bc925a0579512

 src/go.c

@@ -122,7 +122,7 @@
    free(l);
}

void print_board(float *board)
void print_board(float *board, int swap)
{
    int i,j;
    printf("\n\n");
@@ -135,8 +135,8 @@
        printf("%2d ", 19-j);
        for(i = 0; i < 19; ++i){
            int index = j*19 + i;
            if(board[index] > 0) printf("\u25C9 ");
            else if(board[index] < 0) printf("\u25EF ");
            if(board[index]*-swap > 0) printf("\u25C9 ");
            else if(board[index]*-swap < 0) printf("\u25EF ");
            else printf("  ");
        }
        printf("\n");
@@ -161,6 +161,7 @@
    set_batch_network(&net, 1);
    float *board = calloc(19*19, sizeof(float));
    float *move = calloc(19*19, sizeof(float));
    int color = 1;
    while(1){
        float *output = network_predict(net, board);
        copy_cpu(19*19, output, 1, move, 1);
@@ -191,7 +192,7 @@
        int indexes[3];
        int row, col;
        top_k(move, 19*19, 3, indexes);
        print_board(board);
        print_board(board, color);
        for(i = 0; i < 3; ++i){
            int index = indexes[i];
            row = index / 19;
@@ -202,7 +203,9 @@
        int rec_row = index / 19;
        int rec_col = index % 19;

        printf("\u25C9 Enter move: ");
        if(color == 1) printf("\u25EF Enter move: ");
        else printf("\u25C9 Enter move: ");

        char c;
        char *line = fgetl(stdin);
        int num = sscanf(line, "%c %d", &c, &row);
@@ -213,6 +216,7 @@
        }else if (c < 'A' || c > 'T'){
            if (c == 'p'){
                flip_board(board);
                color = -color;
                continue;
            }else{
                char g;
@@ -232,6 +236,7 @@
        }
        update_board(board);
        flip_board(board);
        color = -color;
    }

}