StringArt/HookToEdgeFinder.m at master · bdring/StringArt · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
classdef HookToEdgeFinder < handle
    properties
        hookIndices
        connectionType
    end

    methods
        function obj = HookToEdgeFinder(edgeCodes, connectionType)
             if nargin > 0
                obj.hookIndices = edgeCodes;

                obj.connectionType = [];
                if nargin > 1
                    obj.connectionType = connectionType;
                end
             end
        end

        function I = compute(obj, hookIndex)
            [i, j] = find(obj.hookIndices == hookIndex);
            I = i;

            if ~isempty(obj.connectionType)
                T = obj.connectionType(i);

                switchMask = j == 2;

                oneMask = switchMask & T == 1;
                twoMask = switchMask & T == 2;

                T(oneMask) = 2;
                T(twoMask) = 1;

                I = [I T];
            end
        end
    end
end