onnxruntime

1
// Copyright (c) Microsoft Corporation. All rights reserved.
2
// Licensed under the MIT License.
3

4
#import "error_utils.h"
5

6
NS_ASSUME_NONNULL_BEGIN
7

8
static NSString* const kOrtErrorDomain = @"onnxruntime";
9

10
void ORTSaveCodeAndDescriptionToError(int code, const char* descriptionCstr, NSError** error) {
11
  if (!error) return;
12

13
  NSString* description = [NSString stringWithCString:descriptionCstr
14
                                             encoding:NSASCIIStringEncoding];
15

16
  *error = [NSError errorWithDomain:kOrtErrorDomain
17
                               code:code
18
                           userInfo:@{NSLocalizedDescriptionKey : description}];
19
}
20

21
void ORTSaveCodeAndDescriptionToError(int code, NSString* description, NSError** error) {
22
  if (!error) return;
23

24
  *error = [NSError errorWithDomain:kOrtErrorDomain
25
                               code:code
26
                           userInfo:@{NSLocalizedDescriptionKey : description}];
27
}
28

29
void ORTSaveOrtExceptionToError(const Ort::Exception& e, NSError** error) {
30
  ORTSaveCodeAndDescriptionToError(e.GetOrtErrorCode(), e.what(), error);
31
}
32

33
void ORTSaveExceptionToError(const std::exception& e, NSError** error) {
34
  ORTSaveCodeAndDescriptionToError(ORT_RUNTIME_EXCEPTION, e.what(), error);
35
}
36

37
NS_ASSUME_NONNULL_END
38