diff options
Diffstat (limited to 'weave/scalar_spec.py')
| -rwxr-xr-x | weave/scalar_spec.py | 119 |
1 files changed, 119 insertions, 0 deletions
diff --git a/weave/scalar_spec.py b/weave/scalar_spec.py new file mode 100755 index 000000000..d23620033 --- /dev/null +++ b/weave/scalar_spec.py @@ -0,0 +1,119 @@ +from base_spec import base_specification +import scalar_info +#from Numeric import * +from types import * + +# the following typemaps are for 32 bit platforms. A way to do this +# general case? maybe ask numeric types how long they are and base +# the decisions on that. + +numeric_to_blitz_type_mapping = {} + +numeric_to_blitz_type_mapping['T'] = 'T' # for templates +numeric_to_blitz_type_mapping['F'] = 'std::complex<float> ' +numeric_to_blitz_type_mapping['D'] = 'std::complex<double> ' +numeric_to_blitz_type_mapping['f'] = 'float' +numeric_to_blitz_type_mapping['d'] = 'double' +numeric_to_blitz_type_mapping['1'] = 'char' +numeric_to_blitz_type_mapping['b'] = 'unsigned char' +numeric_to_blitz_type_mapping['s'] = 'short' +numeric_to_blitz_type_mapping['i'] = 'int' +# not strictly correct, but shoulld be fine fo numeric work. +# add test somewhere to make sure long can be cast to int before using. +numeric_to_blitz_type_mapping['l'] = 'int' + +# standard Python numeric type mappings. +numeric_to_blitz_type_mapping[type(1)] = 'int' +numeric_to_blitz_type_mapping[type(1.)] = 'double' +numeric_to_blitz_type_mapping[type(1.+1.j)] = 'std::complex<double> ' +#hmmm. The following is likely unsafe... +numeric_to_blitz_type_mapping[type(1L)] = 'int' + +class scalar_specification(base_specification): + _build_information = [scalar_info.scalar_info()] + + def type_spec(self,name,value): + # factory + new_spec = self.__class__() + new_spec.name = name + new_spec.numeric_type = type(value) + return new_spec + + def declaration_code(self,templatize = 0,inline=0): + if self.compiler == 'msvc': + return self.msvc_decl_code(templatize,inline) + else: + return self.template_decl_code(templatize,inline) + + def template_decl_code(self,template = 0,inline=0): + type = numeric_to_blitz_type_mapping[self.numeric_type] + name = self.name + var_name = self.retrieve_py_variable(inline) + template = '%(type)s %(name)s = '\ + 'py_to_scalar<%(type)s >(%(var_name)s,"%(name)s");\n' + code = template % locals() + return code + + def msvc_decl_code(self,template = 0,inline=0): + # doesn't support template = 1 + if template: + ValueError, 'msvc compiler does not support templated scalar code.'\ + 'try mingw32 instead (www.mingw.org).' + type = numeric_to_blitz_type_mapping[self.numeric_type] + func_type = self.type_name + name = self.name + var_name = self.retrieve_py_variable(inline) + template = '%(type)s %(name)s = '\ + 'py_to_%(func_type)s (%(var_name)s,"%(name)s");\n' + code = template % locals() + return code + + #def c_function_declaration_code(self): + # code = '%s &%s" % \ + # (numeric_to_blitz_type_mapping[self.numeric_type], self.name) + # return code + + def __repr__(self): + msg = "(%s:: name: %s, type: %s)" % \ + (self.type_name,self.name, self.numeric_type) + return msg + def __cmp__(self,other): + #only works for equal + return cmp(self.name,other.name) or \ + cmp(self.numeric_type,other.numeric_type) or \ + cmp(self.__class__, other.__class__) + +class int_specification(scalar_specification): + type_name = 'int' + def type_match(self,value): + return type(value) in [IntType, LongType] + + def local_dict_code(self): + code = 'local_dict["%s"] = Py::Int(%s);\n' % (self.name,self.name) + return code + +class float_specification(scalar_specification): + type_name = 'float' + def type_match(self,value): + return type(value) in [FloatType] + def local_dict_code(self): + code = 'local_dict["%s"] = Py::Float(%s);\n' % (self.name,self.name) + return code + +class complex_specification(scalar_specification): + type_name = 'complex' + def type_match(self,value): + return type(value) in [ComplexType] + def local_dict_code(self): + code = 'local_dict["%s"] = Py::Complex(%s.real(),%s.imag());\n' % \ + (self.name,self.name,self.name) + return code + +def test(): + from scipy_test import module_test + module_test(__name__,__file__) + +def test_suite(): + from scipy_test import module_test_suite + return module_test_suite(__name__,__file__) +
\ No newline at end of file |