# Copyright (C) 2014 Roger A. Light <roger.light@nottingham.ac.uk>
#
# Python wrapper to libvxi11.

# User library for opening, closing, sending to and receiving from
# a device enabled with the VXI11 RPC ethernet protocol. Uses the files
# generated by rpcgen vxi11.x.
#
# 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; either version 2
# of the License, or (at your option) any later version.
#
# 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, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

from ctypes import *
import ctypes.util

DEFAULT_TIMEOUT = 10000
READ_TIMEOUT = 2000
NULL_READ_RESP = 50
NULL_WRITE_RESP = 51


class Vxi11(object):
    def __init__(self, address, device="inst0"):
        if address is None or address == "":
            raise ValueError("address must be defined")

        self._clink = c_void_p()
        self._address = address
        self._device = device
        if address:
            a = c_char_p(address.encode())
        else:
            a = None
        if device:
            d = c_char_p(device.encode())
        else:
            d = None
        if(_vxi11_open_device(byref(self._clink), a, d)):
            raise IOError("unable to open device at address "+address)

    def __del__(self):
        self.close()

    def close(self):
        if self._clink:
            if self._address:
                a = c_char_p(self._address.encode())
            else:
                a = None
            _vxi11_close_device(self._clink, a)
            self._clink = None
            self._address = None
            self._device = None

    def send(self, cmd):
        c = cmd.encode()
        rc = _vxi11_send(self._clink, c_char_p(c), len(c))
        return rc

    def receive(self, max_length=1024, timeout=READ_TIMEOUT):
        buf = create_string_buffer(max_length)
        rc = _vxi11_receive_timeout(self._clink, buf, max_length, timeout)
        if rc < 0:
            return (rc, None)
        else:
            return (rc, buf.value)

    def send_data_block(self, cmd, buf):
        if _vxi11_send_data_block(self._clink, cmd.encode(), buf, len(buf)):
            raise IOError("error sending to the device: "+str(rc))
        return 1

    def receive_data_block(self, max_length=1024, timeout=READ_TIMEOUT):
        buf = create_string_buffer(max_length)
        rc = _vxi11_receive_data_block(self._clink, buf, max_length, timeout)
        if rc < 0:
            return (rc, None)
        else:
            return (rc, buf.value)

    def send_and_receive(self, cmd, max_length=1024, timeout=READ_TIMEOUT):
        buf = create_string_buffer(max_length)
        c = cmd.encode()
        rc = _vxi11_send_and_receive(self._clink, c_char_p(c), buf, max_length, timeout)
        if rc < 0:
            return (rc, None)
        else:
            return (rc, buf.value)

    def obtain_long_value(self, cmd, timeout=READ_TIMEOUT):
        return _vxi11_obtain_long_value_timeout(self._clink, cmd, timeout)

    def obtain_double_value(self, cmd, timeout=READ_TIMEOUT):
        return _vxi11_obtain_double_value_timeout(self._clink, cmd, timeout)


_libvxi11 = cdll.LoadLibrary(ctypes.util.find_library("vxi11"))

_vxi11_open_device = _libvxi11.vxi11_open_device
_vxi11_open_device.argtypes = [POINTER(c_void_p), c_char_p, c_char_p]
_vxi11_open_device.restype = c_int

_vxi11_close_device = _libvxi11.vxi11_close_device
_vxi11_close_device.argtypes = [c_void_p, c_char_p]
_vxi11_close_device.restype = c_int

_vxi11_send = _libvxi11.vxi11_send
_vxi11_send.argtypes = [c_void_p, c_char_p, c_size_t]
_vxi11_send.restype = c_int

_vxi11_receive = _libvxi11.vxi11_receive
_vxi11_receive.argtypes = [c_void_p, c_char_p, c_size_t, c_ulong]
_vxi11_receive.restype = c_ssize_t

_vxi11_send_data_block = _libvxi11.vxi11_send_data_block
_vxi11_send_data_block.argtypes = [c_void_p, c_char_p, c_char_p, c_size_t]
_vxi11_send_data_block.restype = c_int

_vxi11_receive_data_block = _libvxi11.vxi11_receive_data_block
_vxi11_receive_data_block.argtypes = [c_void_p, c_char_p, c_size_t, c_ulong]
_vxi11_receive_data_block.restype = c_ssize_t

_vxi11_send_and_receive = _libvxi11.vxi11_send_and_receive
_vxi11_send_and_receive.argtypes = [c_void_p, c_char_p, c_char_p, c_size_t, c_ulong]
_vxi11_send_and_receive.restype = c_int

_vxi11_obtain_long_value_timeout = _libvxi11.vxi11_obtain_long_value_timeout
_vxi11_obtain_long_value_timeout.argtypes = [c_void_p, c_char_p, c_ulong]
_vxi11_obtain_long_value_timeout.restype = c_long

_vxi11_obtain_double_value_timeout = _libvxi11.vxi11_obtain_double_value_timeout
_vxi11_obtain_double_value_timeout.argtypes = [c_void_p, c_char_p, c_ulong]
_vxi11_obtain_double_value_timeout.restype = c_double