//////////////////////////////////////////////////////////////////////////////
/// deco_volume_test.cpp
/// Unit test for gas consumption c code.
/// Copyright (c) 2015, JD Gascuel, HeinrichsWeikamp, all right reserved.
//////////////////////////////////////////////////////////////////////////////
//  HISTORY
// 2015-05-27 jDG: Creation for gas volum re-introduction in hwOS 1.82

extern "C" {
#   include "p2_deco.c"
}

#include <gtest/gtest.h>
#include <iostream>

//////////////////////////////////////////////////////////////////////////////
/// \brief Defines a default OC gas list
static void setup_gas()
{
    char_I_first_gas = 1;

#define DEFINE_GAS(gas, o2, he, depth, role)    \
    char_I_deco_N2_ratio  [gas-1] = 100 - o2 - he;  \
    char_I_deco_He_ratio  [gas-1] = he;             \
    char_I_deco_gas_change[gas-1] = depth;

    DEFINE_GAS(1, 21,  0, 0, 1);    // Gas#1 : Air       FIRST
    DEFINE_GAS(2, 18, 30, 0, 2);    // Gas#2 : Tx18/30   TRAVEL
    DEFINE_GAS(3, 80,  0, 9, 3);    // Gas#3 : Nx80 @ 9m DECO
    DEFINE_GAS(4, 21, 0,  0, 0);    // Gas#2 : air @ 10m DISABLED
    DEFINE_GAS(5, 21, 0,  0, 0);    // Gas#2 : air @ 40m DISABLED
}

//////////////////////////////////////////////////////////////////////////////
/// \brief Define a default deco plan.
static void setup_plan(const char* stops,
                       const char* gas)
{
    int depth = 3 * (stops ? strlen(stops) : 0);

    if( ! depth )
        std::cout << "             no deco" << std::endl;

    int s = 0;
    while( depth > 0 && s < NUM_STOPS ) {
        std::cout << "             " << std::setw(2) << int(stops[s]) << "' @ " << depth << "m" << std::endl;

        char_O_deco_time [s] = stops[s];
        char_O_deco_depth[s] = depth;
        char_O_deco_gas  [s] = gas ? gas[s] : 1; // Gas#1 by default
        ++s;
        depth -= 3;
    }
    // Done
    for(; s<NUM_STOPS; ++s) {
        char_O_deco_time [s] = 0;
        char_O_deco_depth[s] = 0;
        char_O_deco_gas  [s] = 0;
    }
}

static void setup_dive(int bottom, int depth,
                       const char* stops = 0, const char* gas = 0)
{
    setup_gas();
    setup_plan(stops, gas);

    char_I_bottom_depth = depth;
    char_I_bottom_time  = bottom;
}

//////////////////////////////////////////////////////////////////////////////
/// \brief Gas consumption at a fixed depth
static float fixed(int rmv, int time, int depth) {
    return rmv * time * (1 + 0.1f*depth);
}

TEST(gas_volume, fixed)
{
    EXPECT_EQ(20*30*1, fixed(20,30, 0));    // 30' @  0m
    EXPECT_EQ(20*30*5, fixed(20,30,40));    // 30' @ 40m
}

//////////////////////////////////////////////////////////////////////////////
/// \brief Gas consumption during an ascent at 10m/min.
static float ascent(int rmv, int oldDepth, int newDepth)
{
    return rmv
         * abs(oldDepth-newDepth)*0.1f          // Ascent time
         * (1 + 0.05f*(oldDepth + newDepth));   // Avg pressure.
}

TEST(gas_volume, ascent)
{
    EXPECT_EQ(0,          ascent(20, 30, 30));  // 30m -> 30m : no time, no conso
    EXPECT_EQ(20*4*(1+2), ascent(20, 40,  0));  // 40m ->  0m : 4min, avg 20m
    EXPECT_EQ(20*4*(1+2), ascent(20,  0, 40));  // 0m  -> 40m : 4min, avg 20m
}

static void check_volumes(float G1, const char* L1,
                          float G2, const char* L2,
                          float G3, const char* L3,
                          float G4, const char* L4,
                          float G5, const char* L5)
{
    std::cout << "             " << std::setw(6) << G1 << " = " << L1 << std::endl;
    std::cout << "             " << std::setw(6) << G2 << " = " << L2 << std::endl;
    std::cout << "             " << std::setw(6) << G3 << " = " << L3 << std::endl;
    std::cout << "             " << std::setw(6) << G4 << " = " << L4 << std::endl;
    std::cout << "             " << std::setw(6) << G5 << " = " << L5 << std::endl;

    EXPECT_NEAR(G1, int_O_gas_volumes[0], 1) << L1;
    EXPECT_NEAR(G2, int_O_gas_volumes[1], 1) << L2;
    EXPECT_NEAR(G3, int_O_gas_volumes[2], 1) << L3;
    EXPECT_NEAR(G4, int_O_gas_volumes[3], 1) << L4;
    EXPECT_NEAR(G5, int_O_gas_volumes[4], 1) << L5;
}

