_w1FireCountdown(0), _w2FireCountdown(0), _end(false), _destroyUfo(false), _destroyCraft(false), _ufoBreakingOff(false), _weapon1Enabled(true), _weapon2Enabled(true),

_minimized(false), _endDogfight(false), _animatingHit(false), _waitForPoly(false), _waitForAltitude(false), _ufoSize(0), _craftHeight(0), _currentCraftDamageColor(0),

_interceptionNumber(0), _interceptionsCount(0), _x(0), _y(0), _minimizedIconX(0), _minimizedIconY(0)

{

_screen = false;

_craft->setInDogfight(true);

// Create objects

_window = new Surface(160, 96, _x, _y);

_battle = new Surface(77, 74, _x + 3, _y + 3);

_weapon1 = new InteractiveSurface(15, 17, _x + 4, _y + 52);

_range1 = new Surface(21, 74, _x + 19, _y + 3);

_weapon2 = new InteractiveSurface(15, 17, _x + 64, _y + 52);

_range2 = new Surface(21, 74, _x + 43, _y + 3);

_damage = new Surface(22, 25, _x + 93, _y + 40);

_btnMinimize = new InteractiveSurface(12, 12, _x, _y);

_preview = new InteractiveSurface(160, 96, _x, _y);

_btnStandoff = new ImageButton(36, 15, _x + 83, _y + 4);

_btnCautious = new ImageButton(36, 15, _x + 120, _y + 4);

_btnStandard = new ImageButton(36, 15, _x + 83, _y + 20);

_btnAggressive = new ImageButton(36, 15, _x + 120, _y + 20);

_btnDisengage = new ImageButton(36, 15, _x + 120, _y + 36);

_btnUfo = new ImageButton(36, 17, _x + 120, _y + 52);

_txtAmmo1 = new Text(16, 9, _x + 4, _y + 70);

_txtAmmo2 = new Text(16, 9, _x + 64, _y + 70);

_txtDistance = new Text(40, 9, _x + 116, _y + 72);

_txtStatus = new Text(150, 9, _x + 4, _y + 85);

_btnMinimizedIcon = new InteractiveSurface(32, 20, _minimizedIconX, _minimizedIconY);

_txtInterceptionNumber = new Text(16, 9, _minimizedIconX + 18, _minimizedIconY + 6);

_mode = _btnStandoff;

_craftDamageAnimTimer = new Timer(500);

// Set palette

setInterface("dogfight");

add(_window);

add(_battle);

add(_weapon1);

add(_range1);

add(_weapon2);

add(_range2);

add(_damage);

add(_btnMinimize);

add(_btnStandoff, "standoffButton", "dogfight", _window);

add(_btnCautious, "cautiousButton", "dogfight", _window);

add(_btnStandard, "standardButton", "dogfight", _window);

add(_btnAggressive, "aggressiveButton", "dogfight", _window);

add(_btnDisengage, "disengageButton", "dogfight", _window);

add(_btnUfo, "ufoButton", "dogfight", _window);

add(_txtAmmo1, "numbers", "dogfight", _window);

add(_txtAmmo2, "numbers", "dogfight", _window);

add(_txtDistance, "distance", "dogfight", _window);

add(_preview);

add(_txtStatus, "text", "dogfight", _window);

add(_btnMinimizedIcon);

add(_txtInterceptionNumber, "minimizedNumber", "dogfight");

_btnStandoff->invalidate(false);

_btnCautious->invalidate(false);

_btnStandard->invalidate(false);

_btnAggressive->invalidate(false);

_btnDisengage->invalidate(false);

_btnUfo->invalidate(false);

// Set up objects

RuleInterface *dogfightInterface = _game->getMod()->getInterface("dogfight");

Surface *graphic;

graphic = _game->getMod()->getSurface("INTERWIN.DAT");

graphic->setX(0);

graphic->setY(0);

graphic->getCrop()->x = 0;

graphic->getCrop()->y = 0;

graphic->getCrop()->w = _window->getWidth();

graphic->getCrop()->h = _window->getHeight();

_window->drawRect(graphic->getCrop(), 15);

graphic->blit(_window);

_preview->drawRect(graphic->getCrop(), 15);

graphic->getCrop()->y = dogfightInterface->getElement("previewTop")->y;

graphic->getCrop()->h = dogfightInterface->getElement("previewTop")->h;

graphic->blit(_preview);

graphic->setY(_window->getHeight() - dogfightInterface->getElement("previewBot")->h);

graphic->getCrop()->y = dogfightInterface->getElement("previewBot")->y;

graphic->getCrop()->h = dogfightInterface->getElement("previewBot")->h;

graphic->blit(_preview);

if (ufo->getRules()->getModSprite().empty())

{

graphic->getCrop()->y = dogfightInterface->getElement("previewMid")->y + dogfightInterface->getElement("previewMid")->h * _ufo->getRules()->getSprite();

graphic->getCrop()->h = dogfightInterface->getElement("previewMid")->h;

}

else

{

graphic = _game->getMod()->getSurface(ufo->getRules()->getModSprite());

}

graphic->setX(dogfightInterface->getElement("previewTop")->x);

graphic->setY(dogfightInterface->getElement("previewTop")->h);

graphic->blit(_preview);

_preview->setVisible(false);

_preview->onMouseClick((ActionHandler)&DogfightState::previewClick);

_btnMinimize->onMouseClick((ActionHandler)&DogfightState::btnMinimizeClick);

_btnStandoff->copy(_window);

_btnStandoff->setGroup(&_mode);

_btnStandoff->onMousePress((ActionHandler)&DogfightState::btnStandoffPress);

_btnCautious->copy(_window);

_btnCautious->setGroup(&_mode);

_btnCautious->onMousePress((ActionHandler)&DogfightState::btnCautiousPress);

_btnStandard->copy(_window);

_btnStandard->setGroup(&_mode);

_btnStandard->onMousePress((ActionHandler)&DogfightState::btnStandardPress);

_btnAggressive->copy(_window);

_btnAggressive->setGroup(&_mode);

_btnAggressive->onMousePress((ActionHandler)&DogfightState::btnAggressivePress);

_btnDisengage->copy(_window);

_btnDisengage->onMousePress((ActionHandler)&DogfightState::btnDisengagePress);

_btnDisengage->setGroup(&_mode);

_btnUfo->copy(_window);

_btnUfo->onMouseClick((ActionHandler)&DogfightState::btnUfoClick);

_txtDistance->setText(L"640");

_txtStatus->setText(tr("STR_STANDOFF"));

SurfaceSet *set = _game->getMod()->getSurfaceSet("INTICON.PCK");

// Create the minimized dogfight icon.

Surface *frame = set->getFrame(_craft->getRules()->getSprite());

frame->setX(0);

frame->setY(0);

frame->blit(_btnMinimizedIcon);

_btnMinimizedIcon->onMouseClick((ActionHandler)&DogfightState::btnMinimizedIconClick);

_btnMinimizedIcon->setVisible(false);

// Draw correct number on the minimized dogfight icon.

std::wostringstream ss1;

if (_craft->getInterceptionOrder() == 0)

{

int maxInterceptionOrder = 0;

for (std::vector<Base*>::iterator baseIt = _game->getSavedGame()->getBases()->begin(); baseIt != _game->getSavedGame()->getBases()->end(); ++baseIt)

{

for (std::vector<Craft*>::iterator craftIt = (*baseIt)->getCrafts()->begin(); craftIt != (*baseIt)->getCrafts()->end(); ++craftIt)

{

if ((*craftIt)->getInterceptionOrder() > maxInterceptionOrder)

{

maxInterceptionOrder = (*craftIt)->getInterceptionOrder();

}

}

}

_craft->setInterceptionOrder(++maxInterceptionOrder);

}

ss1 << _craft->getInterceptionOrder();

_txtInterceptionNumber->setText(ss1.str());

_txtInterceptionNumber->setVisible(false);

// define the colors to be used

_colors[CRAFT_MIN] = dogfightInterface->getElement("craftRange")->color;

_colors[CRAFT_MAX] = dogfightInterface->getElement("craftRange")->color2;

_colors[RADAR_MIN] = dogfightInterface->getElement("radarRange")->color;

_colors[RADAR_MAX] = dogfightInterface->getElement("radarRange")->color2;

_colors[DAMAGE_MIN] = dogfightInterface->getElement("damageRange")->color;

_colors[DAMAGE_MAX] = dogfightInterface->getElement("damageRange")->color2;

_colors[BLOB_MIN] = dogfightInterface->getElement("radarDetail")->color;

_colors[RANGE_METER] = dogfightInterface->getElement("radarDetail")->color2;

_colors[DISABLED_WEAPON] = dogfightInterface->getElement("disabledWeapon")->color;

_colors[DISABLED_RANGE] = dogfightInterface->getElement("disabledWeapon")->color2;

_colors[DISABLED_AMMO] = dogfightInterface->getElement("disabledAmmo")->color;

for (unsigned int i = 0; i < _craft->getRules()->getWeapons(); ++i)

{

CraftWeapon *w = _craft->getWeapons()->at(i);

if (w == 0)

continue;

Surface *weapon = 0, *range = 0;

Text *ammo = 0;

int x1, x2;

if (i == 0)

{

weapon = _weapon1;

range = _range1;

ammo = _txtAmmo1;

x1 = 2;

x2 = 0;

}

else

{

weapon = _weapon2;

range = _range2;

ammo = _txtAmmo2;

x1 = 0;

x2 = 18;

}

// Draw weapon icon

frame = set->getFrame(w->getRules()->getSprite() + 5);

frame->setX(0);

frame->setY(0);

frame->blit(weapon);

// Draw ammo

std::wostringstream ss;

ss << w->getAmmo();

ammo->setText(ss.str());

// Draw range (1 km = 1 pixel)

Uint8 color = _colors[RANGE_METER];

range->lock();

int rangeY = range->getHeight() - w->getRules()->getRange(), connectY = 57;

for (int x = x1; x <= x1 + 18; x += 2)

{

range->setPixel(x, rangeY, color);

}

int minY = 0, maxY = 0;

if (rangeY < connectY)

{

minY = rangeY;

maxY = connectY;

}

else if (rangeY > connectY)

{

minY = connectY;

maxY = rangeY;

}

for (int y = minY; y <= maxY; ++y)

{

range->setPixel(x1 + x2, y, color);

}

for (int x = x2; x <= x2 + 2; ++x)

{

range->setPixel(x, connectY, color);

}

range->unlock();

}

if (!(_craft->getRules()->getWeapons() > 0 && _craft->getWeapons()->at(0) != 0))

{

_weapon1->setVisible(false);

_range1->setVisible(false);

_txtAmmo1->setVisible(false);

}

if (!(_craft->getRules()->getWeapons() > 1 && _craft->getWeapons()->at(1) != 0))

{

_weapon2->setVisible(false);

_range2->setVisible(false);

_txtAmmo2->setVisible(false);

}

// Draw damage indicator.

frame = set->getFrame(_craft->getRules()->getSprite() + 11);

frame->setX(0);

frame->setY(0);

frame->blit(_damage);

_craftDamageAnimTimer->onTimer((StateHandler)&DogfightState::animateCraftDamage);

// don't set these variables if the ufo is already engaged in a dogfight

if (!_ufo->getEscapeCountdown())

{

_ufo->setFireCountdown(0);

_ufo->setEscapeCountdown(_ufo->getRules()->getBreakOffTime() + RNG::generate(0, _ufo->getRules()->getBreakOffTime()) - 30 * _game->getSavedGame()->getDifficultyCoefficient());

}

// technically this block is redundant, but i figure better to initialize the variables as SOMETHING

if (_craft->getRules()->getWeapons() > 0 && _craft->getWeapons()->at(0) != 0)

{

_w1FireInterval = _craft->getWeapons()->at(0)->getRules()->getStandardReload();

}

if (_craft->getRules()->getWeapons() > 1 && _craft->getWeapons()->at(1) != 0)

{

_w2FireInterval = _craft->getWeapons()->at(1)->getRules()->getStandardReload();

}

// Set UFO size - going to be moved to Ufo class to implement simultaneous dogfights.

std::string ufoSize = _ufo->getRules()->getSize();

if (ufoSize.compare("STR_VERY_SMALL") == 0)

{

_ufoSize = 0;

}

else if (ufoSize.compare("STR_SMALL") == 0)

{

_ufoSize = 1;

}

else if (ufoSize.compare("STR_MEDIUM_UC") == 0)

{

_ufoSize = 2;

}

else if (ufoSize.compare("STR_LARGE") == 0)

{

_ufoSize = 3;

}

else

{

_ufoSize = 4;

}

// Get crafts height. Used for damage indication.

int x =_damage->getWidth() / 2;

for (int y = 0; y < _damage->getHeight(); ++y)

{

Uint8 pixelColor = _damage->getPixel(x, y);

if (pixelColor >= _colors[CRAFT_MIN] && pixelColor < _colors[CRAFT_MAX])

{

++_craftHeight;

}

}

drawCraftDamage();

// Used for weapon toggling.

_weapon1->onMouseClick((ActionHandler)&DogfightState::weapon1Click);

_weapon2->onMouseClick((ActionHandler)&DogfightState::weapon2Click);

{

if (!_endDogfight)

{

update();

_craftDamageAnimTimer->think(this, 0);

}

if (!_craft->isInDogfight() || _craft->getDestination() != _ufo || _ufo->getStatus() == Ufo::LANDED)

{

endDogfight();

}

{

if (_minimized)

{

return;

}

--_currentCraftDamageColor;

if (_currentCraftDamageColor < _colors[DAMAGE_MIN])

{

_currentCraftDamageColor = _colors[DAMAGE_MAX];

}

drawCraftDamage();

{

if (_craft->getDamagePercentage() != 0)

{

if (!_craftDamageAnimTimer->isRunning())

{

_craftDamageAnimTimer->start();

if (_currentCraftDamageColor < _colors[DAMAGE_MIN])

{

_currentCraftDamageColor = _colors[DAMAGE_MIN];

}

}

int damagePercentage = _craft->getDamagePercentage();

int rowsToColor = (int)floor((double)_craftHeight * (double)(damagePercentage / 100.));

if (rowsToColor == 0)

{

return;

}

int rowsColored = 0;

bool rowColored = false;

for (int y = 0; y < _damage->getHeight(); ++y)

{

rowColored = false;

for (int x = 0; x < _damage->getWidth(); ++x)

{

int pixelColor = _damage->getPixel(x, y);

if (pixelColor >= _colors[DAMAGE_MIN] && pixelColor <= _colors[DAMAGE_MAX])

{

_damage->setPixel(x, y, _currentCraftDamageColor);

rowColored = true;

}

if (pixelColor >= _colors[CRAFT_MIN] && pixelColor < _colors[CRAFT_MAX])

{

_damage->setPixel(x, y, _currentCraftDamageColor);

rowColored = true;

}

}

if (rowColored)

{

++rowsColored;

}

if (rowsColored == rowsToColor)

{

break;

}

}

}

{

// Animate radar waves and other stuff.

for (int x = 0; x < _window->getWidth(); ++x)

{

for (int y = 0; y < _window->getHeight(); ++y)

{

Uint8 radarPixelColor = _window->getPixel(x, y);

if (radarPixelColor >= _colors[RADAR_MIN] && radarPixelColor < _colors[RADAR_MAX])

{

++radarPixelColor;

if (radarPixelColor >= _colors[RADAR_MAX])

{

radarPixelColor = _colors[RADAR_MIN];

}

_window->setPixel(x, y, radarPixelColor);

}

}

}

_battle->clear();

// Draw UFO.

if (!_ufo->isDestroyed())

{

drawUfo();

}

// Draw projectiles.

for (std::vector<CraftWeaponProjectile*>::iterator it = _projectiles.begin(); it != _projectiles.end(); ++it)

{

drawProjectile((*it));

}

// Clears text after a while

if (_timeout == 0)

{

_txtStatus->setText(L"");

}

else

{

_timeout--;

}

// Animate UFO hit.

bool lastHitAnimFrame = false;

if (_animatingHit && _ufo->getHitFrame() > 0)

{

_ufo->setHitFrame(_ufo->getHitFrame() - 1);

if (_ufo->getHitFrame() == 0)

{

_animatingHit = false;

lastHitAnimFrame = true;

}

}

// Animate UFO crash landing.

if (_ufo->isCrashed() && _ufo->getHitFrame() == 0 && !lastHitAnimFrame)

{

--_ufoSize;

}

{

bool finalRun = false;

// Check if craft is not low on fuel when window minimized, and

// Check if crafts destination hasn't been changed when window minimized.

Ufo* u = dynamic_cast<Ufo*>(_craft->getDestination());

if (u != _ufo || !_craft->isInDogfight() || _craft->getLowFuel() || (_minimized && _ufo->isCrashed()))

{

endDogfight();

return;

}

if (!_minimized)

{

animate();

if (!_ufo->isCrashed() && !_ufo->isDestroyed() && !_craft->isDestroyed() && !_ufo->getInterceptionProcessed())

{

_ufo->setInterceptionProcessed(true);

int escapeCounter = _ufo->getEscapeCountdown();

if (escapeCounter > 0 )

{

escapeCounter--;

_ufo->setEscapeCountdown(escapeCounter);

// Check if UFO is breaking off.

if (escapeCounter == 0)

{

_ufo->setSpeed(_ufo->getRules()->getMaxSpeed());

}

}

if (_ufo->getFireCountdown() > 0)

{

_ufo->setFireCountdown(_ufo->getFireCountdown() - 1);

}

}

}

// Crappy craft is chasing UFO.

if (_ufo->getSpeed() > _craft->getRules()->getMaxSpeed())

{

_ufoBreakingOff = true;

finalRun = true;

setStatus("STR_UFO_OUTRUNNING_INTERCEPTOR");

}

else

{

_ufoBreakingOff = false;

}

bool projectileInFlight = false;

if (!_minimized)

{

int distanceChange = 0;

// Update distance

if (!_ufoBreakingOff)

{

if (_currentDist < _targetDist && !_ufo->isCrashed() && !_craft->isDestroyed())

{

distanceChange = 4;

if (_currentDist + distanceChange >_targetDist)

{

distanceChange = _targetDist - _currentDist;

}

}

else if (_currentDist > _targetDist && !_ufo->isCrashed() && !_craft->isDestroyed())

{

distanceChange = -2;

}

// don't let the interceptor mystically push or pull its fired projectiles

for (std::vector<CraftWeaponProjectile*>::iterator it = _projectiles.begin(); it != _projectiles.end(); ++it)

{

if ((*it)->getGlobalType() != CWPGT_BEAM && (*it)->getDirection() == D_UP) (*it)->setPosition((*it)->getPosition() + distanceChange);

}

}

else

{

distanceChange = 4;

// UFOs can try to outrun our missiles, don't adjust projectile positions here

// If UFOs ever fire anything but beams, those positions need to be adjust here though.

}

_currentDist += distanceChange;

std::wostringstream ss;

ss << _currentDist;

_txtDistance->setText(ss.str());

// Move projectiles and check for hits.

for (std::vector<CraftWeaponProjectile*>::iterator it = _projectiles.begin(); it != _projectiles.end(); ++it)

{

CraftWeaponProjectile *p = (*it);

p->move();

// Projectiles fired by interceptor.

if (p->getDirection() == D_UP)

{

// Projectile reached the UFO - determine if it's been hit.

if (((p->getPosition() >= _currentDist) || (p->getGlobalType() == CWPGT_BEAM && p->toBeRemoved())) && !_ufo->isCrashed() && !p->getMissed())

{

// UFO hit.

if (RNG::percent((p->getAccuracy() * (100 + 300 / (5 - _ufoSize)) + 100) / 200))

{

// Formula delivered by Volutar

int damage = RNG::generate(p->getDamage() / 2, p->getDamage());

_ufo->setDamage(_ufo->getDamage() + damage);

if (_ufo->isCrashed())

{

_ufo->setShotDownByCraftId(_craft->getUniqueId());

_ufo->setSpeed(0);

// if the ufo got destroyed here, these no longer apply

_ufoBreakingOff = false;

finalRun = false;

_end = false;

}

if (_ufo->getHitFrame() == 0)

{

_animatingHit = true;

_ufo->setHitFrame(3);

}

setStatus("STR_UFO_HIT");

_game->getMod()->getSound("GEO.CAT", Mod::UFO_HIT)->play();

p->remove();

}

// Missed.

else

{

if (p->getGlobalType() == CWPGT_BEAM)

{

p->remove();

}

else

{

p->setMissed(true);

}

}

}

// Check if projectile passed it's maximum range.

if (p->getGlobalType() == CWPGT_MISSILE && p->getPosition() / 8 >= p->getRange())

{

p->remove();

}

else if (!_ufo->isCrashed())

{

projectileInFlight = true;

}

}

// Projectiles fired by UFO.

else if (p->getDirection() == D_DOWN)

{

if (p->getGlobalType() == CWPGT_MISSILE || (p->getGlobalType() == CWPGT_BEAM && p->toBeRemoved()))

{

if (RNG::percent(p->getAccuracy()))

{

// Formula delivered by Volutar

int damage = RNG::generate(0, _ufo->getRules()->getWeaponPower());

if (damage)

{

_craft->setDamage(_craft->getDamage() + damage);

drawCraftDamage();

setStatus("STR_INTERCEPTOR_DAMAGED");

_game->getMod()->getSound("GEO.CAT", Mod::INTERCEPTOR_HIT)->play(); //10

if (_mode == _btnCautious && _craft->getDamagePercentage() >= 50)

{

_targetDist = STANDOFF_DIST;

}

}

}

p->remove();

}

}

}

// Remove projectiles that hit or missed their target.

for (std::vector<CraftWeaponProjectile*>::iterator it = _projectiles.begin(); it != _projectiles.end();)

{

if ((*it)->toBeRemoved() == true || ((*it)->getMissed() == true && (*it)->getPosition() <= 0))

{

delete *it;

it = _projectiles.erase(it);

}

else

{

++it;

}

}

// Handle weapons and craft distance.

for (unsigned int i = 0; i < _craft->getRules()->getWeapons(); ++i)

{

CraftWeapon *w = _craft->getWeapons()->at(i);

if (w == 0)

{

continue;

}

int wTimer;

if (i == 0)

{

wTimer = _w1FireCountdown;

}

else

{

wTimer = _w2FireCountdown;

}

// Handle weapon firing

if (wTimer == 0 && _currentDist <= w->getRules()->getRange() * 8 && w->getAmmo() > 0 && _mode != _btnStandoff

&& _mode != _btnDisengage && !_ufo->isCrashed() && !_craft->isDestroyed())

{

if (i == 0)

{

if (_weapon1Enabled)

{

fireWeapon1();

projectileInFlight = true;

}

}

else

{

if (_weapon2Enabled)

{

fireWeapon2();

projectileInFlight = true;

}

}

}

else if (wTimer > 0)

{

if (i == 0)

{

_w1FireCountdown--;

}

else

{

_w2FireCountdown--;

}

}

if (w->getAmmo() == 0 && !projectileInFlight && !_craft->isDestroyed())

{

// Handle craft distance according to option set by user and available ammo.

if (_mode == _btnCautious)

{

minimumDistance();

}

else if (_mode == _btnStandard)

{

maximumDistance();

}

}

}

// Handle UFO firing.

if (_currentDist <= _ufo->getRules()->getWeaponRange() * 8 && !_ufo->isCrashed() && !_craft->isDestroyed())

{

if (_ufo->getShootingAt() == 0)

{

_ufo->setShootingAt(_interceptionNumber);

}

if (_ufo->getShootingAt() == _interceptionNumber)

{