Template changes for rep caps

Author: Vaishnavigupta1312

build/helper/__init__.py

@@ -51,6 +51,7 @@
 from build.helper.metadata_filters import filter_library_functions  # noqa: F401
 from build.helper.metadata_filters import filter_parameters  # noqa: F401
 from build.helper.metadata_filters import filter_public_functions  # noqa: F401
+from build.helper.metadata_filters import filter_rep_cap_supported_attributes
 
 from build.helper.metadata_find import find_custom_type  # noqa: F401
 from build.helper.metadata_find import find_session_handle_parameter  # noqa: F401

build/helper/metadata_add_all.py

@@ -12,6 +12,7 @@
 from .helper import get_python_type_for_api_type
 from .metadata_filters import filter_codegen_attributes
 from .metadata_filters import filter_codegen_functions
+from .metadata_filters import filter_rep_cap_supported_attributes
 from .metadata_find import find_custom_type
 from .metadata_find import find_size_parameter
 from .metadata_merge_dicts import merge_helper
@@ -721,6 +722,9 @@ def add_all_config_metadata(config):
     if 'uses_nitclk' not in config:
         config['uses_nitclk'] = False
 
+    if 'rep_cap_expansion' not in config:
+        config['rep_cap_expansion'] = 'session'
+
     return config
 
 

build/helper/metadata_filters.py

@@ -450,4 +450,14 @@ def filter_codegen_enums(enums):
     '''Returns enum metadata only for those enums to be included in codegen'''
     return {k: v for k, v in enums.items() if v['codegen_method'] != 'no'}
 
+def filter_rep_cap_supported_attributes(attributes, rep_cap_name):
+    '''Returns attribute metadata only for those attributes that support the specified repeated capability.
 
+    Args:
+        attributes: Dictionary of attribute metadata.
+        rep_cap_name: The name of the repeated capability to filter by.
+
+    Returns:
+        Dictionary of attributes that support the specified repeated capability.
+    '''
+    return {k: v for k, v in attributes.items() if rep_cap_name in v.get('supported_rep_caps', [])}

build/templates/session.py.mako

@@ -91,6 +91,85 @@ class _Lock(object):
 
 % endif
 % if len(config['repeated_capabilities']) > 0:
+
+% if config['rep_cap_expansion'] == 'custom':
+# Dynamically handle repeated capabilities
+% for rep_cap in config['repeated_capabilities']:
+
+class _RepCap${rep_cap['python_name'].capitalize()}(object):
+    % for attribute in helper.sorted_attrs(helper.filter_rep_cap_supported_attributes(attributes, rep_cap['python_name'])):
+<%
+helper.add_attribute_rep_cap_tip(attributes[attribute], config)
+%>\
+    % if attributes[attribute]['enum']:
+        % if helper.enum_uses_converter(enums[attributes[attribute]['enum']]):
+    ${attributes[attribute]['python_name']} = _attributes.AttributeEnumWithConverter(_attributes.AttributeEnum(_attributes.Attribute${attributes[attribute]['type']}, enums.${enums[attributes[attribute]['enum']]['python_name']}, ${attribute}), _converters.${enums[attributes[attribute]['enum']]['enum_to_converted_value_function_name']}, _converters.${enums[attributes[attribute]['enum']]['converted_value_to_enum_function_name']})
+        % else:
+    ${attributes[attribute]['python_name']} = _attributes.AttributeEnum(_attributes.Attribute${attributes[attribute]['type']}, enums.${enums[attributes[attribute]['enum']]['python_name']}, ${attribute})
+        % endif
+    % else:
+    ${attributes[attribute]['python_name']} = _attributes.${attributes[attribute]['attribute_class']}(${attribute})
+    % endif
+%   if 'documentation' in attributes[attribute] and len(helper.get_documentation_for_node_docstring(attributes[attribute], config, indent=4).strip()) > 0:
+    '''Type: ${attributes[attribute]['type_in_documentation']}
+
+    ${helper.get_documentation_for_node_docstring(attributes[attribute], config, indent=4)}
+    '''
+%   endif
+% endfor
+    def __init__(self, session, repeated_capability_list):
+        object.__setattr__(self, '_session', session)
+        object.__setattr__(self, '_repeated_capability_list', repeated_capability_list)
+        object.__setattr__(self, '_prefix', '${rep_cap["prefix"]}')
+        object.__setattr__(self, '_current_repeated_capability_list', repeated_capability_list if len(repeated_capability_list) > 0 else [''])
+        object.__setattr__(self, '_separator', '')
+
+    def __setattr__(self, key, value):
+        if key not in dir(self):
+            raise AttributeError("'{0}' object has no attribute '{1}'".format(type(self).__name__, key))
+        object.__setattr__(self, key, value)
+
+    def __getitem__(self, repeated_capability):
+        '''Set/get properties or call methods with a repeated capability (i.e. channels)'''
+        rep_caps_list = _converters.convert_repeated_capabilities(repeated_capability, self._prefix)
+        complete_rep_cap_list = [
+            current_rep_cap + self._separator + rep_cap
+            for current_rep_cap in self._current_repeated_capability_list
+            for rep_cap in rep_caps_list
+        ]
+        object.__setattr__(self, '_current_repeated_capability_list', complete_rep_cap_list)
+        self._current_repeated_capability_list = complete_rep_cap_list
+
+        return self
+
+    def _get_attribute_vi_real64(self, attribute):
+        repeated_capability = ','.join(self._current_repeated_capability_list)
+        value = self._session._interpreter.get_attribute_vi_real64(repeated_capability, attribute)
+        return value
+
+    def _set_attribute_vi_real64(self, attribute, value):
+        repeated_capability = ','.join(self._current_repeated_capability_list)
+        self._session._interpreter.set_attribute_vi_real64(repeated_capability, attribute, value)
+
+    def _get_attribute_vi_int32(self, attribute):
+        repeated_capability = ','.join(self._current_repeated_capability_list)
+        value = self._session._interpreter.get_attribute_vi_int32(repeated_capability, attribute)
+        return value
+
+    def _set_attribute_vi_int32(self, attribute, value):
+        repeated_capability = ','.join(self._current_repeated_capability_list)
+        self._session._interpreter.set_attribute_vi_int32(repeated_capability, attribute, value)
+
+    def _get_attribute_vi_string(self, attribute):
+        repeated_capability = ','.join(self._current_repeated_capability_list)
+        value = self._session._interpreter.get_attribute_vi_string(repeated_capability, attribute)
+        return value
+
+    def _set_attribute_vi_string(self, attribute, value):
+        repeated_capability = ','.join(self._current_repeated_capability_list)
+        self._session._interpreter.set_attribute_vi_string(repeated_capability, attribute, value)
+% endfor
+% else:
 class _RepeatedCapabilities(object):
     def __init__(self, session, prefix, current_repeated_capability_list):
         self._session = session

src/nirfsg/metadata/config.py

@@ -39,6 +39,7 @@
         }
     },
     'module_name': 'nirfsg',
+    'rep_cap_expansion': 'custom',
     'repeated_capabilities': [
         {
             'prefix': 'marker',