/*
 * Author: Ruthberg
 * Calculates the range card data
 *
 * Arguments:
 * 0: Scope base angle <NUMBER>
 * 1: Bore height <NUMBER>
 * 2: air friction <NUMBER>
 * 3: muzzle velocity <NUMBER>
 * 4: temperature <NUMBER>
 * 5: barometric pressure <NUMBER>
 * 6: relative humidity <NUMBER>
 * 7: simulation steps <NUMBER>
 * 8: wind speed <NUMBER>
 * 9: target speed <NUMBER>
 * 10: target range <NUMBER>
 * 11: ballistic coefficient <NUMBER>
 * 12: drag model <NUMBER>
 * 13: atmosphere model <STRING>
 * 14: transonicStabilityCoef <NUMBER>
 * 15: Range Card Slot <NUMBER>
 * 16: Use advanced ballistics config? <BOOL>
 *
 * Return Value:
 * Nothing
 *
 * Example:
 * call ace_rangecard_fnc_calculateRangeCard
 *
 * Public: No
 */
#include "script_component.hpp"
params [
    "_scopeBaseAngle", "_boreHeight", "_airFriction", "_muzzleVelocity",
    "_temperature", "_barometricPressure", "_relativeHumidity", "_simSteps",
    "_windSpeed", "_targetSpeed", "_targetRange", "_bc", "_dragModel", "_atmosphereModel",
    "_transonicStabilityCoef", "_rangeCardSlot", "_useABConfig"
];

GVAR(rangeCardDataMVs) set [_rangeCardSlot, format[" %1", round(_muzzleVelocity)]];

private ["_tx", "_tz", "_lastBulletPos", "_bulletPos", "_bulletVelocity", "_bulletAccel", "_bulletSpeed", "_gravity", "_deltaT", "_speedOfSound"];
_tx = 0;
_tz = 0;
_lastBulletPos = [0, 0, 0];
_bulletPos = [0, 0, 0];
_bulletVelocity = [0, 0, 0];
_bulletAccel = [0, 0, 0];
_bulletSpeed = 0;
_gravity = [0, sin(_scopeBaseAngle) * -9.80665, cos(_scopeBaseAngle) * -9.80665];
_deltaT = 1 / _simSteps;
_speedOfSound = 0;
if (missionNamespace getVariable [QEGVAR(advanced_ballistics,enabled), false]) then {
    _speedOfSound = _temperature call EFUNC(weather,calculateSpeedOfSound);
};

private ["_elevation", "_windage", "_lead", "_TOF", "_trueVelocity", "_trueSpeed"];
_elevation = 0;
_windage = 0;
_lead = 0;
_TOF = 0;
_trueVelocity = [0, 0, 0];
_trueSpeed = 0;

private ["_n", "_range"];
_n = 0;
_range = 0;

if (_useABConfig) then {
    _bc = parseNumber(("ace_advanced_ballistics" callExtension format["atmosphericCorrection:%1:%2:%3:%4:%5", _bc, _temperature, _barometricPressure, _relativeHumidity, _atmosphereModel]));
};

private ["_airFrictionCoef", "_airDensity"];
_airFrictionCoef = 1;
if (!_useABConfig && (missionNamespace getVariable [QEGVAR(advanced_ballistics,enabled), false])) then {
    _airDensity = [_temperature, _barometricPressure, _relativeHumidity] call EFUNC(weather,calculateAirDensity);
    _airFrictionCoef = _airDensity / 1.22498;
};

private ["_speedTotal", "_stepsTotal", "_speedAverage"];
_speedTotal = 0;
_stepsTotal = 0;
_speedAverage = 0;

_bulletPos set [0, 0];
_bulletPos set [1, 0];
_bulletPos set [2, -(_boreHeight / 100)];

_bulletVelocity set [0, 0];
_bulletVelocity set [1, Cos(_scopeBaseAngle) * _muzzleVelocity];
_bulletVelocity set [2, Sin(_scopeBaseAngle) * _muzzleVelocity];

while {_TOF < 6 && (_bulletPos select 1) < _targetRange} do {
    _bulletSpeed = vectorMagnitude _bulletVelocity;

    _speedTotal = _speedTotal + _bulletSpeed;
    _stepsTotal = _stepsTotal + 1;
    _speedAverage = (_speedTotal / _stepsTotal);

    if (_transonicStabilityCoef < 1.0 && _speedAverage > 450 && _bulletSpeed < _speedOfSound) exitWith {};

    _trueVelocity = _bulletVelocity vectorDiff [-_windSpeed, 0, 0];
    _trueSpeed = vectorMagnitude _trueVelocity;

    if (_useABConfig) then {
        private _drag = parseNumber(("ace_advanced_ballistics" callExtension format["retard:%1:%2:%3", _dragModel, _bc, _trueSpeed]));
        _bulletAccel = (vectorNormalized _trueVelocity) vectorMultiply (-1 * _drag);
    } else {
        _bulletAccel = _trueVelocity vectorMultiply (_trueSpeed * _airFriction * _airFrictionCoef);
    };

    _bulletAccel = _bulletAccel vectorAdd _gravity;

    _lastBulletPos = _bulletPos;
    _bulletPos = _bulletPos vectorAdd (_bulletVelocity vectorMultiply (_deltaT * 0.5));
    _bulletVelocity = _bulletVelocity vectorAdd (_bulletAccel vectorMultiply _deltaT);
    _bulletPos = _bulletPos vectorAdd (_bulletVelocity vectorMultiply (_deltaT * 0.5));
    
    if (atan((_bulletPos select 2) / (abs(_bulletPos select 1) + 1)) < -2.254) exitWith {};

    _TOF = _TOF + _deltaT;

    _range = GVAR(rangeCardStartRange) + _n * GVAR(rangeCardIncrement);
    if ((_bulletPos select 1) >= _range && _range <= GVAR(rangeCardEndRange)) then {
        if (_range != 0) then {
            _tx = (_lastBulletPos select 0) + (_range - (_lastBulletPos select 1)) * ((_bulletPos select 0) - (_lastBulletPos select 0)) / ((_bulletPos select 1) - (_lastBulletPos select 1));
            _tz = (_lastBulletPos select 2) + (_range - (_lastBulletPos select 1)) * ((_bulletPos select 2) - (_lastBulletPos select 2)) / ((_bulletPos select 1) - (_lastBulletPos select 1));
            _elevation = - atan(_tz / _range);
            _windage = - atan(_tx / _range);
            _lead = (_targetSpeed * _TOF) / (Tan(3.38 / 60) * _range);
        };
        private ["_elevationString", "_windageString", "_leadString"];
        _elevationString = Str(round(-_elevation * 60 / 3.38 * 10) / 10);
        if (_elevationString == "0") then {
            _elevationString = "-0.0";
        };
        if (_elevationString find "." == -1) then {
            _elevationString = _elevationString + ".0";
        };
        _windageString = Str(round(_windage * 60 / 3.38 * 10) / 10);
        if (_windageString find "." == -1) then {
            _windageString = _windageString + ".0";
        };
        _leadString = Str(round(_lead * 10) / 10);
        if (_leadString find "." == -1) then {
            _leadString = _leadString + ".0";
        };
        (GVAR(rangeCardDataElevation) select _rangeCardSlot) set [_n, _elevationString];
        (GVAR(rangeCardDataWindage) select _rangeCardSlot) set [_n, _windageString];
        (GVAR(rangeCardDataLead) select _rangeCardSlot) set [_n, _leadString];
        _n = _n + 1;
    };
};