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_ALLOCATEPSITRAININGSTATE_H
20 #define OPENXCOM_ALLOCATEPSITRAININGSTATE_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 class TextList;
32 class Soldier;
33 
34 /**
35  * Screen shown monthly to allow changing
36  * soldiers currently in psi training.
37  */
38 class AllocatePsiTrainingState : public State
39 {
40 private:
41 	TextButton *_btnOk;
42 	Window *_window;
43 	Text *_txtTitle, *_txtTraining, *_txtName, *_txtRemaining;
44 	Text *_txtPsiStrength, *_txtPsiSkill;
45 	TextList *_lstSoldiers;
46 	std::vector<Soldier*> _soldiers;
47 	size_t _sel;
48 	int _labSpace;
49 	Base *_base;
50 public:
51 	/// Creates the Psi Training state.
52 	AllocatePsiTrainingState(Game *game, Base *base);
53 	/// Cleans up the Psi Training state.
54 	~AllocatePsiTrainingState();
55 	/// Handler for clicking the OK button.
56 	void btnOkClick(Action *action);
57 	/// Handler for clicking the Soldiers list.
58 	void lstSoldiersClick(Action *action);
59 };
60 
61 }
62 
63 #endif
64