// Copyright(c) 2018, Intel Corporation
//
// Redistribution  and  use  in source  and  binary  forms,  with  or  without
// modification, are permitted provided that the following conditions are met:
//
// * Redistributions of  source code  must retain the  above copyright notice,
//   this list of conditions and the following disclaimer.
// * Redistributions in binary form must reproduce the above copyright notice,
//   this list of conditions and the following disclaimer in the documentation
//   and/or other materials provided with the distribution.
// * Neither the name  of Intel Corporation  nor the names of its contributors
//   may be used to  endorse or promote  products derived  from this  software
//   without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTOR."AS ."
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,  BUT NOT LIMITED TO,  THE
// IMPLIED WARRANTIES OF  MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
// ARE DISCLAIMED.  IN NO EVENT  SHALL THE COPYRIGHT OWNER  OR CONTRIBUTORS BE
// LIABLE  FOR  ANY  DIRECT,  INDIRECT,  INCIDENTAL,  SPECIAL,  EXEMPLARY,  OR
// CONSEQUENTIAL  DAMAGES  (INCLUDING,  BUT  NOT LIMITED  TO,  PROCUREMENT  OF
// SUBSTITUTE GOODS OR SERVICES;  LOSS OF USE,  DATA, OR PROFITS;  OR BUSINESS
// INTERRUPTION)  HOWEVER CAUSED  AND ON ANY THEORY  OF LIABILITY,  WHETHER IN
// CONTRACT,  STRICT LIABILITY,  OR TORT  (INCLUDING NEGLIGENCE  OR OTHERWISE)
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,  EVEN IF ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.
#include "pysysobject.h"

using namespace opae::fpga::types;
namespace py = pybind11;

const char *sysobject_doc() {
  return R"doc(
    Wraps the OPAE fpga_object primitive as a Python object.
  )doc";
}

const char *sysobject_doc_token_get() {
  return R"doc(
    Get a sysobject instance from a valid token object.
  )doc";
}

const char *sysobject_doc_handle_get() {
  return R"doc(
    Get a sysobject instance from a valid handle object.
  )doc";
}

const char *sysobject_doc_object_get() {
  return R"doc(
    Get a sysobject instance from a valid sysobject.
    The parent sysobject must be a container type object.
  )doc";
}

const char *sysobject_doc_token_find() {
  return R"doc(
    Find a sysobject instance from a valid token object.
    Args:
      flags: Flags that control behavior of finding sub-objects.
      SYSOBJECT_GLOB is used to indicate that wildcard patterns (*) are allowed.
      SYSOBJECT_RECURSE_ONE is used to indicate that the find routine should recurse one level.
      SYSOBJECT_RECURSE_ALL is used to indicate that the find routine should recurse to all children.
  )doc";
}

const char *sysobject_doc_handle_find() {
  return R"doc(
    Find a sysobject instance from a valid handle object.
  )doc";
}

const char *sysobject_doc_object_find() {
  return R"doc(
    Find a sysobject instance from a valid sysobject.
    The parent sysobject must be a container type object.
  )doc";
}

const char *sysobject_doc_bytes() {
  return R"doc(
    Get bytes from the sysobject.
    Raises `RuntimeError` if the sysobject instance is a container type.
  )doc";
}

const char *sysobject_doc_getitem() {
  return R"doc(
    Get a byte from the sysobject at a given index.
    Raises `RuntimeError` if the sysobject instance is a container type.
  )doc";
}

const char *sysobject_doc_getslice() {
  return R"doc(
    Get a slice of bytes from the sysobject at a given offset.
    Raises `RuntimeError` if the sysobject instance is a container type.
  )doc";
}

sysobject::ptr_t token_get_sysobject(token::ptr_t tok, const std::string &name) {
  return sysobject::get(tok, name, 0);
}

sysobject::ptr_t handle_get_sysobject(handle::ptr_t h, const std::string &name) {
  return sysobject::get(h, name, 0);
}

sysobject::ptr_t sysobject_get_sysobject(sysobject::ptr_t o, const std::string &name) {
  return o->get(name, 0);
}

sysobject::ptr_t sysobject_get_sysobject(sysobject::ptr_t o, int i) {
  return o->get(i);
}

sysobject::ptr_t token_find_sysobject(token::ptr_t tok, const std::string &name,
                                     int flags) {
  return sysobject::get(tok, name, flags);
}

sysobject::ptr_t handle_find_sysobject(handle::ptr_t h, const std::string &name,
                                      int flags) {
  return sysobject::get(h, name, flags);
}

sysobject::ptr_t sysobject_find_sysobject(sysobject::ptr_t o, const std::string &name,
                                      int flags) {
  return o->get(name, flags);
}

std::string sysobject_bytes(sysobject::ptr_t obj) {
  auto bytes = obj->bytes(FPGA_OBJECT_SYNC);
  return std::string(bytes.begin(), bytes.end());
}

py::object sysobject_getitem(sysobject::ptr_t obj, uint32_t offset) {
  switch(obj->type()) {
    case FPGA_OBJECT_ATTRIBUTE:
      return py::cast(obj->bytes(offset, 1, FPGA_OBJECT_SYNC)[0]);
      break;
    case FPGA_OBJECT_CONTAINER:
      return py::cast(obj->get(offset));
      break;
  }
  return py::object();
}

std::string sysobject_getslice(sysobject::ptr_t obj, py::slice slice) {
  size_t start, stop, step, length;
  if (!slice.compute(obj->size(), &start, &stop, &step, &length))
    throw py::error_already_set();
  auto bytes = obj->bytes(FPGA_OBJECT_SYNC);
  std::string buf('\0', bytes.size());
  for (size_t i = start, j = 0; i < stop; i += step, ++j) {
    buf[j] = bytes[i];
  }
  return buf;
}