www.gusucode.com > 电路板故障检测源码程序 > 电路板故障检测源码程序/kdtree/kdtree/kd_demo.m

    % pramod vemulapalli 02/08/2010


%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
% Just a simple demo to show the functionality
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
clear
demo_case = [1,2,3,4];

plot_stuff=1;   % 1 if you want to plot the data
% change to 0 if you donot want to plot anything.

if (plot_stuff) close all; end
clc

rand('seed',1)
dimen=2;
X=rand(200,dimen);
point=0.2*ones(1,dimen);

disp('##### Build Tree #####');

tree = kd_buildtree(X,plot_stuff);

for count=1:max(size(demo_case))
    
    switch demo_case(count)

        case 1

            if (plot_stuff); hold on ; end
            if (plot_stuff); plot(point(1),point(2),'g*','MarkerSize',10); end
            disp('##### Closest Point Fast #####');
            [index_vals,vec_vals,node_number] = kd_closestpointfast(tree,point)
            if (plot_stuff); plot(X(index_vals,1),X(index_vals,2),'y*','MarkerSize',10); end
 
        case 2

            if (plot_stuff); hold on ; end
            if (plot_stuff); plot(point(1),point(2),'go'); end
            disp('##### Closest Point Good #####');
            [index_vals,vec_vals,node_number] = kd_closestpointgood(tree,point)
            if (plot_stuff); plot(X(index_vals,1),X(index_vals,2),'m*','MarkerSize',10); end

        case 3
            point=0.6*ones(1,dimen);
            if (plot_stuff); hold on ; end
            if (plot_stuff); plot(point(1),point(2),'g*','MarkerSize',10); end
            disp('##### N Closest Points #####');
            num_of_points=10;
            [index_vals,dist_vals,vec_vals]  = kd_knn(tree,point,num_of_points,plot_stuff)
            if (plot_stuff);
                plot(X(index_vals,1),X(index_vals,2),'g*');
                dist=sqrt(sum((point-X(index_vals(end),1:2)).^2));
                plot(point(1)+dist*cos(0:0.1:2*pi),point(2)+dist*sin(0:0.1:2*pi),'g-','LineWidth',2)
            end


        case 4

            disp('##### Range Query #####');
            point=0.35*ones(1,dimen);
            range=[-0.1*ones(1,dimen); 0.1*ones(1,dimen)];
            [index_vals,dist_vals,vector_vals] = kd_rangequery(tree,point,range)

            %%% plotting stuff
            if (plot_stuff);
                a=point+range(1,:);
                b=point+range(2,:);
                c=[a(1) a(1) b(1) b(1);a(1) b(1) b(1) a(1)];
                d=[a(2) b(2) b(2) a(2);b(2) b(2) a(2) a(2)];
                plot(point(1),point(2),'k*','MarkerSize',10)
                line(c,d, 'color', 'k','LineWidth',2);
                plot(X(index_vals,1),X(index_vals,2),'k*')
            end

    end

end

if (plot_stuff);
    set(gca,'box','on');
    set(gca,'XTickLabel',[]);
    set(gca,'YTickLabel',[]);    
end