/* * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance * with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, * software distributed under the License is distributed on an * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY * KIND, either express or implied. See the License for the * specific language governing permissions and limitations * under the License. */ /*! * Copyright (c) 2018 by Contributors * \brief gotvm package source for TVMByteArray interface. * \file bytearray.go */ package gotvm //#include "gotvm.h" import "C" import ( "unsafe" ) // ByteArray type wraps the TVMByteArray of C runtime API. // // This can be used to hold raw data like params of a model. type ByteArray uintptr // nativeCPtr returns the type freed unitptr for ByteArray. func (tbytearray ByteArray) nativeCPtr() (retVal uintptr) { retVal = (uintptr)(tbytearray) return } // SetData is used to intialize ByteArray from a golang string object. // // This method initialize both data and data size of the underlaying object. // This function handles freeing old data object if any before allocating new. // // `val` is the golang string object from which the ByteArray is initialized. func (tbytearray ByteArray) setData(val string) { bufPtr := ((*C.TVMByteArray)(unsafe.Pointer(tbytearray))).data if bufPtr == (*C.char)(C.NULL) { C.free(unsafe.Pointer(bufPtr)) } ((*C.TVMByteArray)(unsafe.Pointer(tbytearray))).data = C.CString(val) ((*C.TVMByteArray)(unsafe.Pointer(tbytearray))).size = C.ulong(len(val)) } // getData returns the golang byte slice corresponding to the ByteArray. func (tbytearray ByteArray) getData() (retVal []byte) { val := ((*C.TVMByteArray)(unsafe.Pointer(tbytearray))).data blen := ((*C.TVMByteArray)(unsafe.Pointer(tbytearray))).size retVal = C.GoBytes(unsafe.Pointer(val), C.int(blen)) return } // newByteArray initilizes the native TVMByteArray object with given byte slice // //`val` is the golang byte array used to initialize. // // returns newly created ByteArray. func newByteArray(val []byte) (retVal ByteArray) { handle := ByteArray(C.malloc(C.sizeof_TVMByteArray)) ((*C.TVMByteArray)(unsafe.Pointer(handle))).data = (*C.char)(C.NULL) ((*C.TVMByteArray)(unsafe.Pointer(handle))).size = 0 handle.setData(string(val)) retVal = handle return } // deleteTVMByteArray releases the allocated native object of ByteArray. // // This delete handles freeing of underlaying native data object too. func (tbytearray ByteArray) deleteTVMByteArray() { bufPtr := ((*C.TVMByteArray)(unsafe.Pointer(tbytearray))).data C.free(unsafe.Pointer(bufPtr)) C.free(unsafe.Pointer(tbytearray.nativeCPtr())) }