Octave下操作CH341

 #include <octave/oct.h>
 #include <windows.h>
 #include <cstdint>
 #include <fstream>
 #include <dMatrix.h>
 #include <int32NDArray.h>

 // CH341DLL Functions Prototype
 typedef HANDLE WINAPI(*CH341OpenDevice_fn)(unsigned int iIndex);

 typedef void (*CH341CloseDevice_fn)(unsigned int iIndex);

 typedef BOOL WINAPI(*CH341SetStream_fn)(
     unsigned int     iIndex,
     unsigned int     iMode);

 typedef BOOL WINAPI(*CH341StreamI2C_fn)(
     unsigned int    iIndex,
     unsigned int    iWriteLength,
     void *            iWriteBuffer,
     unsigned int    iReadLength,
     void *            oReadBuffer);

 // Octave function declaration
 DEFUN_DLD(i2c_rd, args, nargout, "USB I2C Read")
 {
     int32NDArray bytes;
     size_t bytes_read;

     if(args.length() < ) {
         octave_stdout << "\n  --i2c_rd I2C_ADR REG_ADR [Bytes to Read]\n" << std::endl;
         return octave_value_list();
     } else if(args.length() == ) {
         bytes = args().matrix_value();
         bytes_read = bytes(, );
     } else {
         bytes(, ) = ;
         bytes_read = ;
     }

     // Load CH341DLL to Current Process
     HMODULE hdll = LoadLibrary("CH341DLL.DLL");

     if(NULL == hdll) {
         octave_stdout <<"USB_I2C CH341DLL Load Failed!" << std::endl;
         return octave_value_list();
     }

     // Get CH341 Functions Handle
     CH341OpenDevice_fn  USBIO_OpenDevice    = (CH341OpenDevice_fn)    GetProcAddress(hdll, "CH341OpenDevice");
     CH341CloseDevice_fn USBIO_CloseDevice    = (CH341CloseDevice_fn)    GetProcAddress(hdll, "CH341CloseDevice");
     CH341SetStream_fn   USBIO_SetStream        = (CH341SetStream_fn)    GetProcAddress(hdll, "CH341SetStream");
     CH341StreamI2C_fn   USBIO_StreamI2C        = (CH341StreamI2C_fn)    GetProcAddress(hdll, "CH341StreamI2C");

     int32NDArray i2c_adr = args().int32_array_value();
     int32NDArray reg_adr = args().int32_array_value();

     Matrix result(, bytes_read);
     uint8_t IBuf[bytes_read];
     uint8_t OBuf[] = {uint8_t(i2c_adr(, )), uint8_t(reg_adr(, ))};

     // Open CH341 Device
     HANDLE devh = USBIO_OpenDevice();

     if(devh == NULL) {
         octave_stdout << "CH341 Open Failed!" << std::endl;
         FreeLibrary(hdll);
         return octave_value_list();
     }

     if(!USBIO_SetStream(, 0x82)) {
         octave_stdout << "CH341 SetStream Failed!" << std::endl;
         USBIO_CloseDevice();
         FreeLibrary(hdll);
         return octave_value_list();
     }

     // If number of bytes to read is not given.
     if(!USBIO_StreamI2C(, , OBuf, bytes_read, IBuf)) {
         octave_stdout << "CH341 Read Failed!" << std::endl;
     } else {
         for(size_t i = ; i < bytes_read; ++i) {
             result(, i) = double(IBuf[i]);
         }
     }

     USBIO_CloseDevice();
     FreeLibrary(hdll);
     return octave_value(result);
 }