Sender_black_and_white.py

import zmq
from cv2 import VideoCapture
import cv2
import numpy as np
import struct
from util import Dither,SHARPEN,Filter,zeros,SMOOTH
import bz2
import time
import sys

ip = "0.0.0.0"
port = 8000
audio = 0


try:
    ip = sys.argv[1] 
    port = sys.argv[2]
    
except:
    print("Please include the sender IP, Port")
    exit()


camNumber = 0
context = zmq.Context()
PublishSocket = context.socket(zmq.PUB)
PublishSocket.set_hwm(2) #Set ZMQ high water mark
PublishSocket.bind("tcp://"+ip+":%s" % port) # Bind server to ip
capture = VideoCapture(camNumber) #Video Source



def AddString(frame1,string):
    cv2.putText(frame1,string,(0,25),cv2.FONT_HERSHEY_SIMPLEX,1,(0,0,0),2, cv2.LINE_AA)
    return frame1


frames = 0
fps = 30
timea = time.time()
bytes = 0
bytesPerSec = 0



while True:
    buffer = b"V"
    frames +=1
    ret,raw_frame = capture.read() #Read camera
    frame2 = cv2.cvtColor(raw_frame, cv2.COLOR_BGR2GRAY)
    #gray = Filter(frame2,SHARPEN)
    #gray = floyd_steinberg(frame2)
    gray =Dither(frame2,zeros(frame2.shape),125) 
    #frameBytes = np.packbits(gray//255) #Convert it to bytes
    
    buffer += struct.pack("HH",gray.shape[0],gray.shape[1]) #Pack frame size into packet
    buffer += gray.tobytes()
    compressed = bz2.compress(buffer,9)
    PublishSocket.send(compressed) #Send Packet
    bytes += len(compressed)
    if(time.time() - timea > 1):
        timea = time.time()
        fps = frames
        frames = 0
        bytesPerSec += bytes/1024/1024
        bytesPerSec = bytesPerSec/2
        bytes = 0
    cv2.imshow("Video",AddString(Filter(gray,SMOOTH),str(fps)+" FPS " + "{:04.2f} Avg MBytes Per Second".format(bytesPerSec))) #Display the frame
    cv2.waitKey(1)