sampleSysAggr.m

function sysStateAggr = sampleSysAggr(self, numEvents, markActivePassive)
% SYSSTATEAGGR = SAMPLESYSAGGR(NUMEVENTS)
 
if GlobalConstants.DummyMode
    sysStateAggr = NaN;
    return
end
 
if strcmp(self.getOptions.lang,'java')
    % Log-based sampling relies on per-node CSV logs written and parsed by the
    % native JMT path; the JLINE (lang=java) delegation does not produce them.
    line_error(mfilename,'SolverJMT log-based sampling (sampleSysAggr/getProbSysAggr) is not supported with lang=''java''. Use lang=''matlab'' or SolverCTMC.');
end
 
if nargin<2 %~exist('numEvents','var')
    numEvents = self.options.samples;
end
 
if nargin<3
    markActivePassive = false;
end
 
sn = self.getStruct;
numEvents = numEvents - 1; % we include the initialization as an event
Q = getAvgQLenHandles(self);
statStateAggr = cell(sn.nstations,1);
 
% create a temp model
modelCopy = self.model.copy;
modelCopy.resetNetwork;
 
% determine the nodes to logs
isNodeClassLogged = false(modelCopy.getNumberOfNodes, modelCopy.getNumberOfClasses);
for i= 1:modelCopy.getNumberOfStations
    ind = self.model.getNodeIndex(modelCopy.getStationNames{i});
    if sn.nodetype(ind) ~= NodeType.Source
        for r=1:modelCopy.getNumberOfClasses
            if ~Q{i,r}.disabled || nargin == 1
                isNodeClassLogged(ind,r) = true;
            else
                isNodeClassLogged(node,r) = true;
            end
        end
    end
end
 
% apply logging to the copied model
Plinked = sn.rtorig;
isNodeLogged = max(isNodeClassLogged,[],2);
logpath = lineTempDir;
modelCopy.linkAndLog(Plinked, isNodeLogged, logpath);
 
% simulate the model copy and retrieve log data
options = self.getOptions;
options.method = 'jsim';  % Force jsim method to avoid recursion with 'replication'
solverjmt = SolverJMT(modelCopy, options);
solverjmt.maxEvents = numEvents*sn.nnodes*sn.nclasses;
solverjmt.runAnalyzer(); % log data
logData = SolverJMT.parseLogs(modelCopy, isNodeLogged, MetricType.toText(MetricType.QLen));
 
% from here convert from nodes in logData to stations
event = {};
sn = modelCopy.getStruct;
for ist= 1:sn.nstations
    isf = sn.stationToStateful(ist);
    ind = sn.stationToNode(ist);
    t = [];
    nir = cell(1,sn.nclasses);
    event{isf,r} = {};
    if sn.nodetype(ind) == NodeType.Source
        nir{r} = [];
    else
        for r=1:sn.nclasses
            if ~isempty(logData{isf,r})
                [~,uniqTSi] = unique(logData{isf,r}.t);
                if isNodeClassLogged(isf,r)
                    if ~isempty(logData{isf,r})
                        t = logData{isf,r}.t(uniqTSi);
                        event{isf,r} = logData{isf,r}.event;
                        %t = [t(2:end);t(end)];
                        nir{r} = logData{isf,r}.QLen(uniqTSi);
                    end
                end
            else
                nir{r} = [];
                event{isf,r} = {};
            end
        end
    end
    if isfinite(self.options.timespan(2))
        stopAt = find(t>self.options.timespan(2),1,'first');
        if ~isempty(stopAt) && stopAt>1
            t = t(1:(stopAt-1));
            for r=1:length(nir)
                nir{r} = nir{r}(1:(stopAt-1));
            end
        end
    end
    statStateAggr{ist} = struct();
    statStateAggr{ist}.handle = self.model.stations{ist};
    statStateAggr{ist}.t = t;
    statStateAggr{ist}.state = cell2mat(nir);
    statStateAggr{ist}.event = {event{isf,:}};
    statStateAggr{ist}.isaggregate = true;
    %sysStateAggr.arv_job_id = logData{2}.arvID;
    %sysStateAggr.dep_job_id = logData{2}.depID;
end
 
tranSysStateAggr = cell(1,1+sn.nstations);
 
tranSysStateAggr{1} = []; % timestamps
for i=1:sn.nstations % stations
    if isempty(tranSysStateAggr{1})
        tranSysStateAggr{1} = statStateAggr{i}.t;
    else
        tumax = min(max(tranSysStateAggr{1}),max(statStateAggr{i}.t));
        tranSysStateAggr{1} = union(tranSysStateAggr{1}, statStateAggr{i}.t);
        tranSysStateAggr{1} = tranSysStateAggr{1}(tranSysStateAggr{1}<=tumax);
        tranSysStateAggr{1} = union(tranSysStateAggr{1}, statStateAggr{i}.t);
    end
