stretching codegen

hysop/backend/device/codegen/base/opencl_codegen.py

@@ -27,6 +27,8 @@ class OpenClCodeGenerator(CodeGenerator):
         '\n': '\n',
         ' ': ' '
     }
+        
+    _global, _local = '__global', '__local'
     
     def __init__(self,name,typegen,ext='.cl',
             known_vars=None, declare_cl_exts=True,
@@ -46,8 +48,6 @@ class OpenClCodeGenerator(CodeGenerator):
             for cl_ext in typegen.cl_requirements():
                 if cl_ext is not None:
                     self.declare_cl_extension(cl_ext)
-        
-        self._global, self._local = '__global', '__local'
     
 
     def test_compile(self, contexts=None):

hysop/backend/device/codegen/base/variables.py

@@ -160,12 +160,13 @@ class CodegenVariable(object):
 
 
     def alias(self, name, ctype, 
-            const=None, volatile=None):
+            const=None, volatile=None, restrict=None):
         handle = copy.copy(self)
         handle.ctype = ctype
         handle.name=name
         handle.const = const if (const is not None) else handle.const
         handle.volatile = volatile if (volatile is not None) else handle.volatile
+        handle.restrict = restrict if (restrict is not None) else handle.restrict
         handle.init='({})({})'.format(handle.full_ctype(), self)
         return handle
 

hysop/backend/device/opencl/opencl_array_backend.py

@@ -2967,5 +2967,42 @@ class OpenClArrayBackend(ArrayBackend):
         """
         self._not_implemented_yet('nanvar')
 
+    @staticmethod
+    def build_codegen_argument(args, name, typegen, ctype, 
+            itype='unsigned long', 
+            const=False, ptr=True, **kargs):
+
+        from hysop.backend.device.codegen.base.utils import ArgDict
+        from hysop.backend.device.codegen.base.variables import CodegenVariable
+        check_instance(args, ArgDict)
+
+        base   = '{}_base'.format(name)
+        offset = '{}_offset'.format(name)
+        assert base not in args
+        assert offset not in args
+        
+        assert 'nl' not in kargs
+        assert 'add_impl_const' not in kargs
+        assert 'init' not in kargs
+
+        args[base] = CodegenVariable(name=base,
+                typegen=typegen, ctype=ctype, ptr=ptr, const=const,
+                add_impl_const=True, nl=False, **kargs)
+
+        args[offset] = CodegenVariable(name=offset,
+                typegen=typegen, ctype=itype,
+                add_impl_const=True,nl=True)
+        
+        char_alias  = args[base].alias(None, ctype='char', restrict=False, volatile=False).full_ctype()
+        ctype_alias = args[base].alias(None, ctype=ctype,  restrict=False, volatile=False).full_ctype()
+        init = '({})(({})({})+{})'.format(ctype_alias, char_alias, base, offset)
+
+        var = CodegenVariable(name=name,
+                typegen=typegen, ctype=ctype, ptr=ptr, const=const,
+                add_impl_const=True, nl=False, 
+                init=init, **kargs)
+
+        return var
+
 
 ArrayBackend._register_backend(clArray.Array, OpenClArrayBackend)