from .otsu import *
#from FeatureExtract import *
import cv2
import numpy as np
from skimage import io
import scipy.misc

def truncate_hu(image_array, max = 400, min = -900):
    image = image_array.copy()
    image[image > max] = max
    image[image < min] = min
    return image
'''
edge不用这个函数了，比较慢
'''
def edge_detection(img):
    # gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY) # 转换为灰度图
    # img = truncate_hu(gray)
    img = truncate_hu(img)

    x = cv2.Sobel(img, cv2.CV_16S, 1, 0)
    y = cv2.Sobel(img, cv2.CV_16S, 0, 1)
    #z = cv2.Sobel(img, cv2.CV_16S, 1, 1)

    absX = cv2.convertScaleAbs(x)   # 转回uint8
    absY = cv2.convertScaleAbs(y)
    #absz = cv2.convertScaleAbs(z)

    dst = cv2.addWeighted(absX,0.5,absY,0.5,0)
    return dst
    
'''
otsu还没找到代替的
'''
def otsu(img):
    # gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY) # 转换为灰度图
    # otsu_img = otsu_helper(gray, upper=118, down = 45,categories=1)
    otsu_img = otsu_helper(img, categories=1)
    return otsu_img