//////////////////////////////////////////////////////////////////////////////
// v1.82 ZH-L16+GF, OC, 30%/85%
TEST(gas_volume, OC_13min30m)
{
    char_I_const_ppO2 = 0;  // OC
    setup_dive(13, 30);     // 13' @ 30m --> no deco

    ASSERT_NO_THROW( deco_gas_volumes() );
    check_volumes(fixed(20,13,30) + ascent(20,30,0), "Gas1: 1190 L",
                  0, "",
                  0, "",
                  0, "",
                  0, "");
}

//////////////////////////////////////////////////////////////////////////////
// v1.82 ZH-L16+GF, OC, 30%/85%
TEST(gas_volume, OC_15min30m)
{
    char_I_const_ppO2 = 0;  // OC
    char stops[] = {1, 0};
    char gas[]   = {3, 0};
    setup_dive(15, 30, stops, gas);     // 15' @ 30m --> 1min at 3m

    ASSERT_NO_THROW( deco_gas_volumes() );
    check_volumes(fixed(20,15,30) + ascent(20,30,3), "Gas1: 1343 L",
                  0, "",
                  fixed(20, 1, 3) + ascent(20, 3,0), "Gas3: 33",
                  0, "",
                  0, "");
}

//////////////////////////////////////////////////////////////////////////////
// v1.82 ZH-L16+GF, OC, 30%/85%
TEST(gas_volume, OC_29min30m)
{
    char_I_const_ppO2 = 0;  // OC
    char stops[] = {1, 1, 2, 4, 0};
    char gas[]   = {1, 3, 3, 3, 0};
    setup_dive(29, 30, stops, gas);     // 29' @ 30m --> 1' 1' 2' 4'

    ASSERT_NO_THROW( deco_gas_volumes() );
    check_volumes(fixed(20,29,30) + ascent(20,30,12) +
                  fixed(20, 1,12) + ascent(20,12, 9), "Gas1: 2488 L",
                  0, "",
                  fixed(20, 1, 9) + ascent(20, 9, 6) +
                  fixed(20, 2, 6) + ascent(20, 6, 3) +
                  fixed(20, 4, 3) + ascent(20, 3, 0), "Gas3: 232 L",
                  0, "",
                  0, "");
}

//////////////////////////////////////////////////////////////////////////////
// v1.82 ZH-L16+GF, OC, 30%/85%
TEST(gas_volume, OC_15min60m)
{
    char_I_const_ppO2 = 0;  // OC
    char stops[] = {2, 1, 2, 4, 3, 4, 9, 0};
    char gas[]   = {1, 1, 1, 1, 3, 3, 3, 0};
    setup_dive(15, 60, stops, gas);     // 15' @ 60m --> DTR 32'

    ASSERT_NO_THROW( deco_gas_volumes() );
    check_volumes(fixed(20,15,60) + ascent(20,60,21) +
                  fixed(20, 2,21) + ascent(20,21,18) +
                  fixed(20, 1,18) + ascent(20,18,15) +
                  fixed(20, 2,15) + ascent(20,15,12) +
                  fixed(20, 4,12) + ascent(20,12, 9), "Gas1: 3010 L",
                  0, "",
                  fixed(20, 3, 9) + ascent(20, 9, 6) +
                  fixed(20, 4, 6) + ascent(20, 6, 3) +
                  fixed(20, 9, 3) + ascent(20, 3, 0), "Gas3: 502 L",
                  0, "",
                  0, "");
}

//////////////////////////////////////////////////////////////////////////////
// v1.82 ZH-L16+GF, CCR, 30%/85%
TEST(gas_volume, CCR_23min30m)
{
    char_I_const_ppO2 = 140;// SP 1.4 bar
    setup_dive(23, 30);     // 23' @ 30m --> no deco / no BAIL deco

    ASSERT_NO_THROW( deco_gas_volumes() );
    check_volumes(/*NO BTM CONSO*/ ascent(20,30,0), "Gas1: 150 L",
                  0, "",
                  0, "",
                  0, "",
                  0, "");
}

