schoolpost · matthewverde · Jun 15, 2024 · Jun 15, 2024 · Jun 17, 2024 · Jun 19, 2024
diff --git a/examples/analyzePixels.py b/examples/analyzePixels.py
@@ -0,0 +1,94 @@
+from pidng.core import RPICAM2DNG, DNGTags, Tag
+from pidng.camdefs import *
+import numpy as np
+import matplotlib.pyplot as plt
+
+# load raw image data from file into numpy array. RAW frame from HQ camera. 
+img = '../rpicam-raw-20-6-2024--13.0.24\BGGRss1200gain0.1---00000.raw'
+data = np.fromfile(img, dtype=np.uint8)
+# file size 4669440
+# (1520 height) * (((2028 width * 12 bits per pixel) / 8 bits per byte) + 30 extra bytes*) = 4669440 bytes
+# (((2028 * 12) / 8) + 30) = 3072
+# bytes*: strolls with dog claims it should only by 28 extra bytes https://www.strollswithmydog.com/open-raspberry-pi-high-quality-camera-raw/
+# tuple is in the form row, col
+# data = data.reshape((1520, 3072))
+data = data.reshape((3040, 6112))
+
+# choose a predefined camera model, set the sensor mode and bayer layout. 
+# this camera model class sets the appropriate DNG's tags needed based on the camera sensor. ( needed for bit unpacking, color matrices )
+# camera = RaspberryPiHqCamera(3, CFAPattern.BGGR)
+# [0, 0, 0, 0] is equal to a RRRR monochrome filter
+# in order for this to work, we need to rewrite the blue and green pixels
+camera = RaspberryPiHqCamera(3, [0, 0, 0, 0])
+
+# if self.mode == 1:
+#     width = 2028
+#     height = 1080
+# if self.mode == 2:
+#     width = 2028
+#     height = 1520
+# if self.mode == 3:
+#     width = 4056
+#     height = 3040
+# if self.mode == 4:
+#     width = 1012
+#     height = 760
+
+# example of adding custom DNG tags to predefined tags from camera model
+camera.tags.set(Tag.ApertureValue, [[4,1]])             # F 4.0
+camera.tags.set(Tag.ExposureTime, [[1,300]])             # SHUTTER 1/400
+camera.tags.set(Tag.PhotographicSensitivity, [1000])     # ISO 400
+camera.fmt = dict({
+    # tuple is in the form width height
+    'size': (4056,3040),
+    'bpp': 12,
+    'format': 'SRGGB12_CSI2P'
+})
+
+globalBlueArray = 0
+globalGreenArray = 0
+globalRedArray = 0
+def getColorArrays(rawFrame):
+    height, width = rawFrame.shape
+
+    blueArray = np.zeros((int(height / 2), int(width / 2)), dtype=np.uint16)
+    blueArray[:, :] = rawFrame[::2, ::2]
+    print(blueArray[1000,1000])
+
+    redArray = np.zeros((int(height / 2), int(width / 2)), dtype=np.uint16)
+    redArray[:, :] = rawFrame[1::2, 1::2]
+    print(redArray[1000,1000])
+
+    greenArray = np.zeros((int(height), int(width / 2)), dtype=np.uint16)
+    # get even rows (BG rows)
+    greenArray[::2, :] = rawFrame[::2, 1::2]
+    # get odd rows (GR rows)
+    greenArray[1::2, :] = rawFrame[1::2, ::2]
+
+    global globalBlueArray
+    globalBlueArray = blueArray
+
+    global globalRedArray
+    globalRedArray = redArray
+
+    global globalGreenArray
+    globalGreenArray = greenArray
+
+    return rawFrame
+
+# pass camera reference into the converter.
+r = RPICAM2DNG(camera)
+unpackedPixels = r.__unpack_pixels__(data)
+getColorArrays(unpackedPixels)
+
+plt.figure(figsize=(12, 8))
+# set blue green or red global array here
+plt.boxplot(globalGreenArray)
+plt.show()
+
+# B G B G B G
+# G R G R G R
+# B G B G B G
+# G R G R G R
+# B G B G B G
+# G R G R G R
diff --git a/examples/app.py b/examples/app.py
@@ -0,0 +1,62 @@
+from flask import Flask,request
+import subprocess
+
+app = Flask(__name__)
+
+wifi_device = "wlan0"
+
+@app.route('/')
+def index():
+    result = subprocess.check_output(["nmcli", "--colors", "no", "-m", "multiline", "--get-value", "SSID", "dev", "wifi", "list", "ifname", wifi_device])
+    ssids_list = result.decode().split('\n')
+    dropdowndisplay = f"""
+        <!DOCTYPE html>
+        <html>
+        <head>
+            <title>Wifi Control</title>
+        </head>
+        <body>
+            <h1>Wifi Control</h1>
+            <form action="/submit" method="post">
+                <label for="ssid">Choose a WiFi network:</label>
+                <select name="ssid" id="ssid">
+        """
+    for ssid in ssids_list:
+        only_ssid = ssid.removeprefix("SSID:")
+        if len(only_ssid) > 0:
+            dropdowndisplay += f"""
+                    <option value="{only_ssid}">{only_ssid}</option>
+            """
+    dropdowndisplay += f"""
+                </select>
+                <p/>
+                <label for="password">Password: <input type="password" name="password"/></label>
+                <p/>
+                <input type="submit" value="Connect">
+            </form>
+        </body>
+        </html>
+        """
+    return dropdowndisplay
+
+
+@app.route('/submit',methods=['POST'])
+def submit():
+    if request.method == 'POST':
+        print(*list(request.form.keys()), sep = ", ")
+        ssid = request.form['ssid']
+        password = request.form['password']
+        connection_command = ["nmcli", "--colors", "no", "device", "wifi", "connect", ssid, "ifname", wifi_device]
+        if len(password) > 0:
+          connection_command.append("password")
+          connection_command.append(password)
+        result = subprocess.run(connection_command, capture_output=True)
+        if result.stderr:
+            return "Error: failed to connect to wifi network: <i>%s</i>" % result.stderr.decode()
+        elif result.stdout:
+            return "Success: <i>%s</i>" % result.stdout.decode()
+        return "Error: failed to connect."
+
+
+if __name__ == '__main__':
+    app.run(debug=True, host='0.0.0.0', port=80)