#include <Autocoders/Python/test/enum_xml/Component1ComponentAc.hpp>
#include <Autocoders/Python/test/enum_xml/Component1Impl.hpp>
#include <Autocoders/Python/test/enum_xml/Enum1EnumAc.hpp>
#include <Autocoders/Python/test/enum_xml/Port1PortAc.hpp>
#include <Autocoders/Python/test/enum_xml/Serial1SerializableAc.hpp>

#include <Fw/Types/SerialBuffer.hpp>
#include <Fw/Types/Assert.hpp>

#include "Fw/Test/UnitTestAssert.hpp"
#include "STest/Pick/Pick.hpp"
#include "STest/Random/Random.hpp"
#include "gtest/gtest.h"

#include <iostream>
#include <string.h>

using namespace std;

// Component instance pointers
Example::ExampleEnumImpl* inst1 = 0;
Example::ExampleEnumImpl* inst2 = 0;

void constructArchitecture(void) {

    // Instantiate the inst1 and inst2
    inst1   = new Example::ExampleEnumImpl("inst1");
    inst2   = new Example::ExampleEnumImpl("inst2");

    // Connect inst1 to inst2
    inst1->set_EnumOut_OutputPort(0, inst2->get_EnumIn_InputPort(0));

    // Connect inst2 to inst1
    inst2->set_EnumOut_OutputPort(0, inst1->get_EnumIn_InputPort(0));

    // Instantiate components
    inst1->init(100);
    inst2->init(100);

}

Example::Enum1::t getEnumConstant() {
    Example::Enum1::t c = Example::Enum1::Item1;
    const U32 i = STest::Pick::lowerUpper(0, 4);

    switch(i) {
        case 0:
            c = Example::Enum1::Item1;
            break;
        case 1:
            c = Example::Enum1::Item2;
            break;
        case 2:
            c = Example::Enum1::Item3;
            break;
        case 3:
            c = Example::Enum1::Item4;
            break;
        case 4:
            c = Example::Enum1::Item5;
            break;
        default:
            FW_ASSERT(0, i);
            break;
    }

    return c;
}

Example::Enum1::t getNonnegativeConstant() {
    Example::Enum1::t c = Example::Enum1::Item1;
    const U32 i = STest::Pick::lowerUpper(0, 2);

    switch(i) {
        case 0:
            c = Example::Enum1::Item2;
            break;
        case 1:
            c = Example::Enum1::Item3;
            break;
        case 2:
            c = Example::Enum1::Item4;
            break;
        default:
            FW_ASSERT(0, i);
            break;
    }

    return c;
}

Example::Enum1::t getNegativeConstant() {
    Example::Enum1::t c = Example::Enum1::Item1;
    const U32 i = STest::Pick::lowerUpper(0, 1);

    switch(i) {
        case 0:
            c = Example::Enum1::Item1;
            break;
        case 1:
            c = Example::Enum1::Item5;
            break;
        default:
            FW_ASSERT(0, i);
            break;
    }

    return c;
}

Example::Enum1 getEnum() {
    const Example::Enum1 e = getEnumConstant();
    return e;
}

Example::Enum1 getEnumFromI32() {
    Example::Enum1 e = getEnumConstant();
    e = static_cast<I32>(getEnumConstant());
    return e;
}

Example::Enum1 getEnumFromU32() {
    Example::Enum1 e = getNonnegativeConstant();
    e = static_cast<U32>(getNonnegativeConstant());
    return e;
}

void checkAssertionFailure(
    const ::Test::UnitTestAssert& uta,
    const U32 expectedLineNumber,
    const U32 expectedArg1
) {
  ASSERT_TRUE(uta.assertFailed());
#if FW_ASSERT_LEVEL == FW_FILEID_ASSERT
  NATIVE_UINT_TYPE file = 0;
#else
  U8 file[256];
  memset(file, 0, sizeof(file));
#endif
  NATIVE_UINT_TYPE lineNo;
  NATIVE_UINT_TYPE numArgs;
  AssertArg arg1;
  AssertArg arg2;
  AssertArg arg3;
  AssertArg arg4;
  AssertArg arg5;
  AssertArg arg6;
  uta.retrieveAssert(file, lineNo, numArgs, arg1, arg2, arg3, arg4, arg5, arg6);
  ASSERT_EQ(expectedLineNumber, lineNo);
  ASSERT_EQ(1U, numArgs);
  ASSERT_EQ(expectedArg1, arg1);
}

