import socket
import numpy as np
import cv2
from classifer import WoodClass
import time
import os
from utils import PreSocket, process_cmd



def main():
    socket_receive = PreSocket(socket.AF_INET, socket.SOCK_STREAM)
    socket_receive.connect(('127.0.0.1', 21122))
    socket_send = PreSocket(socket.AF_INET, socket.SOCK_STREAM)
    socket_send.connect(('127.0.0.1', 21123))
    process_cmd(recv_sock=socket_receive , send_sock=socket_send)


if __name__ == '__main__':
    # 2个端口
    # 接受端口21122
    # 发送端口21123
    # 接收到图片 n_rows * n_bands * n_cols, float32
    # 发送图片 n_rows * n_cols, uint8
    main()
    # test(r"D:\build-tobacco-Desktop_Qt_5_9_0_MSVC2015_64bit-Release\calibrated15.raw")
    # main()
    # debug_main()
    # test_run(all_data_dir=r'D:\数据')
    # with open(r'D:\数据\虫子\valid2.raw', 'rb') as f:
    #     data = np.frombuffer(f.read(), dtype=np.float32).reshape(600, 29, 1024).transpose(0, 2, 1)
    # plt.matshow(data[:, :, 10])
    # plt.show()
    # detector = SpecDetector('model_spec/model_29.p')
    # result = detector.predict(data)
    #
    # plt.matshow(result)
    # plt.show()
    # result = result.reshape((600, 1024))