1 files changed, 54 insertions, 0 deletions

diff --git a/SD-VBS/benchmarks/disparity/src/matlab/getDisparity.m b/SD-VBS/benchmarks/disparity/src/matlab/getDisparity.m
new file mode 100755
index 0000000..db64cac
--- /dev/null
+++ b/SD-VBS/benchmarks/disparity/src/matlab/getDisparity.m
@@ -0,0 +1,54 @@
+function [retDisparity retSAD minSAD]=getDisparity(Ileft, Iright, win_sz, max_shift)
+Ileft=double(Ileft);
+Iright=double(Iright);
+[nr,nc,nb]=size(Ileft);
+retSAD=zeros(nr, nc, max_shift);
+
+if(win_sz > 1)
+    half_win_sz=win_sz/2;
+    IleftPadded=padarray(Ileft, [half_win_sz, half_win_sz]);
+    IrightPadded=padarray(Iright, [half_win_sz, half_win_sz]);
+    for i=1:max_shift   
+        retSAD(:,:,i)=correlateSAD(IleftPadded,IrightPadded, win_sz, i-1);
+    end
+    [minSAD retDisparity]=min(retSAD,[],3);
+else
+    IleftPadded = Ileft;
+    IrightPadded = Iright;
+    retSAD=correlateSAD(IleftPadded,IrightPadded, win_sz, 0);
+    [minSAD retDisparity]=min(retSAD,[],3);
+end
+end
+
+function retSAD=correlateSAD(Ileft, Iright, win_sz, disparity)
+
+Iright_moved=padarray(Iright, [0,disparity], 'pre');    
+Iright_moved=Iright_moved(:, 1:end-disparity,:);
+
+[rows, cols] = size(Ileft);
+for i=1:rows
+    for j=1:cols
+        diff = Ileft(i,j) - Iright_moved(i,j);
+        SAD(i,j) = diff * diff;
+    end
+end
+
+%2D scan.
+integralImg=integralImage2D(SAD);
+retSAD=integralImg(win_sz+1:end,win_sz+1:end,:) +integralImg(2:end-win_sz+1,2:end-win_sz+1,:)-integralImg(2:end-win_sz+1,win_sz+1:end,:)-integralImg(win_sz+1:end,2:end-win_sz+1,:);
+
+end
+
+function retImg=integralImage2D(I)
+[nr,nc,nb]=size(I);
+retImg=zeros(nr,nc,nb);
+retImg(1,:,:)=I(1,:,:);
+for i=2:nr
+    retImg(i,:,:)=retImg(i-1,:,:)+I(i,:,:);
+end
+%vtuneResumeMex;
+for j=2:nc
+    retImg(:,j,:)=retImg(:,j-1,:)+retImg (:,j,:);
+end
+%vtunePauseMex;
+end