Initial commit

.envrc

@@ -0,0 +1 @@
+use nix

.gitattributes

@@ -0,0 +1,2 @@
+*.deb filter=lfs diff=lfs merge=lfs -text
+*.wav filter=lfs diff=lfs merge=lfs -text

boot.txt

@@ -0,0 +1,63 @@
+fdt_addr=0x43000000
+image=Image
+script=boot.scr
+mmc_root="PARTUUID=70672ec3-5eee-49ff-b3b1-eb1fbd406bf5"
+sd_root="/dev/mmcblk1p2"
+
+# Check boot device to decide which root partition to use
+if test ${bootdev} = 1; then
+    setenv root ${sd_root}
+else
+    setenv root ${mmc_root}
+fi
+
+if test ${baseboardid} = 1; then
+    setenv fdt_file fsl-imx8mq-phanbell.dtb
+elif test ${baseboardid} = 2; then
+    setenv fdt_file fsl-imx8mq-yorktown.dtb
+else
+    # EVT Boards have quirky baseboard ID resistors, so fall back
+    # to phanbell if we're unsure what the board is.
+    echo "Unknown baseboard revision, selecting phanbell dtb."
+    setenv fdt_file fsl-imx8mq-phanbell.dtb
+fi
+
+cmdline="console=ttymxc0,115200 console=tty0 earlycon=ec_imx6q,0x30860000,115200 root=${root} rootfstype=ext4 rw rootwait init=/sbin/init net.ifnames=0 pci=pcie_bus_perf"
+
+ext2load mmc ${bootdev}:1 ${loadaddr} ${image}
+setenv bootargs ${cmdline} ${extra_bootargs}
+
+#Load base device tree.
+ext2load mmc ${bootdev}:1 ${fdt_addr} ${fdt_file}
+fdt addr ${fdt_addr}
+fdt resize 4096 # PORTILOOP: Add extra 4096 bytes to fit overlay
+
+# Load overlays to arbitrary address beyond fdt location (will apply them later).
+setexpr fdtovaddr ${fdt_addr} + F000
+
+# Parse overlays.txt and apply individual overlays.
+if ext2load mmc ${bootdev}:1 ${fdtovaddr} overlays.txt && env import -t ${fdtovaddr} ${filesize} && test -n ${overlay}; then
+	echo loaded overlay.txt: ${overlay}
+	for ov in ${overlay}; do
+		if ext2load mmc ${bootdev}:1 ${fdtovaddr} ${ov}.dtbo && fdt apply ${fdtovaddr}; then
+			echo overlay of ${ov} successful
+		else
+			echo overlay failed, reverting to original device tree
+			ext2load mmc ${bootdev}:1 ${fdt_addr} ${fdt_file}
+			booti ${loadaddr} - ${fdt_addr}
+		fi
+	done
+fi
+
+if printenv m4enabled; then
+	if test ${m4enabled} = 1; then
+		if ext2load mmc ${bootdev}:1 ${fdtovaddr} fsl-imx8mq-phanbell-m4.dtbo && fdt apply ${fdtovaddr}; then
+			echo M4 overlay loaded, starting core
+			run m4boot
+		else
+			echo M4 overlay failed
+		fi
+	fi
+fi
+
+booti ${loadaddr} - ${fdt_addr}

commands.md

@@ -0,0 +1,5 @@
+# Build scr
+sudo mkimage -A arm -T script -O linux -d boot.txt /boot/boot.scr
+
+# Edit device tree overlay
+sudo vim /boot/portiloop.dts && sudo dtc -I dts -O dtb -o /boot/portiloop.dtbo /boot/portiloop.dts

demo.py

