// Copyright (C) 2013   Stefan Mahr     <dac922@gmx.de>
// Copyright (C) 2012   Andrius Sutas   <andrius.sutas@gmail.com>
//
// 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 3 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, see <http://www.gnu.org/licenses/>.

#include <octave/oct.h>

#ifdef HAVE_CONFIG_H
#include "../config.h"
#endif

#ifdef BUILD_USBTMC
#include <errno.h>

#include "usbtmc_class.h"

static bool type_loaded = false;
#endif

DEFUN_DLD (usbtmc_write, args, nargout,
        "-*- texinfo -*-\n\
@deftypefn {Loadable Function} {@var{n} = } usbtmc_write (@var{usbtmc}, @var{data})\n \
\n\
Write data to a usbtmc slave device.\n \
\n\
@var{usbtmc} - instance of @var{octave_usbtmc} class.\n \
@var{data} - data, of type uint8, to be written to the slave device.\n \
\n\
Upon successful completion, usbtmc_write() shall return the number of bytes written as the result @var{n}.\n \
@end deftypefn")
{
#ifndef BUILD_USBTMC
    error("usbtmc: Your system doesn't support the USBTMC interface");
    return octave_value();
#else
    if (!type_loaded)
    {
        octave_usbtmc::register_type();
        type_loaded = true;
    }

    if (args.length() != 2 || args(0).type_id() != octave_usbtmc::static_type_id())
    {
        print_usage();
        return octave_value(-1);
    }

    octave_usbtmc* usbtmc = NULL;
    int retval;

    const octave_base_value& rep = args(0).get_rep();
    usbtmc = &((octave_usbtmc &)rep);

    const octave_base_value& data = args(1).get_rep();

    if (data.is_string())
    {
        string buf = data.string_value();
        retval = usbtmc->write((uint8_t*)buf.c_str(), buf.length());
    }
    else if (data.byte_size() == data.numel())
    {
        NDArray dtmp = data.array_value();
        uint8_t *buf = NULL;
        buf = new uint8_t[dtmp.length()];

        if (buf == NULL)
        {
            error("usbtmc_write: cannot allocate requested memory: %s\n", strerror(errno));
            return octave_value(-1);
        }

        for (int i = 0; i < dtmp.length(); i++)
            buf[i] = static_cast<uint8_t>(dtmp(i));

        retval = usbtmc->write(buf, dtmp.length());

        delete[] buf;
    }
    else
    {
        print_usage();
        return octave_value(-1);
    }

    return octave_value(retval);
#endif
}