LSD直线提取算法MATLAB.rar

function [linesInfo1,linesInfo2]=LineRefiner(linesInfo1,linesInfo2,valIntPts,distThre) % [linesInfo1,linesInfo2]=LineRefiner(linesInfo1,linesInfo2,valIntPts) % 用于将各线段的端点延伸至最近的有效交点处。 % 输入： % 输入包含3个必要参数。 % linesInfo1为主方向直线段结构体数组<1xM struct>。 % linesInfo2为次方向直线段结构体数组<1xN struct>。 % valIntPts为有效交点结构体矩阵<MxN struct>。 % 输出： % 输出包含2个必要参数。 % linesInfo1为修正后的主方向直线段结构体数组<1xM struct>。 % linesInfo2为修正后的次方向直线段结构体数组<1xM struct>。 % 示例： % priLinesInfo=LinesSorter(priLinesInfo); [M,N]=size(valIntPts); used1=cell(1,M); used2=cell(1,N); for m=1:M used1{m}=zeros(1,length(linesInfo1(m).x1)); end; for n=1:N used2{n}=zeros(1,length(linesInfo2(n).x1)); end; for i=1:M for j=1:N switch (valIntPts(i,j).type) case 2 %% 在主方向直线上且距离次方向线段端点在阈值内(在线段上) for n=1:length(linesInfo2(j).x1) dist1=sqrt((valIntPts(i,j).x-linesInfo2(j).x1(n))^2+(valIntPts(i,j).y-linesInfo2(j).y1(n))^2); dist2=sqrt((valIntPts(i,j).x-linesInfo2(j).x2(n))^2+(valIntPts(i,j).y-linesInfo2(j).y2(n))^2); if ((dist1>distThre)&&(dist2>distThre)) continue; end; if (dist1<dist2) linesInfo2(j).x1(n)=valIntPts(i,j).x; linesInfo2(j).y1(n)=valIntPts(i,j).y; else %% 判断右端点是否已经延长过，如果延长过则不再延长，除非再延长的长度小于2 if (used2{j}(n)==0) linesInfo2(j).x2(n)=valIntPts(i,j).x; linesInfo2(j).y2(n)=valIntPts(i,j).y; used2{j}(n)=1; elseif (dist2<2) linesInfo2(j).x2(n)=valIntPts(i,j).x; linesInfo2(j).y2(n)=valIntPts(i,j).y; end; end; linesInfo2(j).len(n)=sqrt((linesInfo2(j).x2(n)-linesInfo2(j).x1(n))^2+(linesInfo2(j).y2(n)-linesInfo2(j).y1(n))^2); end; case 3 %% 在次方向直线上且距离主方向线段端点在阈值内(在线段上) for m=1:length(linesInfo1(i).x1) dist1=sqrt((valIntPts(i,j).x-linesInfo1(i).x1(m))^2+(valIntPts(i,j).y-linesInfo1(i).y1(m))^2); dist2=sqrt((valIntPts(i,j).x-linesInfo1(i).x2(m))^2+(valIntPts(i,j).y-linesInfo1(i).y2(m))^2); if ((dist1>distThre)&&(dist2>distThre)) continue; end; if (dist1<dist2) linesInfo1(i).x1(m)=valIntPts(i,j).x; linesInfo1(i).y1(m)=valIntPts(i,j).y; else %% 判断右端点是否已经延长过，如果延长过则不再延长，除非再延长的长度小于2 if (used1{i}(m)==0) linesInfo1(i).x2(m)=valIntPts(i,j).x; linesInfo1(i).y2(m)=valIntPts(i,j).y; used1{i}(m)=1; elseif (dist2<2) linesInfo1(i).x2(m)=valIntPts(i,j).x; linesInfo1(i).y2(m)=valIntPts(i,j).y; end; end; linesInfo1(i).len(m)=sqrt((linesInfo1(i).x2(m)-linesInfo1(i).x1(m))^2+(linesInfo1(i).y2(m)-linesInfo1(i).y1(m))^2); end; case 4 %% 距离俩线段端点均在阈值内(不在线段上) for m=1:length(linesInfo1(i).x1) dist1=sqrt((valIntPts(i,j).x-linesInfo1(i).x1(m))^2+(valIntPts(i,j).y-linesInfo1(i).y1(m))^2); dist2=sqrt((valIntPts(i,j).x-linesInfo1(i).x2(m))^2+(valIntPts(i,j).y-linesInfo1(i).y2(m))^2); if ((dist1>distThre)&&(dist2>distThre)) continue; end; if (dist1<dist2) linesInfo1(i).x1(m)=valIntPts(i,j).x; linesInfo1(i).y1(m)=valIntPts(i,j).y; else %% 判断右端点是否已经延长过，如果延长过则不再延长，除非再延长的长度小于2 if (used1{i}(m)==0) linesInfo1(i).x2(m)=valIntPts(i,j).x; linesInfo1(i).y2(m)=valIntPts(i,j).y; used1{i}(m)=1; elseif (dist2<2) linesInfo1(i).x2(m)=valIntPts(i,j).x; linesInfo1(i).y2(m)=valIntPts(i,j).y; end; end; linesInfo1(i).len(m)=sqrt((linesInfo1(i).x2(m)-linesInfo1(i).x1(m))^2+(linesInfo1(i).y2(m)-linesInfo1(i).y1(m))^2); end; for n=1:length(linesInfo2(j).x1) dist3=sqrt((valIntPts(i,j).x-linesInfo2(j).x1(n))^2+(valIntPts(i,j).y-linesInfo2(j).y1(n))^2); dist4=sqrt((valIntPts(i,j).x-linesInfo2(j).x2(n))^2+(valIntPts(i,j).y-linesInfo2(j).y2(n))^2); if ((dist3>distThre)&&(dist4>distThre)) continue; end; if (dist3<dist4) linesInfo2(j).x1(n)=valIntPts(i,j).x; linesInfo2(j).y1(n)=valIntPts(i,j).y; else %% 判断右端点是否已经延长过，如果延长过则不再延长，除非再延长的长度小于2 if (used2{j}(n)==0) linesInfo2(j).x2(n)=valIntPts(i,j).x; linesInfo2(j).y2(n)=valIntPts(i,j).y; used2{j}(n)=1; elseif (dist4<2) linesInfo2(j).x2(n)=valIntPts(i,j).x; linesInfo2(j).y2(n)=valIntPts(i,j).y; end; end; linesInfo2(j).len(n)=sqrt((linesInfo2(j).x2(n)-linesInfo2(j).x1(n))^2+(linesInfo2(j).y2(n)-linesInfo2(j).y1(n))^2); end; otherwise end; end; end; end