TEST(EnumXML, InvalidNegativeConstant) {
  ::Test::UnitTestAssert uta;
  Example::Enum1 enum1 = getEnum();
  // Get a valid negative constant
  const I32 negativeConstant = getNegativeConstant();
  const U32 expectedLineNumber = 65;
  // Turn it into a U32
  const U32 expectedArg1 = negativeConstant;
  // As a U32, the constant is not valid
  // This should cause an assertion failure
  enum1 = expectedArg1;
  checkAssertionFailure(uta, expectedLineNumber, expectedArg1);
}

TEST(EnumXML, InvalidConstant) {
  ::Test::UnitTestAssert uta;
  Example::Enum1 enum1 = getEnum();
  // Get an invalid constant
  const I32 invalidConstant = 42;
  // This should cause an assertion failure
  enum1 = invalidConstant;
  const U32 expectedLineNumber = 58;
  const U32 expectedArg1 = invalidConstant;
  checkAssertionFailure(uta, expectedLineNumber, expectedArg1);
}

TEST(EnumXML, OK) {

    Example::Enum1 enum1;
    Example::Enum1 enum2;
    Example::Enum1 enum3;
    Example::Serial1 serial1;

    enum1 = getEnumFromI32();
    cout << "Created first enum: " << enum1 << endl;

    enum2 = getEnum();
    cout << "Created second enum: " << enum2 << endl;

    enum3 = getEnumFromU32();
    cout << "Created third enum: " << enum3 << endl;

    // Save copy of enums to test against post-serialization
    Example::Enum1 enum1Save = enum1;
    Example::Enum1 enum2Save = enum2;

    int serial_arg1 = 0;

    int serial_arg2 = 0;

    // Serialize enums
    U8 buffer1[1024];
    U8 buffer2[1024];
    Fw::SerialBuffer enumSerial1 = Fw::SerialBuffer(buffer1, sizeof(buffer1));
    Fw::SerialBuffer enumSerial2 = Fw::SerialBuffer(buffer2, sizeof(buffer2));
    ASSERT_EQ(enumSerial1.serialize(enum1), Fw::FW_SERIALIZE_OK);
    cout << "Serialized enum1" << endl;

    ASSERT_EQ(enumSerial2.serialize(enum2), Fw::FW_SERIALIZE_OK);
    cout << "Serialized enum2" << endl;

    cout << "Serialized enums" << endl;

    // Deserialize enums
    ASSERT_EQ(enumSerial1.deserialize(enum1Save), Fw::FW_SERIALIZE_OK);
    cout << "Deserialized enum1" << endl;

    ASSERT_EQ(enumSerial2.deserialize(enum2Save), Fw::FW_SERIALIZE_OK);
    cout << "Deserialized enum2" << endl;

    ASSERT_EQ(enum1, enum1Save);
    ASSERT_FALSE(enum1 != enum1Save);
    cout << "Successful enum1 check" << endl;

    ASSERT_EQ(enum2, enum2Save);
    ASSERT_FALSE(enum2 != enum2Save);
    cout << "Successful enum2 check" << endl;

    cout << "Deserialized enums" << endl;

    // Create serializable and test that enum is saved
    serial1 = Example::Serial1(serial_arg1, serial_arg2, enum2);
    ASSERT_EQ(serial1.getMember3(), enum2);
    cout << "Created serializable with enum arg" << endl;

    cout << "Created serializable" << endl;

    // Invoke ports to test enum usage
    cout << "Invoking inst1..." << endl;
    inst1->get_ExEnumIn_InputPort(0)->invoke(enum1, serial1);
    inst1->doDispatch();
    inst1->get_ExEnumIn_InputPort(0)->invoke(enum2, serial1);
    inst1->doDispatch();

    cout << "Invoking inst2..." << endl;
    inst2->get_ExEnumIn_InputPort(0)->invoke(enum1, serial1);
    inst2->doDispatch();
    inst2->get_ExEnumIn_InputPort(0)->invoke(enum2, serial1);
    inst2->doDispatch();

    cout << "Invoked ports" << endl;
}

int main(int argc, char* argv[]) {
    ::testing::InitGoogleTest(&argc, argv);
    STest::Random::seed();

    constructArchitecture();

    int status = RUN_ALL_TESTS();

    cout << "Deleting components..." << endl;
    delete inst1;
    delete inst2;

    cout << "Completed..." << endl;

    return status;
}