snowmix/camera1/cam.py

#!/usr/bin/python3
import sys
import gi
gi.require_version('Gst', '1.0')
from gi.repository import Gst
import time
import threading
import os
import stat

WIDTH  = 1280
HEIGHT = 720

SCREEN_WIDTH = 800
SCREEN_HEIGHT = 600

def bus_call(bus, msg, *args):
	if msg.type == Gst.MessageType.EOS:
		print("End-of-stream")
		loop.quit()
		return
	elif msg.type == Gst.MessageType.ERROR:
		print("GST ERROR", msg.parse_error())
		loop.quit()
		return
	return True

zoom = 1
def quit():
	pipeline.send_event(Gst.Event.new_eos())

def set_preview(val):
	if val == 0:
		videosrc.set_property("preview-x", SCREEN_WIDTH)
		videosrc.set_property("preview-y", SCREEN_HEIGHT)
	else:
		videosrc.set_property("preview-x", 0)
		videosrc.set_property("preview-y", 0)

def set_zoom(target):
	global zoom
	if target < zoom:
		direction = -1
	else:
		direction = 1

	target=min(target, 1)
	target=max(target, 0.40)

	while abs(target-zoom) > 0.01: #fuck IEEE floats
		zoom = zoom + direction*0.015
		zoom = min(zoom, 1)
		zoom = max(zoom, 0.4)
		videosrc.set_property("roi-x", (1-zoom)/2)
		videosrc.set_property("roi-y", (1-zoom)/2)
		videosrc.set_property("roi-w", zoom)
		videosrc.set_property("roi-h", zoom)
		time.sleep(0.03)

def add_client(c):
	if "192.168" not in c or ":" not in c:
		print("Invalid client! format is 192.168.Y.Z:XXXX")
		return
	cur_clients=sink.get_property("clients").split(",")
	print(cur_clients)
	if c in cur_clients:
		print("Client exists!")
		return
	cur_clients.append(c)
	strclients = ",".join(cur_clients)
	sink.set_property("clients", strclients)
	
def control():
	FIFO="/tmp/picamfifo"
	if os.path.exists(FIFO):
		if stat.S_ISFIFO(os.stat(FIFO).st_mode):
			pass
		else:
			os.remove(FIFO)
			os.mkfifo(FIFO)
	else:
		os.mkfifo(FIFO)

	while True:
		with open(FIFO) as fifo:
			line=fifo.read().strip()
			print("Got: ", line)
			if "=" not in line:
				print("Invalid line.")
				continue
			key = line.split("=")[0]
			val = line.split("=")[1]

			if key == "preview":
				set_preview(int(val))

			if key == "zoom":
				set_zoom(float(val))

			if key == "client" :
				add_client(val)

			
if __name__ == "__main__":
	if 'LD_LIBRARY_PATH' not in os.environ:
		os.environ['LD_LIBRARY_PATH'] = '/usr/local/lib'
		os.execv(sys.argv[0], sys.argv)
	# initialization
	Gst.init(None)

	videocodec = "queue ! video/x-h264, width=%d, height=%d, framerate=25/1, profile=high ! h264parse ! rtph264pay mtu=1360 config-interval=1" % (WIDTH, HEIGHT)
	p = "rpicamsrc sensor-mode=4 do-timestamp=true name=src ! "+videocodec+" ! multiudpsink name=out"
	pipeline = Gst.parse_launch (p)
	if pipeline == None:
		print ("Failed to create pipeline")
		sys.exit(0)

	videosrc = pipeline.get_by_name ("src")
	videosrc.set_property("preview", 1)
	videosrc.set_property("preview-w", WIDTH)
	videosrc.set_property("preview-h", HEIGHT)
	videosrc.set_property("fullscreen", False)
	videosrc.set_property("bitrate", 16000000)
	videosrc.set_property("rotation", 180)
	set_preview(0)

	sink = pipeline.get_by_name ("out")
	# sink.set_property("clients", ",".join(["192.168.2.120:5000", "192.168.2.21:5000", "192.168.2.84:5000"]))
	sink.set_property("clients", ",".join(["192.168.2.21:5000"]))
	print(sink.get_property("clients"))

	# run
	pipeline.set_state(Gst.State.PLAYING)
	t = threading.Thread(target=control, daemon=True)
	t.start()
	while True:
		time.sleep(10000)

	# cleanup
	pipeline.set_state(Gst.State.NULL)