end
 
for i=1:sn.nstations % stations
    ind = sn.stationToNode(i);
    tranSysStateAggr{1+i} = [];
    [~,uniqTSi] = unique(statStateAggr{i}.t);
    if sn.nodetype(ind) ~= NodeType.Source
        for j=1:sn.nclasses % classes
            % we floor the interpolation as we hold the last state
            if ~isempty(uniqTSi)
                Qijt = interp1(statStateAggr{i}.t(uniqTSi), statStateAggr{i}.state(uniqTSi,j), tranSysStateAggr{1},'previous');
                if isnan(Qijt(end))
                    Qijt(end)=Qijt(end-1);
                end
                tranSysStateAggr{1+i} = [tranSysStateAggr{1+i}, Qijt];
            else
                Qijt = NaN*ones(length(tranSysStateAggr{1}),1);
                tranSysStateAggr{1+i} = [tranSysStateAggr{1+i}, Qijt];
            end
        end
    else
        tranSysStateAggr{1+i} = [Inf];
    end
end
 
sysStateAggr = struct();
sysStateAggr.handle = self.model.stations';
sysStateAggr.t = tranSysStateAggr{1};
sysStateAggr.state = {tranSysStateAggr{2:end}};
 
% % % now put the events in the .event cell
% eventSysStateAggr = cell(sn.nstations, sn.nclasses); % timestamps
% for i=1:sn.nstations % stations
%     for r=1:sn.nclasses
%         if isempty(statStateAggr{i}.event)
%             eventSysStateAggr{i,r} = struct();
%         else
%             eventSysStateAggr{i,r} = struct();
%             for e=1:size(statStateAggr{i}.event{r},1)
%                 if ~isempty(statStateAggr{i}.event{r}{e})
%                     %statStateAggr{i}.event{r}{e}.t
%                     %etpos = find(sysStateAggr.t == statStateAggr{i}.event{r}{e}.t);
%                     %if ~isempty(etpos)
%                     evtype = statStateAggr{i}.event{r}{e}.event;
%                     evfield = EventType.toText(evtype);
%                     if ~isfield(eventSysStateAggr{i,r},evfield)
%                         eventSysStateAggr{i,r}.(EventType.toText(evtype)) = {};
%                     end
%                     eventSysStateAggr{i,r}.(EventType.toText(evtype)){end+1,1} = statStateAggr{i}.event{r}{e};
%                     %end
%                 end
%             end
%             %eventSysStateAggr{i,r} = statStateAggr{i}.event{r};
%         end
%     end
% end
% %sysStateAggr.t = [0; sysStateAggr.t(1:end-1)];
% sysStateAggr.event = eventSysStateAggr;
 
% Flatten the 2D `event` cell into a column cell of Event objects, sorted by
% time. Robust to the fact that some `event{i,j}` slots may be `[]` (numeric)
% rather than `{}` (cell) when nclasses>1, due to implicit cell-array growth.
flatEvents = {};
for ii=1:size(event,1)
    for jj=1:size(event,2)
        cellij = event{ii,jj};
        if iscell(cellij) && ~isempty(cellij)
            for kk=1:numel(cellij)
                if isa(cellij{kk}, 'Event')
                    flatEvents{end+1,1} = cellij{kk};
                end
            end
        end
    end
end
if isempty(flatEvents)
    sysStateAggr.event = cell(0,1);
else
    event_t = cellfun(@(c) c.t, flatEvents);
    [~,I] = sort(event_t);
    sysStateAggr.event = flatEvents(I);
end
sysStateAggr.isaggregate = true;
%sysStateAggr.arv_job_id = logData{2}.arvID;
%sysStateAggr.dep_job_id = logData{2}.depID;
 
if markActivePassive
    apevent = cell(1,length(sysStateAggr.t)-1);
    for ti = 1:length(apevent)
        apevent{ti} = struct('active',[],'passive',[]);
    end
    for e=1:length(sysStateAggr.event)
        ti = find(sysStateAggr.event{e}.t == sysStateAggr.t);
        if ~isempty(ti)
        switch sysStateAggr.event{e}.event
            case EventType.ARV
                apevent{ti-1}.passive = sysStateAggr.event{e};
            otherwise
                apevent{ti-1}.active = sysStateAggr.event{e};
        end
        end
    end
    sysStateAggr.event = apevent';
end
end