source: trunk/src/civ_uvmat.m @ 251

Last change on this file since 251 was 251, checked in by sommeria, 13 years ago

bug corrected in mouse_motion (display of z)
civ_uvmat corrected to use mask in fix (not implemented in civx fortran programmes)

File size: 22.3 KB
Line 
1% To develop....
2function [Data,errormsg]= civ_uvmat(Param,ncfile)
3errormsg='';
4Data.ListGlobalAttribute={'Conventions','Program','CivStage'};
5Data.Conventions='uvmat/civdata';% states the conventions used for the description of field variables and attributes
6Data.Program='civ_uvmat';
7Data.CivStage=0;%default
8ListVarCiv1={'Civ1_X','Civ1_Y','Civ1_U','Civ1_V','Civ1_C','Civ1_F'};
9ListVarFix1={'Civ1_X','Civ1_Y','Civ1_U','Civ1_V','Civ1_C','Civ1_F','Civ1_FF'};
10mask='';
11maskname='';%default
12test_civx=0;%default
13
14%% Civ1
15if isfield (Param,'Civ1')
16    par_civ1=Param.Civ1;
17    image1=imread(par_civ1.filename_ima_a);
18    image2=imread(par_civ1.filename_ima_b);
19    stepx=str2num(par_civ1.dx);
20    stepy=str2num(par_civ1.dy);
21    ibx2=ceil(str2num(par_civ1.ibx)/2);
22    iby2=ceil(str2num(par_civ1.iby)/2);
23    isx2=ceil(str2num(par_civ1.isx)/2);
24    isy2=ceil(str2num(par_civ1.isy)/2);
25    shiftx=str2num(par_civ1.shiftx);
26    shifty=str2num(par_civ1.shifty);
27    miniy=max(1+isy2+shifty,1+iby2);
28    minix=max(1+isx2-shiftx,1+ibx2);
29    maxiy=min(size(image1,1)-isy2+shifty,size(image1,1)-iby2);
30    maxix=min(size(image1,2)-isx2-shiftx,size(image1,2)-ibx2);
31    [GridX,GridY]=meshgrid(minix:stepx:maxix,miniy:stepy:maxiy);
32    PointCoord(:,1)=reshape(GridX,[],1);
33    PointCoord(:,2)=reshape(GridY,[],1);
34    if ~isempty(par_civ1.maskname)
35        maskname=par_civ1.maskname;
36        mask=imread(maskname);
37    end
38    % caluclate velocity data (y and v in indices, reverse to y component)
39    [xtable ytable utable vtable ctable F] = pivlab (image1,image2,ibx2,iby2,isx2,isy2,shiftx,-shifty,PointCoord,str2num(par_civ1.rho), mask);
40    list_param=(fieldnames(par_civ1))';
41    list_remove={'pxcmx','pxcmy','npx','npy','gridflag','maskflag','term_a','term_b','T0'};
42    index=zeros(size(list_param));
43    for ilist=1:length(list_remove)
44        index=strcmp(list_remove{ilist},list_param);
45        if ~isempty(find(index,1))
46            list_param(index)=[];
47        end
48    end
49    for ilist=1:length(list_param)
50        Civ1_param{ilist}=['Civ1_' list_param{ilist}];
51        eval(['Data.Civ1_' list_param{ilist} '=Param.Civ1.' list_param{ilist} ';'])
52    end
53    if isfield(Data,'Civ1_gridname') && strcmp(Data.Civ1_gridname(1:6),'noFile')
54        Data.Civ1_gridname='';
55    end
56    if isfield(Data,'Civ1_maskname') && strcmp(Data.Civ1_maskname(1:6),'noFile')
57        Data.Civ1_maskname='';
58    end
59    Data.ListGlobalAttribute=[Data.ListGlobalAttribute Civ1_param {'Civ1_Time','Civ1_Dt'}];
60    Data.Civ1_Time=str2double(par_civ1.T0);
61    Data.Civ1_Dt=str2double(par_civ1.Dt);
62    Data.ListVarName={'Civ1_X','Civ1_Y','Civ1_U','Civ1_V','Civ1_C','Civ1_F'};%  cell array containing the names of the fields to record
63    Data.VarDimName={'nbvec1','nbvec1','nbvec1','nbvec1','nbvec1','nbvec1'};
64    Data.VarAttribute{1}.Role='coord_x';
65    Data.VarAttribute{2}.Role='coord_y';
66    Data.VarAttribute{3}.Role='vector_x';
67    Data.VarAttribute{4}.Role='vector_y';
68    Data.VarAttribute{5}.Role='warnflag';
69    Data.Civ1_X=reshape(xtable,[],1);
70    Data.Civ1_Y=reshape(size(image1,1)-ytable+1,[],1);
71    Data.Civ1_U=reshape(utable,[],1);
72    Data.Civ1_V=reshape(-vtable,[],1);
73    Data.Civ1_C=reshape(ctable,[],1);
74    Data.Civ1_F=reshape(F,[],1);
75    Data.CivStage=1;
76else
77    Data=nc2struct(ncfile,'ListGlobalAttribute','absolut_time_T0');
78   
79    % read Civx data
80    if ~isempty(Data.absolut_time_T0')%read civx file
81%         var={'Civ1_X','Civ1_Y','Civ1_U','Civ1_V','Civ1_C','Civ1_F','Civ1_FF';...
82%             'vec_X','vec_Y','vec_U','vec_V','vec_C','vec_F','vec_FixFlag'};
83
84        %var=varcivx_generator('velocity','Civ1');%determine the names of constants and variables to read
85        test_civx=1;
86        [Data,vardetect,ichoice]=nc2struct(ncfile);%read the variables in the netcdf file
87%         Data.ListGlobalAttribute=[{'Conventions','Program','CivStage'} Data.ListGlobalAttribute {'Civ1_Time','Civ1_Dt'}];
88%         Data.Conventions='uvmat/civdata';% states the conventions used for the description of field variables and attributes
89%         Data.Program='civ_uvmat';
90%         Data.Civ1_Time=double(Data.absolut_time_T0);
91%         Data.Civ1_Dt=double(Data.dt);
92%         Data.VarDimName={'nbvec1','nbvec1','nbvec1','nbvec1','nbvec1','nbvec1','nbvec1'};
93%         Data.VarAttribute{1}.Role='coord_x';
94%         Data.VarAttribute{2}.Role='coord_y';
95%         Data.VarAttribute{3}.Role='vector_x';
96%         Data.VarAttribute{4}.Role='vector_y';
97%         Data.VarAttribute{5}.Role='ancillary';
98%         Data.VarAttribute{6}.Role='warnflag';
99%         Data.VarAttribute{7}.Role='errorflag';
100%         Data.CivStage=1;
101    else
102
103        if isfield(Param,'Fix1')
104            Data=nc2struct(ncfile,ListVarCiv1);%read civ1 data in the existing netcdf file
105        else
106            Data=nc2struct(ncfile,ListVarFix1);%read civ1 and fix1 data in the existing netcdf file
107        end
108    end
109    if isfield(Data,'Txt')
110        msgbox_uvmat('ERROR',Data.Txt)
111        return
112    end
113end
114
115%% Fix1
116if isfield (Param,'Fix1')
117    ListFixParam=fieldnames(Param.Fix1);
118    for ilist=1:length(ListFixParam)
119        ParamName=ListFixParam{ilist};
120        ListName=['Fix1_' ParamName];
121        eval(['Data.ListGlobalAttribute=[Data.ListGlobalAttribute ''' ParamName '''];'])
122        eval(['Data.' ListName '=Param.Fix1.' ParamName ';'])
123    end
124%     Data.ListGlobalAttribute=[Data.ListGlobalAttribute {'Fix1_WarnFlags','Fix1_TreshCorr','Fix1_TreshVel','Fix1_UpperBoundTest'}];
125%     Data.Fix1_WarnFlags=Param.Fix1.WarnFlags;
126%     Data.Fix1_ThreshCorr=Param.Fix1.ThreshCorr;
127%     Data.Fix1_ThreshVel=Param.Fix1.ThreshVel;
128%     Data.Fix1_UpperBoundTest=Param.Fix1.UpperBoundTest;
129
130    if test_civx
131        if ~isfield(Data,'fix')
132            Data.ListGlobalAttribute=[Data.ListGlobalAttribute 'fix'];
133            Data.fix=1;
134            Data.ListVarName=[Data.ListVarName {'vec_FixFlag'}];
135            Data.VarDimName=[Data.VardimName {'nb_vectors'}];
136        end
137        Data.vec_FixFlag=fix_uvmat(Param.Fix1,Data.vec_F,Data.vec_C,Data.vec_U,Data.vec_V,Data.vec_X,Data.vec_Y);
138    else
139        Data.ListVarName=[Data.ListVarName {'Civ1_FF'}];
140        Data.VarDimName=[Data.VarDimName {'nbvec1'}];
141        nbvar=length(Data.ListVarName);
142        Data.VarAttribute{nbvar}.Role='errorflag';   
143        Data.Civ1_FF=fix_uvmat(Param.Fix1,Data.Civ1_F,Data.Civ1_C,Data.Civ1_U,Data.Civ1_V);
144        Data.CivStage=2;   
145    end
146end   
147%% Patch1
148if isfield (Param,'Patch1')
149    Data.ListGlobalAttribute=[Data.ListGlobalAttribute {'Patch1_Rho','Patch1_Threshold','Patch1_SubDomain'}];
150    Data.Patch1_Rho=str2double(Param.Patch1.Rho);
151    Data.Patch1_Threshold=str2double(Param.Patch1.Threshold);
152    Data.Patch1_SubDomain=str2double(Param.Patch1.SubDomain);
153    Data.ListVarName=[Data.ListVarName {'Civ1_U_Diff','Civ1_V_Diff','Civ1_X_SubRange','Civ1_Y_SubRange','Civ1_NbSites','Civ1_X_tps','Civ1_Y_tps','Civ1_U_tps','Civ1_V_tps'}];
154    Data.VarDimName=[Data.VarDimName {'NbVec1','NbVec1',{'NbSubDomain1','Two'},{'NbSubDomain1','Two'},'NbSubDomain1',...
155             {'NbVec1Sub','NbSubDomain1'},{'NbVec1Sub','NbSubDomain1'},{'Nbtps1','NbSubDomain1'},{'Nbtps1','NbSubDomain1'}}];
156    nbvar=length(Data.ListVarName);
157    Data.VarAttribute{nbvar-1}.Role='vector_x';
158    Data.VarAttribute{nbvar}.Role='vector_y';
159    Data.Civ1_U_Diff=zeros(size(Data.Civ1_X));
160    Data.Civ1_V_Diff=zeros(size(Data.Civ1_X));
161    if isfield(Data,'Civ1_FF')
162        ind_good=find(Data.Civ1_FF==0);
163    else
164        ind_good=1:numel(Data.Civ1_X);
165    end
166    [Data.Civ1_X_SubRange,Data.Civ1_Y_SubRange,Data.Civ1_NbSites,FFres,Ures, Vres,Data.Civ1_X_tps,Data.Civ1_Y_tps,Data.Civ1_U_tps,Data.Civ1_V_tps]=...
167                            patch_uvmat(Data.Civ1_X(ind_good)',Data.Civ1_Y(ind_good)',Data.Civ1_U(ind_good)',Data.Civ1_V(ind_good)',Data.Patch1_Rho,Data.Patch1_Threshold,Data.Patch1_SubDomain);
168      Data.Civ1_U_Diff(ind_good)=Data.Civ1_U(ind_good)-Ures;
169      Data.Civ1_V_Diff(ind_good)=Data.Civ1_V(ind_good)-Vres;
170      Data.Civ1_FF(ind_good)=FFres;
171      Data.CivStage=3;                             
172end   
173
174%% Civ2
175if isfield (Param,'Civ2')
176    par_civ2=Param.Civ2
177    image1=imread(par_civ2.filename_ima_a);
178    image2=imread(par_civ2.filename_ima_b);
179    stepx=str2num(par_civ2.dx);
180    stepy=str2num(par_civ2.dy);
181    ibx2=ceil(str2num(par_civ2.ibx)/2);
182    iby2=ceil(str2num(par_civ2.iby)/2);
183    isx2=4;
184    isy2=4;
185%     shiftx=str2num(par_civ1.shiftx);
186%     shifty=str2num(par_civ1.shifty);
187% TO GET shift from par_civ2.filename_nc1
188    miniy=max(1+isy2+shifty,1+iby2);
189    minix=max(1+isx2-shiftx,1+ibx2);
190    maxiy=min(size(image1,1)-isy2+shifty,size(image1,1)-iby2);
191    maxix=min(size(image1,2)-isx2-shiftx,size(image1,2)-ibx2);
192    [GridX,GridY]=meshgrid(minix:stepx:maxix,miniy:stepy:maxiy);
193    PointCoord(:,1)=reshape(GridX,[],1);
194    PointCoord(:,2)=reshape(GridY,[],1);
195    if ~isempty(par_civ2.maskname)&& ~strcmp(maskname,par_civ2.maskname)% mask exist, not already read in civ1
196        mask=imread(par_civ2.maskname);
197    end
198    % caluclate velocity data (y and v in indices, reverse to y component)
199    [xtable ytable utable vtable ctable F] = pivlab (image1,image2,ibx2,iby2,isx2,isy2,shiftx,-shifty,PointCoord,str2num(par_civ1.rho),mask);
200    list_param=(fieldnames(par_civ1))';
201    list_remove={'pxcmx','pxcmy','npx','npy','gridflag','maskflag','term_a','term_b','T0'};
202    index=zeros(size(list_param));
203    for ilist=1:length(list_remove)
204        index=strcmp(list_remove{ilist},list_param);
205        if ~isempty(find(index,1))
206            list_param(index)=[];
207        end
208    end
209    for ilist=1:length(list_param)
210        Civ1_param{ilist}=['Civ1_' list_param{ilist}];
211        eval(['Data.Civ1_' list_param{ilist} '=Param.Civ1.' list_param{ilist} ';'])
212    end
213    if isfield(Data,'Civ1_gridname') && strcmp(Data.Civ1_gridname(1:6),'noFile')
214        Data.Civ1_gridname='';
215    end
216    if isfield(Data,'Civ1_maskname') && strcmp(Data.Civ1_maskname(1:6),'noFile')
217        Data.Civ1_maskname='';
218    end
219    Data.ListGlobalAttribute=[Data.ListGlobalAttribute Civ1_param {'Civ1_Time','Civ1_Dt'}];
220    Data.Civ1_Time=str2double(par_civ1.T0);
221    Data.Civ1_Dt=str2double(par_civ1.Dt);
222    Data.ListVarName={'Civ1_X','Civ1_Y','Civ1_U','Civ1_V','Civ1_C','Civ1_F'};%  cell array containing the names of the fields to record
223    Data.VarDimName={'nbvec1','nbvec1','nbvec1','nbvec1','nbvec1','nbvec1'};
224    Data.VarAttribute{1}.Role='coord_x';
225    Data.VarAttribute{2}.Role='coord_y';
226    Data.VarAttribute{3}.Role='vector_x';
227    Data.VarAttribute{4}.Role='vector_y';
228    Data.VarAttribute{5}.Role='warnflag';
229    Data.Civ1_X=reshape(xtable,[],1);
230    Data.Civ1_Y=reshape(size(image1,1)-ytable+1,[],1);
231    Data.Civ1_U=reshape(utable,[],1);
232    Data.Civ1_V=reshape(-vtable,[],1);
233    Data.Civ1_C=reshape(ctable,[],1);
234    Data.Civ1_F=reshape(F,[],1);
235    Data.CivStage=Data.CivStage+1;
236end
237%% write result
238% 'TESTcalc'
239% [DataOut,errormsg]=calc_field('velocity',Data)
240if exist('ncfile','var')
241errormsg=struct2nc(ncfile,Data);
242end
243
244
245%'RUN_FIX': function for fixing velocity fields:
246%-----------------------------------------------
247% RUN_FIX(filename,field,flagindex,thresh_vecC,thresh_vel,iter,flag_mask,maskname,fileref,fieldref)
248%
249%filename: name of the netcdf file (used as input and output)
250%field: structure specifying the names of the fields to fix (depending on civ1 or civ2)
251    %.vel_type='civ1' or 'civ2';
252    %.nb=name of the dimension common to the field to fix ('nb_vectors' for civ1);
253    %.fixflag=name of fix flag variable ('vec_FixFlag' for civ1)
254%flagindex: flag specifying which values of vec_f are removed:
255        % if flagindex(1)=1: vec_f=-2 vectors are removed
256        % if flagindex(2)=1: vec_f=3 vectors are removed
257        % if flagindex(3)=1: vec_f=2 vectors are removed (if iter=1) or vec_f=4 vectors are removed (if iter=2)
258%iter=1 for civ1 fields and iter=2 for civ2 fields
259%thresh_vecC: threshold in the image correlation vec_C
260%flag_mask: =1 mask used to remove vectors (0 else)
261%maskname: name of the mask image file for fix
262%thresh_vel: threshold on velocity, or on the difference with the reference file fileref if exists
263%inf_sup=1: remove values smaller than threshold thresh_vel, =2, larger than threshold
264%fileref: .nc file name for a reference velocity (='': refrence 0 used)
265%fieldref: 'civ1','filter1'...feld used in fileref
266
267function FF=fix_uvmat(Param,F,C,U,V,X,Y)
268%error=[]; %default
269Param
270FF=zeros(size(F));%default
271
272%criterium on warn flags
273if isfield (Param,'WarnFlags')
274    for iflag=1:numel(Param.WarnFlags)
275        FF=(FF==1| F==Param.WarnFlags(iflag));
276    end
277end
278
279%criterium on correlation values
280if isfield (Param,'LowerBoundCorr')
281    FF=FF==1 | C<Param.LowerBoundCorr;
282end
283
284if isfield (Param,'LowerBoundVel')&& ~isequal(Param.LowerBoundVel,0)
285    thresh=Param.LowerBoundVel*Param.LowerBoundVel;
286    FF=FF==1 | (U.*U+V.*V)<thresh;
287end
288if isfield (Param,'UpperBoundVel')&& ~isequal(Param.UpperBoundVel,0)
289    thresh=Param.UpperBoundVel*Param.UpperBoundVel;
290    FF=FF==1 | (U.*U+V.*V)>thresh;
291end
292if isfield(Param,'MaskName')
293   M=imread(Param.MaskName);
294   nxy=size(M);
295   M=reshape(M,1,[]);
296   rangx0=[0.5 nxy(2)-0.5];
297   rangy0=[0.5 nxy(1)-0.5];
298   vec_x1=X-U/2;%beginning points
299   vec_x2=X+U/2;%end points of vectors
300   vec_y1=Y-V/2;%beginning points
301   vec_y2=Y+V/2;%end points of vectors
302   indx=1+round((nxy(2)-1)*(vec_x1-rangx0(1))/(rangx0(2)-rangx0(1)));% image index x at abcissa vec_x1
303   indy=1+round((nxy(1)-1)*(vec_y1-rangy0(1))/(rangy0(2)-rangy0(1)));% image index y at ordinate vec_y1   
304   test_in=~(indx < 1 |indy < 1 | indx > nxy(2) |indy > nxy(1)); %=0 out of the mask image, 1 inside
305   indx=indx.*test_in+(1-test_in); %replace indx by 1 out of the mask range
306   indy=indy.*test_in+(1-test_in); %replace indy by 1 out of the mask range
307   ICOMB=((indx-1)*nxy(1)+(nxy(1)+1-indy));%determine the indices in the image reshaped in a Matlab vector
308   Mvalues=M(ICOMB);
309   flag7b=((20 < Mvalues) & (Mvalues < 200))| ~test_in';
310   indx=1+round((nxy(2)-1)*(vec_x2-rangx0(1))/(rangx0(2)-rangx0(1)));% image index x at abcissa vec_x2
311   indy=1+round((nxy(1)-1)*(vec_y2-rangy0(1))/(rangy0(2)-rangy0(1)));% image index y at ordinate vec_y2
312   test_in=~(indx < 1 |indy < 1 | indx > nxy(2) |indy > nxy(1)); %=0 out of the mask image, 1 inside
313   indx=indx.*test_in+(1-test_in); %replace indx by 1 out of the mask range
314   indy=indy.*test_in+(1-test_in); %replace indy by 1 out of the mask range
315   ICOMB=((indx-1)*nxy(1)+(nxy(1)+1-indy));%determine the indices in the image reshaped in a Matlab vector
316   Mvalues=M(ICOMB);
317   flag7e=((Mvalues > 20) & (Mvalues < 200))| ~test_in';
318   FF=FF==1 |(flag7b|flag7e)';
319end
320%    flag7=0;
321% end   
322
323
324FF=double(FF);
325%
326% % criterium on velocity values
327% delta_u=Field.U;%default without ref file
328% delta_v=Field.V;
329% if exist('fileref','var') && ~isempty(fileref)
330%     if ~exist(fileref,'file')
331%         error='reference file not found in RUN_FIX.m';
332%         display(error);
333%         return
334%     end
335%     FieldRef=read_civxdata(fileref,[],fieldref);   
336%     if isfield(FieldRef,'FF')
337%         index_true=find(FieldRef.FF==0);
338%         FieldRef.X=FieldRef.X(index_true);
339%         FieldRef.Y=FieldRef.Y(index_true);
340%         FieldRef.U=FieldRef.U(index_true);
341%         FieldRef.V=FieldRef.V(index_true);
342%     end
343%     if ~isfield(FieldRef,'X') || ~isfield(FieldRef,'Y') || ~isfield(FieldRef,'U') || ~isfield(FieldRef,'V')
344%         error='reference file is not a velocity field in RUN_FIX.m '; %bad input file
345%         return
346%     end
347%     if length(FieldRef.X)<=1
348%         errordlg('reference field with one vector or less in RUN_FIX.m')
349%         return
350%     end
351%     vec_U_ref=griddata_uvmat(FieldRef.X,FieldRef.Y,FieldRef.U,Field.X,Field.Y);  %interpolate vectors in the ref field
352%     vec_V_ref=griddata_uvmat(FieldRef.X,FieldRef.Y,FieldRef.V,Field.X,Field.Y);  %interpolate vectors in the ref field to the positions  of the main field     
353%     delta_u=Field.U-vec_U_ref;%take the difference with the interpolated ref field
354%     delta_v=Field.V-vec_V_ref;
355% end
356% thresh_vel_x=thresh_vel;
357% thresh_vel_y=thresh_vel;
358% if isequal(inf_sup,1)
359%     flag5=abs(delta_u)<thresh_vel_x & abs(delta_v)<thresh_vel_y &(flag1~=1)&(flag2~=1)&(flag3~=1)&(flag4~=1);
360% elseif isequal(inf_sup,2)
361%     flag5=(abs(delta_u)>thresh_vel_x | abs(delta_v)>thresh_vel_y) &(flag1~=1)&(flag2~=1)&(flag3~=1)&(flag4~=1);
362% end
363%
364%             % flag7 introduce a grey mask, matrix M
365
366% flagmagenta=flag1|flag2|flag3|flag4|flag5|flag7;
367% fixflag_unit=Field.FF-10*floor(Field.FF/10); %unity term of fix_flag
368
369
370
371%------------------------------------------------------------------------
372% patch function
373function [SubRangx,SubRangy,nbpoints,FF,U_smooth,V_smooth,X_tps,Y_tps,U_tps,V_tps] =patch_uvmat(X,Y,U,V,Rho,Threshold,SubDomain)
374%subdomain decomposition
375warning off
376U=reshape(U,[],1);
377V=reshape(V,[],1);
378X=reshape(X,[],1);
379Y=reshape(Y,[],1);
380nbvec=numel(X);
381NbSubDomain=ceil(nbvec/SubDomain);
382MinX=min(X);
383MinY=min(Y);
384MaxX=max(X);
385MaxY=max(Y);
386RangX=MaxX-MinX;
387RangY=MaxY-MinY;
388AspectRatio=RangY/RangX;
389NbSubDomainX=max(floor(sqrt(NbSubDomain/AspectRatio)),1);
390NbSubDomainY=max(floor(sqrt(NbSubDomain*AspectRatio)),1);
391NbSubDomain=NbSubDomainX*NbSubDomainY;
392SizX=RangX/NbSubDomainX;%width of subdomains
393SizY=RangY/NbSubDomainY;%height of subdomains
394CentreX=linspace(MinX+SizX/2,MaxX-SizX/2,NbSubDomainX);
395CentreY=linspace(MinY+SizY/2,MaxY-SizY/2,NbSubDomainY);
396[CentreX,CentreY]=meshgrid(CentreX,CentreY);
397CentreY=reshape(CentreY,1,[]);
398CentreX=reshape(CentreX,1,[]);
399rho=SizX*SizY*Rho/1000000;%optimum rho increase as the area of the subdomain (division by 10^6 to reach good values with the default GUI input)
400U_tps_sub=zeros(length(X),NbSubDomain);%default spline
401V_tps_sub=zeros(length(X),NbSubDomain);%default spline
402U_smooth=zeros(length(X),1);
403V_smooth=zeros(length(X),1);
404
405nb_select=zeros(length(X),1);
406FF=zeros(length(X),1);
407test_empty=zeros(1,NbSubDomain);
408for isub=1:NbSubDomain
409    SubRangx(isub,:)=[CentreX(isub)-SizX/2 CentreX(isub)+SizX/2];
410    SubRangy(isub,:)=[CentreY(isub)-SizY/2 CentreY(isub)+SizY/2];
411    ind_sel_previous=[];
412    ind_sel=0;
413    while numel(ind_sel)>numel(ind_sel_previous) %increase the subdomain during four iterations at most
414        ind_sel_previous=ind_sel;
415        ind_sel=find(X>SubRangx(isub,1) & X<SubRangx(isub,2) & Y>SubRangy(isub,1) & Y<SubRangy(isub,2));
416        % if no vector in the subdomain, skip the subdomain
417        if isempty(ind_sel)
418            test_empty(isub)=1;   
419            U_tps(1,isub)=0;%define U_tps and V_tps by default
420            V_tps(1,isub)=0;
421            break
422            % if too few selected vectors, increase the subrange for next iteration
423        elseif numel(ind_sel)<SubDomain/4 && ~isequal( ind_sel,ind_sel_previous);
424            SubRangx(isub,1)=SubRangx(isub,1)-SizX/4;
425            SubRangx(isub,2)=SubRangx(isub,2)+SizX/4;
426            SubRangy(isub,1)=SubRangy(isub,1)-SizY/4;
427            SubRangy(isub,2)=SubRangy(isub,2)+SizY/4;
428        else
429            [U_smooth_sub,U_tps_sub]=tps_coeff(X(ind_sel),Y(ind_sel),U(ind_sel),rho);
430            [V_smooth_sub,V_tps_sub]=tps_coeff(X(ind_sel),Y(ind_sel),V(ind_sel),rho);
431            UDiff=U_smooth_sub-U(ind_sel);
432            VDiff=V_smooth_sub-V(ind_sel);
433            NormDiff=UDiff.*UDiff+VDiff.*VDiff;
434            FF(ind_sel)=20*(NormDiff>Threshold);%put FF value to 20 to identify the criterium of elimmination
435            ind_ind_sel=find(FF(ind_sel)==0); % select the indices of ind_sel corresponding to the remaining vectors
436            % no value exceeds threshold, the result is recorded
437            if isequal(numel(ind_ind_sel),numel(ind_sel))
438                U_smooth(ind_sel)=U_smooth(ind_sel)+U_smooth_sub;
439                V_smooth(ind_sel)=V_smooth(ind_sel)+V_smooth_sub;
440                nbpoints(isub)=numel(ind_sel);
441                X_tps(1:nbpoints(isub),isub)=X(ind_sel);
442                Y_tps(1:nbpoints(isub),isub)=Y(ind_sel);
443                U_tps(1:nbpoints(isub)+3,isub)=U_tps_sub;
444                V_tps(1:nbpoints(isub)+3,isub)=V_tps_sub;         
445                nb_select(ind_sel)=nb_select(ind_sel)+1;
446                 display('good')
447                break
448                % too few selected vectors, increase the subrange for next iteration
449            elseif numel(ind_ind_sel)<SubDomain/4 && ~isequal( ind_sel,ind_sel_previous);
450                SubRangx(isub,1)=SubRangx(isub,1)-SizX/4;
451                SubRangx(isub,2)=SubRangx(isub,2)+SizX/4;
452                SubRangy(isub,1)=SubRangy(isub,1)-SizY/4;
453                SubRangy(isub,2)=SubRangy(isub,2)+SizY/4;
454%                 display('fewsmooth')
455                % interpolation-smoothing is done again with the selected vectors
456            else
457                [U_smooth_sub,U_tps_sub]=tps_coeff(X(ind_sel(ind_ind_sel)),Y(ind_sel(ind_ind_sel)),U(ind_sel(ind_ind_sel)),rho);
458                [V_smooth_sub,V_tps_sub]=tps_coeff(X(ind_sel(ind_ind_sel)),Y(ind_sel(ind_ind_sel)),V(ind_sel(ind_ind_sel)),rho);
459                U_smooth(ind_sel(ind_ind_sel))=U_smooth(ind_sel(ind_ind_sel))+U_smooth_sub;
460                V_smooth(ind_sel(ind_ind_sel))=V_smooth(ind_sel(ind_ind_sel))+V_smooth_sub;
461                nbpoints(isub)=numel(ind_ind_sel);
462                X_tps(1:nbpoints(isub),isub)=X(ind_sel(ind_ind_sel));
463                Y_tps(1:nbpoints(isub),isub)=Y(ind_sel(ind_ind_sel));
464                U_tps(1:nbpoints(isub)+3,isub)=U_tps_sub;
465                V_tps(1:nbpoints(isub)+3,isub)=V_tps_sub;
466                nb_select(ind_sel(ind_ind_sel))=nb_select(ind_sel(ind_ind_sel))+1;
467                display('good2')
468                break
469            end
470        end
471    end
472end
473ind_empty=find(test_empty);
474%remove empty subdomains
475if ~isempty(ind_empty)
476    SubRangx(ind_empty,:)=[];
477    SubRangy(ind_empty,:)=[];
478    X_tps(:,ind_empty)=[];
479    Y_tps(:,ind_empty)=[];
480    U_tps(:,ind_empty)=[];
481    V_tps(:,ind_empty)=[];
482end
483nb_select(nb_select==0)=1;%ones(size(find(nb_select==0)));
484U_smooth=U_smooth./nb_select;
485V_smooth=V_smooth./nb_select;
486
487
488
489
490
491
492% U_patch = EM * spline_coeff;
493% U_patch=reshape(U_patch,npy,npx);
494% PM = [ones(size(dsites,1),1) dsites];
495% EM = [IM_sites PM];
496% U(test_false)=[];
497% U_nodes=EM * spline_coeff;
498
499%exact = testfunctions(epoints);
500%maxerr = norm(Pf-exact,inf);
501% PlotSurf(xe,ye,Pf,neval,exact,maxerr,[160,20]);
502% PlotError2D(xe,ye,Pf,exact,maxerr,neval,[160,20]);
503
504
505
506  % DM = DistanceMatrix(dsites,ctrs)
507% Forms the distance matrix of two sets of points in R^s,
508% i.e., DM(i,j) = || datasite_i - center_j ||_2.
509% Input
510%   dsites: Mxs matrix representing a set of M data sites in R^s
511%              (i.e., each row contains one s-dimensional point)
512%   ctrs:   Nxs matrix representing a set of N centers in R^s
513%              (one center per row)
514% Output
515
516
517
518
Note: See TracBrowser for help on using the repository browser.