1 files changed, 101 insertions, 0 deletions
diff --git a/rgbautocam.py b/rgbautocam.py
new file mode 100644
index 0000000..8e53cdd
--- /dev/null
+++ b/rgbautocam.py
@@ -0,0 +1,101 @@
+import cv2
+import numpy as np
+import random
+import namemaker3
+print 'Press q to quit while the video is playing'
+
+rec = raw_input('Record [y/n]? ').lower()[0] == 'y'
+
+single = ['np.cos', 'np.sin', 'np.tan'] #Operations on a single number
+binary = ['*', '+', '-'] #Operations for 2 numbers
+numlist = ['frame', 't', 'Constant']
+
+def randFunction(functionLength, singleweight, numberweight):
+    hasframe = False
+    hast = False
+    while not(hasframe and hast):
+        #Types: b for binary, s for single, f for first, n for number
+        function = ''
+        lasttype = 'f'
+        thistype = 0
+        hasframe = False
+        hast = False
+        chanceend = 0
+        length = 1 #Number of operations done so far
+        while True:
+            chanceend = (1.0 - (1.0 / length)) ** (float(functionLength)/7)
+            if lasttype == 'n':
+                number = random.random()
+                if number < chanceend:
+                    break
+                function = '(' + function + ')' + random.choice(binary)
+                lasttype = 'b'
+            elif lasttype == 's' or lasttype == 'b' or lasttype == 'f':
+                function += '('
+                thistype = random.random()
+                if thistype < singleweight / (singleweight + numberweight):
+                    function += random.choice(single)
+                    lasttype = 's'
+                else:
+                    what = random.choice(numlist)
+                    if what == 'Constant':
+                        function += str(random.gauss(150, 50))
+                    else:
+                        function += what
+                        if what == 'frame':
+                            hasframe = True
+                        elif what == 't':
+                            hast = True
+                    lasttype = 'n'
+                    function += ')'
+            length += 1
+    if function.count('(') > function.count(')'):
+        function += ')' * (function.count('(') - function.count(')'))
+    return function
+
+t = 0
+cap = cv2.VideoCapture(0)
+try:
+    w = cap.get(cv2.CV_CAP_PROP_FRAME_WIDTH)
+    h = cap.get(cv2.CV_CAP_PROP_FRAME_HEIGHT)
+except:
+    w = cap.get(cv2.cv.CV_CAP_PROP_FRAME_WIDTH)
+    h = cap.get(cv2.cv.CV_CAP_PROP_FRAME_HEIGHT)
+rfunction = randFunction(80, 1, 1)
+gfunction = randFunction(80, 1, 1)
+bfunction = randFunction(80, 1, 1)
+
+if rec:
+    try:
+        fourcc = cv2.cv.CV_FOURCC(*'XVID')
+    except:
+        fourcc = cv2.VideoWriter_fourcc(*'XVID')
+    out = cv2.VideoWriter(' '.join([namemaker3.generate() for i in range(3)]) + '.avi', fourcc, 10, (int(w),int(h)))
+
+
+while True:
+    t += 9.1
+    ret, fr = cap.read()
+    b, g, r = cv2.split(fr)
+    r = r.astype(float)
+    g = g.astype(float)
+    b = b.astype(float)
+    frame = r
+    r = eval(rfunction)
+    frame = g
+    g = eval(gfunction)
+    frame = b
+    b = eval(bfunction)
+    fr = np.dstack((r, g, b)).astype(np.uint8)
+    if rec:
+        out.write(fr)
+    cv2.imshow('Video', fr)
+    if cv2.waitKey(1) & 0xFF == ord('q'):
+            break
+
+cap.release()
+if rec:
+    out.release()
+cv2.destroyAllWindows()
+
+