ExternalVendorCode/trick
1
0
Fork 0
mirror of https://github.com/nasa/trick.git synced 2024-12-19 05:07:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
899b0d3efe
trick/trick_source/trick_swig/swig_int_templates.hh
Alex Lin 14a75508a3 Cleaning up once include variables and copyright cleanup. 
Changed all header file once include variables to follow the same naming
convention and not start with any underscores.  Also deleted old
incorrect copyright notices.  Also removed $Id: tags from all files.

Fixes #14.  Fixes #22.
2015-03-23 16:03:14 -05:00

321 lines
11 KiB
C++
Raw Blame History

/*
This is the 1D array/pointer support code for SWIG
*/
#ifndef SWIG_INT_TEMPLATES_HH
#define SWIG_INT_TEMPLATES_HH
template< class S , typename T > static int convert_and_set( T & output , void * my_argp , std::string to_units ) {
int ret = 0 ;
S * temp_m = reinterpret_cast< S * >(my_argp) ;
if ( temp_m->units.compare("--") ) {
try {
Unit converter(temp_m->units.c_str()) ;
output = (T)converter.Convert_to( temp_m->value , to_units.c_str()) ;
}
catch (Unit::CONVERSION_ERROR & ce_err ) {
PyErr_SetString(PyExc_TypeError,"Units conversion error");
ret = -1;
}
} else {
output = (T)temp_m->value ;
}
return ret ;
}
template<typename T > static int typemap_in_scalar( T & output , PyObject *input , const char * symname ) {
// SCALAR IN
std::string temp_name ;
std::string left_units ;
void * my_argp ;
int ret = 0 ;
temp_name = symname ;
if ( temp_name.length() > 4 ) {
temp_name.erase(temp_name.length() - 4) ;
}
left_units = Trick::UnitsMap::units_map()->get_units(temp_name) ;
//std::cerr << "\033[33mleft side units = " << left_units << "\033[00m" << std::endl ;
if (SWIG_IsOK(SWIG_ConvertPtr(input, &my_argp,SWIG_TypeQuery("_p_swig_int"), 0 ))) {
ret = convert_and_set< swig_int , T >( output , my_argp , left_units ) ;
} else if (SWIG_IsOK(SWIG_ConvertPtr(input, &my_argp,SWIG_TypeQuery("_p_swig_double"), 0 ))) {
ret = convert_and_set< swig_double , T >( output , my_argp , left_units ) ;
} else if ( PyFloat_Check(input) ) {
output = (T)PyFloat_AsDouble(input) ;
} else if ( PyInt_Check(input) ) {
output = (T)PyInt_AsLong(input) ;
} else if ( PyString_Check(input) ) {
// scalar char as a string. Set the value of the output to the value of the first char.
if ( PyString_Size(input) == 1 ) {
char * temp_str = PyString_AsString(input) ;
output = (T)temp_str[0] ;
}
} else {
ret = -1 ;
}
return ret ;
}
template<typename T > static T * typemap_in_1d( PyObject *input , unsigned int out_size, const char * symname ) {
void * argp2 ;
unsigned int ii ;
T * new_array ;
int ret ;
// convert list to tuple so we only have to deal with tuples in code below.
if (PyList_Check(input)) {
input = PyList_AsTuple(input) ;
}
if (PyTuple_Check(input)) {
unsigned int size = PyTuple_Size(input) ;
if ( size > out_size ) {
PyErr_SetString(PyExc_TypeError,"List too long to fit.");
return NULL ;
}
std::string temp_name ;
std::string left_units ;
temp_name = symname ;
if ( temp_name.length() > 4 ) {
temp_name.erase(temp_name.length() - 4) ;
}
left_units = Trick::UnitsMap::units_map()->get_units(temp_name) ;
new_array = (T *)calloc( out_size , sizeof(T));
for( ii = 0 ; ii < size ; ii++ ) {
PyObject *o = PyTuple_GetItem(input,ii) ;
if (PyFloat_Check(o)) {
new_array[ii] = (T)PyFloat_AsDouble(o) ;
} else if (SWIG_IsOK(SWIG_ConvertPtr(o, &argp2,SWIG_TypeQuery("_p_swig_int"), 0 ))) {
ret = convert_and_set< swig_int , T >( new_array[ii] , argp2 , left_units ) ;
if ( ret != 0 ) {
free(new_array) ;
return NULL;
}
} else if (SWIG_IsOK(SWIG_ConvertPtr(o, &argp2,SWIG_TypeQuery("_p_swig_double"), 0 ))) {
ret = convert_and_set< swig_double , T >( new_array[ii] , argp2 , left_units ) ;
if ( ret != 0 ) {
free(new_array) ;
return NULL;
}
} else if (PyInt_Check(o)) {
new_array[ii] = (T)PyInt_AsLong(o) ;
} else {
PyErr_SetString(PyExc_TypeError,"List must contain numerical values");
free(new_array) ;
return NULL;
}
}
} else if ( PyString_Check(input) ) {
unsigned int size = PyString_Size(input) ;
char * temp_str = PyString_AsString(input) ;
if ( size > out_size ) {
PyErr_SetString(PyExc_TypeError,"List too long to fit.");
return NULL ;
}
new_array = (T *)calloc( out_size , sizeof(T));
for( ii = 0 ; ii < size ; ii++ ){
new_array[ii] = (T)temp_str[ii] ;
}
} else {
PyErr_SetString(PyExc_TypeError,"Input must be of type List");
return NULL;
}
return new_array ;
}
template<typename T > static int typemap_in_1dp( PyObject *input , const char * basetype ,
const char * symname , T ** output ) {
// INT *
void * argp2 ;
int ret ;
// convert list to tuple so we only have to deal with tuples in code below.
if (PyList_Check(input)) {
input = PyList_AsTuple(input) ;
}
if (PyTuple_Check(input)) {
unsigned int size = PyTuple_Size(input) ;
unsigned int ii = 0;
std::string temp_name ;
std::string left_units ;
temp_name = symname ;
if ( temp_name.length() > 4 ) {
temp_name.erase(temp_name.length() - 4) ;
}
left_units = Trick::UnitsMap::units_map()->get_units(temp_name) ;
*output = (T *)TMM_declare_var_1d(basetype , size) ;
for( ii = 0 ; ii < size ; ii++ ) {
PyObject *o = PyTuple_GetItem(input,ii) ;
if (PyFloat_Check(o)) {
(*output)[ii] = (T)PyFloat_AsDouble(o) ;
} else if (SWIG_IsOK(SWIG_ConvertPtr(o, &argp2,SWIG_TypeQuery("_p_swig_int"), 0 ))) {
ret = convert_and_set< swig_int , T >( (*output)[ii] , argp2 , left_units ) ;
if ( ret != 0 ) {
TMM_delete_var_a(*output) ;
return -1;
}
} else if (SWIG_IsOK(SWIG_ConvertPtr(o, &argp2,SWIG_TypeQuery("_p_swig_double"), 0 ))) {
ret = convert_and_set< swig_double , T >( (*output)[ii] , argp2 , left_units ) ;
if ( ret != 0 ) {
TMM_delete_var_a(*output) ;
return -1;
}
} else if (PyInt_Check(o)) {
(*output)[ii] = (T)PyInt_AsLong(o) ;
} else {
PyErr_SetString(PyExc_TypeError,"List must contain int");
return -1;
}
}
} else if ( SWIG_IsOK(SWIG_ConvertPtr(input, &argp2,SWIG_TypeQuery("_p_swig_ref"), 0)) ) {
// Array to pointer assignment
swig_ref * temp_swig_ref = reinterpret_cast< swig_ref * >(argp2);
if ( temp_swig_ref != NULL ) {
*output = (T *)temp_swig_ref->ref.address ;
}
// package the array address as a void
} else if ( SWIG_IsOK(SWIG_ConvertPtr(input, &argp2,SWIG_TypeQuery("_p_REF2"), 0)) ) {
// We have an address coming in, we don't have to do any translation
REF2 * temp_ref = reinterpret_cast< REF2 * >(argp2) ;
if ( temp_ref != NULL ){
*output = (T *)temp_ref->address ;
}
} else if ( SWIG_IsOK(SWIG_ConvertPtr(input, &argp2,SWIG_TypeQuery("_p_void"), 0)) ) {
// We have an address coming in, we don't have to do any translation
*output = reinterpret_cast< T * >(argp2) ;
} else {
if ( !strncmp( basetype , "char" , 4 )) {
if ( PyString_Check(input) ) {
*output = (T *)TMM_strdup(PyString_AsString(input)) ;
} else {
PyErr_SetString(PyExc_TypeError,"swig_int (char): Input must be of type List, string, or a pointer type");
return -1;
}
} else {
PyErr_SetString(PyExc_TypeError,"swig_int: Input must be of type List or a pointer type");
return -1;
}
}
return 0 ;
}
template<typename T, typename baseT > static void * typemap_in_2d( PyObject *input , unsigned int out_dim0, unsigned int out_dim1, const char * symname ) {
//INT[ANY][ANY] IN
void * argp2 ;
unsigned int ii , jj ;
unsigned int size0, size1 ;
T * new_array ;
int ret ;
// convert list to tuple so we only have to deal with tuples in code below.
if (PyList_Check(input)) {
input = PyList_AsTuple(input) ;
}
if (PyTuple_Check(input)) {
size0 = PyTuple_Size(input) ;
if ( size0 > out_dim0 ) {
PyErr_SetString(PyExc_TypeError,"Outer list too long to fit.");
return NULL ;
}
std::string temp_name ;
std::string left_units ;
temp_name = symname ;
if ( temp_name.length() > 4 ) {
temp_name.erase(temp_name.length() - 4) ;
}
left_units = Trick::UnitsMap::units_map()->get_units(temp_name) ;
new_array = (T *)calloc( out_dim0 , sizeof(T));
for( ii = 0 ; ii < size0 ; ii++ ) {
PyObject *o = PyTuple_GetItem( input, ii ) ;
// convert list to tuple so we only have to deal with tuples in code below.
if (PyList_Check(o)) {
o = PyList_AsTuple(o) ;
}
if (PyTuple_Check(o)) {
size1 = PyTuple_Size(o) ;
if ( size1 > out_dim1 ) {
PyErr_SetString(PyExc_TypeError,"List too long to fit.");
return NULL ;
}
for( jj = 0 ; jj < size1 ; jj++ ){
PyObject *o2 = PyTuple_GetItem( o , jj ) ;
if (PyFloat_Check(o2)) {
new_array[ii][jj] = (baseT)PyFloat_AsDouble(o2) ;
} else if (SWIG_IsOK(SWIG_ConvertPtr(o2, &argp2, SWIG_TypeQuery("_p_swig_int"), 0 ))) {
ret = convert_and_set< swig_int , baseT >( new_array[ii][jj] , argp2 , left_units ) ;
if ( ret != 0 ) {
free(new_array) ;
return NULL;
}
} else if (SWIG_IsOK(SWIG_ConvertPtr(o2, &argp2, SWIG_TypeQuery("_p_swig_double"), 0 ))) {
ret = convert_and_set< swig_double , baseT >( new_array[ii][jj] , argp2 , left_units ) ;
if ( ret != 0 ) {
free(new_array) ;
return NULL;
}
} else if (PyInt_Check(o2)) {
new_array[ii][jj] = (baseT)PyInt_AsLong(o2) ;
}
}
} else if ( PyString_Check(o) ) {
unsigned int size = PyString_Size(o) ;
char * temp_str = PyString_AsString(o) ;
if ( size > out_dim1 ) {
PyErr_SetString(PyExc_TypeError,"String too long to fit.");
return NULL ;
}
for( jj = 0 ; jj < size ; jj++ ){
new_array[ii][jj] = (baseT)temp_str[jj] ;
}
} else {
PyErr_SetString(PyExc_TypeError,"Input must be list of lists");
return NULL;
}
}
} else {
PyErr_SetString(PyExc_TypeError,"Input must be of type List");
return NULL;
}
return new_array ;
}
#endif
Reference in New Issue View Git Blame Copy Permalink