openmv/src/omv/modules/py_assert.h
2020-12-18 00:59:28 +02:00

65 lines
2.8 KiB
C

/*
* This file is part of the OpenMV project.
*
* Copyright (c) 2013-2019 Ibrahim Abdelkader <iabdalkader@openmv.io>
* Copyright (c) 2013-2019 Kwabena W. Agyeman <kwagyeman@openmv.io>
*
* This work is licensed under the MIT license, see the file LICENSE for details.
*
* MP assertions.
*/
#ifndef __PY_ASSERT_H__
#define __PY_ASSERT_H__
#include "mp.h"
#define PY_ASSERT_TRUE(cond) \
do { \
if ((cond) == 0) { \
nlr_raise(mp_obj_new_exception_msg( \
&mp_type_OSError, \
"Operation not supported")); \
} \
} while(0)
#define PY_ASSERT_TRUE_MSG(cond, msg) \
do { \
if ((cond) == 0) { \
nlr_raise(mp_obj_new_exception_msg( \
&mp_type_OSError, msg)); \
} \
} while(0)
#define PY_ASSERT_FALSE_MSG(cond, msg) \
do { \
if ((cond) == 1) { \
nlr_raise(mp_obj_new_exception_msg( \
&mp_type_OSError, msg)); \
} \
} while(0)
#define PY_ASSERT_TYPE(obj, type) \
do { \
__typeof__ (obj) _a = (obj); \
__typeof__ (type) _b = (type); \
if (!MP_OBJ_IS_TYPE(_a, _b)) { \
nlr_raise(mp_obj_new_exception_msg_varg( \
&mp_type_TypeError, \
"Can't convert %s to %s", \
mp_obj_get_type_str(_a), \
mp_obj_get_type_str(_b))); \
} \
} while(0)
/* IS_TYPE doesn't work for str objs */
#define PY_ASSERT_STR(obj) \
do { \
__typeof__ (obj) _a = (obj); \
if (!MP_OBJ_IS_STR(_a)) { \
nlr_raise(mp_obj_new_exception_msg_varg( \
&mp_type_TypeError, \
"Can't convert %s to %s", \
mp_obj_get_type_str(_a), \
str_type.name)); \
} \
} while(0)
#endif // __PY_ASSERT_H__