# -*- coding: utf-8 -*-

import numpy as np

def bezier_curve(p0, p1, p2, p3, inserted):
 """
 三階貝塞爾曲線
 
 p0, p1, p2, p3 - 點(diǎn)坐標(biāo)，tuple、list或numpy.ndarray類型
 inserted  - p0和p3之間插值的數(shù)量
 """
 
 assert isinstance(p0, (tuple, list, np.ndarray)), u'點(diǎn)坐標(biāo)不是期望的元組、列表或numpy數(shù)組類型'
 assert isinstance(p0, (tuple, list, np.ndarray)), u'點(diǎn)坐標(biāo)不是期望的元組、列表或numpy數(shù)組類型'
 assert isinstance(p0, (tuple, list, np.ndarray)), u'點(diǎn)坐標(biāo)不是期望的元組、列表或numpy數(shù)組類型'
 assert isinstance(p0, (tuple, list, np.ndarray)), u'點(diǎn)坐標(biāo)不是期望的元組、列表或numpy數(shù)組類型'
 
 if isinstance(p0, (tuple, list)):
  p0 = np.array(p0)
 if isinstance(p1, (tuple, list)):
  p1 = np.array(p1)
 if isinstance(p2, (tuple, list)):
  p2 = np.array(p2)
 if isinstance(p3, (tuple, list)):
  p3 = np.array(p3)
 
 points = list()
 for t in np.linspace(0, 1, inserted+2):
  points.append(p0*np.power((1-t),3) + 3*p1*t*np.power((1-t),2) + 3*p2*(1-t)*np.power(t,2) + p3*np.power(t,3))
 
 return np.vstack(points)


