forked from CelsoReyes/zmap7
-
Notifications
You must be signed in to change notification settings - Fork 7
/
MyProgressMonitorNew.m
102 lines (90 loc) · 3.73 KB
/
MyProgressMonitorNew.m
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
classdef MyProgressMonitorNew < matlab.net.http.ProgressMonitor
properties
ProgHandle
Direction matlab.net.http.MessageType
Value uint64
NewDir matlab.net.http.MessageType = matlab.net.http.MessageType.Request
end
methods
function obj = MyProgressMonitorNew
obj.Interval = .01;
end
function done(obj)
obj.closeit();
end
function delete(obj)
obj.closeit();
end
function set.Direction(obj, dir)
obj.Direction = dir;
obj.changeDir();
end
function set.Value(obj, value)
obj.Value = value;
obj.update();
end
end
methods (Access = private)
function update(obj,~)
% called when Value is set
import matlab.net.http.*
if ~isempty(obj.Value)
if isempty(obj.Max)
% no maximum means we don't know length, so message changes on
% every call
value = 0;
if obj.Direction == MessageType.Request
msg = sprintf('Sent %d bytes...', obj.Value);
else
msg = sprintf('Received %d bytes...', obj.Value);
end
else
% maximum known; update proportional value
value = double(obj.Value)/double(obj.Max);
if obj.NewDir == MessageType.Request
% message changes only on change of direction
if obj.Direction == MessageType.Request
msg = 'Sending...';
else
msg = 'Receiving...';
end
end
end
if isempty(obj.ProgHandle)
% if we don't have a progress bar, display it for first time
obj.ProgHandle = ...
waitbar(value, msg, 'CreateCancelBtn', @(~,~)cancelAndClose(obj));
obj.NewDir = MessageType.Response;
elseif obj.NewDir == MessageType.Request || isempty(obj.Max)
% on change of direction or if no maximum known, change message
waitbar(value, obj.ProgHandle, msg);
obj.NewDir = MessageType.Response;
else
% no direction change else just update proportional value
waitbar(value, obj.ProgHandle);
end
end
function cancelAndClose(obj)
% Call the required CancelFcn and then close our progress bar. This is
% called when user clicks cancel or closes the window.
obj.CancelFcn();
obj.closeit();
end
end
function changeDir(obj,~)
% Called when Direction is set or changed. Leave the progress bar displayed.
obj.NewDir = matlab.net.http.MessageType.Request;
end
end
methods (Access=private)
function closeit(obj)
% Close the progress bar by deleting the handle so CloseRequestFcn isn't
% called, because waitbar calls our cancelAndClose(), which would cause
% recursion.
if ~isempty(obj.ProgHandle)
delete(obj.ProgHandle);
obj.ProgHandle = [];
end
end
end
end