@@ -0,0 +1,115 @@
+from periphery import GPIO, PWM, SPI, I2C
+from spidev import SpiDev
+from time import sleep
+
+led1_R = PWM(0, 0)
+led1_B = PWM(2, 0)
+led1_gnd = PWM(1, 0)
+
+led2_R = GPIO("/dev/gpiochip0", 8, "out")
+led2_B = GPIO("/dev/gpiochip0", 6, "out")
+led2_gnd = GPIO("/dev/gpiochip0", 7, "out")
+
+led3_R = GPIO("/dev/gpiochip2", 0, "out")
+#led3_B = GPIO("/dev/gpiochip2", 5, "out")
+led3_gnd = GPIO("/dev/gpiochip2", 8, "out")
+
+aquisition = GPIO("/dev/gpiochip2", 20, "out")
+
+frontend_nrst = GPIO("/dev/gpiochip2", 9, "out")
+frontend_npwdn = GPIO("/dev/gpiochip2", 12, "out")
+frontend_start = GPIO("/dev/gpiochip3", 29, "out")
+frontend_ = SpiDev()
+frontend_.open(0, 0)
+frontend_.max_speed_hz = 8000000
+frontend_.mode = 0b01
+
+audio = I2C("/dev/i2c-1")
+AUDIO_ADDR = 0x68
+
+aquisition.write(True)
+sleep(0.5)
+aquisition.write(False)
+sleep(0.5)
+aquisition.write(True)
+
+try:
+  frontend_npwdn.write(True)
+  frontend_start.write(False)
+  sleep(0.1)
+  frontend_nrst.write(False)
+  sleep(0.01)
+  frontend_nrst.write(True)
+  sleep(0.1)
+  frontend_.xfer([0x11])
+
+  data_out = [0x20, 0x00, 0x00]
+  data_in = frontend_.xfer(list(data_out))
+
+  assert data_in[2] == 0x3E, "Wrong output"
+  print("EEG Frontend responsive")
+finally:
+  frontend_.close()
+
+try:
+  for reg in [0x0000, 0x0002, 0x0004, 0x0006, 0x000A, 0x000E, 0x0010, 0x0014, 0x0020, 0x0022, 0x0024, 0x0026, 0x0028, 0x002A, 0x002C, 0x002E, 0x0030, 0x0032, 0x0034, 0x0036, 0x0038, 0x003A, 0x003C, 0x0100, 0x0102, 0x0104, 0x0106, 0x0108, 0x010A, 0x010C, 0x010E, 0x0110, 0x0116]:
+    msgs = [I2C.Message([reg >> 8, reg & 0xFF]), I2C.Message([0x00, 0x00], read=True)]
+    audio.transfer(AUDIO_ADDR, msgs)
+    print("0x{:04x}: 0x{:02x} 0x{:02x}".format(reg, msgs[1].data[0], msgs[1].data[1]))
+finally:
+  audio.close()
+
+state = 0
+try:
+  led1_R.frequency = 1e3
+  led1_R.enable()
+  led1_B.frequency = 1e3
+  led1_B.enable()
+  led1_gnd.frequency = 1e3
+  led1_gnd.enable()
+  led1_gnd.duty_cycle = 0.0
+  for i in range(200):
+    led1_R.duty_cycle = (state % 10) / 10
+    led1_B.duty_cycle = (state // 10) / 10
+    state = (state + 1) % 100
+    sleep(0.02)
+finally:
+  led1_R.duty_cycle = 0.0
+  led1_R.close()
+  led1_B.duty_cycle = 0.0
+  led1_B.close()
+  led1_gnd.duty_cycle = 0.0
+  led1_gnd.close()
+
+state = 0
+try:
+  led2_gnd.write(False)
+  for i in range(12):
+    led2_R.write(state % 2 == 1)
+    led2_B.write(state // 2 == 1)
+    state = (state + 1) % 4
+    sleep(0.2)
+finally:
+  led2_R.write(False)
+  led2_R.close()
+  led2_B.write(False)
+  led2_B.close()
+  led2_gnd.write(False)
+  led2_gnd.close()
+
+state = 0
+try:
+  led3_gnd.write(False)
+  for i in range(12):
+    state = 3
+    led3_R.write(state % 2 == 1)
+    #led3_B.write(state // 2 == 1)
+    state = (state + 1) % 4
+    sleep(0.2)
+finally:
+  led3_R.write(False)
+  led3_R.close()
+  #led3_B.write(False)
+  #led3_B.close()
+  led3_gnd.write(False)
+  led3_gnd.close()

linux-image-4.14.98-imx_12-4_arm64.deb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:74d6a69d76d03fb59bffd60d6c1d3229bb9307f5e64a1f2c8af6ce2233dffa0d
+size 11093740

net.py

@@ -0,0 +1,40 @@
+import tensorflow as tf
+import tensorflow.keras.layers as layers
+
+input_shape = (1,54,1)
+
+model = tf.keras.models.Sequential()
+model.add(layers.Conv1D(31, 7, activation='relu', input_shape=input_shape[1:]))
+model.add(layers.MaxPooling1D(7, data_format='channels_first'))
+model.add(layers.Conv1D(31, 7, activation='relu'))
+model.add(layers.MaxPooling1D(7, data_format='channels_first'))
+model.add(layers.Conv1D(31, 7, activation='relu'))
+model.add(layers.MaxPooling1D(7, data_format='channels_first'))
+model.add(layers.GRU(7))
+model.add(layers.Dense(18))
+
+model.compile(optimizer='sgd', loss='mse')
+
+print("\n===============\n")
+
+x = tf.random.normal(input_shape)
+#print("x: ", x)
+
+y = model.evaluate(x)
+#print("y: ", y)
+
+model = layers.Conv1D(31, 7, activation='relu', input_shape=input_shape[1:])(x)
+print(model.shape)
+model = layers.MaxPooling1D(7, data_format='channels_first')(model)
+print(model.shape)
+model = layers.Conv1D(31, 7, activation='relu')(model)
+print(model.shape)
+model = layers.MaxPooling1D(7, data_format='channels_first')(model)
+print(model.shape)
+model = layers.Conv1D(31, 7, activation='relu')(model)
+print(model.shape)
+model = layers.MaxPooling1D(7, data_format='channels_first')(model)
+print(model.shape)
+model = layers.GRU(7)(model)
+print(model.shape)
+

overlays.txt

@@ -0,0 +1,2 @@
+# List of device tree overlays to load. Format: overlay=<dtbo name, no extenstion> <dtbo2> ...
+overlay=portiloop

portiloop.dts

@@ -0,0 +1,141 @@
+/dts-v1/;
+/plugin/;
+
+///include/ "fsl-imx8mq-phanbell.dtb";
+
+/ {
+	compatible = "fsl,imx8mq-phanbell", "fsl,imx8mq";
+
+	fragment@0 {
+		target-path = "/hdmi@32c00000";
+		__overlay__ {
+			status = "disabled";
+		};
+	};
+
+	fragment@1 {
+		target-path = "/dcss@0x32e00000";
+		__overlay__ {
+			status = "disabled";
+		};
+	};
+
+	fragment@2 {
+		target-path = "/mipi_csi1@30a70000";
+		__overlay__ {
+			status = "disabled";
+		};
+	};
+
+	fragment@3 {
+		target-path = "/mipi_csi2@30b60000";
+		__overlay__ {
+			status = "disabled";
+		};
+	};
+
+	fragment@4 {
+		target-path = "/dsi_phy@30A00300";
+		__overlay__ {
+			status = "disabled";
+		};
+	};
+
+	fragment@5 {
+		target-path = "/mipi_dsi@30A00000";
+		__overlay__ {
+			status = "disabled";
+		};
+	};
+
+	fragment@6 {
+		target-path = "/csi1_bridge@30a90000";
+		__overlay__ {
+			status = "disabled";
+		};
+	};
+
+	fragment@7 {
+		target-path = "/sound-rt5645";
+		__overlay__ {
+			status = "disabled";
+		};
+	};
+
+	fragment@8 {
+		target-path = "/i2c@30a40000/rt5645@1a/";
+		__overlay__ {
+			status = "disabled";
+		};
+	};
+
+	fragment@9 {
+		target-path = "/i2c@30a30000";
+		__overlay__ {
+			status = "okay";
+			
+			sgtl5000: audio-codec@a {
+				#sound-dai-cells = <0>;
+				compatible = "fsl,sgtl5000";
+				reg = <0x0a>;
+
+				clocks = <&clk IMX8MQ_CLK_SAI2_ROOT>;
+				assigned-clocks = <&clk IMX8MQ_CLK_SAI2>;
+				assigned-clock-parents = <&clk IMX8MQ_AUDIO_PLL1_OUT>;
+				assigned-clock-rates = <24576000>;
+
+				status = "okay";
+			};
+		};
+	};
+
+	fragment@a {
+		target-path = "/sai@308b0000";
+		__overlay__ {
+			#sound-dai-cells = <0>;
+			fsl,sai-mclk-direction-output;
+		};
+	};
+
+	fragment@b {
+		target-path = "/";
+		__overlay__ {
+			sound-sgtl5000 {
+				compatible = "simple-audio-card";
+				
+				simple-audio-card,name = "sgtl5000";
+				simple-audio-card,format = "i2s";
+				simple-audio-card,widgets = "Headphone", "Headphones";
+				simple-audio-card,routing = "Headphones", "HPOUTL",
+				                            "Headphones", "HPOUTR";
+
+				status = "okay";
+		
+				simple-audio-card,cpu {
+					sound-dai = <&sai2>;
+				};
+		
+				audio_codec: simple-audio-card,codec {
+					sound-dai = <&sgtl5000>;
+					clocks = <&clk IMX8MQ_CLK_SAI2_ROOT>;
+					frame-master;
+					bitclock-master;
+				};
+			};
+		};
+	};
+
+	fragment@c {
+		target-path = "/dummy-dai";
+		__overlay__ {
+			status = "disabled";
+		};
+	};
+
+	fragment@d {
+		target-path = "/sound-header";
+		__overlay__ {
+			status = "disabled";
+		};
+	};
+};

shell.nix

@@ -0,0 +1,17 @@
+with import <nixpkgs> {};
+mkShell {
+  name = "tensorflow-cuda-shell";
+  buildInputs = with python3.pkgs; [
+    pip
+    setuptools
+    python39Packages.tensorflow
+    python39Packages.numpy
+  ];
+  shellHook = ''
+    export LD_LIBRARY_PATH=${pkgs.stdenv.cc.cc.lib}/lib:$LD_LIBRARY_PATH
+    alias pip="PIP_PREFIX='$(pwd)/_build/pip_packages' TMPDIR='$HOME' \pip"
+    export PYTHONPATH="$(pwd)/_build/pip_packages/lib/python3.7/site-packages:$PYTHONPATH"
+    export PATH="$(pwd)/_build/pip_packages/bin:$PATH"
+    unset SOURCE_DATE_EPOCH
+  '';
+}