Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
ai_onnxruntime_OnnxSequence.c 4.12 KB
Newer Older
gaoqiong's avatar
update v1.14.0  
gaoqiong committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105 
/*
 * Copyright (c) 2019, 2022, Oracle and/or its affiliates. All rights reserved.
 * Licensed under the MIT License.
 */
#include <jni.h>
#include "onnxruntime/core/session/onnxruntime_c_api.h"
#include "OrtJniUtil.h"
#include "ai_onnxruntime_OnnxSequence.h"

/*
 * Class:     ai_onnxruntime_OnnxSequence
 * Method:    getMaps
 * Signature: (JJJ)[Lai/onnxruntime/OnnxMap;
 */
JNIEXPORT jobjectArray JNICALL Java_ai_onnxruntime_OnnxSequence_getMaps(JNIEnv* jniEnv, jobject jobj,
                                                                        jlong apiHandle, jlong handle,
                                                                        jlong allocatorHandle) {
  (void)jobj;  // Required JNI parameter not needed by functions which don't need to access their host object.
  const OrtApi* api = (const OrtApi*)apiHandle;
  OrtValue* sequence = (OrtValue*)handle;
  OrtAllocator* allocator = (OrtAllocator*)allocatorHandle;

  jobjectArray outputArray = NULL;

  // Get the element count of this sequence
  size_t count;
  OrtErrorCode code = checkOrtStatus(jniEnv, api, api->GetValueCount(sequence, &count));
  if (code == ORT_OK) {
    jclass tensorClazz = (*jniEnv)->FindClass(jniEnv, "ai/onnxruntime/OnnxMap");
    outputArray = (*jniEnv)->NewObjectArray(jniEnv, safecast_size_t_to_jsize(count), tensorClazz, NULL);
    for (size_t i = 0; i < count; i++) {
      // Extract element
      OrtValue* element;
      code = checkOrtStatus(jniEnv, api, api->GetValue(sequence, (int)i, allocator, &element));
      if (code == ORT_OK) {
        jobject str = createJavaMapFromONNX(jniEnv, api, allocator, element);
        if (str == NULL) {
          api->ReleaseValue(element);
          // bail out as exception has been thrown
          return NULL;
        }
        (*jniEnv)->SetObjectArrayElement(jniEnv, outputArray, (jsize)i, str);
      } else {
        // bail out as exception has been thrown
        return NULL;
      }
    }
  }
  return outputArray;
}

/*
 * Class:     ai_onnxruntime_OnnxSequence
 * Method:    getTensors
 * Signature: (JJJ)[Lai/onnxruntime/OnnxTensor;
 */
JNIEXPORT jobjectArray JNICALL Java_ai_onnxruntime_OnnxSequence_getTensors(JNIEnv* jniEnv, jobject jobj,
                                                                           jlong apiHandle, jlong handle,
                                                                           jlong allocatorHandle) {
  (void)jobj;  // Required JNI parameter not needed by functions which don't need to access their host object.
  const OrtApi* api = (const OrtApi*)apiHandle;
  OrtValue* sequence = (OrtValue*)handle;
  OrtAllocator* allocator = (OrtAllocator*)allocatorHandle;

  jobjectArray outputArray = NULL;

  // Get the element count of this sequence
  size_t count;
  OrtErrorCode code = checkOrtStatus(jniEnv, api, api->GetValueCount(sequence, &count));
  if (code == ORT_OK) {
    jclass tensorClazz = (*jniEnv)->FindClass(jniEnv, "ai/onnxruntime/OnnxTensor");
    outputArray = (*jniEnv)->NewObjectArray(jniEnv, safecast_size_t_to_jsize(count), tensorClazz, NULL);
    for (size_t i = 0; i < count; i++) {
      // Extract element
      OrtValue* element;
      code = checkOrtStatus(jniEnv, api, api->GetValue(sequence, (int)i, allocator, &element));
      if (code == ORT_OK) {
        jobject str = createJavaTensorFromONNX(jniEnv, api, allocator, element);
        if (str == NULL) {
          api->ReleaseValue(element);
          // bail out as exception has been thrown
          return NULL;
        }
        (*jniEnv)->SetObjectArrayElement(jniEnv, outputArray, (jsize)i, str);
      } else {
        // bail out as exception has been thrown
        return NULL;
      }
    }
  }
  return outputArray;
}

/*
 * Class:     ai_onnxruntime_OnnxSequence
 * Method:    close
 * Signature: (J)V
 */
JNIEXPORT void JNICALL Java_ai_onnxruntime_OnnxSequence_close(JNIEnv* jniEnv, jobject jobj, jlong apiHandle,
                                                              jlong handle) {
  (void)jniEnv;
  (void)jobj;  // Required JNI parameters not needed by functions which don't need to access their host object.
  const OrtApi* api = (const OrtApi*)apiHandle;
  api->ReleaseValue((OrtValue*)handle);
}