1 /*
2  * Copyright 2010-2014 OpenXcom Developers.
3  *
4  * This file is part of OpenXcom.
5  *
6  * OpenXcom is free software: you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation, either version 3 of the License, or
9  * (at your option) any later version.
10  *
11  * OpenXcom is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with OpenXcom.  If not, see <http://www.gnu.org/licenses/>.
18  */
19 #ifndef OPENXCOM_PSITRAININGSTATE_H
20 #define OPENXCOM_PSITRAININGSTATE_H
21 
22 #include "../Engine/State.h"
23 
24 namespace OpenXcom
25 {
26 
27 class TextButton;
28 class Window;
29 class Text;
30 class Base;
31 
32 /**
33  * Screen shown monthly when the player has psi labs available.
34  */
35 class PsiTrainingState : public State
36 {
37 private:
38 	TextButton *_btnOk;
39 	Window *_window;
40 	Text *_txtTitle;
41 	std::vector<TextButton*> _btnBases;
42 	std::vector<Base*> _bases;
43 public:
44 	/// Creates the Psi Training state.
45 	PsiTrainingState(Game *game);
46 	/// Cleans up the Psi Training state.
47 	~PsiTrainingState();
48 	/// Handler for clicking the OK button.
49 	void btnOkClick(Action *action);
50 	/// Handler for clicking a Base button.
51 	void btnBaseXClick(Action *action);
52 };
53 
54 }
55 
56 #endif
57