#!/usr/bin/python3

# Python Remote I/O Protocol PWM Output Test

# Copyright (C)2021-2023, Philip Munts dba Munts Technologies.
#
# 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.
#
# 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 HOLDER 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.

import ctypes
import os
import time

print('\nPython Remote I/O Protocol PWM Output Test\n')

# Select the OS appropriate shared library file

if os.name == 'nt':
  libremoteio = ctypes.cdll.LoadLibrary(os.environ.get('LIBSIMPLEIO') + '/win64/libremoteio.dll')
elif os.name == 'posix':
  libremoteio = ctypes.cdll.LoadLibrary('/usr/local/lib/libremoteio.so')

# Open Remote I/O Protocol Server

handle = ctypes.c_int()
error  = ctypes.c_int()

libremoteio.open_udp('usbgadget.munts.net'.encode(), 8087, 1000, ctypes.byref(handle), ctypes.byref(error))

if error.value != 0:
  print('ERROR: open_udp() failed, error=' + str(error.value))
  quit()

# Display version information

vers = ctypes.create_string_buffer(64)

libremoteio.get_version(handle, vers, ctypes.byref(error))

if error.value != 0:
  print('ERROR: get_version() failed, error=' + str(error.value))
  quit()

print(vers.raw.decode())

# Display capabilities

caps = ctypes.create_string_buffer(64)

libremoteio.get_capability(handle, caps, ctypes.byref(error))

if error.value != 0:
  print('ERROR: get_capability() failed, error=' + str(error.value))
  quit()

print(caps.raw.decode())
print()

# Probe available PWM channels

channels    = ctypes.create_string_buffer(128)

libremoteio.pwm_channels(handle, channels, ctypes.byref(error))

if error.value != 0:
  print('ERROR: pwm_channels() failed, error=' + str(error.value))
  quit()

# Convert byte array to set

pwmoutputs = set()

for c in range(len(channels)):
  if channels[c] == b'\x01':
    pwmoutputs.add(c)

del channels

print('Available PWM output channels: ' + str(pwmoutputs))
print()

# Configure PWM output

outp = list(pwmoutputs)[0]

libremoteio.pwm_configure(handle, outp, 1000, ctypes.byref(error))

if error.value != 0:
  print('ERROR: pwm_configure() failed, error=' + str(error.value))
  quit()

# Write to PWM output

duty = ctypes.c_float(0.0)

while True:
  for s in range(3000):
    duty.value = s/30.0
    libremoteio.pwm_write(handle, outp, duty, ctypes.byref(error))

    if error.value != 0:
      print('ERROR: pwm_write() failed, error=' + str(error.value))

  for s in reversed(range(3000)):
    duty.value = s/30.0
    libremoteio.pwm_write(handle, outp, duty, ctypes.byref(error))

    if error.value != 0:
      print('ERROR: pwm_write() failed, error=' + str(error.value))
      quit()