adding initial files

3 files changed, 253 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;
+            }
+        }
+    }
+}
diff --git a/src/chess_piece.cpp b/src/chess_piece.cpp
new file mode 100644
index 0000000..cd28ef8
--- /dev/null
+++ b/src/chess_piece.cpp
@@ -0,0 +1,66 @@
+#include "../include/chess_ai.hpp"
+
+chess_ai::chess_piece::chess_piece() {
+    type = empty;
+    colour = none;
+    x = -1;
+    y = -1;
+}
+
+chess_ai::chess_piece::chess_piece(piece_type type, piece_colour colour) {
+    this->type = type;
+    this->colour = colour;
+
+    if(colour == black) {
+        y = 0;
+    } else {
+        y = 7;
+    }
+    
+    if(type == king) {
+        x = 4;
+    } else {
+        x = 3;
+    }
+}
+
+chess_ai::chess_piece::chess_piece(piece_type type, piece_colour colour,
+                                   unsigned x, unsigned y) {
+    this->type = type;
+    this->colour = colour;
+    this->x = x;
+    this->y = y;
+    
+}
+
+void chess_ai::chess_piece::set_type(piece_type type) {
+    this->type = type;
+}
+
+void chess_ai::chess_piece::set_colour(piece_colour colour) {
+    this->colour = colour;
+}
+
+void chess_ai::chess_piece::set_x(unsigned x) {
+    this->x = x;
+}
+
+void chess_ai::chess_piece::set_y(unsigned y) {
+    this->y = y;
+}
+
+std::string chess_ai::chess_piece::str() {
+    if(type == empty)
+        return " ";
+    else if(type == pawn)
+        return "p";
+    else if(type == rook)
+        return "r";
+    else if(type == knight)
+        return "n";
+    else if(type == bishop)
+        return "b";
+    else if(type == queen)
+        return "q";
+    return "k";
+}
diff --git a/src/main.cpp b/src/main.cpp
new file mode 100644
index 0000000..b41c919
--- /dev/null
+++ b/src/main.cpp
@@ -0,0 +1,30 @@
+/*
+
+  description:
+
+  author: Yann Herklotz <ymherklotz@gmail.com>
+  date created: DD-MM-YYYY
+
+ */
+
+#include "../include/chess_ai.hpp"
+
+#include <iostream>
+
+using namespace std;
+using namespace chess_ai;
+
+int main(int argc, char** argv) {
+    (void)argc;
+    (void)argv;
+
+    chess_board board(initial);
+    board.print_board();
+
+    chess_piece piece(rook, white, 5, 3);
+    board.set_piece(piece);
+
+    board.print_board();
+    
+    return 0;
+}