Variable-Sampling-Region-RRT/VSR_RRT_optimized.m at main · advicement/Variable-Sampling-Region-RRT · 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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
%% RRT algorithm
% YU CHUANGYANG  Waseda University
% Code for Robot Path Planning using Rapidly-exploring Random Trees
%%
map=im2bw(imread('Maps/23279020_15.tif')); % input map read from a bmp file. for new maps write the file name here
map2=imread('Maps/23279020_15.tiff');

image((map==0).*0 + (map==1).*255);
axis image
colormap(gray(256))

source=[ 1390 22 ]; % source position in Y, X format 23279020 scenario1
goal=[ 680 990 ]; % goal position in Y, X format
% source=[ 1360 1250 ]; % source position in Y, X format 23279020 scenario2
% goal=[ 155 145 ]; % goal position in Y, X format
% source=[ 150 1360 ]; % source position in Y, X format 23129050 scenario3
% goal=[ 1120 110 ]; % goal position in Y, X format
% source=[ 760 1290 ]; % source position in Y, X format 23129050 scenario4
% goal=[ 180 420 ]; % goal position in Y, X format
% source=[ 1210 320 ]; % source position in Y, X format 23729065 scenario5
% goal=[ 180 960 ]; % goal position in Y, X format
% source=[ 1210 1040 ]; % source position in Y, X format 23729065 scenario6
% goal=[ 180 960 ]; % goal position in Y, X format

Gb=1;% set goalbias
goalbias=Gb; % set goalbias
loc=2; % set size of each step of the RRT (local stepsize)
max=5; % set size of each step of the RRT (global stepsize)
disTh=2; % nodes closer than this threshold are taken as almost the same

maxFailedAttempts = 1000000;
display=true; % display of RRT
show=false; % display of the tree
rectanglesize=size(map);
rectanglesize=[rectanglesize(2) rectanglesize(1)];
%%%%% parameters end here %%%%%

tic;
if ~feasiblePoint(source,map), error('source lies on an obstacle or outside map'); end
if ~feasiblePoint(goal,map), error('goal lies on an obstacle or outside map'); end
if display, imshow(map);rectangle('position',[1, 1, rectanglesize-1],'edgecolor','k'); end
RRTree=double([source -1]); % RRT rooted at the source, representation node and parent index
failedAttempts=0;
counter=0;
pathFound=false;
newPoint=source;

while failedAttempts<=maxFailedAttempts  % loop to grow RRTs
    if rand > goalbias
        sample=rand(1,2) .* 10 + newPoint - 5; % random sample
        if sample == newPoint
            continue;
        end
        stepsize=loc;
    else
        stepsize=max;
        sample=goal; % sample taken as goal to bias tree generation to goal
    end
    [A, I]=min(distanceCost(RRTree(:,1:2),sample) ,[],1); % find closest as per the function in the metric node to the sample
    closestNode = RRTree(I(1),1:2);
    theta=atan2(sample(1)-closestNode(1),sample(2)-closestNode(2));  % direction to extend sample to produce new node
    newPoint = double(int32(closestNode(1:2) + stepsize * [sin(theta)  cos(theta)]));
    if ~checkPath(closestNode(1:2), newPoint, map) % if extension of closest node in tree to the new point is feasible
        failedAttempts=failedAttempts+1;
        goalbias=0;
        continue;
    end
    RRTree=[RRTree;newPoint I(1)]; % add node

    if show
        line([closestNode(2);newPoint(2)],[closestNode(1);newPoint(1)],'Color','blue','LineWidth',1.5);
        counter=counter+1;M(counter)=getframe;
    end

    if distanceCost(newPoint,goal)<=max && checkPath(newPoint,goal,map), pathFound=true;break; end % goal reached
    goalbias=Gb;
end

if pathFound && show
    line([newPoint(2);goal(2)],[newPoint(1);goal(1)],'Color','blue','LineWidth',1.5);
    counter=counter+1;M(counter)=getframe;
end

if ~pathFound, error('no path found. maximum attempts reached'); end
path=[newPoint;goal];
prev=I(1);

while prev>0
    path=[RRTree(prev,1:2);path];
    prev=RRTree(prev,3);
end

path_optimized=path;
pathsize=size(path,1);
start_node=1;
curcheck_node=pathsize;

while start_node<size(path_optimized,1)-1
    while curcheck_node-start_node>1
        if checkPathFast(path_optimized(curcheck_node,1:2), path_optimized(start_node,1:2),map,max)
            id_X=path_optimized(curcheck_node,2);
            id_Y=path_optimized(curcheck_node,1);
            path_optimized(start_node+1:curcheck_node-1,:) =[];
            start_node=find(path_optimized(:,1)==id_Y & path_optimized(:,2)==id_X);
            curcheck_node=size(path_optimized,1);
            break;
        else
            curcheck_node=curcheck_node-1;
            if ~(curcheck_node-start_node>1)
                start_node=start_node+1;
                curcheck_node=size(path_optimized,1);
            end
        end
    end
end

fprintf('processing time=%d \n', toc);
pathLength_optimiazed=0;
for i=1:length(path_optimized)-1, pathLength_optimiazed=pathLength_optimiazed+distanceCost(path_optimized(i,1:2),path_optimized(i+1,1:2)); end
fprintf('optimized path length=%d \n', pathLength_optimiazed);
imshow(map);rectangle('position',[1 1 rectanglesize-1],'edgecolor','k');
line(path_optimized(:,2),path_optimized(:,1),'Color','red','LineWidth',2.5);

disp('press any key');
waitforbuttonpress;
f=figure;
imshow(map2);rectangle('position',[1 1 rectanglesize-1],'edgecolor','k');
line(path_optimized(:,2),path_optimized(:,1),'Color','red','LineWidth',2.5);