photomod.py

import time
from time import sleep
import datetime
import os
import sys
import subprocess 
from PIL import Image # to make thumbnail
from subprocess import call
from shutil import copyfile



def videodevlist():
	folderpath="/dev"
	videolist=[]
	filelist=os.listdir(folderpath)
	for files in filelist:
		if "video" in files:
			videolist.append(files)
	return videolist # item1 (path) item2 (name) item3 (datetime)
	

def checkPIcam(device):
	cmd = ['v4l2-ctl', '-d', '/dev/'+device, '-D']
	wordtofind="bm2835"
	isfound=executeandsearch(cmd,wordtofind)
	return isfound
	
def findPIcam():
	devicelist=videodevlist()
	# v4l2-ctl -d /dev/video0 -D
	for device in devicelist:
		cmd = ['v4l2-ctl', '-d', '/dev/'+device, '-D']
		wordtofind="bm2835"
		isfound=executeandsearch(cmd,wordtofind)
		if isfound:
			return device
	return ""

def executeandsearch(cmd,wordtofind):
	try:
		scanoutput = subprocess.check_output(cmd).decode('utf-8')
	except:
		print "error to execute the command" , cmd
		logger.error("error to execute the command %s",cmd)
		return False

	for line in scanoutput.split('\n'):
		#print " line ",line
		strstart=line.find(wordtofind)
		if strstart>-1:
			#found
			return True
	return False
	
	
	
	
	
	
 
def saveshot(filepath, video, realshot, resolution, positionvalue, vdirection):
	shottaken=False
	print "take photo"
	
	if vdirection=="neg":
		rotdeg="180"
	else:
		rotdeg="0"
	
	currentdate=datetime.datetime.now().strftime("%y-%m-%d,%H:%M")
	print "Current date and time: " , currentdate
	if realshot:
		filenamenopath=currentdate+"@"+video+"@"+positionvalue+".jpg"
		filenamenopath2=currentdate+"@"+video+"@"+positionvalue+"F.jpg"
		filenamenopath3=currentdate+"@"+video+"@"+positionvalue+"R.jpg"
	else:
		filenamenopath="testimage.jpg"
		filenamenopath2=filenamenopath
		filenamenopath3=filenamenopath
		
	filename=os.path.join(filepath, filenamenopath)
	print "Start Photo procedure: ", video , " ************************************************"
	print "FILE : ", filename		

	
	cam_list = "/dev/" + video
	if not (video==""):
		

		filexist=os.path.isfile(filename)
		print "file already exist = ", filexist
		
		if (filexist)and(not realshot):
			os.rename(filename, filename + ".bak")
		


		shottaken=False
		w=resolution.split("x")[0]
		h=resolution.split("x")[1]			

		

		filenamebase=filenamenopath.split(".")[0]
		extension=filename.split(".")[1]				
			
		# capture image using V4l2
		# http://www.geeetech.com/wiki/index.php/Raspberry_Pi_Camera_Module
		# v4l2-ctl --set-fmt-video=width=2592,height=1944,pixelformat=3
		# v4l2-ctl --stream-mmap=3 --stream-count=1 --stream-to=somefile.jpg
		# https://www.raspberrypi.org/forums/viewtopic.php?f=43&t=62364&start=450

		# check the v4l2 setting: v4l2-ctl -d /dev/video0 --list-ctrls
		# setting
		# v4l2-ctl --set-ctrl=gain=00
		# sudo v4l2-ctl -d /dev/video0 --set-ctrl=auto_exposure=1
		# (auto exposure=0 ->auto; exposure=1 ->manual, each camera has its own name of the parameter, auto_exposure, exposure_auto)
		# v4l2-ctl --set-ctrl=exposure_absolute=10
		
		# raspistill provides way better photo than the fswebcam when using the raspbery camera
		# raspberry camera is on video0 only
		# there is no reliable way to detect the raspicam, then just try to get a picture first with raspistill
		
		
		if checkPIcam(video):
			print "The video device should be PI camera"
			shottaken=takeshotandsave_raspistill(filepath,filenamenopath3, video, resolution,rotdeg)
			if not shottaken: # gives it a second chance :)
				shottaken=takeshotandsave_fswebcam(filepath,filenamenopath2, video, resolution,rotdeg)	
		else:
			print " The video device should be USB camera"
			shottaken=takeshotandsave_fswebcam(filepath,filenamenopath2, video, resolution,rotdeg)		
		
		#shottaken=takeshotandsave_mjpg_streamer(filepath,filenamenopath, video, resolution)	
				
		if (not shottaken)and(not realshot):
			if filexist:
				os.rename(filename + ".bak", filename)
		
		print "Picture acknowledge return = " ,shottaken
		
				

	else:
		print "camera not connected"	
	return shottaken



