// Copyright 2012 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <gtest/gtest.h>

#include "include/gestures.h"
#include "include/non_linearity_filter_interpreter.h"
#include "include/unittest_util.h"
#include "include/util.h"

const char kTestNonlinearData[] =
    "data/non_linearity_data/testing_non_linearity_data.dat";

namespace gestures {

class NonLinearityFilterTest : public ::testing::Test {};

class NonLinearityFilterInterpreterTestInterpreter : public Interpreter {
 public:
  NonLinearityFilterInterpreterTestInterpreter()
      : Interpreter(nullptr, nullptr, false) {}

  virtual void SyncInterpret(HardwareState& hwstate, stime_t* timeout) {}
};

TEST(NonLinearityFilterInterpreterTest, DisablingTest) {
  FingerState finger_state = { 0, 0, 0, 0, 35, 0, 999, 500, 1, 0 };
  HardwareState hwstate = make_hwstate(200000, 0, 2, 2, &finger_state);

  NonLinearityFilterInterpreterTestInterpreter* base =
                            new NonLinearityFilterInterpreterTestInterpreter;
  NonLinearityFilterInterpreter interpreter(nullptr, base, nullptr);
  TestInterpreterWrapper wrapper(&interpreter);

  // Nothing should change since it is disabled by default without data
  EXPECT_EQ(nullptr, wrapper.SyncInterpret(hwstate, nullptr));
  EXPECT_FLOAT_EQ(hwstate.fingers[0].position_x, 999);
  EXPECT_FLOAT_EQ(hwstate.fingers[0].position_y, 500);

  // Nothing should change even though it's "enabled" since there is no data
  interpreter.enabled_.val_ = 1;
  EXPECT_EQ(nullptr, wrapper.SyncInterpret(hwstate, nullptr));
  EXPECT_FLOAT_EQ(hwstate.fingers[0].position_x, 999);
  EXPECT_FLOAT_EQ(hwstate.fingers[0].position_y, 500);

  // Even with data loaded, if it is not enabled nothing should change
  interpreter.data_location_.val_ = kTestNonlinearData;
  interpreter.LoadData();
  interpreter.enabled_.val_ = 0;
  EXPECT_EQ(nullptr, wrapper.SyncInterpret(hwstate, nullptr));
  EXPECT_FLOAT_EQ(hwstate.fingers[0].position_x, 999);
  EXPECT_FLOAT_EQ(hwstate.fingers[0].position_y, 500);
}

TEST(NonLinearityFilterInterpreterTest, HWstateModificationTest) {
  FingerState finger_state = { 0, 0, 0, 0, 0.2, 0, 0.1, 0.3, 1, 0 };
  HardwareState hwstate = make_hwstate(200000, 0, 1, 1, &finger_state);

  NonLinearityFilterInterpreterTestInterpreter* base =
                            new NonLinearityFilterInterpreterTestInterpreter;
  NonLinearityFilterInterpreter interpreter(nullptr, base, nullptr);
  TestInterpreterWrapper wrapper(&interpreter);
  interpreter.enabled_.val_ = 1;
  interpreter.data_location_.val_ = kTestNonlinearData;
  interpreter.LoadData();

  // This reading should be modified slightly by the testing filter
  // with errors of (0.325000, -0.325000)
  EXPECT_EQ(nullptr, wrapper.SyncInterpret(hwstate, nullptr));
  EXPECT_FLOAT_EQ((float)hwstate.fingers[0].position_x, 0.1 - 0.325);
  EXPECT_FLOAT_EQ((float)hwstate.fingers[0].position_y, 0.3 + 0.325);
}

TEST(NonLinearityFilterInterpreterTest, HWstateNoChangesNeededTest) {
  FingerState finger_states[] = {
    { 0, 0, 0, 0, 0.5, 0, 0.5, 0.5, 1, 0 },
    { 0, 0, 0, 0, 0.12, 0, 0.78, 0.34, 2, 0 },
  };

  HardwareState hwstates[] = {
    make_hwstate(200000, 0, 2, 2, finger_states),
    make_hwstate(200100, 0, 1, 1, finger_states),
  };

  NonLinearityFilterInterpreterTestInterpreter* base =
                            new NonLinearityFilterInterpreterTestInterpreter;
  NonLinearityFilterInterpreter interpreter(nullptr, base, nullptr);
  TestInterpreterWrapper wrapper(&interpreter);
  interpreter.enabled_.val_ = 1;
  interpreter.data_location_.val_ = kTestNonlinearData;
  interpreter.LoadData();

  // Nothing should change since 2 fingers are on the touchpad
  EXPECT_EQ(nullptr, wrapper.SyncInterpret(hwstates[0], nullptr));
  EXPECT_FLOAT_EQ(hwstates[0].fingers[0].position_x, 0.5);
  EXPECT_FLOAT_EQ(hwstates[0].fingers[0].position_y, 0.5);
  EXPECT_FLOAT_EQ(hwstates[0].fingers[1].position_x, 0.78);
  EXPECT_FLOAT_EQ(hwstates[0].fingers[1].position_y, 0.34);

  // This finger is at (0.5, 0.5, 0.5) which has 0 error in the test readings
  EXPECT_EQ(nullptr, wrapper.SyncInterpret(hwstates[1], nullptr));
  EXPECT_FLOAT_EQ(hwstates[1].fingers[0].position_x, 0.5);
  EXPECT_FLOAT_EQ(hwstates[1].fingers[0].position_y, 0.5);
}

}  // namespace gestures