# distutils: language = c++
# cython: c_string_encoding=ascii, language_level=3

from libcpp.string cimport string

from .runmodel cimport USE_CPU_RUNTIME, USE_GPU_RUNTIME, USE_DSP_RUNTIME
from selfdrive.modeld.models.commonmodel_pyx cimport CLMem
import numpy as np

class Runtime:
  CPU = USE_CPU_RUNTIME
  GPU = USE_GPU_RUNTIME
  DSP = USE_DSP_RUNTIME

cdef class RunModel:
  def __dealloc__(self):
    del self.model

  def addInput(self, string name, float[:] buffer):
    if buffer is not None:
      self.model.addInput(name, &buffer[0], len(buffer))
    else:
      self.model.addInput(name, NULL, 0)

  def setInputBuffer(self, string name, unsigned char[:] input_buffer):
    cdef int num_floats = len(input_buffer) // sizeof(float)
    cdef float* float_ptr = <float*> &input_buffer[0]
    cdef float[:] float_buffer_view = <float[:num_floats]> float_ptr
    if float_buffer_view is not None:
      self.model.setInputBuffer(name, &float_buffer_view[0], num_floats)

  def getCLBuffer(self, string name):
    cdef void * cl_buf = self.model.getCLBuffer(name)
    if not cl_buf:
      return None
    return CLMem.create(cl_buf)

  def execute(self):
    self.model.execute()