def smoothing_base_bezier(date_x, date_y, k=0.5, inserted=10, closed=False):
 """
 基于三階貝塞爾曲線的數(shù)據(jù)平滑算法
 
 date_x  - x維度數(shù)據(jù)集，list或numpy.ndarray類型
 date_y  - y維度數(shù)據(jù)集，list或numpy.ndarray類型
 k   - 調(diào)整平滑曲線形狀的因子，取值一般在0.2~0.6之間。默認(rèn)值為0.5
 inserted - 兩個(gè)原始數(shù)據(jù)點(diǎn)之間插值的數(shù)量。默認(rèn)值為10
 closed  - 曲線是否封閉，如是，則首尾相連。默認(rèn)曲線不封閉
 """
 
 assert isinstance(date_x, (list, np.ndarray)), u'x數(shù)據(jù)集不是期望的列表或numpy數(shù)組類型'
 assert isinstance(date_y, (list, np.ndarray)), u'y數(shù)據(jù)集不是期望的列表或numpy數(shù)組類型'
 
 if isinstance(date_x, list) and isinstance(date_y, list):
  assert len(date_x)==len(date_y), u'x數(shù)據(jù)集和y數(shù)據(jù)集長(zhǎng)度不匹配'
  date_x = np.array(date_x)
  date_y = np.array(date_y)
 elif isinstance(date_x, np.ndarray) and isinstance(date_y, np.ndarray):
  assert date_x.shape==date_y.shape, u'x數(shù)據(jù)集和y數(shù)據(jù)集長(zhǎng)度不匹配'
 else:
  raise Exception(u'x數(shù)據(jù)集或y數(shù)據(jù)集類型錯(cuò)誤')
 
 # 第1步：生成原始數(shù)據(jù)折線中點(diǎn)集
 mid_points = list()
 for i in range(1, date_x.shape[0]):
  mid_points.append({
   'start': (date_x[i-1], date_y[i-1]),
   'end':  (date_x[i], date_y[i]),
   'mid':  ((date_x[i]+date_x[i-1])/2.0, (date_y[i]+date_y[i-1])/2.0)
  })
 
 if closed:
  mid_points.append({
   'start': (date_x[-1], date_y[-1]),
   'end':  (date_x[0], date_y[0]),
   'mid':  ((date_x[0]+date_x[-1])/2.0, (date_y[0]+date_y[-1])/2.0)
  })
 
 # 第2步：找出中點(diǎn)連線及其分割點(diǎn)
 split_points = list()
 for i in range(len(mid_points)):
  if i < (len(mid_points)-1):
   j = i+1
  elif closed:
   j = 0
  else:
   continue
  
  x00, y00 = mid_points[i]['start']
  x01, y01 = mid_points[i]['end']
  x10, y10 = mid_points[j]['start']
  x11, y11 = mid_points[j]['end']
  d0 = np.sqrt(np.power((x00-x01), 2) + np.power((y00-y01), 2))
  d1 = np.sqrt(np.power((x10-x11), 2) + np.power((y10-y11), 2))
  k_split = 1.0*d0/(d0+d1)
  
  mx0, my0 = mid_points[i]['mid']
  mx1, my1 = mid_points[j]['mid']
  
  split_points.append({
   'start': (mx0, my0),
   'end':  (mx1, my1),
   'split': (mx0+(mx1-mx0)*k_split, my0+(my1-my0)*k_split)
  })
 
 # 第3步：平移中點(diǎn)連線，調(diào)整端點(diǎn)，生成控制點(diǎn)
 crt_points = list()
 for i in range(len(split_points)):
  vx, vy = mid_points[i]['end'] # 當(dāng)前頂點(diǎn)的坐標(biāo)
  dx = vx - split_points[i]['split'][0] # 平移線段x偏移量
  dy = vy - split_points[i]['split'][1] # 平移線段y偏移量
  
  sx, sy = split_points[i]['start'][0]+dx, split_points[i]['start'][1]+dy # 平移后線段起點(diǎn)坐標(biāo)
  ex, ey = split_points[i]['end'][0]+dx, split_points[i]['end'][1]+dy # 平移后線段終點(diǎn)坐標(biāo)
  
  cp0 = sx+(vx-sx)*k, sy+(vy-sy)*k # 控制點(diǎn)坐標(biāo)
  cp1 = ex+(vx-ex)*k, ey+(vy-ey)*k # 控制點(diǎn)坐標(biāo)
  
  if crt_points:
   crt_points[-1].insert(2, cp0)
  else:
   crt_points.append([mid_points[0]['start'], cp0, mid_points[0]['end']])
  
  if closed:
   if i < (len(mid_points)-1):
    crt_points.append([mid_points[i+1]['start'], cp1, mid_points[i+1]['end']])
   else:
    crt_points[0].insert(1, cp1)
  else:
   if i < (len(mid_points)-2):
    crt_points.append([mid_points[i+1]['start'], cp1, mid_points[i+1]['end']])
   else:
    crt_points.append([mid_points[i+1]['start'], cp1, mid_points[i+1]['end'], mid_points[i+1]['end']])
    crt_points[0].insert(1, mid_points[0]['start'])
 
 # 第4步：應(yīng)用貝塞爾曲線方程插值
 out = list()
 for item in crt_points:
  group = bezier_curve(item[0], item[1], item[2], item[3], inserted)
  out.append(group[:-1])
 
 out.append(group[-1:])
 out = np.vstack(out)
 
 return out.T[0], out.T[1]


if __name__ == '__main__':
 import matplotlib.pyplot as plt
 
 x = np.array([2,4,4,3,2])
 y = np.array([2,2,4,3,4])
	
	plt.plot(x, y, 'ro')
 x_curve, y_curve = smoothing_base_bezier(x, y, k=0.3, closed=True)
 plt.plot(x_curve, y_curve, label='$k=0.3$')
 x_curve, y_curve = smoothing_base_bezier(x, y, k=0.4, closed=True)
 plt.plot(x_curve, y_curve, label='$k=0.4$')
 x_curve, y_curve = smoothing_base_bezier(x, y, k=0.5, closed=True)
 plt.plot(x_curve, y_curve, label='$k=0.5$')
 x_curve, y_curve = smoothing_base_bezier(x, y, k=0.6, closed=True)
 plt.plot(x_curve, y_curve, label='$k=0.6$')
 plt.legend(loc='best')
 
 plt.show()