python openCV 基於TCP的socket網絡傳輸視頻（三）

  在這個版本中，我實現了兩個客戶端，一個服務器的操作，其實有時候解決問題，就差這麼一點，一點就通。我來說說我這個版本，一個客戶端可以採集視頻，另一個客戶端可以觀看視頻，服務器可以建在本地，也可以搭建在雲平臺上，就是可以被外網訪問。
  服務器的思路:採用多線程，採集視頻一個客戶端，觀看視頻一個客戶端，其中有個要關注的點，就是有一個線程要加延時，不然的話觀看視頻會太卡。另外我這只是實現了，有需要注意的地方很多，代碼還不完善，先開服務器，再開採集，最後開觀看客戶端，錯了順序就不好使了，等我把多線程旅順了，再更新整理。主要是這個思路。

服務器

#!/usr/bin/python
# -*-coding:utf-8 -*-
import socket
import threading
import cv2
import numpy
from time import sleep


def recv_all(sock, count):
    buf = ''
    while count:
        newbuf = sock.recv(count)
        if not newbuf: return None
        buf += newbuf
        count -= len(newbuf)
    return buf


# 線程鎖
threadLock = threading.Lock()
# 視頻buf
# videoDatastr = ''
# 客戶端套接字
conn_list = []

def robotVideoThread(sock):
    global videoDatastr
    # 接受TCP鏈接並返回（conn, addr），其中conn是新的套接字對象，可以用來接收和發送數據，addr是鏈接客戶端的地址。
    conn, addr = sock.accept()
    print 'robot Connected with' + ' ' + addr[0] + ':' + str(addr[1])
    conn_list.append(conn)
    while True:
        length = recv_all(conn, 16)  # 首先接收來自客戶端發送的大小信息
        if len(length) == 16:
            # 若成功接收到大小信息，進一步再接收整張圖片
            #threadLock.acquire()
            videoDatastr = recv_all(conn, int(length))
            #threadLock.release()


def userVideoThread(sock):
    global videoDatastr
    # 接受TCP鏈接並返回（conn, addr），其中conn是新的套接字對象，可以用來接收和發送數據，addr是鏈接客戶端的地址。
    conn, addr = sock.accept()
    print 'user Connected with' + ' ' + addr[0] + ':' + str(addr[1])
    conn_list.append(conn)
    while True:
        #threadLock.acquire()
        conn.send(str(len(videoDatastr)).ljust(16))
        conn.send(videoDatastr)
        #threadLock.release()
        sleep(0.1)


if __name__ == '__main__':

    s_robot = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    address = ('10.0.0.30', 8888)
    s_robot.bind(address)
    s_robot.listen(True)
    print 'robot服務器初始化成功'

    s_user = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    address = ('10.0.0.30', 9999)
    s_user.bind(address)
    s_user.listen(True)
    print 'user服務器初始化成功'

    threading.Thread(target=robotVideoThread, args=(s_robot, )).start()
    threading.Thread(target=userVideoThread, args=(s_user,)).start()

採集客戶端

#!/usr/bin/python
# -*-coding:utf-8 -*-
import socket
import cv2
import numpy
from time import sleep

# socket.AF_INET 用於服務器與服務器之間的網絡通信
# socket.SOCK_STREAM 代表基於TCP的流式socket通信
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 連接服務端
address_server = ('10.0.0.30', 8888)
sock.connect(address_server)

# 從攝像頭採集圖像
# 參數是0,表示打開筆記本的內置攝像頭,參數是視頻文件路徑則打開視頻
capture = cv2.VideoCapture(0)
# capture.read() 按幀讀取視頻
# ret,frame 是capture.read()方法的返回值
# 其中ret是布爾值，如果讀取幀正確，返回True;如果文件讀到末尾，返回False。
# frame 就是每一幀圖像，是個三維矩陣
ret, frame = capture.read()
encode_param = [int(cv2.IMWRITE_JPEG_QUALITY), 50]
while ret:
    # 首先對圖片進行編碼，因爲socket不支持直接發送圖片
    # '.jpg'表示把當前圖片frame按照jpg格式編碼
    # result, img_encode = cv2.imencode('.jpg', frame)
    # img_encode = cv2.imencode('.jpg', frame, encode_param)[1]
    result, img_encode = cv2.imencode('.jpg', frame)
    # data = numpy.array(img_encode)
    # stringData = data.tostring()
    stringData = img_encode.tostring()
    # 首先發送圖片編碼後的長度
    sock.send(str(len(stringData)).ljust(16))
    # 然後一個字節一個字節發送編碼的內容
    # 如果是python對python那麼可以一次性發送，如果發給c++的server則必須分開發因爲編碼裏面有字符串結束標誌位，c++會截斷
    # for i in range(0, len(stringData)):
    #     sock.send(stringData[i])
    sock.send(stringData)
    # sleep(1)
    ret, frame = capture.read()
    cv2.resize(frame, (640, 480))

sock.close()
cv2.destroyAllWindows()

觀看客戶端

#!usr/bin/python
# coding=utf-8

import socket
import cv2
import numpy


# 接受圖片大小的信息
def recv_size(sock, count):
    buf = ''
    while count:
        newbuf = sock.recv(count)
        if not newbuf: return None
        buf += newbuf
        count -= len(newbuf)
    return buf


# socket.AF_INET 用於服務器與服務器之間的網絡通信
# socket.SOCK_STREAM 代表基於TCP的流式socket通信
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 連接服務端
address_server = ('10.0.0.30', 9999)
sock.connect(address_server)

while True:
    length = recv_size(sock, 16)  # 首先接收來自客戶端發送的大小信息
    if len(length) == 16:  # 若成功接收到大小信息，進一步再接收整張圖片
        stringData = recv_size(sock, int(length))
        data = numpy.fromstring(stringData, dtype='uint8')
        decimg = cv2.imdecode(data, 1)  # 解碼處理，返回mat圖片
        img = cv2.resize(decimg, (640, 480))
        cv2.imshow('SERVER', img)
        if cv2.waitKey(1) == 27:
            break
    # print('Image recieved successfully!')



sock.close()
cv2.destroyAllWindows()