[c++/m] Add MultiTag referencesFilter

Author: mpsonntag

+nix/MultiTag.m

@@ -78,6 +78,11 @@
             c = nix_mx('MultiTag::referenceCount', obj.nix_handle);
         end
 
+        function filtered = filter_references(obj, filter, val)
+            filtered = nix.Utils.filter(obj, filter, val, ...
+                'MultiTag::referencesFiltered', @nix.DataArray);
+        end
+
         % ------------------
         % Features methods
         % ------------------

nix_mx.cc

@@ -372,6 +372,7 @@ void mexFunction(int            nlhs,
         methods->add("MultiTag::openSourceIdx", nixmultitag::openSourceIdx);
         methods->add("MultiTag::compare", nixmultitag::compare);
         methods->add("MultiTag::sourcesFiltered", nixmultitag::sourcesFiltered);
+        methods->add("MultiTag::referencesFiltered", nixmultitag::referencesFiltered);
 
         classdef<nix::Section>("Section", methods)
             .desc(&nixsection::describe)

src/nixmultitag.cc

@@ -117,4 +117,13 @@ namespace nixmultitag {
         output.set(0, res);
     }
 
+    void referencesFiltered(const extractor &input, infusor &output) {
+        nix::MultiTag currObj = input.entity<nix::MultiTag>(1);
+        std::vector<nix::DataArray> res = filterEntity<nix::DataArray>(input,
+                                                [currObj](const nix::util::Filter<nix::DataArray>::type &filter) {
+            return currObj.references(filter);
+        });
+        output.set(0, res);
+    }
+
 } // namespace nixmultitag

src/nixmultitag.h

@@ -41,6 +41,8 @@ namespace nixmultitag {
 
     void sourcesFiltered(const extractor &input, infusor &output);
 
+    void referencesFiltered(const extractor &input, infusor &output);
+
 } // namespace nixmultitag
 
 #endif

tests/TestMultiTag.m

@@ -46,6 +46,7 @@
     funcs{end+1} = @test_attrs;
     funcs{end+1} = @test_compare;
     funcs{end+1} = @test_filter_source;
+    funcs{end+1} = @test_filter_reference;
 end
 
 %% Test: Add sources by entity and id
@@ -856,3 +857,78 @@
     assert(size(filtered, 1) == 1);
     assert(strcmp(filtered{1}.name, mainName));
 end
+
+%% Test: filter references
+function [] = test_filter_reference( varargin )
+    filterName = 'filterMe';
+    filterType = 'filterType';
+    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 7]);
+    t = b.create_multi_tag('testMultiTag', 'nixMultiTag', d);
+    d = b.create_data_array(filterName, 'nixDataArray', nix.DataType.Double, [1 2]);
+    t.add_reference(d);
+    filterID = d.id;
+	d = b.create_data_array('testDataArray1', filterType, nix.DataType.Double, [1 2]);
+    t.add_reference(d);
+    filterIDs = {filterID, d.id};
+    d = b.create_data_array('testDataArray2', filterType, nix.DataType.Double, [1 2]);
+    t.add_reference(d);
+
+    % test empty id filter
+    assert(isempty(f.blocks{1}.multiTags{1}.filter_references(nix.Filter.id, 'IdoNotExist')));
+
+    % test nix.Filter.accept_all
+    filtered = f.blocks{1}.multiTags{1}.filter_references(nix.Filter.accept_all, '');
+    assert(size(filtered, 1) == 3);
+
+    % test nix.Filter.id
+    filtered = f.blocks{1}.multiTags{1}.filter_references(nix.Filter.id, filterID);
+    assert(size(filtered, 1) == 1);
+    assert(strcmp(filtered{1}.id, filterID));
+
+    % test nix.Filter.ids
+    filtered = f.blocks{1}.multiTags{1}.filter_references(nix.Filter.ids, filterIDs);
+    assert(size(filtered, 1) == 2);
+    assert(strcmp(filtered{1}.id, filterIDs{1}) || strcmp(filtered{1}.id, filterIDs{2}));
+    
+    % test nix.Filter.name
+    filtered  = f.blocks{1}.multiTags{1}.filter_references(nix.Filter.name, filterName);
+    assert(size(filtered, 1) == 1);
+    assert(strcmp(filtered{1}.name, filterName));
+    
+    % test nix.Filter.type
+    filtered = f.blocks{1}.multiTags{1}.filter_references(nix.Filter.type, filterType);
+    assert(size(filtered, 1) == 2);
+
+    % test nix.Filter.metadata
+    mainName = 'testSubSection';
+    main = b.create_data_array(mainName, 'nixDataArray', nix.DataType.Bool, [2 2]);
+    t.add_reference(main);
+    subName = 'testSubSection1';
+    s = f.create_section(subName, 'nixSection');
+    main.set_metadata(s);
+    subID = s.id;
+
+    assert(isempty(f.blocks{1}.multiTags{1}.filter_references(nix.Filter.metadata, 'Do not exist')));
+    filtered = f.blocks{1}.multiTags{1}.filter_references(nix.Filter.metadata, subID);
+    assert(size(filtered, 1) == 1);
+    assert(strcmp(filtered{1}.name, mainName));
+
+    % test nix.Filter.source
+    mainName = 'testSubSource';
+    main = b.create_data_array(mainName, 'nixDataArray', nix.DataType.Bool, [2 2]);
+    t.add_reference(main);
+    mainID = main.id;
+    subName = 'testSubSource1';
+    s = b.create_source(subName, 'nixSource');
+    main.add_source(s);
+    subID = s.id;
+
+    assert(isempty(f.blocks{1}.multiTags{1}.filter_references(nix.Filter.source, 'Do not exist')));
+
+    % filter works only for ID, not for name
+    filtered = f.blocks{1}.multiTags{1}.filter_references(nix.Filter.source, subID);
+    assert(size(filtered, 1) == 1);
+    assert(strcmp(filtered{1}.name, mainName));
+end