Node.m

classdef Node < NetworkElement
    % An abstract for a node in a Network model
    %
    % Copyright (c) 2012-2026, Imperial College London
    % All rights reserved.
 
    properties
        model;
        input;
        server;
        output;
        index;
    end
 
    methods(Hidden)
        %Constructor
        function self = Node(name)
            % SELF = NODE(NAME)
 
            self@NetworkElement(char(name));
            self.index = NaN;
        end
 
        function self = setModel(self, model)
            % SELF = SETMODEL(MODEL)
            %
            % Add a pointer to the model object
 
            self.model = model;
        end
 
        function self = link(self, nodeTo)
            % SELF = LINK(NODETO)
            %
            %
 
            self.model.addLink(self,nodeTo);
        end
 
        function self = reset(self)
            % SELF = RESET()
            %
            % Reset internal data structures when the network model is
            % reset
 
        end
    end
 
    methods
 
        function sections = getSections(self)
            % SECTIONS = GETSECTIONS()
 
            sections = {self.input, self.server, self.output};
        end
 
        function setProbRouting(self, class, destination, probability)
            % SETPROBROUTING(CLASS, DESTINATION, PROBABILITY)
 
            setRouting(self, class, RoutingStrategy.PROB, destination, probability);
        end
 
        function setRouting(self, class, strategy, par1, par2)
            % SETROUTING(CLASS, STRATEGY, PARAM)
            % SETROUTING(CLASS, STRATEGY, DESTINATION, PROBABILITY)
 
            %global GlobalConstants.CoarseTol
 
            if self.model.isJavaNative()
                jline_classes = self.model.obj.getClasses();
                switch strategy
                    case RoutingStrategy.RAND
                        self.obj.setRouting(jline_classes.get(class.index-1),jline.lang.constant.RoutingStrategy.RAND);
                    case RoutingStrategy.RROBIN
                        self.obj.setRouting(jline_classes.get(class.index-1),jline.lang.constant.RoutingStrategy.RROBIN);
                    case RoutingStrategy.WRROBIN
                        node_target =  self.model.obj.getNodeByName(par1.getName());
                        weight = par2;
                        self.obj.setRouting(jline_classes.get(class.index-1),jline.lang.constant.RoutingStrategy.WRROBIN, node_target, weight);
                    case RoutingStrategy.DISABLED
                        self.obj.setRouting(jline_classes.get(class.index-1),jline.lang.constant.RoutingStrategy.DISABLED);
                    case RoutingStrategy.PROB
                        line_error(mfilename, 'Use setProbRouting to assign routing probabilities for a JNetwork node.');
                end
                return
            end
            
            if isa(self,'Cache')
                switch strategy
                    case {RoutingStrategy.KCHOICES, RoutingStrategy.WRROBIN, RoutingStrategy.RROBIN}
                        line_error(mfilename,'State-dependent routing not supported with caches. Add instead a Router node after the cache.');
                end
            end
            switch strategy
                case RoutingStrategy.KCHOICES
                    if nargin < 4
                        par1 = 2; % param_k
                    end
                    if nargin < 5
                        par2 = false; % param_memory
                    end
                    param_k = par1;
                    param_memory = par2;
                    self.output.outputStrategy{1, class.index}{2} = RoutingStrategy.toText(strategy);
                    self.output.outputStrategy{1, class.index}{3}{1} = param_k;
                    self.output.outputStrategy{1, class.index}{3}{2} = param_memory;
                case RoutingStrategy.WRROBIN                    
                    destination = par1;
                    weight = par2;
                    if abs(weight-round(weight)) < GlobalConstants.CoarseTol
                        self.output.outputStrategy{1, class.index}{2} = RoutingStrategy.toText(strategy);
                        if length(self.output.outputStrategy{1, class.index})<3
                            self.output.outputStrategy{1, class.index}{3}{1} = {destination, weight};
                        else
                            self.output.outputStrategy{1, class.index}{3}{end+1} = {destination, weight};
                        end
                    else
                        line_error(mfilename,'Weighted round robin weights must be integers.')
                    end
                case RoutingStrategy.RL
                    self.output.outputStrategy{1, class.index}{2} = RoutingStrategy.toText(strategy);
                    if nargin < 4
                        par1 = -1;
                        par2 = {-1, -1};
                    end
                    self.output.outputStrategy{1, class.index}{3} = par1;      % part1 is value function (tabular or FA)
                    self.output.outputStrategy{1, class.index}{4} = par2{1};      % part2{2} is nodes that need action
                    self.output.outputStrategy{1, class.index}{5} = par2{2};      % part2{2} is state size (truncation_value + 1)
                otherwise
                    switch nargin
                        case 3 % no destination specified
                            self.output.outputStrategy{1, class.index}{2} = RoutingStrategy.toText(strategy);
                        case 5
                            destination = par1;
                            probability = par2;
                            self.output.outputStrategy{1, class.index}{2} = RoutingStrategy.toText(strategy);
                            if length(self.output.outputStrategy{1, class.index})<3
                                self.output.outputStrategy{1, class.index}{3}{1} = {destination, probability};
                            else
                                self.output.outputStrategy{1, class.index}{3}{end+1} = {destination, probability};
                            end
                    end
            end
        end
 
        function bool = hasClassSwitching(self)
            % BOOL = HASCLASSSWITCHING()
 
            bool = isa(self.server,'ClassSwitcher');
        end
 
        function bool = isStateful(self)
            % BOOL = ISSTATEFUL()
 
            bool = isa(self,'StatefulNode');
        end
 
        function bool = isStation(self)
            % BOOL = ISSTATION()
 
            bool = isa(self,'Station');
        end
    end
 
    methods(Access = protected)
        % Override copyElement method:
        function clone = copyElement(self)
            % CLONE = COPYELEMENT()
 
            % Make a shallow copy of all properties
            clone = copyElement@Copyable(self);
            % Make a deep copy of each object
            clone.input = self.input.copy;
            clone.server = self.server.copy;
            clone.output = self.output.copy;
        end
    end
 
    methods (Access = public)
        function ind = subsindex(self)
            % IND = SUBSINDEX()
            if isa(self.model,'Network')
                % Handle the new delegation pattern
                node_idx = self.model.getNodeIndex(self.name);
                if isempty(node_idx) || isnan(node_idx) || node_idx <= 0
                    error('Node:subsindex', 'Invalid node index for %s: %g', self.name, node_idx);
                end
                ind = double(node_idx)-1; % 0 based for MATLAB indexing
            elseif isa(self.model,'MNetwork') 
                node_idx = self.model.getNodeIndex(self.name);
                if isempty(node_idx) || isnan(node_idx) || node_idx <= 0
                    error('Node:subsindex', 'Invalid node index for %s: %g', self.name, node_idx);
                end
                ind = double(node_idx)-1; % 0 based
            elseif isa(self.model,'JNetwork') 
                ind = self.model.obj.getNodeIndex(self.obj);
            else
                error('Node:subsindex', 'Unsupported model type: %s', class(self.model));
            end
        end
 
        function V = horzcat(self, varargin)
            % V = HORZCAT(VARARGIN)
 
            V = zeros(1, length(varargin) + 1);
            try
                self_idx = self.subsindex();
                if numel(self_idx) ~= 1
                    error('Node:horzcat', 'subsindex returned non-scalar value for %s: %s', self.name, mat2str(self_idx));
                end
                V(1) = 1+ self_idx;
                
                for v=1:length(varargin)
                    if isa(varargin{v}, 'Node')
                        node_idx = varargin{v}.subsindex();
                        if numel(node_idx) ~= 1
                            error('Node:horzcat', 'subsindex returned non-scalar value for %s: %s', varargin{v}.name, mat2str(node_idx));
                        end
                        V(1+v) = 1+ node_idx;
                    else
                        error('Node:horzcat', 'Element %d is not a Node object', v);
                    end
                end
            catch e
                error('Node:horzcat', 'Error in horizontal concatenation: %s', e.message);
            end
        end
 
        function V = vertcat(self, varargin)
            % V = VERTCAT(VARARGIN)
 
            V = zeros(length(varargin) + 1, 1);
            try
                V(1) = 1+ self.subsindex;
                for v=1:length(varargin)
                    if isa(varargin{v}, 'Node')
                        V(1+v) = 1+varargin{v}.subsindex;
                    else
                        error('Node:vertcat', 'Element %d is not a Node object', v);
                    end
                end
            catch e
                error('Node:vertcat', 'Error in vertical concatenation: %s', e.message);
            end
        end
 
        function summary(self)
            % SUMMARY()
 
            line_printf('\nNode: <strong>%s</strong>',self.getName);
            %self.input.summary;
            %            self.server.summary;
            %            self.output.summary;
        end
    end
end