army.h

// START A3HEADER
//
// This source file is part of the Atlantis PBM game program.
// Copyright (C) 1995-1999 Geoff Dunbar
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program, in the file license.txt. If not, write
// to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
// Boston, MA 02111-1307, USA.
//
// See the Atlantis Project web page for details:
// http://www.prankster.com/project
//
// END A3HEADER
#ifndef ARMY_CLASS
#define ARMY_CLASS

#include <functional>
#include <map>
using namespace std;

class Soldier;
class Army;

#include "unit.h"
#include "alist.h"
#include "items.h"
#include "object.h"
#include "shields.h"
#include "helper.h"

class Soldier {
	public:
		Soldier(Unit *unit, Object *object, int regType, int race, int ass=0);

		void SetupSpell();
		void SetupCombatItems();

		//
		// SetupHealing is actually game-specific, and appears in specials.cpp
		//
		void SetupHealing();

		int HasEffect(char const *);
		void SetEffect(char const *);
		void ClearEffect(char const *);
		void ClearOneTimeEffects(void);
		int ArmorProtect(int weaponClass );

		void RestoreItems();
		void Alive(int);
		void Dead();

		/* Unit info */
		AString name;
		Unit * unit;
		int race;
		int riding;
		int building;

		/* Healing information */
		int healing;
		int healtype;
		int healitem;
		int canbehealed;
		int regen;

		/* Attack info */
		int weapon;
		int attacktype;
		int askill;
		int attacks;
		char const *special;
		int slevel;

		/* Defense info */
		int dskill[NUM_ATTACK_TYPES];
		int protection[NUM_ATTACK_TYPES];
		int armor;
		int hits;
		int maxhits;
		int damage;

		BITFIELD battleItems;
		int amuletofi;

		/* Effects */
		map< char const *, int > effects;
};

typedef Soldier * SoldierPtr;

class Army
{
	public:
		Army(Unit *,AList *,int,int = 0);
		~Army();

		void WriteLosses(Battle *);
		void Lose(Battle *,ItemList *);
		void Win(Battle *,ItemList *);
		void Tie(Battle *);
		int CanBeHealed();
		void DoHeal(Battle *);
		void DoHealLevel(Battle *,int,int useItems );
		void Regenerate(Battle *);

		void GetMonSpoils(ItemList *,int, int);

		int Broken();
		int NumAlive();
		int NumSpoilers();
		int CanAttack();
		int NumFront();
		Soldier *GetAttacker( int, int & );
		int GetEffectNum(char const *effect);
		int GetTargetNum(char const *special = NULL);
		Soldier *GetTarget( int );
		int RemoveEffects(int num, char const *effect);
		int DoAnAttack(char const *special, int numAttacks, int attackType,
				int attackLevel, int flags, int weaponClass, char const *effect,
				int mountBonus, Soldier *attacker);
		void Kill(int);
		void Reset();

		//
		// These funcs are in specials.cpp
		//
		int CheckSpecialTarget(char const *,int);

		SoldierPtr * soldiers;
		Unit * leader;
		ShieldList shields;
		int round;
		int tac;
		int canfront;
		int canbehind;
		int notfront;
		int notbehind;
		int count;

		int hitsalive; // current number of "living hits"
		int hitstotal; // Number of hits at start of battle.
};

#endif