def takeshotandsave_raspistill(filepath,filenamenopath, video, resolution, rotdeg):
	shottaken=False
	if rotdeg=="180":
		vflip="-vf -hf"
	else:
		vflip=""
	print "flip ", vflip
		

	cam_list = "/dev/" + video			

	i=0
	while (not shottaken)and(i<3):
		i=i+1
		filename=os.path.join(filepath, filenamenopath)
		print "FILE : ", filename		

		shottaken=False
		w=resolution.split("x")[0]
		h=resolution.split("x")[1]			

		print "try raspistill"


		filenamebase=filenamenopath.split(".")[0]
		extension=filename.split(".")[1]

		# create the picture files
		try:
			myproc = subprocess.check_output("raspistill "+vflip+" -w "+w+" -h "+h+" -q 95 -a 12 -a \"%Y-%m-%d %X (UTC)\" -o " + filename, shell=True, stderr=subprocess.STDOUT)
		except:
			print "problem to execute command"
			myproc = "error"

		newfilexist=os.path.isfile(filename)
		print "file was created = ", newfilexist
				

		if (myproc=="")and(newfilexist):
			print "raspistill got picture"
			shottaken=True
			# make thumbnail
			ExistandThumb(filepath,filenamenopath,shottaken)			

		else:
			print "raspistill not able to get picture"
			shottaken=False


		print "RASPISTILL Picture take = " ,shottaken, "  Attempt ", i
	

	return shottaken






def takeshotandsave_fswebcam(filepath,filenamenopath, video, resolution, rotdeg):
	shottaken=False

	if not (video==""):
		cam_list = "/dev/" + video			
	
		i=0
		while (not shottaken)and(i<3):
			i=i+1
			filename=os.path.join(filepath, filenamenopath)
			print "FILE : ", filename		

	

			shottaken=False
			w=resolution.split("x")[0]
			h=resolution.split("x")[1]			

			print "try fswebcam"


			filenamebase=filenamenopath.split(".")[0]
			extension=filename.split(".")[1]
			
			#fswebcam option
			if i==1:
				S="15"
			else:
				S="5"
			
			# create the picture files
			#fswebcam option
			try:
				myproc = subprocess.check_output("fswebcam -q -d "+ cam_list +" -r "+resolution+" -S "+S+" --rotate "+rotdeg+" -s brightness=50% -s Contrast=50% --jpeg 95 " + filename, shell=True, stderr=subprocess.STDOUT)
			except:
				print "problem to execute command"
				myproc = "error"
			# -R use read() method -- NOT WORKING ---
			# -D delay before taking frames
			# -S skip the first frames
			# -q quiet output mode
			# -d device
			# -r resoltion
			# -F takes frames

			print "output from subprocess: " , myproc

			newfilexist=os.path.isfile(filename)
			print "file was created = ", newfilexist
					

			if (myproc=="")and(newfilexist):
				print "fswebcam got picture"
				shottaken=True
				# make thumbnail
				ExistandThumb(filepath,filenamenopath,shottaken)			

			else:
				print "fswebcam not able to get picture"
				shottaken=False


			print "FSWEBCAM Picture take = " ,shottaken, "  Attempt ", i
	
	else:
		print "camera not connected"	
	return shottaken

