// 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 CONTRIBUTORS "AS IS"
// 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.
#pragma once
#include <Python.h>
#include <opae/cxx/core/shared_buffer.h>
#include <pybind11/pybind11.h>
#include <chrono>
#include "pyhandle.h"

const char *shared_buffer_doc();

const char *shared_buffer_doc_allocate();
opae::fpga::types::shared_buffer::ptr_t shared_buffer_allocate(
    opae::fpga::types::handle::ptr_t hndl, size_t size);
const char *shared_buffer_doc_size();

const char *shared_buffer_doc_wsid();

const char *shared_buffer_doc_io_address();

const char *shared_buffer_doc_fill();

const char *shared_buffer_doc_compare();

const char *shared_buffer_doc_getitem();
uint8_t shared_buffer_getitem(opae::fpga::types::shared_buffer::ptr_t buf,
                              uint32_t offset);

const char *shared_buffer_doc_setitem();
void shared_buffer_setitem(opae::fpga::types::shared_buffer::ptr_t buf,
                           uint32_t offset, pybind11::int_ item);

const char *shared_buffer_doc_getslice();
pybind11::list shared_buffer_getslice(
    opae::fpga::types::shared_buffer::ptr_t buf, pybind11::slice slice);

const char *shared_buffer_doc_read32();
const char *shared_buffer_doc_read64();
const char *shared_buffer_doc_write32();
const char *shared_buffer_doc_write64();
const char *shared_buffer_doc_copy();
void shared_buffer_copy(opae::fpga::types::shared_buffer::ptr_t self,
                        opae::fpga::types::shared_buffer::ptr_t other,
                        size_t size);
const char *shared_buffer_doc_split();
std::vector<opae::fpga::types::shared_buffer::ptr_t> shared_buffer_split(
    opae::fpga::types::shared_buffer::ptr_t buf, pybind11::args args);

template <typename T>
bool shared_buffer_poll(opae::fpga::types::shared_buffer::ptr_t self,
                        size_t offset, T value, T mask = 0,
                        uint64_t timeout_usec = 1000) {
  using hrc = std::chrono::high_resolution_clock;
  auto ptr = self->c_type();
  auto begin = hrc::now();
  std::chrono::microseconds timeout(timeout_usec);
  if (!mask) {
    mask = ~mask;
  }

  while ((*reinterpret_cast<volatile T *>(ptr + offset) & mask) != value) {
    if (std::chrono::duration_cast<std::chrono::microseconds>(
            hrc::now() - begin) >= timeout) {
      return false;
    }
  }
  return true;
}