#!/usr/bin/env python
# coding: utf-8

# In[1]:


# testing extended search area piv from pyprocess
# this won't show any difference since 0.23 version
# because we incorporate the extended_seach_piv
# 


# In[2]:


import numpy as np
from numpy.fft import rfft2, irfft2, fftshift
from numpy import log
import matplotlib.pyplot as plt
from openpiv.tools import imread
from openpiv.pyprocess import extended_search_area_piv, \
                            moving_window_array, \
                            fft_correlate_strided_images, \
                            find_subpixel_peak_position, \
                            normalize_intensity, \
                            get_field_shape


# In[3]:


frame_a = imread('../test1/exp1_001_a.bmp')
frame_b = imread('../test1/exp1_001_b.bmp')


# In[4]:


# frame_a = frame_a[:128,:64]
# frame_b = frame_b[:128,:64]


# In[5]:


def show_pair(I,J):
    fig, ax = plt.subplots(1,2,figsize=(12,8))
    ax[0].imshow(I,cmap=plt.cm.gray)
    ax[1].imshow(J,cmap=plt.cm.gray)


# In[6]:


show_pair(frame_a, frame_b)


# In[7]:


frame_a = normalize_intensity(frame_a)
frame_b = normalize_intensity(frame_b)


# In[8]:


show_pair(frame_a, frame_b)


# In[9]:


window_size = 32
overlap = 16
dt=1.0
search_area_size = 32
correlation_method="circular"
subpixel_method="gaussian"
sig2noise_method='peak2peak'


# In[10]:


get_ipython().run_cell_magic('time', '', "vel1 = extended_search_area_piv(frame_a, frame_b, window_size=window_size,\n               search_area_size=search_area_size, overlap=overlap,dt=dt,\n                               correlation_method='circular',\n                               subpixel_method=subpixel_method,\n                               sig2noise_method=sig2noise_method)\n")


# In[11]:


get_ipython().run_cell_magic('time', '', "window_size = 24\nsearch_area_size = 32\nvel2 = extended_search_area_piv(frame_a, frame_b, window_size=window_size,\n               search_area_size=search_area_size, overlap=overlap,dt=dt,\n                               correlation_method='circular',\n                               normalized_correlation=True,\n                               subpixel_method=subpixel_method,\n                               sig2noise_method=sig2noise_method)\n")


# In[12]:


get_ipython().run_cell_magic('time', '', "window_size = 24\nsearch_area_size = 32\nvel2 = extended_search_area_piv(frame_a, frame_b, window_size=window_size,\n               search_area_size=search_area_size, overlap=overlap,dt=dt,\n                               correlation_method='linear',\n                               subpixel_method=subpixel_method,\n                               sig2noise_method=sig2noise_method)\n")


# In[13]:


get_ipython().run_cell_magic('time', '', "window_size = 24\nsearch_area_size = 32\nvel2 = extended_search_area_piv(frame_a, frame_b, window_size=window_size,\n               search_area_size=search_area_size, overlap=overlap,dt=dt,\n                               correlation_method='linear',\n                               normalized_correlation=True,\n                               subpixel_method=subpixel_method,\n                               sig2noise_method=sig2noise_method)\n")


# In[14]:


plt.figure(figsize=(20,20))
plt.quiver(vel1[0],vel1[1],scale=100,color='b',alpha=0.2)


# In[ ]: