AWGF-code/scatterbar.m at master · Jeaninezpp/AWGF-code · 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
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
103
104
105
106
107
108
function scatterbar(x,y,z,scale)
%   根据散点数据绘制3维彩色柱状图
%   scatterbar(x,y,z,scale)  x,y,z是实值数组，用来指定柱子顶面中心点三维坐标。
%              scale是大于0的标量，用来指定柱子的粗细，scale越大，柱子越细，默
%              认情况下根据坐标点自动计算柱子的粗细。
%
%   CopyRight:xiezhh（谢中华）
%   2011.10.31
%   Example：
%       [x,y] = meshgrid(-6:6,-3:0.5:3);
%       z = mvnpdf([x(:),y(:)],[0,0],[4,0;0,1]);
%       scatterbar(x,y,z)
%       scatterbar(x,y,z,50);

% 输入参数类型判断
if nargin < 3
    error('至少需要三个输入参数');
end
if ~isreal(x) || ~isreal(y) || ~isreal(z)
    error('前三个输入应为实值数组');
end

% 提取x,y,z等长部分的元素
x = x(:);
y = y(:);
z = z(:);
n = min([numel(x) numel(y) numel(z)]);
x = x(1:n);
y = y(1:n);
z = z(1:n);

% 计算极差和差分值
rx = range(x);
ry = range(y);
dx = abs(diff(x));
dx = min(dx(dx>0));
dy = abs(diff(y));
dy = min(dy(dy>0));

% 自动计算柱子的粗细
if nargin == 3
    if ~isempty(dx)
        hx = dx/2;
    else
        hx = 0.5;
    end
    if ~isempty(dy)
        hy = dy/2;
    else
        hy = 0.5;
    end
end

% 根据用户输入参数scale计算柱子的粗细
if nargin == 4
    if ~isreal(scale) || scale < 0
        error('第四个输入应为正的标量');
    end
    if rx == 0 && ry == 0
        rx = 0.5*scale;
        ry = rx;
    elseif rx == 0 || ry == 0
        rx = max(rx,ry);
        ry = rx;
    end
    hx = rx/scale;
    hy = ry/scale;
end

% 通过循环绘制三维彩色柱状图
figure
hold on
Xp = [];
Yp = [];
Zp = [];
for i = 1:n
    [xp,yp,zp] = Vertices(x(i),y(i),z(i));
    Xp = [Xp;xp];
    Yp = [Yp;yp];
    Zp = [Zp;zp];
end
%通过surf函数生成彩色的立方体盒子
h = surf(Xp,Yp,Zp,Zp,'FaceColor','interp');
%set(h,'FaceAlpha',0.25);    %设置立方体盒子透明度
grid on
view(3)
hold off

%--------------------------------------------------
% 求柱子顶点的子函数
%--------------------------------------------------
function [xp,yp,zp] = Vertices(x,y,z)
    % 由长方体底面中心坐标求顶点坐标
    xp = [x-hx x-hx x+hx x+hx x-hx
          x-hx x-hx x+hx x+hx x-hx
          x-hx x-hx x+hx x+hx x-hx
          x-hx x-hx x+hx x+hx x-hx
          x    x    x    x    x
          NaN  NaN  NaN  NaN  NaN];
    yp = [y-hy y+hy y+hy y-hy y-hy
          y-hy y+hy y+hy y-hy y-hy
          y-hy y+hy y+hy y-hy y-hy
          y-hy y+hy y+hy y-hy y-hy
          y    y    y    y    y
          NaN  NaN  NaN  NaN  NaN];
    zp = [repmat(linspace(0,z,4)',[1,5]);z z z z z;NaN NaN NaN NaN NaN];
end
end