[image,probs,shs,scores]=sr_one_step_wb(image,t,method,ds,sr,sigma,th_prob,wb_handle,offset,scale,orig_sh)

function [image,probs,shs,scores]=sr_one_step_wb(image,t,method,ds,sr,sigma,th_prob,wb_handle,offset,scale,orig_sh)

if ~exist('th_prob','var')
    th_prob=0.9;
end

shs=zeros(length(t),2);

orig_image=image;
if exist('orig_sh','var') % cheat mode
   scores=[];
   for tid=1:length(t)
     shs(tid,:)=orig_sh(tid,:);
     probs(tid)=1;
   end
else
  for tid=1:length(t)
      waitbar(tid/2/length(t)*scale+offset,wb_handle);
        [tmp_sh,tmp_prob,tmp_scores]=subpixel_register(image,t{tid},ds,sr,sigma);
        probs(tid)=tmp_prob;
        shs(tid,:)=tmp_sh;
        scores{tid}=tmp_scores;
  end
end

if strcmp(method,'max_pocs') % only project to the one with highest score
    [dummy,max_id]=max(probs);
    image=pocs(image,t{max_id},ds,shs(max_id,:));
elseif strcmp(method,'pocs')
  for tid=1:length(t)
      waitbar((length(t)+tid)/2/length(t)*scale+offset,wb_handle);
     if (probs(tid) > th_prob)
            image=pocs(image,t{tid},ds,shs(tid,:));        
     end
  end
elseif strcmp(method,'sort_pocs')
    [dummy,sort_ids]=sort(probs(:),1,'descend');
    for tid=1:length(t)
      waitbar((length(t)+tid)/2/length(t)*scale+offset,wb_handle);
      if (probs(sort_ids(tid)) > th_prob)
            image=pocs(image,t{tid},ds,shs(tid,:));        
      end
    end
elseif strcmp(method,'average')
  image=zeros(size(image));
  for tid=1:length(t)
      waitbar((length(t)+tid)/2/length(t)*scale+offset,wb_handle);      
     if (probs(tid) > th_prob)
            sh_image=shift_image(kron(t{tid},ones(ds)),-shs(tid,:));
            image=image+sh_image;
     end
  end   
  if length(find(probs>th_prob))>0
      image=image/length(find(probs>th_prob));
  else
      image=orig_image; % can't find suitable one
  end
else
  error('method is not recognized');
end