def takeshotandsave_mjpg_streamer(filepath,filenamenopath, video, resolution , rotdeg):
	shottaken=False
	
	
	filename=os.path.join(filepath, filenamenopath)
	print "FILE : ", filename		

	

	if not (video==""):
		cam_list = "/dev/" + video			

		shottaken=False
		w=resolution.split("x")[0]
		h=resolution.split("x")[1]			

		print "try mjpg_streamer"


		filenamebase=filenamenopath.split(".")[0]
		extension=filename.split(".")[1]
		
		pathmjpg=os.path.join(filepath,"mjpg")
		if not os.path.exists(pathmjpg):
			# fi folder do not exist, create it
			os.makedirs(pathmjpg)
		else:
			#remove all files in folder
			for the_file in os.listdir(pathmjpg):
				file_path = os.path.join(pathmjpg, the_file)
				try:
					if os.path.isfile(file_path):
						os.unlink(file_path)
				except Exception as e:
					print(e)
		
		# create the picture files
		fps="20"
		
		if (video=="video0")and(int(w)>1024):
			print "mjpg_streamer using the raspicam"
			stream="mjpg_streamer -i '/usr/local/lib/mjpg-streamer/input_raspicam.so -d /dev/"+video+" -x "+w+" -y "+h+" -fps "+fps+" -rot "+rotdeg+"' -o '/usr/local/lib/mjpg-streamer/output_file.so -f "+pathmjpg+" -d 100' &"
		else:
			stream="mjpg_streamer -i '/usr/local/lib/mjpg-streamer/input_uvc.so -d /dev/"+video+" -r "+w+"x"+h+" -f "+fps+" -rot "+rotdeg+"' -o '/usr/local/lib/mjpg-streamer/output_file.so -f "+pathmjpg+" -d 100' &"
		call ([stream], shell=True)
		time.sleep(2)
		call (["sudo pkill mjpg_streamer"], shell=True)
		
		# take last saved file in the folder
		folderpath=pathmjpg

		
		filenamelist=[]
		sortedlist=sorted([f for f in os.listdir(folderpath) if os.path.isfile(os.path.join(folderpath, f))])
		sortedlist.reverse()
		lastfile=""
		for files in sortedlist:
			if (files.endswith(".jpg") or files.endswith(".png")):
				lastfile=files
				break

		#copy lastfile to filepath
		if not (lastfile==""):
			shottaken=True
			#copy file to the right folder and right name
			src=os.path.join(pathmjpg, lastfile)
			dst=filename
			copyfile(src, dst)		
			# make thumbnail
			ExistandThumb(filepath,filenamenopath,shottaken)			
		else:
			print "mjpg_streame not able to get picture"
			shottaken=False


		print "MJPG_STREAMER Picture take = " ,shottaken
		

					

	else:
		print "camera not connected"	
	return shottaken




def ExistandThumb(filepath,filenamenopath,shottaken):
	filename=os.path.join(filepath, filenamenopath)
	newfilexist=os.path.isfile(filename)
	# make thumbnail
	if (shottaken and newfilexist):
		paththumb=os.path.join(filepath,"thumb")
		if not os.path.exists(paththumb):
			os.makedirs(paththumb)
		try:
			image = Image.open(filename)
			image.thumbnail((300, 300))
			thumbname=os.path.join(paththumb,filenamenopath)
			image.save(thumbname)
		except:
			print "not able to make thumbnail"
	return newfilexist
	

def thumbconsistency(apprunningpath):
	# check if there is a thumbnail without corresponding image
	
	filepath=os.path.join(apprunningpath, "static")
	filepath=os.path.join(filepath, "hydropicture")
	# control if the folder hydropicture exist otherwise create it
	if not os.path.exists(filepath):
		os.makedirs(filepath)
		print "Hydropicture folder has been created"
	paththumb=os.path.join(filepath,"thumb")
	if not os.path.exists(paththumb):
		os.makedirs(paththumb)
		print "Hydropicture thumbnail folder has been created"
	
	filenamelist=os.listdir(filepath)

	thumbnamelist=os.listdir(paththumb)
	
	for thumbnail in thumbnamelist:
		if thumbnail not in filenamelist:
			print "thumbnail has no corresponding image, delete"
			os.remove(os.path.join(paththumb, thumbnail))
	
	# create thumbnail in case picture has no coresponding thumbnail
	
	
	for fileimage in filenamelist:
		if os.path.isfile(os.path.join(filepath, fileimage)):
			if fileimage not in thumbnamelist:
				print "image has no corresponding thumbnail, create"
				#create thumbnail
				try:
					image = Image.open(os.path.join(filepath,fileimage))
					image.thumbnail((300, 300))
					thumbname=os.path.join(paththumb,os.path.basename(fileimage))
					image.save(thumbname)
				except:
					not "able to make thumbnail"
			
	return True







if __name__ == '__main__':
	
	"""
	prova funzioni di camera
	"""
	print "PI cam device :" , findPIcam()
	#saveshot()