source: trunk/src/civ_matlab.m @ 538

Last change on this file since 538 was 497, checked in by sommeria, 12 years ago

cleaning and small bug repair.
pb of histogram for filter data solved
display of uicontrol by right mouse selection improved

File size: 34.7 KB
RevLine 
[315]1%'civ_matlab': Matlab version of the PIV programs CivX
2% --- call the sub-functions:
3%   civ: PIV function itself
4%   fix: removes false vectors after detection by various criteria
[382]5%   filter_tps: make interpolation-smoothing
[315]6%------------------------------------------------------------------------
7% function [Data,errormsg,result_conv]= civ_uvmat(Param,ncfile)
8%
9%OUTPUT
10% Data=structure containing the PIV results and information on the processing parameters
11% errormsg=error message char string, default=''
12% resul_conv: image inter-correlation function for the last grid point (used for tests)
13%
14%INPUT:
15% Param: input images and processing parameters
[387]16%     .Civ1: for civ1
17%     .Fix1:
18%     .Patch1:
19%     .Civ2: for civ2
20%     .Fix2:
21%     .Patch2:
[315]22% ncfile: name of a netcdf file to be created for the result (extension .nc)
23%
24%AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
25%  Copyright  2011, LEGI / CNRS-UJF-INPG, joel.sommeria@legi.grenoble-inp.fr.
26%AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
27%     This is part of the toolbox UVMAT.
28%
29%     UVMAT is free software; you can redistribute it and/or modify
30%     it under the terms of the GNU General Public License as published by
31%     the Free Software Foundation; either version 2 of the License, or
32%     (at your option) any later version.
33%
34%     UVMAT is distributed in the hope that it will be useful,
35%     but WITHOUT ANY WARRANTY; without even the implied warranty of
36%     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
37%     GNU General Public License (open UVMAT/COPYING.txt) for more details.
38%AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
39
40function [Data,errormsg,result_conv]= civ_matlab(Param,ncfile)
41errormsg='';
42Data.ListGlobalAttribute={'Conventions','Program','CivStage'};
43Data.Conventions='uvmat/civdata';% states the conventions used for the description of field variables and attributes
44Data.Program='civ_matlab';
45Data.CivStage=0;%default
46ListVarCiv1={'Civ1_X','Civ1_Y','Civ1_U','Civ1_V','Civ1_C','Civ1_F'}; %variables to read
47ListVarFix1={'Civ1_X','Civ1_Y','Civ1_U','Civ1_V','Civ1_C','Civ1_F','Civ1_FF'};
48mask='';
49maskname='';%default
50check_civx=0;%default
51check_civ1=0;%default
52check_patch1=0;%default
[466]53%         ImageFileA=Param.Civ1.ImageA;
54%         ImageFileB=Param.Civ1.ImageB;
55%         if isfield(Param,'Civ2')
56%             ImageFileA_civ2=Param.Civ2.ImageA;
57%              ImageFileB_civ2=Param.Civ2.ImageB;
58%         end
[435]59% case of input Param set by an xml file (batch mode)
[319]60if ischar(Param)
[387]61    Param=xml2struct(Param); %if Param is the name of an xml file, read this file as a Matlab structure
[493]62%     if isfield(Param,'Civ1')
63%         if strcmp(Param.Civ1.FileTypeA,'video')
64%             Param.Civ1.ImageA=VideoReader(regexprep(Param.Civ1.ImageA,'''','\'));% remove spurious ' appearing in Windows;
65%         elseif strcmp(Param.Civ1.FileTypeA,'mmreader')
66%             Param.Civ1.ImageA=mmreader(regexprep(Param.Civ1.ImageA,'''','\'));% remove spurious ' appearing in Windows
67%         end
68%         if strcmp(Param.Civ1.FileTypeB,'video')
69%             Param.Civ1.ImageB=VideoReader(regexprep(Param.Civ1.ImageB,'''','\'));% remove spurious ' appearing in Windows
70%         elseif strcmp(Param.Civ1.FileTypeB,'mmreader')
71%             Param.Civ1.ImageB=mmreader(regexprep(Param.Civ1.ImageB,'''','\'));% remove spurious ' appearing in Windows
72%         end
73%     end
74%     if isfield(Param,'Civ2')
75%         if strcmp(Param.Civ2.FileTypeA,'video')
76%             Param.Civ2.ImageA=VideoReader(regexprep(Param.Civ2.ImageA,'''','\'));% remove spurious ' appearing in Windows;
77%         elseif strcmp(Param.Civ2.FileTypeA,'mmreader')
78%             Param.Civ2.ImageA=mmreader(regexprep(Param.Civ2.ImageA,'''','\'));% remove spurious ' appearing in Windows
79%             if strcmp(Param.Civ2.FileTypeB,'video')
80%                 Param.Civ2.ImageB=VideoReader(regexprep(Param.Civ2.ImageB,'''','\'));% remove spurious ' appearing in Windows
81%             elseif strcmp(Param.Civ2.FileTypeB,'mmreader')
82%                 Param.Civ2.ImageB=mmreader(regexprep(Param.Civ2.ImageB,'''','\'));% remove spurious ' appearing in Windows
83%             end;
84%         end
85%     end
[319]86end
87
[315]88%% Civ1
89if isfield (Param,'Civ1')
[364]90    par_civ1=Param.Civ1;
[494]91    if isfield(par_civ1,'reverse_pair')% A REVOIR
[364]92        if par_civ1.reverse_pair
93            if ischar(par_civ1.ImageB)
94                temp=par_civ1.ImageA;
[388]95                par_civ1.ImageA=imread(par_civ1.ImageB);
[364]96            end
97            if ischar(temp)
[388]98                par_civ1.ImageB=imread(temp);
[364]99            end
100        end
101    else
[494]102        if isfield(Param.Civ1,'ImageA')%&&...   
[493]103             Param.Civ1.ImageA=regexprep(Param.Civ1.ImageA,'''','\');
[494]104            [par_civ1.ImageA,VideoObject] = read_image(Param.Civ1.ImageA,par_civ1.FileTypeA,[],par_civ1.FrameIndexA);
[364]105        end
[494]106        if isfield(Param.Civ1,'ImageB')%&& ...
[493]107             Param.Civ1.ImageB=regexprep(Param.Civ1.ImageB,'''','\');
[494]108             if strcmp(Param.Civ1.ImageA,Param.Civ1.ImageB)% use the same movie object
109                 [par_civ1.ImageB,VideoObject] = read_image(Param.Civ1.ImageB,par_civ1.FileTypeB,VideoObject,par_civ1.FrameIndexB);
110             else
111            [par_civ1.ImageB,VideoObject] = read_image(Param.Civ1.ImageB,par_civ1.FileTypeB,par_civ1.ImageB,par_civ1.FrameIndexB);
112             end
[364]113        end
114    end
115   
[315]116    % caluclate velocity data (y and v in indices, reverse to y component)
[364]117    [xtable ytable utable vtable ctable F result_conv errormsg] = civ (par_civ1);
[337]118   
[364]119    % to try the reverse_pair method, uncomment below
120    %     [xtable1 ytable1 utable1 vtable1 ctable1 F1 result_conv1 errormsg1] = civ (Param.Civ1);
121    %     Param.Civ1.reverse_pair=1;
122    %     [xtable2 ytable2 utable2 vtable2 ctable2 F2 result_conv2 errormsg2] = civ (Param.Civ1);
123    %     xtable=[xtable1; xtable2];
124    %     ytable=[ytable1; ytable2];
125    %     utable=[utable1; -utable2];
126    %     vtable=[vtable1; -vtable2];
127    %     ctable=[ctable1; ctable2];
128    %     F=[F1; F2];
129    %     result_conv=[result_conv1; result_conv2];
130    %     errormsg=[errormsg1; errormsg2];
[315]131    if ~isempty(errormsg)
132        return
133    end
134    list_param=(fieldnames(Param.Civ1))';
135    Civ1_param=list_param;%default
[406]136    %set the values of all the global attributes in list_param
[315]137    for ilist=1:length(list_param)
138        Civ1_param{ilist}=['Civ1_' list_param{ilist}];
139        Data.(['Civ1_' list_param{ilist}])=Param.Civ1.(list_param{ilist});
140    end
141    Data.ListGlobalAttribute=[Data.ListGlobalAttribute Civ1_param];% {'Civ1_Time','Civ1_Dt'}];
[364]142    Data.ListVarName={'Civ1_X','Civ1_Y','Civ1_U','Civ1_V','Civ1_F','Civ1_C'};%  cell array containing the names of the fields to record
[399]143    Data.VarDimName={'nb_vec_1','nb_vec_1','nb_vec_1','nb_vec_1','nb_vec_1','nb_vec_1'};
[319]144    Data.VarAttribute{1}.Role='coord_x';
145    Data.VarAttribute{2}.Role='coord_y';
146    Data.VarAttribute{3}.Role='vector_x';
147    Data.VarAttribute{4}.Role='vector_y';
148    Data.VarAttribute{5}.Role='warnflag';
[315]149    Data.Civ1_X=reshape(xtable,[],1);
150    Data.Civ1_Y=reshape(Param.Civ1.ImageHeight-ytable+1,[],1);
151    Data.Civ1_U=reshape(utable,[],1);
152    Data.Civ1_V=reshape(-vtable,[],1);
153    Data.Civ1_C=reshape(ctable,[],1);
154    Data.Civ1_F=reshape(F,[],1);
[435]155    Data.CivStage=1;
[315]156else
[319]157    if exist('ncfile','var')
158        CivFile=ncfile;
159    elseif isfield(Param.Patch1,'CivFile')
160        CivFile=Param.Patch1.CivFile;
161    end
[363]162    Data=nc2struct(CivFile,'ListGlobalAttribute','absolut_time_T0'); %look for the constant 'absolut_time_T0' to detect old civx data format
[317]163    if isfield(Data,'Txt')
[319]164        errormsg=Data.Txt;
[317]165        return
166    end
[315]167    if ~isempty(Data.absolut_time_T0')%read civx file
168        check_civx=1;% test for old civx data format
[319]169        [Data,vardetect,ichoice]=nc2struct(CivFile);%read the variables in the netcdf file
[315]170    else
[363]171        Data=nc2struct(CivFile);%read civ1 and fix1 data in the existing netcdf file
[315]172    end
173end
174
175%% Fix1
176if isfield (Param,'Fix1')
177    ListFixParam=fieldnames(Param.Fix1);
178    for ilist=1:length(ListFixParam)
179        ParamName=ListFixParam{ilist};
180        ListName=['Fix1_' ParamName];
181        eval(['Data.ListGlobalAttribute=[Data.ListGlobalAttribute ''' ParamName '''];'])
182        eval(['Data.' ListName '=Param.Fix1.' ParamName ';'])
183    end
184    if check_civx
185        if ~isfield(Data,'fix')
186            Data.ListGlobalAttribute=[Data.ListGlobalAttribute 'fix'];
187            Data.fix=1;
188            Data.ListVarName=[Data.ListVarName {'vec_FixFlag'}];
189            Data.VarDimName=[Data.VarDimName {'nb_vectors'}];
190        end
191        Data.vec_FixFlag=fix(Param.Fix1,Data.vec_F,Data.vec_C,Data.vec_U,Data.vec_V,Data.vec_X,Data.vec_Y);
192    else
193        Data.ListVarName=[Data.ListVarName {'Civ1_FF'}];
[399]194        Data.VarDimName=[Data.VarDimName {'nb_vec_1'}];
[315]195        nbvar=length(Data.ListVarName);
[435]196        Data.VarAttribute{nbvar}.Role='errorflag';
[315]197        Data.Civ1_FF=fix(Param.Fix1,Data.Civ1_F,Data.Civ1_C,Data.Civ1_U,Data.Civ1_V);
[435]198        Data.CivStage=2;
[315]199    end
[435]200end
[315]201%% Patch1
202if isfield (Param,'Patch1')
[326]203    if check_civx
204        errormsg='Civ Matlab input needed for patch';
205        return
206    end
[389]207   
[315]208    Data.ListGlobalAttribute=[Data.ListGlobalAttribute {'Patch1_Rho','Patch1_Threshold','Patch1_SubDomain'}];
[437]209    Data.Patch1_Rho=Param.Patch1.FieldSmooth;
[316]210    Data.Patch1_Threshold=Param.Patch1.MaxDiff;
211    Data.Patch1_SubDomain=Param.Patch1.SubdomainSize;
[397]212    nbvar=length(Data.ListVarName);
[389]213    Data.ListVarName=[Data.ListVarName {'Civ1_U_smooth','Civ1_V_smooth','Civ1_SubRange','Civ1_NbSites','Civ1_Coord_tps','Civ1_U_tps','Civ1_V_tps'}];
[435]214    Data.VarDimName=[Data.VarDimName {'nb_vec_1','nb_vec_1',{'nb_coord','nb_bounds','nb_subdomain_1'},{'nb_subdomain_1'},...
215        {'nb_tps_1','nb_coord','nb_subdomain_1'},{'nb_tps_1','nb_subdomain_1'},{'nb_tps_1','nb_subdomain_1'}}];
[397]216    Data.VarAttribute{nbvar+1}.Role='vector_x';
217    Data.VarAttribute{nbvar+2}.Role='vector_y';
218    Data.VarAttribute{nbvar+5}.Role='coord_tps';
[492]219    Data.VarAttribute{nbvar+6}.Role='vector_x';
220    Data.VarAttribute{nbvar+7}.Role='vector_y';
[389]221    Data.Civ1_U_smooth=zeros(size(Data.Civ1_X));
222    Data.Civ1_V_smooth=zeros(size(Data.Civ1_X));
[315]223    if isfield(Data,'Civ1_FF')
224        ind_good=find(Data.Civ1_FF==0);
[435]225    else
[315]226        ind_good=1:numel(Data.Civ1_X);
227    end
[382]228    [Data.Civ1_SubRange,Data.Civ1_NbSites,Data.Civ1_Coord_tps,Data.Civ1_U_tps,Data.Civ1_V_tps,tild,Ures, Vres,tild,FFres]=...
[435]229        filter_tps([Data.Civ1_X(ind_good) Data.Civ1_Y(ind_good)],Data.Civ1_U(ind_good),Data.Civ1_V(ind_good),[],Data.Patch1_SubDomain,Data.Patch1_Rho,Data.Patch1_Threshold);
230    Data.Civ1_U_smooth(ind_good)=Ures;
231    Data.Civ1_V_smooth(ind_good)=Vres;
232    Data.Civ1_FF(ind_good)=FFres;
233    Data.CivStage=3;
234end
[315]235
236%% Civ2
237if isfield (Param,'Civ2')
238    par_civ2=Param.Civ2;
[494]239    par_civ2.ImageA=[];
240    par_civ2.ImageB=[];
241    if isfield(Param.Civ2,'ImageA') && isfield(Param.Civ2,'ImageB')
242        Param.Civ2.ImageA=regexprep(Param.Civ2.ImageA,'''','\');
243        Param.Civ2.ImageB=regexprep(Param.Civ2.ImageB,'''','\');
244        if isfield (Param,'Civ1')
245            Param.Civ2.ImageA=regexprep(Param.Civ2.ImageA,'''','\');
246            if strcmp(Param.Civ1.ImageA,Param.Civ2.ImageA)
247                if isequal(Param.Civ1.FrameIndexA,Param.Civ2.FrameIndexA)
248                    par_civ2.ImageA=par_civ1.ImageA;
249                else % read another frame of the same movie object
250                    par_civ2.ImageA = read_image(Param.Civ2.ImageA,Param.Civ2.FileTypeA,VideoObject,Param.Civ2.FrameIndexA);
251                end
[435]252            end
[494]253            Param.Civ2.ImageB=regexprep(Param.Civ2.ImageB,'''','\');
254            if strcmp(Param.Civ1.ImageB,Param.Civ2.ImageB)
255                if isequal(Param.Civ1.FrameIndexB,Param.Civ2.FrameIndexB)
256                    par_civ2.ImageB=par_civ1.ImageB;
257                else % read another frame of the same movie object
258                    par_civ2.ImageB = read_image(Param.Civ2.ImageB,Param.Civ2.FileTypeB,VideoObject,Param.Civ2.FrameIndexB);
259                end
[435]260            end
[494]261        end
262        if isempty(par_civ2.ImageA) && isfield(Param.Civ2,'ImageA')
263            [par_civ2.ImageA,VideoObject] = read_image(Param.Civ2.ImageA,Param.Civ2.FileTypeA,[],Param.Civ2.FrameIndexA);
264        end
265        if isempty(par_civ2.ImageB)&& isfield(Param.Civ2,'ImageB')
266            if strcmp(Param.Civ2.ImageA,Param.Civ2.ImageB)
267                par_civ2.ImageB = read_image(Param.Civ2.ImageB,Param.Civ2.FileTypeB,VideoObject,Param.Civ2.FrameIndexB);
268            else
269                par_civ2.ImageB = read_image(Param.Civ2.ImageB,Param.Civ2.FileTypeB,[],Param.Civ2.FrameIndexB);
270            end
271        end
[315]272    end
[437]273    ibx2=ceil(par_civ2.CorrBoxSize(1)/2);
274    iby2=ceil(par_civ2.CorrBoxSize(2)/2);
[389]275    isx2=ibx2+4;% search ara +-4 pixels around the guess
276    isy2=iby2+4;
[363]277    % shift from par_civ2.filename_nc1
278    % shiftx=velocity interpolated at position
279    miniy=max(1+isy2,1+iby2);
280    minix=max(1+isx2,1+ibx2);
281    maxiy=min(size(par_civ2.ImageA,1)-isy2,size(par_civ2.ImageA,1)-iby2);
282    maxix=min(size(par_civ2.ImageA,2)-isx2,size(par_civ2.ImageA,2)-ibx2);
[315]283    [GridX,GridY]=meshgrid(minix:par_civ2.Dx:maxix,miniy:par_civ2.Dy:maxiy);
[363]284    GridX=reshape(GridX,[],1);
[435]285    GridY=reshape(GridY,[],1);
[363]286    Shiftx=zeros(size(GridX));% shift expected from civ1 data
287    Shifty=zeros(size(GridX));
288    nbval=zeros(size(GridX));
289    if par_civ2.CheckDeformation
290        DUDX=zeros(size(GridX));
291        DUDY=zeros(size(GridX));
292        DVDX=zeros(size(GridX));
293        DVDY=zeros(size(GridX));
294    end
[383]295    NbSubDomain=size(Data.Civ1_SubRange,3);
[363]296    % get the guess from patch1
297    for isub=1:NbSubDomain
298        nbvec_sub=Data.Civ1_NbSites(isub);
[383]299        ind_sel=find(GridX>=Data.Civ1_SubRange(1,1,isub) & GridX<=Data.Civ1_SubRange(1,2,isub) & GridY>=Data.Civ1_SubRange(2,1,isub) & GridY<=Data.Civ1_SubRange(2,2,isub));
[363]300        epoints = [GridX(ind_sel) GridY(ind_sel)];% coordinates of interpolation sites
[383]301        ctrs=Data.Civ1_Coord_tps(1:nbvec_sub,:,isub) ;%(=initial points) ctrs
[363]302        nbval(ind_sel)=nbval(ind_sel)+1;% records the number of values for eacn interpolation point (in case of subdomain overlap)
303        EM = tps_eval(epoints,ctrs);
304        Shiftx(ind_sel)=Shiftx(ind_sel)+EM*Data.Civ1_U_tps(1:nbvec_sub+3,isub);
305        Shifty(ind_sel)=Shifty(ind_sel)+EM*Data.Civ1_V_tps(1:nbvec_sub+3,isub);
306        if par_civ2.CheckDeformation
307            [EMDX,EMDY] = tps_eval_dxy(epoints,ctrs);%2D matrix of distances between extrapolation points epoints and spline centres (=site points) ctrs
308            DUDX(ind_sel)=DUDX(ind_sel)+EMDX*Data.Civ1_U_tps(1:nbvec_sub+3,isub);
309            DUDY(ind_sel)=DUDY(ind_sel)+EMDY*Data.Civ1_U_tps(1:nbvec_sub+3,isub);
310            DVDX(ind_sel)=DVDX(ind_sel)+EMDX*Data.Civ1_V_tps(1:nbvec_sub+3,isub);
311            DVDY(ind_sel)=DVDY(ind_sel)+EMDY*Data.Civ1_V_tps(1:nbvec_sub+3,isub);
312        end
313    end
314    mask='';
[387]315    if par_civ2.CheckMask&&~isempty(par_civ2.Mask)&& ~strcmp(maskname,par_civ2.Mask)% mask exist, not already read in civ1
316        mask=imread(par_civ2.Mask);
[315]317    end
[437]318    par_civ2.SearchBoxSize(1)=2*isx2+1;
319    par_civ2.SearchBoxSize(2)=2*isy2+1;
320    par_civ2.SearchBoxShift=[Shiftx(nbval>=1)./nbval(nbval>=1) Shifty(nbval>=1)./nbval(nbval>=1)];
321%     par_civ2.SearchBoxShift(2)=Shifty(nbval>=1)./nbval(nbval>=1);
322    par_civ2.Grid=[GridX(nbval>=1)-par_civ2.SearchBoxShift(:,1)/2 GridY(nbval>=1)-par_civ2.SearchBoxShift(:,2)/2];% grid taken at the extrapolated origin of the displacement vectors
[363]323    if par_civ2.CheckDeformation
[370]324        par_civ2.DUDX=DUDX./nbval;
325        par_civ2.DUDY=DUDY./nbval;
326        par_civ2.DVDX=DVDX./nbval;
327        par_civ2.DVDY=DVDY./nbval;
[363]328    end
[315]329    % caluclate velocity data (y and v in indices, reverse to y component)
[363]330    [xtable ytable utable vtable ctable F] = civ (par_civ2);
[435]331    %     diff_squared=(utable-par_civ2.Shiftx).*(utable-par_civ2.Shiftx)+(vtable+par_civ2.Shifty).*(vtable+par_civ2.Shifty);
332    %     F(diff_squared>=4)=4; %flag vectors whose distance to the guess exceeds 2 pixels
[363]333    list_param=(fieldnames(Param.Civ2))';
[315]334    list_remove={'pxcmx','pxcmy','npx','npy','gridflag','maskflag','term_a','term_b','T0'};
335    for ilist=1:length(list_remove)
336        index=strcmp(list_remove{ilist},list_param);
337        if ~isempty(find(index,1))
338            list_param(index)=[];
339        end
340    end
341    for ilist=1:length(list_param)
[363]342        Civ2_param{ilist}=['Civ2_' list_param{ilist}];
343        eval(['Data.Civ2_' list_param{ilist} '=Param.Civ2.' list_param{ilist} ';'])
[315]344    end
[363]345    if isfield(Data,'Civ2_gridname') && strcmp(Data.Civ1_gridname(1:6),'noFile')
[315]346        Data.Civ1_gridname='';
347    end
[363]348    if isfield(Data,'Civ2_maskname') && strcmp(Data.Civ1_maskname(1:6),'noFile')
349        Data.Civ2_maskname='';
[315]350    end
[363]351    Data.ListGlobalAttribute=[Data.ListGlobalAttribute Civ2_param {'Civ2_Time','Civ2_Dt'}];
[435]352    %     Data.Civ2_Time=par_civ2.Time;
353    %     Data.Civ2_Dt=par_civ2.Dt;
[364]354    nbvar=numel(Data.ListVarName);
355    Data.ListVarName=[Data.ListVarName {'Civ2_X','Civ2_Y','Civ2_U','Civ2_V','Civ2_F','Civ2_C'}];%  cell array containing the names of the fields to record
[399]356    Data.VarDimName=[Data.VarDimName {'nb_vec_2','nb_vec_2','nb_vec_2','nb_vec_2','nb_vec_2','nb_vec_2'}];
[364]357    Data.VarAttribute{nbvar+1}.Role='coord_x';
358    Data.VarAttribute{nbvar+2}.Role='coord_y';
359    Data.VarAttribute{nbvar+3}.Role='vector_x';
360    Data.VarAttribute{nbvar+4}.Role='vector_y';
361    Data.VarAttribute{nbvar+5}.Role='warnflag';
[363]362    Data.Civ2_X=reshape(xtable,[],1);
363    Data.Civ2_Y=reshape(size(par_civ2.ImageA,1)-ytable+1,[],1);
364    Data.Civ2_U=reshape(utable,[],1);
365    Data.Civ2_V=reshape(-vtable,[],1);
366    Data.Civ2_C=reshape(ctable,[],1);
367    Data.Civ2_F=reshape(F,[],1);
[315]368    Data.CivStage=Data.CivStage+1;
369end
370
371%% Fix2
372if isfield (Param,'Fix2')
373    ListFixParam=fieldnames(Param.Fix2);
374    for ilist=1:length(ListFixParam)
375        ParamName=ListFixParam{ilist};
[364]376        ListName=['Fix2_' ParamName];
[315]377        eval(['Data.ListGlobalAttribute=[Data.ListGlobalAttribute ''' ParamName '''];'])
378        eval(['Data.' ListName '=Param.Fix2.' ParamName ';'])
379    end
380    if check_civx
381        if ~isfield(Data,'fix2')
382            Data.ListGlobalAttribute=[Data.ListGlobalAttribute 'fix2'];
383            Data.fix2=1;
384            Data.ListVarName=[Data.ListVarName {'vec2_FixFlag'}];
385            Data.VarDimName=[Data.VarDimName {'nb_vectors2'}];
386        end
387        Data.vec_FixFlag=fix(Param.Fix2,Data.vec2_F,Data.vec2_C,Data.vec2_U,Data.vec2_V,Data.vec2_X,Data.vec2_Y);
388    else
389        Data.ListVarName=[Data.ListVarName {'Civ2_FF'}];
[399]390        Data.VarDimName=[Data.VarDimName {'nb_vec_2'}];
[315]391        nbvar=length(Data.ListVarName);
[435]392        Data.VarAttribute{nbvar}.Role='errorflag';
[315]393        Data.Civ2_FF=fix(Param.Fix2,Data.Civ2_F,Data.Civ2_C,Data.Civ2_U,Data.Civ2_V);
[435]394        Data.CivStage=Data.CivStage+1;
[315]395    end
396   
[435]397end
[315]398
399%% Patch2
400if isfield (Param,'Patch2')
401    Data.ListGlobalAttribute=[Data.ListGlobalAttribute {'Patch2_Rho','Patch2_Threshold','Patch2_SubDomain'}];
[437]402    Data.Patch2_Rho=Param.Patch2.FieldSmooth;
[364]403    Data.Patch2_Threshold=Param.Patch2.MaxDiff;
404    Data.Patch2_SubDomain=Param.Patch2.SubdomainSize;
[397]405    nbvar=length(Data.ListVarName);
406    Data.ListVarName=[Data.ListVarName {'Civ2_U_smooth','Civ2_V_smooth','Civ2_SubRange','Civ2_NbSites','Civ2_Coord_tps','Civ2_U_tps','Civ2_V_tps'}];
[399]407    Data.VarDimName=[Data.VarDimName {'nb_vec_2','nb_vec_2',{'nb_coord','nb_bounds','nb_subdomain_2'},{'nb_subdomain_2'},...
[435]408        {'nb_tps_2','nb_coord','nb_subdomain_2'},{'nb_tps_2','nb_subdomain_2'},{'nb_tps_2','nb_subdomain_2'}}];
409   
410    Data.VarAttribute{nbvar+1}.Role='vector_x';
[397]411    Data.VarAttribute{nbvar+2}.Role='vector_y';
412    Data.VarAttribute{nbvar+5}.Role='coord_tps';
[492]413    Data.VarAttribute{nbvar+6}.Role='vector_x';
414    Data.VarAttribute{nbvar+7}.Role='vector_y';
[397]415    Data.Civ2_U_smooth=zeros(size(Data.Civ2_X));
416    Data.Civ2_V_smooth=zeros(size(Data.Civ2_X));
[315]417    if isfield(Data,'Civ2_FF')
418        ind_good=find(Data.Civ2_FF==0);
419    else
420        ind_good=1:numel(Data.Civ2_X);
[435]421    end
[382]422    [Data.Civ2_SubRange,Data.Civ2_NbSites,Data.Civ2_Coord_tps,Data.Civ2_U_tps,Data.Civ2_V_tps,tild,Ures, Vres,tild,FFres]=...
[435]423        filter_tps([Data.Civ2_X(ind_good) Data.Civ2_Y(ind_good)],Data.Civ2_U(ind_good),Data.Civ2_V(ind_good),[],Data.Patch2_SubDomain,Data.Patch2_Rho,Data.Patch2_Threshold);
[397]424    Data.Civ2_U_smooth(ind_good)=Ures;
425    Data.Civ2_V_smooth(ind_good)=Vres;
426    Data.Civ2_FF(ind_good)=FFres;
[435]427    Data.CivStage=Data.CivStage+1;
428end
[315]429
430%% write result in a netcdf file if requested
[319]431if exist('ncfile','var')
[315]432    errormsg=struct2nc(ncfile,Data);
[494]433    if isempty(errormsg)
434        disp([ncfile ' written'])
435    else
436        disp(errormsg)
437    end
[315]438end
439
440% 'civ': function piv.m adapted from PIVlab http://pivlab.blogspot.com/
441%--------------------------------------------------------------------------
442% function [xtable ytable utable vtable typevector] = civ (image1,image2,ibx,iby step, subpixfinder, mask, roi)
443%
444% OUTPUT:
445% xtable: set of x coordinates
446% ytable: set of y coordiantes
447% utable: set of u displacements (along x)
448% vtable: set of v displacements (along y)
449% ctable: max image correlation for each vector
450% typevector: set of flags, =1 for good, =0 for NaN vectors
451%
452%INPUT:
[493]453% par_civ: structure of input parameters, with fields:
454%  .CorrBoxSize
455%  .SearchBoxSize
456%  .SearchBoxShift
457%  .ImageHeight
458%  .ImageWidth
459%  .Dx, Dy
460%  .Grid
461%  .Mask
462%  .MinIma
463%  .MaxIma
464%  .image1:first image (matrix)
[315]465% image2: second image (matrix)
466% ibx2,iby2: half size of the correlation box along x and y, in px (size=(2*iby2+1,2*ibx2+1)
467% isx2,isy2: half size of the search box along x and y, in px (size=(2*isy2+1,2*isx2+1)
468% shiftx, shifty: shift of the search box (in pixel index, yshift reversed)
469% step: mesh of the measurement points (in px)
470% subpixfinder=1 or 2 controls the curve fitting of the image correlation
471% mask: =[] for no mask
472% roi: 4 element vector defining a region of interest: x position, y position, width, height, (in image indices), for the whole image, roi=[];
473function [xtable ytable utable vtable ctable F result_conv errormsg] = civ (par_civ)
474%this funtion performs the DCC PIV analysis. Recent window-deformation
475%methods perform better and will maybe be implemented in the future.
476
477%% prepare grid
[437]478ibx2=ceil(par_civ.CorrBoxSize(1)/2);
479iby2=ceil(par_civ.CorrBoxSize(2)/2);
480isx2=ceil(par_civ.SearchBoxSize(1)/2);
481isy2=ceil(par_civ.SearchBoxSize(2)/2);
482shiftx=round(par_civ.SearchBoxShift(:,1));
483shifty=-round(par_civ.SearchBoxShift(:,2));% sign minus because image j index increases when y decreases
[315]484if isfield(par_civ,'Grid')
[364]485    if ischar(par_civ.Grid)%read the drid file if the input is a file name
[315]486        par_civ.Grid=dlmread(par_civ.Grid);
487        par_civ.Grid(1,:)=[];%the first line must be removed (heading in the grid file)
488    end
489else% automatic measurement grid
[437]490%     ibx2=ceil(par_civ.Bx/2);
491%     iby2=ceil(par_civ.By/2);
492%     isx2=ceil(par_civ.Searchx/2);
493%     isy2=ceil(par_civ.Searchy/2);
[315]494    miniy=max(1+isy2+shifty,1+iby2);
495    minix=max(1+isx2-shiftx,1+ibx2);
496    maxiy=min(par_civ.ImageHeight-isy2+shifty,par_civ.ImageHeight-iby2);
497    maxix=min(par_civ.ImageWidth-isx2-shiftx,par_civ.ImageWidth-ibx2);
498    [GridX,GridY]=meshgrid(minix:par_civ.Dx:maxix,miniy:par_civ.Dy:maxiy);
499    par_civ.Grid(:,1)=reshape(GridX,[],1);
500    par_civ.Grid(:,2)=reshape(GridY,[],1);
501end
[363]502nbvec=size(par_civ.Grid,1);
503if numel(shiftx)==1
[365]504    shiftx=shiftx*ones(nbvec,1);
505    shifty=shifty*ones(nbvec,1);
[363]506end
[315]507%% Default output
508xtable=par_civ.Grid(:,1);
509ytable=par_civ.Grid(:,2);
510utable=zeros(nbvec,1);
511vtable=zeros(nbvec,1);
512ctable=zeros(nbvec,1);
513F=zeros(nbvec,1);
514result_conv=[];
515errormsg='';
516
517%% prepare mask
518if isfield(par_civ,'Mask') && ~isempty(par_civ.Mask)
519    if strcmp(par_civ.Mask,'all')
520        return    % get the grid only, no civ calculation
521    elseif ischar(par_civ.Mask)
522        par_civ.Mask=imread(par_civ.Mask);
523    end
524end
525check_MinIma=isfield(par_civ,'MinIma');% test for image luminosity threshold
526check_MaxIma=isfield(par_civ,'MaxIma') && ~isempty(par_civ.MaxIma);
527
[364]528% %% prepare images
529% if isfield(par_civ,'reverse_pair')
530%     if par_civ.reverse_pair
531%         if ischar(par_civ.ImageB)
532%             temp=par_civ.ImageA;
533%             par_civ.ImageA=imread(par_civ.ImageB);
534%         end
535%         if ischar(temp)
536%             par_civ.ImageB=imread(temp);
537%         end
538%     end
539% else
540%     if ischar(par_civ.ImageA)
541%         par_civ.ImageA=imread(par_civ.ImageA);
542%     end
543%     if ischar(par_civ.ImageB)
544%         par_civ.ImageB=imread(par_civ.ImageB);
545%     end
546% end
[388]547par_civ.ImageA=sum(double(par_civ.ImageA),3);%sum over rgb component for color images
548par_civ.ImageB=sum(double(par_civ.ImageB),3);
[315]549[npy_ima npx_ima]=size(par_civ.ImageA);
550if ~isequal(size(par_civ.ImageB),[npy_ima npx_ima])
551    errormsg='image pair with unequal size';
552    return
553end
554
555%% Apply mask
[411]556    % Convention for mask IDEAS TO IMPLEMENT ?
[315]557    % mask >200 : velocity calculated
558    %  200 >=mask>150;velocity not calculated, interpolation allowed (bad spots)
559    % 150>=mask >100: velocity not calculated, nor interpolated
[411]560    %  100>=mask> 20: velocity not calculated, impermeable (no flux through mask boundaries)
[315]561    %  20>=mask: velocity=0
562checkmask=0;
563if isfield(par_civ,'Mask') && ~isempty(par_civ.Mask)
564   checkmask=1;
565   if ~isequal(size(par_civ.Mask),[npy_ima npx_ima])
566        errormsg='mask must be an image with the same size as the images';
567        return
568   end
569  %  check_noflux=(par_civ.Mask<100) ;%TODO: to implement
[411]570    check_undefined=(par_civ.Mask<200 & par_civ.Mask>=20 );
[315]571    par_civ.ImageA(check_undefined)=min(min(par_civ.ImageA));% put image A to zero (i.e. the min image value) in the undefined  area
572    par_civ.ImageB(check_undefined)=min(min(par_civ.ImageB));% put image B to zero (i.e. the min image value) in the undefined  area
573end
574
575%% compute image correlations: MAINLOOP on velocity vectors
[317]576corrmax=0;
577sum_square=1;% default
[368]578mesh=1;% default
579CheckDecimal=isfield(par_civ,'CheckDecimal')&& par_civ.CheckDecimal==1;
580if CheckDecimal
581    mesh=0.2;%mesh in pixels for subpixel image interpolation
[370]582    CheckDeformation=isfield(par_civ,'CheckDeformation')&& par_civ.CheckDeformation==1;
[368]583end
[315]584% vector=[0 0];%default
585for ivec=1:nbvec
[370]586    iref=round(par_civ.Grid(ivec,1)+0.5);% xindex on the image A for the middle of the correlation box
587    jref=round(par_civ.ImageHeight-par_civ.Grid(ivec,2)+0.5);% yindex on the image B for the middle of the correlation box
[315]588    if ~(checkmask && par_civ.Mask(jref,iref)<=20) %velocity not set to zero by the black mask
[363]589        if jref-iby2<1 || jref+iby2>par_civ.ImageHeight|| iref-ibx2<1 || iref+ibx2>par_civ.ImageWidth||...
590              jref+shifty(ivec)-isy2<1||jref+shifty(ivec)+isy2>par_civ.ImageHeight|| iref+shiftx(ivec)-isx2<1 || iref+shiftx(ivec)+isx2>par_civ.ImageWidth  % we are outside the image
[315]591            F(ivec)=3;
592        else
593            image1_crop=par_civ.ImageA(jref-iby2:jref+iby2,iref-ibx2:iref+ibx2);%extract a subimage (correlation box) from image A
[363]594            image2_crop=par_civ.ImageB(jref+shifty(ivec)-isy2:jref+shifty(ivec)+isy2,iref+shiftx(ivec)-isx2:iref+shiftx(ivec)+isx2);%extract a larger subimage (search box) from image B
[315]595            image1_mean=mean(mean(image1_crop));
596            image2_mean=mean(mean(image2_crop));
597            %threshold on image minimum
598            if check_MinIma && (image1_mean < par_civ.MinIma || image2_mean < par_civ.MinIma)
599                F(ivec)=3;
600            end
[319]601            %threshold on image maximum
602            if check_MaxIma && (image1_mean > par_civ.MaxIma || image2_mean > par_civ.MaxIma)
603                F(ivec)=3;
604            end
[366]605        end     
[315]606        if F(ivec)~=3
607            image1_crop=image1_crop-image1_mean;%substract the mean
608            image2_crop=image2_crop-image2_mean;
[370]609            if CheckDecimal
[368]610                xi=(1:mesh:size(image1_crop,2));
611                yi=(1:mesh:size(image1_crop,1))';
[370]612                if CheckDeformation
613                    [XI,YI]=meshgrid(xi-ceil(size(image1_crop,2)/2),yi-ceil(size(image1_crop,1)/2));
614                    XIant=XI-par_civ.DUDX(ivec)*XI-par_civ.DUDY(ivec)*YI+ceil(size(image1_crop,2)/2);
615                    YIant=YI-par_civ.DVDX(ivec)*XI-par_civ.DVDY(ivec)*YI+ceil(size(image1_crop,1)/2);
616                    image1_crop=interp2(image1_crop,XIant,YIant);
617                else
618                    image1_crop=interp2(image1_crop,xi,yi);
619                end
[369]620                xi=(1:mesh:size(image2_crop,2));
621                yi=(1:mesh:size(image2_crop,1))';
[368]622                image2_crop=interp2(image2_crop,xi,yi);
623            end
[315]624            sum_square=sum(sum(image1_crop.*image1_crop));
625            %reference: Oliver Pust, PIV: Direct Cross-Correlation
626            result_conv= conv2(image2_crop,flipdim(flipdim(image1_crop,2),1),'valid');
627            corrmax= max(max(result_conv));
628            result_conv=(result_conv/corrmax)*255; %normalize, peak=always 255
629            %Find the correlation max, at 255
630            [y,x] = find(result_conv==255,1);
631            if ~isempty(y) && ~isempty(x)
632                try
[437]633                    if par_civ.CorrSmooth==1
[315]634                        [vector,F(ivec)] = SUBPIXGAUSS (result_conv,x,y);
[437]635                    elseif par_civ.CorrSmooth==2
[315]636                        [vector,F(ivec)] = SUBPIX2DGAUSS (result_conv,x,y);
637                    end
[368]638                    utable(ivec)=vector(1)*mesh+shiftx(ivec);
639                    vtable(ivec)=vector(2)*mesh+shifty(ivec);                 
[370]640                    xtable(ivec)=iref+utable(ivec)/2-0.5;% convec flow (velocity taken at the point middle from imgae 1 and 2)
641                    ytable(ivec)=jref+vtable(ivec)/2-0.5;% and position of pixel 1=0.5 (convention for image coordinates=0 at the edge)
[315]642                    iref=round(xtable(ivec));% image index for the middle of the vector
643                    jref=round(ytable(ivec));
644                    if checkmask && par_civ.Mask(jref,iref)<200 && par_civ.Mask(jref,iref)>=100
645                        utable(ivec)=0;
646                        vtable(ivec)=0;
647                        F(ivec)=3;
648                    end
649                    ctable(ivec)=corrmax/sum_square;% correlation value
650                catch ME
651                    F(ivec)=3;
652                end
653            else
654                F(ivec)=3;
655            end
656        end
657    end
658   
659    %Create the vector matrix x, y, u, v
660end
661result_conv=result_conv*corrmax/(255*sum_square);% keep the last correlation matrix for output
662
663%------------------------------------------------------------------------
664% --- Find the maximum of the correlation function after interpolation
665function [vector,F] = SUBPIXGAUSS (result_conv,x,y)
666%------------------------------------------------------------------------
667vector=[0 0]; %default
668F=0;
669[npy,npx]=size(result_conv);
670
671% if (x <= (size(result_conv,1)-1)) && (y <= (size(result_conv,1)-1)) && (x >= 1) && (y >= 1)
[337]672    %the following 8 lines are copyright (c) 1998, Uri Shavit, Roi Gurka, Alex Liberzon, Technion ï¿œ Israel Institute of Technology
[315]673    %http://urapiv.wordpress.com
674    peaky = y;
675    if y <= npy-1 && y >= 1
676        f0 = log(result_conv(y,x));
677        f1 = real(log(result_conv(y-1,x)));
678        f2 = real(log(result_conv(y+1,x)));
679        peaky = peaky+ (f1-f2)/(2*f1-4*f0+2*f2);
680    else
681        F=-2; % warning flag for vector truncated by the limited search box
682    end
683    peakx=x;
684    if x <= npx-1 && x >= 1
685        f0 = log(result_conv(y,x));
686        f1 = real(log(result_conv(y,x-1)));
687        f2 = real(log(result_conv(y,x+1)));
688        peakx = peakx+ (f1-f2)/(2*f1-4*f0+2*f2);
689    else
690        F=-2; % warning flag for vector truncated by the limited search box
691    end
692    vector=[peakx-floor(npx/2)-1 peaky-floor(npy/2)-1];
693
694%------------------------------------------------------------------------
695% --- Find the maximum of the correlation function after interpolation
696function [vector,F] = SUBPIX2DGAUSS (result_conv,x,y)
697%------------------------------------------------------------------------
698vector=[0 0]; %default
699F=-2;
700peaky=y;
701peakx=x;
702[npy,npx]=size(result_conv);
703if (x <= npx-1) && (y <= npy-1) && (x >= 1) && (y >= 1)
704    F=0;
705    for i=-1:1
706        for j=-1:1
707            %following 15 lines based on
[337]708            %H. Nobach ï¿œ M. Honkanen (2005)
[315]709            %Two-dimensional Gaussian regression for sub-pixel displacement
710            %estimation in particle image velocimetry or particle position
711            %estimation in particle tracking velocimetry
[337]712            %Experiments in Fluids (2005) 38: 511ï¿œ515
[315]713            c10(j+2,i+2)=i*log(result_conv(y+j, x+i));
714            c01(j+2,i+2)=j*log(result_conv(y+j, x+i));
715            c11(j+2,i+2)=i*j*log(result_conv(y+j, x+i));
716            c20(j+2,i+2)=(3*i^2-2)*log(result_conv(y+j, x+i));
717            c02(j+2,i+2)=(3*j^2-2)*log(result_conv(y+j, x+i));
718        end
719    end
720    c10=(1/6)*sum(sum(c10));
721    c01=(1/6)*sum(sum(c01));
722    c11=(1/4)*sum(sum(c11));
723    c20=(1/6)*sum(sum(c20));
724    c02=(1/6)*sum(sum(c02));
725    deltax=(c11*c01-2*c10*c02)/(4*c20*c02-c11^2);
726    deltay=(c11*c10-2*c01*c20)/(4*c20*c02-c11^2);
727    if abs(deltax)<1
728        peakx=x+deltax;
729    end
730    if abs(deltay)<1
731        peaky=y+deltay;
732    end
733end
734vector=[peakx-floor(npx/2)-1 peaky-floor(npy/2)-1];
735
736%'RUN_FIX': function for fixing velocity fields:
737%-----------------------------------------------
738% RUN_FIX(filename,field,flagindex,thresh_vecC,thresh_vel,iter,flag_mask,maskname,fileref,fieldref)
739%
740%filename: name of the netcdf file (used as input and output)
741%field: structure specifying the names of the fields to fix (depending on civ1 or civ2)
742    %.vel_type='civ1' or 'civ2';
743    %.nb=name of the dimension common to the field to fix ('nb_vectors' for civ1);
744    %.fixflag=name of fix flag variable ('vec_FixFlag' for civ1)
745%flagindex: flag specifying which values of vec_f are removed:
746        % if flagindex(1)=1: vec_f=-2 vectors are removed
747        % if flagindex(2)=1: vec_f=3 vectors are removed
748        % if flagindex(3)=1: vec_f=2 vectors are removed (if iter=1) or vec_f=4 vectors are removed (if iter=2)
749%iter=1 for civ1 fields and iter=2 for civ2 fields
750%thresh_vecC: threshold in the image correlation vec_C
751%flag_mask: =1 mask used to remove vectors (0 else)
752%maskname: name of the mask image file for fix
753%thresh_vel: threshold on velocity, or on the difference with the reference file fileref if exists
754%inf_sup=1: remove values smaller than threshold thresh_vel, =2, larger than threshold
755%fileref: .nc file name for a reference velocity (='': refrence 0 used)
756%fieldref: 'civ1','filter1'...feld used in fileref
757
758function FF=fix(Param,F,C,U,V,X,Y)
759FF=zeros(size(F));%default
760
761%criterium on warn flags
762FlagName={'CheckFmin2','CheckF2','CheckF3','CheckF4'};
763FlagVal=[-2 2 3 4];
764for iflag=1:numel(FlagName)
765    if isfield(Param,FlagName{iflag}) && Param.(FlagName{iflag})
766        FF=(FF==1| F==FlagVal(iflag));
767    end
768end
769%criterium on correlation values
770if isfield (Param,'MinCorr')
771    FF=FF==1 | C<Param.MinCorr;
772end
[316]773if (isfield(Param,'MinVel')&&~isempty(Param.MinVel))||(isfield (Param,'MaxVel')&&~isempty(Param.MaxVel))
[315]774    Umod= U.*U+V.*V;
[316]775    if isfield (Param,'MinVel')&&~isempty(Param.MinVel)
[315]776        FF=FF==1 | Umod<(Param.MinVel*Param.MinVel);
777    end
778    if isfield (Param,'MaxVel')&&~isempty(Param.MaxVel)
779        FF=FF==1 | Umod>(Param.MaxVel*Param.MaxVel);
780    end
781end
782
783
784
[326]785
[315]786
787
788
Note: See TracBrowser for help on using the repository browser.