[c++/m] Add DataArray getSourceByIndex function

Author: mpsonntag

nix_mx.cc

@@ -246,6 +246,7 @@ void mexFunction(int            nlhs,
         methods->add("DataArray::setPolynomCoefficients", nixdataarray::polynomCoefficients);
         methods->add("DataArray::dataType", nixdataarray::dataType);
         methods->add("DataArray::setDataExtent", nixdataarray::setDataExtent);
+        methods->add("DataArray::openSourceIdx", nixdataarray::openSourceIdx);
 
         classdef<nix::Source>("Source", methods)
             .desc(&nixsource::describe)

src/nixdataarray.cc

@@ -136,4 +136,10 @@ namespace nixdataarray {
         da.dataExtent(size);
     }
 
+    void openSourceIdx(const extractor &input, infusor &output) {
+        nix::DataArray currObj = input.entity<nix::DataArray>(1);
+        nix::ndsize_t idx = (nix::ndsize_t)input.num<double>(2);
+        output.set(0, currObj.getSource(idx));
+    }
+
 } // namespace nixdataarray

src/nixdataarray.h

@@ -41,6 +41,8 @@ namespace nixdataarray {
 
     void setDataExtent(const extractor &input, infusor &output);
 
+    void openSourceIdx(const extractor &input, infusor &output);
+
 } // namespace nixdataarray
 
 #endif

tests/TestDataArray.m

@@ -23,6 +23,7 @@
     funcs{end+1} = @test_add_source;
     funcs{end+1} = @test_add_sources;
     funcs{end+1} = @test_open_source;
+    funcs{end+1} = @test_open_source_idx;
     funcs{end+1} = @test_remove_source;
     funcs{end+1} = @test_has_source;
     funcs{end+1} = @test_source_count;
@@ -368,6 +369,23 @@
     assert(isempty(d.open_source('I do not exist')));
 end
 
+function [] = test_open_source_idx( varargin )
+%% Test Open Source by index
+    f = nix.File(fullfile(pwd, 'tests', 'testRW.h5'), nix.FileMode.Overwrite);
+    b = f.create_block('testBlock', 'nixBlock');
+    d = b.create_data_array('testDataArray', 'nixDataArray', nix.DataType.Double, [2 9]);
+    s1 = b.create_source('testSource1', 'nixSource');
+    s2 = b.create_source('testSource2', 'nixSource');
+    s3 = b.create_source('testSource3', 'nixSource');
+    d.add_source(s1);
+    d.add_source(s2);
+    d.add_source(s3);
+
+    assert(strcmp(f.blocks{1}.dataArrays{1}.open_source_idx(0).name, s1.name));
+    assert(strcmp(f.blocks{1}.dataArrays{1}.open_source_idx(1).name, s2.name));
+    assert(strcmp(f.blocks{1}.dataArrays{1}.open_source_idx(2).name, s3.name));
+end
+
 %% Test: Remove sources by entity and id
 function [] = test_remove_source ( varargin )
     f = nix.File(fullfile(pwd, 'tests', 'testRW.h5'), nix.FileMode.Overwrite);