//////////////////////////////////////////////////////////////////////////////
// v1.82 ZH-L16+GF, CCR, 30%/85%
TEST(gas_volume, CCR_25min30m)
{
    char_I_const_ppO2 = 140;// SP 1.4 bar
    char stops[] = {1, 0};
    char gas[]   = {3, 0};
    setup_dive(25, 30, stops, gas);     // 25' @ 30m --> no deco / BAIL 1' @ 3m

    ASSERT_NO_THROW( deco_gas_volumes() );
    check_volumes(/*NO BTM CONSO*/  ascent(20,30,3), "Gas1: 143 L",
                  0, "",
                  fixed(20, 1, 3) + ascent(20, 3,0), "Gas3:  33 L",
                  0, "",
                  0, "");
}

//////////////////////////////////////////////////////////////////////////////
// v1.82 ZH-L16+GF, CCR, 30%/85%
TEST(gas_volume, CCR_45min30m)
{
    char_I_const_ppO2 = 140;    // SP 1.4 bar
    char stops[] = {1, 2, 5, 0};
    char gas[]   = {3, 3, 3, 0};
    setup_dive(45, 30, stops, gas);     // 45' @ 30m

    ASSERT_NO_THROW( deco_gas_volumes() );
    check_volumes(/*NO BTM CONSO*/  ascent(20,30, 9), "Gas1: 124 L",
                  0, "",
                  fixed(20, 1, 9) + ascent(20, 9, 6) +
                  fixed(20, 2, 6) + ascent(20, 6, 3) +
                  fixed(20, 5, 3) + ascent(20, 3, 0), "Gas3: 258 L",
                  0, "",
                  0, "");
}

//////////////////////////////////////////////////////////////////////////////
// v1.82 ZH-L16+GF, CCR, 30%/85%
TEST(gas_volume, CCR_19min51m)
{
    char_I_const_ppO2 = 140;    // SP 1.4 bar
    char stops[] = {1, 2, 2, 4, 3, 4, 9, 0};
    char gas[]   = {1, 1, 1, 1, 3, 3, 3, 0};
    setup_dive(19, 51, stops, gas);     // 19' @ 51m --> 20' CCR / 31' BAIL

    ASSERT_NO_THROW( deco_gas_volumes() );
    check_volumes(/*NO BTM CONSO*/  ascent(20,51,21) +
                  fixed(20, 1,21) + ascent(20,21,18) +
                  fixed(20, 2,18) + ascent(20,18,15) +
                  fixed(20, 2,15) + ascent(20,15,12) +
                  fixed(20, 4,12) + ascent(20,12, 9), "Gas1: 786 L",
                  0, "",
                  fixed(20, 3, 9) + ascent(20, 9, 6) +
                  fixed(20, 4, 6) + ascent(20, 6, 3) +
                  fixed(20, 9, 3) + ascent(20, 3, 0), "Gas3: 502 L",
                  0, "",
                  0, "");
}

//////////////////////////////////////////////////////////////////////////////
// v1.82 ZH-L16+GF, CCR, 30%/85%
TEST(gas_volume, CCR_15min60m)
{
    char_I_const_ppO2 = 140;    // SP 1.4 bar
    char stops[] = {2, 1, 1, 3, 5, 4, 8, 0};   // BAILOUT mode
    char gas[]   = {3, 3, 3, 3, 3, 5, 5, 0};
    setup_dive(15, 60, stops, gas);     // 15' @ 60m --> DTR 32'

    DEFINE_GAS(3, 32, 0, 40, 3);    // Gas#3 : Nx32 @ 40m DECO
    DEFINE_GAS(5, 90, 0,  6, 3);    // Gas#5 : Nx90 @  6m DECO

    ASSERT_NO_THROW( deco_gas_volumes() );
    check_volumes(/*NO BTM CONSO*/  ascent(20,60,21), "Gas1: 394 L",
                  0, "",
                  fixed(20, 2,21) + ascent(20,21,18) +
                  fixed(20, 1,18) + ascent(20,18,15) +
                  fixed(20, 1,15) + ascent(20,15,12) +
                  fixed(20, 3,12) + ascent(20,12, 9) +
                  fixed(20, 5, 9) + ascent(20, 9, 6), "Gas3: 623 L",
                  0, "",
                  fixed(20, 4, 6) + ascent(20, 6, 3) +
                  fixed(20, 8, 3) + ascent(20, 3, 0), "Gas5: 352 L");
}