/* -*- mode: c++ -*-
 * Copyright (C) 2020  Eric Paniagua (epaniagua@google.com)
 *
 * This program is free software: you can redistribute it and/or modify it under
 * the terms of the GNU General Public License as published by the Free Software
 * Foundation, version 3.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include "testing/VirtualDeviceTest.h"

#include "HIDReportObserver.h"
#include "testing/HIDState.h"

namespace kaleidoscope {
namespace testing {

void VirtualDeviceTest::SetUp() {
  HIDReportObserver::resetHook(&internal::HIDStateBuilder::ProcessHidReport);
}

std::unique_ptr<State> VirtualDeviceTest::RunCycle() {
  sim_.RunCycle();
  return State::Snapshot();
}

// =============================================================================
void VirtualDeviceTest::Run(Cycles n) {
  sim_.RunCycles(n.value);
}

void VirtualDeviceTest::Run(Millis t) {
  sim_.RunForMillis(t);
}

void VirtualDeviceTest::RunUntil(Millis end) {
  uint32_t t0 = Runtime.millisAtCycleStart();
  Run(Millis{end - t0});
}

void VirtualDeviceTest::RunFrom(Millis start, Millis t) {
  RunUntil(Millis{start + t});
}

// =============================================================================
void VirtualDeviceTest::LoadState() {
  output_state_ = State::Snapshot();
}

void VirtualDeviceTest::ClearState() {
  output_state_ = nullptr;
  input_timestamps_.clear();
  expected_reports_.clear();
}

const HIDState* VirtualDeviceTest::HIDReports() const {
  if (output_state_ == nullptr) return nullptr;
  return output_state_->HIDReports();
}

Millis VirtualDeviceTest::ReportTimestamp(size_t index) const {
  uint32_t t = output_state_->HIDReports()->Keyboard(index).Timestamp();
  return Millis{t};
}

// -----------------------------------------------------------------------------
Millis VirtualDeviceTest::EventTimestamp(size_t index) const {
  return Millis{input_timestamps_[index]};
}

// =============================================================================
void VirtualDeviceTest::PressKey(KeyAddr addr) {
  sim_.Press(addr);
  input_timestamps_.push_back(Millis{Runtime.millisAtCycleStart()});
}
void VirtualDeviceTest::ReleaseKey(KeyAddr addr) {
  sim_.Release(addr);
  input_timestamps_.push_back(Millis{Runtime.millisAtCycleStart()});
}

// -----------------------------------------------------------------------------
void VirtualDeviceTest::PressKey(Cycles n, KeyAddr addr) {
  sim_.RunCycles(n.value);
  PressKey(addr);
}
void VirtualDeviceTest::ReleaseKey(Cycles n, KeyAddr addr) {
  sim_.RunCycles(n.value);
  ReleaseKey(addr);
}

// -----------------------------------------------------------------------------
void VirtualDeviceTest::PressKey(Millis t, KeyAddr addr) {
  sim_.RunForMillis(t);
  PressKey(addr);
}
void VirtualDeviceTest::ReleaseKey(Millis t, KeyAddr addr) {
  sim_.RunForMillis(t);
  ReleaseKey(addr);
}


// =============================================================================
void VirtualDeviceTest::ExpectReport(Keycodes keys,
                                     std::string description) {
  size_t report_timestamp{Runtime.millisAtCycleStart()};
  ClearReport();
  for (Key key : keys) {
    AddToReport(key);
  }
  ExpectedKeyboardReport new_report(report_timestamp,
                                    current_keyboard_keycodes_,
                                    description);
  expected_reports_.push_back(new_report);
}

// =============================================================================
void VirtualDeviceTest::ExpectReport(AddKeycodes added_keys,
                                     RemoveKeycodes removed_keys,
                                     std::string description) {
  Millis report_timestamp{Runtime.millisAtCycleStart()};
  for (Key key : added_keys) {
    AddToReport(key);
  }
  for (Key key : removed_keys) {
    RemoveFromReport(key);
  }
  ExpectedKeyboardReport new_report(report_timestamp,
                                    current_keyboard_keycodes_,
                                    description);
  expected_reports_.push_back(new_report);
}

// -----------------------------------------------------------------------------
void VirtualDeviceTest::ExpectReport(AddKeycodes added_keys,
                                     std::string description) {
  ExpectReport(added_keys, RemoveKeycodes{}, description);
}
void VirtualDeviceTest::ExpectReport(RemoveKeycodes removed_keys,
                                     std::string description) {
  ExpectReport(AddKeycodes{}, removed_keys, description);
}

// -----------------------------------------------------------------------------
void VirtualDeviceTest::ExpectReport(Cycles n,
                                     AddKeycodes added_keys,
                                     std::string description) {
  sim_.RunCycles(n.value);
  ExpectReport(added_keys, description);
}
void VirtualDeviceTest::ExpectReport(Cycles n,
                                     RemoveKeycodes removed_keys,
                                     std::string description) {
  sim_.RunCycles(n.value);
  ExpectReport(removed_keys, description);
}

// -----------------------------------------------------------------------------
void VirtualDeviceTest::ExpectReport(Millis t,
                                     AddKeycodes added_keys,
                                     std::string description) {
  sim_.RunForMillis(t);
  ExpectReport(added_keys, description);
}
void VirtualDeviceTest::ExpectReport(Millis t,
                                     RemoveKeycodes removed_keys,
                                     std::string description) {
  sim_.RunForMillis(t);
  ExpectReport(removed_keys, description);
}

// =============================================================================
void VirtualDeviceTest::ClearReport() {
  current_keyboard_keycodes_.clear();
}
void VirtualDeviceTest::AddToReport(Key key) {
  current_keyboard_keycodes_.insert(key.getKeyCode());
}
void VirtualDeviceTest::RemoveFromReport(Key key) {
  current_keyboard_keycodes_.erase(key.getKeyCode());
}

}  // namespace testing
}  // namespace kaleidoscope