First actual commit
[progcomp2012.git] / manager / stratego.cpp
1 #include "common.h"
2
3 #include "stratego.h"
4
5 using namespace std;
6
7 /**
8  * Static variables
9  */
10 Board Board::theBoard(14,14);
11 //nothing, boulder, flag, spy, scout, miner, sergeant, lietenant, captain, major, colonel, general, marshal, bomb, error
12 char  Piece::tokens[] = {'.','+','F','y','s','n','S','L','c','m','C','G','M','B','?'};
13 int Piece::maxUnits[] = {0,0,1,1,8,5,4,4,4,3,2,1,1,6,0};
14
15
16
17 #ifdef GRAPHICS
18         Piece::TextureManager Piece::textures;
19 #endif //GRAPHICS
20
21
22 #ifdef GRAPHICS
23
24 Piece::TextureManager::~TextureManager()
25 {
26         Array<Texture*>::Iterator i(*this);
27         while (i.Good())
28         {
29                 delete (*i);
30                 ++i;
31         }
32 }
33
34 Texture & Piece::TextureManager::operator[](const LUint & at)
35 {
36         while (Array<Texture*>::Size() <= at)
37         {
38                 char buffer[BUFSIZ];
39                 sprintf(buffer, "images/piece%lu.bmp", Array<Texture*>::Size());
40                 Array<Texture*>::Add(new Texture(buffer, false));
41                 
42         }
43         return *(Array<Texture*>::operator[](at));
44 }
45 #endif //GRAPHICS
46
47 /**
48  * Gets the type of a piece, based off a character token
49  * @param fromToken - character identifying the piece
50  * @returns The type of the piece
51  */
52 Piece::Type Piece::GetType(char fromToken)
53 {
54         for (int ii=0; ii <= (int)(Piece::BOMB); ++ii)
55         {
56                 if (tokens[ii] == fromToken)
57                 {
58                         return Type(Piece::NOTHING + ii);
59                 }
60         }
61         return Piece::BOULDER;
62 }
63
64 /**
65  * Construct a new, empty board
66  * @param newWidth - the width of the board
67  * @param newHeight - the height of the board
68  */
69 Board::Board(int newWidth, int newHeight) : winner(Piece::NONE), width(newWidth), height(newHeight), board(NULL)
70 {
71         board = new Piece**[width];
72         for (int x=0; x < width; ++x)
73         {
74                 board[x] = new Piece*[height];
75                 for (int y=0; y < height; ++y)
76                         board[x][y] = NULL;
77         }
78 }
79
80 /**
81  * Cleanup a board
82  */
83 Board::~Board()
84 {
85         for (int x=0; x < width; ++x)
86         {
87                 for (int y=0; y < height; ++y)
88                         delete board[x][y];
89                 delete [] board[x];
90         }
91 }
92
93 /**
94  * Print textual representation of the board to a stream
95  * @param stream - the stream to print information to
96  * @param reveal - Pieces matching this colour will have their identify revealed, other pieces will be shown as '#' or '*' for RED or BLUE respectively.
97  */
98 void Board::Print(FILE * stream, const Piece::Colour & reveal)
99 {
100         for (int y=0; y < height; ++y)
101         {
102                 for (int x=0; x < width; ++x)
103                 {
104                         Piece * piece = board[x][y];
105                         if (piece == NULL)
106                         {
107                                 fprintf(stream, ".");
108                         }
109                         else if (piece->colour != Piece::NONE && (piece->colour == reveal || reveal == Piece::BOTH))
110                         {
111                                 fprintf(stream, "%c", Piece::tokens[piece->type]);
112                         }
113                         else
114                         {
115                                 switch (piece->colour)
116                                 {
117                                         case Piece::RED:
118                                                 fprintf(stream, "#");
119                                                 break;
120                                         case Piece::BLUE:
121                                                 fprintf(stream, "*");
122                                                 break;
123                                         case Piece::NONE:
124                                                 fprintf(stream, "+"); 
125                                                 break;
126                                         case Piece::BOTH:
127                                                 fprintf(stream, "$"); //Should never see these!
128                                                 break;
129                                 }
130                         }
131                 }
132                 fprintf(stream, "\n");
133         }
134         
135 }
136
137
138 #ifdef GRAPHICS
139 /**
140  * Draw the board state to graphics
141  * @param reveal - Pieces matching this colour will be revealed. All others will be shown as blank coloured squares.
142  */
143 void Board::Draw(const Piece::Colour & reveal)
144 {
145         if (!Graphics::Initialised())
146         {
147                 Graphics::Initialise("Stratego", width*32, height*32);
148         }
149
150         Graphics::ClearScreen();
151         
152         for (int y=0; y < height; ++y)
153         {
154                 for (int x=0; x < width; ++x)
155                 {
156                         Piece * piece = board[x][y];
157                         if (piece == NULL)
158                         {
159                                 //Don't display anything
160
161                         }
162                         else if (piece->colour != Piece::NONE && (piece->colour == reveal || reveal == Piece::BOTH))
163                         {
164                                 //Display the piece
165                                 Piece::textures[(int)(piece->type)].DrawColour(x*32,y*32,0,1, Piece::GetGraphicsColour(piece->colour));
166                                 
167                         }
168                         else
169                         {
170                                 switch (piece->colour)
171                                 {
172                                         case Piece::RED:
173                                                 Piece::textures[(int)(Piece::BOULDER)].DrawColour(x*32,y*32,0,1, Piece::GetGraphicsColour(piece->colour));
174                                                 break;
175                                         case Piece::BLUE:
176                                                 Piece::textures[(int)(Piece::BOULDER)].DrawColour(x*32,y*32,0,1, Piece::GetGraphicsColour(piece->colour));
177                                                 break;
178                                         case Piece::NONE:
179                                                 Piece::textures[(int)(Piece::BOULDER)].DrawColour(x*32,y*32,0,1, Piece::GetGraphicsColour(piece->colour));
180                                                 break;
181                                         case Piece::BOTH:
182                                                 Piece::textures[(int)(Piece::BOULDER)].DrawColour(x*32,y*32,0,1, Piece::GetGraphicsColour(piece->colour));
183                                                 break;
184                                 }
185                         }
186                 }
187                 
188         }
189         Graphics::UpdateScreen();
190         
191 }
192 #endif //GRAPHICS
193
194 /**
195  * Adds a piece to the board
196  * @param x - x-coord to place the piece at, starting at zero, must be less than board width
197  * @param y - y-coord to place the piece at, starting at zero, must be less than board height
198  * @param newType - the Type of the piece
199  * @param newColour - the Colour of the piece
200  * @returns true if and only if the piece could be successfully added.
201  */
202 bool Board::AddPiece(int x, int y, const Piece::Type & newType, const Piece::Colour & newColour)
203 {
204         if (board == NULL || x < 0 || y < 0 || x >= width || y >= width || board[x][y] != NULL)
205                 return false;
206
207         Piece * piece = new Piece(newType, newColour);
208         board[x][y] = piece;
209         return true;
210 }
211
212 /**
213  * Gets a pointer to a piece at a board location
214  * UNUSED
215  * @param x - x-coord of the piece
216  * @param y - y-coord of the piece
217  * @returns pointer to the piece, or NULL if the board location was empty
218  * @throws error if board is null or coords are invalid
219  */
220 Piece * Board::GetPiece(int x, int y)
221 {
222         assert(board != NULL);
223         assert(x >= 0 && x < width && y >= 0 && y < height);
224         return board[x][y];
225 }
226
227 /**
228  * Moves a piece at a specified position in the specified direction, handles combat if necessary
229  * @param x - x-coord of the piece
230  * @param y - y-coord of the piece
231  * @param direction - Direction in which to move (UP, DOWN, LEFT or RIGHT)
232  * @param colour - Colour which the piece must match for the move to be valid
233  * @returns A MovementResult which indicates the result of the move - OK is good, VICTORY means that a flag was captured, anything else is an error
234  */
235 Board::MovementResult Board::MovePiece(int x, int y, const Direction & direction, int multiplier,const Piece::Colour & colour)
236 {
237         if (board == NULL) 
238         {
239                 return NO_BOARD;
240         }
241         if (!(x >= 0 && x < width && y >= 0 && y < height)) 
242         {
243                 return INVALID_POSITION;
244         }
245         Piece * target = board[x][y];
246         if (target == NULL) 
247         {
248                 return NO_SELECTION;
249         }
250         if (!(colour == Piece::NONE || target->colour == colour)) 
251         {
252                 return NOT_YOUR_UNIT;
253         }
254         if (target->type == Piece::FLAG || target->type == Piece::BOMB || target->type == Piece::BOULDER) 
255         {
256                 return IMMOBILE_UNIT;
257         }
258         if (multiplier > 1 && target->type != Piece::SCOUT)
259         {
260                 return INVALID_DIRECTION; //Can only move a scout multiple times.
261         }
262         int x2 = x; int y2 = y;
263
264         for (int ii=0; ii < multiplier; ++ii)
265         {
266                 switch (direction)
267                 {
268                         case UP:
269                                 --y2;
270                                 break;
271                         case DOWN:
272                                 ++y2;
273                                 break;
274                         case LEFT:
275                                 --x2;
276                                 break;
277                         case RIGHT:
278                                 ++x2;
279                                 break;
280                 }
281                 if (!(x2 >= 0 && x2 < width && y2 >= 0 && y2 < height)) 
282                 {
283                         return INVALID_DIRECTION;
284                 }
285                 if (ii < multiplier-1 && board[x2][y2] != NULL)
286                 {
287                         return POSITION_FULL;
288                 }
289         }
290         Piece * defender = board[x2][y2];
291         if (defender == NULL)
292         {
293                 board[x][y] = NULL;
294                 board[x2][y2] = target;
295         }
296         else if (defender->colour != target->colour)
297         {
298                 if (defender->colour == Piece::NONE) 
299                 {
300                         return POSITION_FULL;
301                 }
302                 if (defender->type == Piece::FLAG)
303                 {
304                         winner = target->colour;
305                         return VICTORY;
306                 }
307                 else if (defender->type == Piece::BOMB)
308                 {
309                         if (target->type == Piece::MINER)
310                         {
311                                 delete defender;
312                                 board[x][y] = NULL;
313                                 board[x2][y2] = target;
314                                 return KILLS;
315                         }
316                         else
317                         {
318                                 delete defender;
319                                 delete target;
320                                 board[x][y] = NULL;
321                                 board[x2][y2] = NULL;
322                                 return BOTH_DIE;
323                         }
324                 }
325                 else if (defender->type == Piece::MARSHAL && target->type == Piece::SPY)
326                 {
327                         delete defender;
328                         board[x][y] = NULL;
329                         board[x2][y2] = target;
330                         return KILLS;
331                 }
332                 else if (target->operator > (*defender))
333                 {
334                         delete defender;
335                         board[x][y] = NULL;
336                         board[x2][y2] = target;
337                         return KILLS;
338                 }
339                 else if (target->operator==(*defender) && rand() % 2 == 0)
340                 {
341                         delete defender;
342                         board[x][y] = NULL;
343                         board[x2][y2] = target; 
344                         return KILLS;
345                 }
346                 else
347                 {
348                         delete target;
349                         board[x][y] = NULL;
350                         return DIES;
351                 }
352         }
353         else
354         {
355                 return POSITION_FULL;
356         }
357         return OK;
358 }       
359
360
361

UCC git Repository :: git.ucc.asn.au