aboutsummaryrefslogtreecommitdiff
path: root/python/pyarmnn/src/pyarmnn/swig/standard_header.i
blob: c412dc3bffb560218a37e51b25c3f2d22a7c9158 (plain)
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
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
%include "stl.i"
%include "cstring.i"
%include "std_string.i"
%include "std_vector.i"
%include "std_unordered_set.i"
%include "std_pair.i"
%include "stdint.i"
%include "carrays.i"
%include "exception.i"
%include "typemaps.i"
%include "std_iostream.i"

%ignore *::operator=;
%ignore *::operator[];


// Define exception typemap to wrap armnn exception into python exception.

%exception{
    try {
        $action
    } catch (armnn::Exception &e) {
        SWIG_exception(SWIG_RuntimeError, const_cast<char*>(e.what()));
    }
};

%exception __getitem__ {
    try {
        $action
    } catch (armnn::InvalidArgumentException &e) {
        SWIG_exception(SWIG_ValueError, const_cast<char*>(e.what()));
    } catch (const std::out_of_range &e) {
        SWIG_exception(SWIG_IndexError, const_cast<char*>(e.what()));
    } catch (const std::exception &e) {
        SWIG_exception(SWIG_RuntimeError, const_cast<char*>(e.what()));
    }
};

%exception __setitem__ {
    try {
        $action
    } catch (armnn::InvalidArgumentException &e) {
        SWIG_exception(SWIG_ValueError, const_cast<char*>(e.what()));
    } catch (const std::out_of_range &e) {
        SWIG_exception(SWIG_IndexError, const_cast<char*>(e.what()));
    } catch (const std::exception &e) {
        SWIG_exception(SWIG_RuntimeError, const_cast<char*>(e.what()));
    }
};