adding initial files

1 files changed, 157 insertions, 0 deletions

diff --git a/src/chess_board.cpp b/src/chess_board.cpp
new file mode 100644
index 0000000..9a6848b
--- /dev/null
+++ b/src/chess_board.cpp
@@ -0,0 +1,157 @@
+#include "../include/chess_ai.hpp"
+
+typedef std::vector<std::vector<chess_ai::chess_piece>>::
+iterator vector_iterator;
+typedef std::vector<chess_ai::chess_piece>::iterator square_iterator;
+
+chess_ai::chess_board::chess_board() {
+    init_board_vector();
+}
+
+chess_ai::chess_board::chess_board(board_state state) {
+    unsigned vec_index, sqr_index;
+
+    init_board_vector();
+    
+    if(state == initial) {
+
+        for(vector_iterator it_vec = grid.begin(); it_vec != grid.end();
+            ++it_vec) {
+            
+            for(square_iterator it_sqr = (*it_vec).begin();
+                it_sqr != (*it_vec).end(); ++it_sqr) {
+                chess_piece piece;
+
+                piece_colour colour;
+                    
+                if(vec_index < 3) {
+                    colour = black;
+                } else {
+                    colour = white;
+                }
+                
+                vec_index = it_vec - grid.begin();
+                sqr_index = it_sqr - (*it_vec).begin();
+                
+                if(vec_index == 1 || vec_index == 6) {
+                    piece.set_type(pawn);
+                } else if(vec_index == 0 || vec_index == 7) {
+
+                    if(sqr_index == 0 || sqr_index == 7)
+                        piece.set_type(rook);
+                    else if(sqr_index == 1 || sqr_index == 6)
+                        piece.set_type(knight);
+                    else if(sqr_index == 2 || sqr_index == 5)
+                        piece.set_type(bishop);
+                    else if(sqr_index == 3)
+                        piece.set_type(queen);
+                    else if(sqr_index == 4)
+                        piece.set_type(king);
+                }
+
+                piece.set_colour(colour);
+                piece.set_x(sqr_index);
+                piece.set_y(vec_index);
+                
+                *it_sqr = piece;
+            }
+        }
+    } else {
+        for(auto& row : grid) {
+            for(auto& square : row) {
+                chess_piece piece;
+                square = piece;
+            }
+        }
+    }
+}
+
+void chess_ai::chess_board::init_board_vector() {
+    for(unsigned i = 0; i < SIZE; ++i) {
+        std::vector<chess_ai::chess_piece> tmp_vec;
+        for(unsigned j = 0; j < SIZE; ++j) {
+            chess_piece piece;
+            tmp_vec.push_back(piece);
+        }
+        grid.push_back(tmp_vec);
+    }
+}
+
+void chess_ai::chess_board::print_board() {
+    std::cout << "  |";
+    for(unsigned i = 0; i < SIZE; ++i) {
+        std::cout << "---|";
+    }
+    std::cout << std::endl << "8 |";
+    
+    for(unsigned y = 0; y < SIZE; ++y) {
+        for(unsigned x = 0; x < SIZE; ++x) {
+            std::cout << " " << grid[y][x].str() << " |";
+        }
+        
+        std::cout << std::endl << "  |";
+        
+
+        if(y == SIZE-1) {
+            std::cout << "---+";
+            for(unsigned i = 0; i < SIZE-2; ++i) {
+                std::cout << "---+";
+            }
+            std::cout << "---|" << std::endl;
+        } else {
+            for(unsigned i = 0; i < SIZE-1; ++i) {
+                std::cout << "---+";
+            }
+            std::cout << "---|" << std::endl;
+            std::cout << 7-y << " |";
+        }
+    }
+    std::cout << "    a   b   c   d   e   f   g   h" << std::endl;
+}
+
+void chess_ai::chess_board::set_piece(chess_piece piece) {
+    unsigned vec_index, sqr_index;
+    for(vector_iterator it_vec = grid.begin(); it_vec != grid.end(); ++it_vec) {
+        for(square_iterator it_sqr = (*it_vec).begin();
+            it_sqr != (*it_vec).end(); ++it_sqr) {
+            vec_index = it_vec - grid.begin();
+            sqr_index = it_sqr - (*it_vec).begin();
+            
+            if(vec_index == piece.y && sqr_index == piece.x) {
+                *it_sqr = piece;
+            }
+        }
+    }
+}
+
+void chess_ai::chess_board::remove_piece(chess_piece piece) {
+    unsigned vec_index, sqr_index;
+    for(vector_iterator it_vec = grid.begin(); it_vec != grid.end(); ++it_vec) {
+        for(square_iterator it_sqr = (*it_vec).begin();
+            it_sqr != (*it_vec).end(); ++it_sqr) {
+            vec_index = it_vec - grid.begin();
+            sqr_index = it_sqr - (*it_vec).begin();
+            
+            if(vec_index == piece.y && sqr_index == piece.x) {
+                chess_piece empty_piece;
+                *it_sqr = empty_piece;
+            }
+        }
+    }
+}
+
+void chess_ai::chess_board::remove_piece(unsigned x, unsigned y) {
+    unsigned vec_index, sqr_index;
+    for(vector_iterator it_vec = grid.begin(); it_vec != grid.end(); ++it_vec) {
+        for(square_iterator it_sqr = (*it_vec).begin();
+            it_sqr != (*it_vec).end(); ++it_sqr) {
+            vec_index = it_vec - grid.begin();
+            sqr_index = it_sqr - (*it_vec).begin();
+            
+            if(vec_index == y && sqr_index == x) {
+                chess_piece empty_piece;
+                *it_sqr = empty_piece;
+            }
+        }
+    }
+}