def main():
try:
#time.sleep(60)
CASC_PATH = "haarcascade_frontalface_default.xml"
URI = 'mongodb://*****:*****@ds139322.mlab.com:39322/net_photographs'
client = pymongo.MongoClient(URI)
db = client['net_photographs']
simulation = db.sessions
config = db.config
my_photo_id = {'photoID': 8}
my_friend_id = {'photoID': 2}
my_file = open("log.txt", "w")
cap = cv2.VideoCapture(0)
cap.set(3, 320)
cap.set(4, 240)
pygame.mixer.init()
pygame.mixer.music.load("8 Dostoevsky.mp3")
face_time = 0
# init face_time from DB
cursor = simulation.find(my_photo_id)
for doc in cursor:
face_time = doc['accumulateViewersPerDay']
# init bridge ip
cursor_ip = config.find({'id': 1})
for doc in cursor_ip:
ip = doc['bridgeIP']
prev_faces = 0
prev_power = 0
sound_time = 0
my_friend_faces = 0
min_volum = volum = 0.05
max_volum = 1.0
volum_jump = 0.336
try:
while True:
# Create the haar cascade xml file
face_cascade = cv2.CascadeClassifier(CASC_PATH)
# Loop until the camera is working
rval = False
while not rval:
# Read the image
(rval, image) = cap.read()
if not rval:
cap.release()
cap = cv2.VideoCapture(0)
cap.set(3, 320)
cap.set(4, 240)
print("Failed to open webcam. Trying again...")
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
# Detect faces in the image
faces = face_cascade.detectMultiScale(
gray,
scaleFactor=1.1,
minNeighbors=5,
minSize=(30, 30)
# flags = cv2.CV_HAAR_SCALE_IMAGE
)
cursor_friend = simulation.find(my_friend_id)
for doc in cursor_friend:
my_friend_faces = doc['currentViewers']
# call light func with the photo's properties
(prev_faces, prev_power) = light.change_light(
prev_faces, (len(faces) + (my_friend_faces / 2)),
prev_power, ip)
simulation.update(my_photo_id,
{'$set': {
'currentLightning': prev_power
}})
# update curr faces on db
simulation.update(my_photo_id,
{'$set': {
'currentViewers': len(faces)
}})
# to do if there are ppl
if len(faces) > 0:
face_time += len(faces)
# if no recording is on, play
if pygame.mixer.music.get_busy() == 0:
pygame.mixer.music.play()
# inc FaceTime & update
simulation.update(
my_photo_id,
{'$set': {
'accumulateViewersPerDay': face_time
}})
# calculate volume power
volum = len(faces) * volum_jump
# logs
lastUpdate = log.log(faces, face_time, my_file)
simulation.update(my_photo_id,
{'$set': {
'lastUpdate': lastUpdate
}})
# draw a rectangle around the faces
#for (x, y, w, h) in faces:
#cv2.rectangle(image, (x, y), (x+w, y+h), (0, 255, 0), 2)
# calibrate if overflow
if volum > max_volum:
volum = max_volum
if volum < min_volum:
volum = min_volum
pygame.mixer.music.set_volume(volum)
#cv2.imshow("Faces found", image)
k = cv2.waitKey(1) & 0xFF
if k == 27:
break
except KeyboardInterrupt:
pass
except Exception as e:
print(e)
cap.release()
doomsDay.uponUs()
client.close()
cv2.destroyAllWindows()
cap.release()
my_file.close()
# Detect faces in the image
faces = faceCascade.detectMultiScale(gray,
scaleFactor=1.1,
minNeighbors=5,
minSize=(30, 30)
# flags = cv2.CV_HAAR_SCALE_IMAGE
)
facetime += len(faces)
simulation.update(myphotoid, {'$set': {'faceTime': facetime}})
soundtime += len(faces)
simulation.update(myphotoid, {'$set': {'prevLightPower': prevpower}})
# call light func that deside light power
light.change_light(prevfaces, len(faces), prevpower)
#simulation.update(myphotoid, {'$set': {'lightPower': power}})
simulation.update(myphotoid, {'$set': {'faces': len(faces)}})
simulation.update(myphotoid, {'$set': {'prevFaces': prevfaces}})
#if no recording is on, play
if len(faces) > 0:
if pygame.mixer.music.get_busy() == 0:
pygame.mixer.music.play()
#calculate volum power
simulation.update(myphotoid, {'$set': {'prevSoundPower': volum}})
volum = len(faces) * volumjump
#logs
rwfile.rw_file(faces, facetime, file)
