3L_neonLights.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "abba9a71",
   "metadata": {},
   "source": [
    "## **Neonlights**\n",
    "\n",
    "**关键词：树莓派、Python、Scratch、CodeLab Adapter、NeoPixel（ws2812）**"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "a5d9d9f0",
   "metadata": {},
   "source": [
    "</br>\n"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "b1ab263a",
   "metadata": {},
   "source": [
    "### **P1：使用 Scratch 彩虹按钮调控灯光颜色**\n",
    "\n",
    "配合 Scratch 使用，在 Scratch 内按动某种颜色的按钮，灯光相应变化颜色。\n",
    "\n",
    "Scratch 项目[在此](https://create.codelab.club/projects/11063/)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "2a0652fc",
   "metadata": {},
   "outputs": [],
   "source": [
    "import time\n",
    "from codelab_adapter_client import AdapterNode\n",
    "from rpi_ws281x import PixelStrip, Color\n",
    "\n",
    "\n",
    "class MyNode(AdapterNode):\n",
    "    NODE_ID = \"linda/color\"\n",
    "\n",
    "    def __init__(self):\n",
    "        super().__init__()\n",
    "    \n",
    "\n",
    "node = MyNode()\n",
    "node.receive_loop_as_thread()\n",
    "time.sleep(0.1)\n",
    "\n",
    "# 下面是 NeoPixel 的基本参数，参考 rpi_ws281x 库的用例设置\n",
    "LED_COUNT = 64              # 灯珠数量\n",
    "LED_PIN = 18                # GPIO 18\n",
    "LED_FREQ_HZ = 800000        # 频率\n",
    "LED_DMA = 10\n",
    "LED_BRIGHTNESS = 200         # 亮度\n",
    "LED_INVERT = False\n",
    "LED_CHANNEL = 0\n",
    "\n",
    "strip = PixelStrip(LED_COUNT, LED_PIN, LED_FREQ_HZ, LED_DMA, LED_INVERT, LED_BRIGHTNESS, LED_CHANNEL)\n",
    "strip.begin()\n",
    "\n",
    "\n",
    "# 设计并定义一种霓虹效果\n",
    "def colorWipe(strip, color):\n",
    "    for i in range(strip.numPixels()):\n",
    "        strip.setPixelColor(i, color)\n",
    "        strip.show() \n",
    "\n",
    "# 持续等待 Scratch 项目通过 Linda 发送过来的颜色消息，灯光相应做出变化\n",
    "try: \n",
    "    while True:\n",
    "        lindaIn = node.linda_in([\"color\", \"*\"]) \n",
    "        color = lindaIn[1]\n",
    "        if color == 'red':\n",
    "            colorWipe(strip, Color(255, 0, 0))\n",
    "        if color == 'blue':\n",
    "            colorWipe(strip, Color(0, 0, 255))\n",
    "        if color == 'green':\n",
    "            colorWipe(strip, Color(0, 255, 0))\n",
    "        if color == 'purple':\n",
    "            colorWipe(strip, Color(255, 0, 255))\n",
    "        if color == 'yellow':\n",
    "            colorWipe(strip, Color(255, 255, 0))\n",
    "        if color == 'cyan':\n",
    "            colorWipe(strip, Color(0, 180, 100))\n",
    "        if color == 'violet':\n",
    "            colorWipe(strip, Color(187, 51, 133))\n",
    "        if color == 'orange':\n",
    "            colorWipe(strip, Color(255, 85, 0))\n",
    "\n",
    "except KeyboardInterrupt:\n",
    "    ColorWipe(strip, Color(0, 0, 0))\n",
    "        \n"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "690724a5",
   "metadata": {},
   "source": [
    "</br>"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "01f76eb7",
   "metadata": {},
   "source": [
    "### **P2：你想要什么颜色的棒棒糖？**\n",
    "\n",
    "配合 Scratch 使用，在 Scratch 中修改 RGB 三个数值，颜色信息会同步给 Python，进而的灯光颜色也相应变化。\n",
    "\n",
    "Scratch 项目[在此](https://create.codelab.club/projects/11067/)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d4e7d702",
   "metadata": {},
   "outputs": [],
   "source": [
    "import time\n",
    "from codelab_adapter_client import AdapterNode\n",
    "from rpi_ws281x import PixelStrip, Color\n",
    "\n",
    "\n",
    "\n",
    "class MyNode(AdapterNode):\n",
    "    NODE_ID = \"linda/color\"\n",
    "\n",
    "    def __init__(self):\n",
    "        super().__init__()\n",
    "    \n",
    "\n",
    "node = MyNode()\n",
    "node.receive_loop_as_thread()\n",
    "time.sleep(0.1)\n",
    "\n",
    "# 下面是 NeoPixel 的基本参数，参考 rpi_ws281x 库的用例设置\n",
    "LED_COUNT = 64              # 灯珠数量\n",
    "LED_PIN = 18                # GPIO 18\n",
    "LED_FREQ_HZ = 800000        # 频率\n",
    "LED_DMA = 10\n",
    "LED_BRIGHTNESS = 200         # 亮度\n",
    "LED_INVERT = False\n",
    "LED_CHANNEL = 0\n",
    "\n",
    "strip = PixelStrip(LED_COUNT, LED_PIN, LED_FREQ_HZ, LED_DMA, LED_INVERT, LED_BRIGHTNESS, LED_CHANNEL)\n",
    "strip.begin()\n",
    "\n",
    "# 持续等待 Scratch 项目通过 Linda 发送过来的颜色消息，灯光相应做出变化\n",
    "\n",
    "r = 255\n",
    "g = 255\n",
    "b = 255\n",
    "\n",
    "def ColorWipe(strip, color):\n",
    "    for i in range(strip.numPixels()):\n",
    "        strip.setPixelColor(i, color)\n",
    "        strip.show() \n",
    "        \n",
    "ColorWipe(strip, Color(r,g,b))\n",
    "\n",
    "\n",
    "try: \n",
    "    while True:\n",
    "        lindaIn = node.linda_in([\"Color\", \"*\", \"*\"]) \n",
    "        RGB = lindaIn[1]\n",
    "        num = int(lindaIn[2])\n",
    "        color = (r, g, b)\n",
    "        if RGB == 'R':\n",
    "            r = num\n",
    "        if RGB == 'G':\n",
    "            g = num\n",
    "        if RGB == 'B':\n",
    "            b = num\n",
    "        ColorWipe(strip, Color(r, g, b))\n",
    "\n",
    "\n",
    "except KeyboardInterrupt:\n",
    "    ColorWipe(strip, Color(0, 0, 0))"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "0056f8e8",
   "metadata": {},
   "source": [
    "</bm>\n"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "400aac88",
   "metadata": {},
   "source": [
    "### **P3：音乐节拍与灯光**\n",
    "\n",
    "配合 Toio，Scratch 使用。当把 Toio 放在垫子上的时候，会启动 Scratch 播放音乐，以下 Python 代码会实时分析音乐的节拍，同时把节拍结果反馈给灯光与 Scratch 舞台上的彩虹按钮，灯光闪烁，按钮跳动，二者颜色相同。\n",
    "\n",
    "Scratch 项目[在此](https://create.codelab.club/projects/11093/)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d2b00c28",
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import time\n",
    "import random \n",
    "from rpi_ws281x import PixelStrip, Color\n",
    "from madmom.features.beats import DBNBeatTrackingProcessor, RNNBeatProcessor\n",
    "from madmom.models import BEATS_LSTM\n",
    "from madmom.processors import IOProcessor, process_online\n",
    "from codelab_adapter_client import AdapterNode\n",
    "\n",
    "\n",
    "\n",
    "class MyNode(AdapterNode):\n",
    "    NODE_ID = \"eim/beats\"\n",
    "    \n",
    "    def __init__(self):\n",
    "        super().__init__()\n",
    "        self.is_ready = False\n",
    "        \n",
    "    def send_data(self, content):\n",
    "        message = self.message_template()\n",
    "        message[\"payload\"][\"content\"] = content\n",
    "        self.publish(message)\n",
    "        \n",
    "    def extension_message_handle(self, topic, payload):\n",
    "        self.logger.info(f'the message payload from scratch: {payload}')\n",
    "        content = payload[\"content\"]\n",
    "        if content == \"ready\":\n",
    "            self.is_ready = True\n",
    "        \n",
    "node = MyNode()\n",
    "node.receive_loop_as_thread()\n",
    "time.sleep(0.1)\n",
    "\n",
    "while not node.is_ready:\n",
    "    print(\"等待开始\")\n",
    "    time.sleep(0.5)\n",
    "\n",
    "print(\"go\")\n",
    "\n",
    "# 下面是 NeoPixel 的基本参数，参考 rpi_ws281x 库的用例设置\n",
    "LED_COUNT = 64              # 灯珠数量\n",
    "LED_PIN = 18                # GPIO 18\n",
    "LED_FREQ_HZ = 800000        # 频率\n",
    "LED_DMA = 10\n",
    "LED_BRIGHTNESS = 200         # 亮度\n",
    "LED_INVERT = False\n",
    "LED_CHANNEL = 0\n",
    "\n",
    "strip = PixelStrip(LED_COUNT, LED_PIN, LED_FREQ_HZ, LED_DMA, LED_INVERT, LED_BRIGHTNESS, LED_CHANNEL)\n",
    "strip.begin()\n",
    "\n",
    "\n",
    "\n",
    "kwargs = dict(\n",
    "    fps = 100,\n",
    "    correct = True,\n",
    "    infile = None,\n",
    "    outfile = None,\n",
    "    #max_bpm = 170,\n",
    "    #min_bpm = 100,\n",
    "    nn_files = [BEATS_LSTM[0],BEATS_LSTM[1], BEATS_LSTM[2]],\n",
    "    num_frames = 1,\n",
    "    online = True,\n",
    "    #verbose = 1\n",
    ")\n",
    "\n",
    "color_list = [\"red\", \"orange\", \"yellow\", \"green\", \"cyan\", \"blue\", \"purple\", \"violet\"]\n",
    "\n",
    "\n",
    "\n",
    "def ColorWipe(strip, color):\n",
    "    for i in range(strip.numPixels()):\n",
    "        strip.setPixelColor(i, color)\n",
    "    strip.show() \n",
    "        \n",
    "        \n",
    "def beat_callback(beats, output=None):\n",
    "    if len(beats) > 0:\n",
    "        #global n\n",
    "        # Do something with the beat (for now, just print the array to stdout)\n",
    "        #n = n + 1\n",
    "        color = random.choice(color_list)\n",
    "        print(color)\n",
    "        if color == 'red':\n",
    "            node.send_data(\"red\")\n",
    "            ColorWipe(strip, Color(255, 0, 0))\n",
    "        if color == 'blue':\n",
    "            node.send_data(\"blue\")\n",
    "            ColorWipe(strip, Color(0, 0, 255))\n",
    "        if color == 'green':\n",
    "            node.send_data(\"green\")\n",
    "            ColorWipe(strip, Color(0, 255, 0))\n",
    "            \n",
    "        if color == 'purple':\n",
    "            node.send_data(\"purple\")\n",
    "            ColorWipe(strip, Color(255, 0, 255))\n",
    "            \n",
    "        if color == 'yellow':\n",
    "            node.send_data(\"yellow\")\n",
    "            ColorWipe(strip, Color(255, 255, 0))\n",
    "            \n",
    "        if color == 'cyan':\n",
    "            node.send_data(\"cyan\")\n",
    "            ColorWipe(strip, Color(0, 180, 100))\n",
    "            \n",
    "        if color == 'violet':\n",
    "            node.send_data(\"violet\")\n",
    "            ColorWipe(strip, Color(187, 51, 133))\n",
    "            \n",
    "        if color == 'orange':\n",
    "            node.send_data(\"orange\")\n",
    "            ColorWipe(strip, Color(255, 85, 0))\n",
    "            \n",
    "        \n",
    "    \n",
    "in_processor = RNNBeatProcessor(**kwargs)\n",
    "beat_processor = DBNBeatTrackingProcessor(**kwargs)\n",
    "out_processor = [beat_processor, beat_callback]\n",
    "processor = IOProcessor(in_processor, out_processor)\n",
    "process_online(processor, **kwargs)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "c51ce5b6",
   "metadata": {},
   "source": [
    "</br>"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "f47de258",
   "metadata": {},
   "source": [
    "### **P4：Shinning for Toio**\n",
    "\n",
    "配合 Toio 和 Scratch 使用，使用 Scratch 积木控制 Toio 在垫子上移动，垫子被分为 5 行，7 列，Toio 的位置信息会每隔一段时间反馈给 Python，下面代码将灯光的颜色变化与位置信息做了映射，所以灯光会随着 Toio 在垫子上位置的改变而变化颜色。\n",
    "\n",
    "Scratch 项目[在此](https://create.codelab.club/projects/11124/)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "ccf94b49",
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import time\n",
    "import random \n",
    "from rpi_ws281x import PixelStrip, Color\n",
    "from codelab_adapter_client import AdapterNode\n",
    "\n",
    "class MyNode(AdapterNode):\n",
    "    NODE_ID = \"linda/color\"\n",
    "\n",
    "    def __init__(self):\n",
    "        super().__init__()\n",
    "    \n",
    "\n",
    "node = MyNode()\n",
    "node.receive_loop_as_thread()\n",
    "time.sleep(0.1)\n",
    "\n",
    "# 下面是 NeoPixel 的基本参数，参考 rpi_ws281x 库的用例设置\n",
    "LED_COUNT = 60              # 灯珠数量\n",
    "LED_PIN = 18                # GPIO 18\n",
    "LED_FREQ_HZ = 800000        # 频率\n",
    "LED_DMA = 10\n",
    "LED_BRIGHTNESS = 20         # 亮度\n",
    "LED_INVERT = False\n",
    "LED_CHANNEL = 0\n",
    "\n",
    "strip = PixelStrip(LED_COUNT, LED_PIN, LED_FREQ_HZ, LED_DMA, LED_INVERT, LED_BRIGHTNESS, LED_CHANNEL)\n",
    "strip.begin()\n",
    "\n",
    "\n",
    "colors = list(range(0, 255, 7))[0:35]\n",
    "\n",
    "def pos_generator(row, column):\n",
    "        if row == 0 and column == 2:\n",
    "            pos = int(colors[0])\n",
    "            return pos\n",
    "        if row == 0 and column == 1:\n",
    "            pos = int(colors[1])\n",
    "            return pos\n",
    "        if row == 0 and column == 0:\n",
    "            pos = int(colors[2])\n",
    "            return pos\n",
    "        if row == 0 and column == -1:\n",
    "            pos = int(colors[3])\n",
    "            return pos\n",
    "        if row == 0 and column == -2:\n",
    "            pos = int(colors[4])\n",
    "            return pos        \n",
    "        if row == 0 and column == -3:\n",
    "            pos = int(colors[5])\n",
    "            return pos        \n",
    "        if row == 0 and column == -4:\n",
    "            pos = int(colors[6])\n",
    "            return pos\n",
    "        if row == 1 and column == -4:\n",
    "            pos = int(colors[7])\n",
    "            return pos\n",
    "        if row == 2 and column == -4:\n",
    "            pos = int(colors[8])\n",
    "            return pos\n",
    "        if row == 3 and column == -4:\n",
    "            pos = int(colors[9])\n",
    "            return pos\n",
    "        if row == 4 and column == -4:\n",
    "            pos = int(colors[10])\n",
    "            return pos        \n",
    "        if row == 4 and column == -3:\n",
    "            pos = int(colors[11])\n",
    "            return pos          \n",
    "        if row == 4 and column == -2:\n",
    "            pos = int(colors[12])\n",
    "            return pos\n",
    "        if row == 4 and column == -1:\n",
    "            pos = int(colors[13])\n",
    "            return pos\n",
    "        if row == 4 and column == 0:\n",
    "            pos = int(colors[14])\n",
    "            return pos\n",
    "        if row == 4 and column == 1:\n",
    "            pos = int(colors[15])\n",
    "            return pos\n",
    "        if row == 4 and column == 2:\n",
    "            pos = int(colors[16])\n",
    "            return pos        \n",
    "        if row == 3 and column == 2:\n",
    "            pos = int(colors[17])\n",
    "            return pos        \n",
    "        if row == 2 and column == 2:\n",
    "            pos = int(colors[18])\n",
    "            return pos\n",
    "        if row == 1 and column == 2:\n",
    "            pos = int(colors[19])\n",
    "            return pos\n",
    "        if row == 1 and column == 1:\n",
    "            pos = int(colors[20])\n",
    "            return pos\n",
    "        if row == 1 and column == 0:\n",
    "            pos = int(colors[21])\n",
    "            return pos\n",
    "        if row == 1 and column == -1:\n",
    "            pos = int(colors[22])\n",
    "            return pos        \n",
    "        if row == 1 and column == -2:\n",
    "            pos = int(colors[23])\n",
    "            return pos         \n",
    "        if row == 1 and column == -3:\n",
    "            pos = int(colors[24])\n",
    "            return pos        \n",
    "        if row == 2 and column == -3:\n",
    "            pos = int(colors[25])\n",
    "            return pos        \n",
    "        if row == 3 and column == -3:\n",
    "            pos = int(colors[26])\n",
    "            return pos\n",
    "        if row == 3 and column == -2:\n",
    "            pos = int(colors[27])\n",
    "            return pos\n",
    "        if row == 3 and column == -1:\n",
    "            pos = int(colors[28])\n",
    "            return pos\n",
    "        if row == 3 and column == 0:\n",
    "            pos = int(colors[29])\n",
    "            return pos\n",
    "        if row == 3 and column == 1:\n",
    "            pos = int(colors[30])\n",
    "            return pos        \n",
    "        if row == 2 and column == 1:\n",
    "            pos = int(colors[31])\n",
    "            return pos                 \n",
    "        if row == 2 and column == 0:\n",
    "            pos = int(colors[32])\n",
    "            return pos\n",
    "        if row == 2 and column == -1:\n",
    "            pos = int(colors[33])\n",
    "            return pos        \n",
    "        if row == 2 and column == -2:\n",
    "            pos = int(colors[34])\n",
    "            return pos                 \n",
    "                  \n",
    "def wheel(pos):\n",
    "    \"\"\"Generate rainbow colors across 0-255 positions.\"\"\"\n",
    "    if pos < 85:\n",
    "        return Color(pos * 3, 255 - pos * 3, 0)\n",
    "    elif pos < 170:\n",
    "        pos -= 85\n",
    "        return Color(255 - pos * 3, 0, pos * 3)\n",
    "    else:\n",
    "        pos -= 170\n",
    "        return Color(0, pos * 3, 255 - pos * 3)\n",
    "\n",
    "\n",
    "def ColorWipe(strip, color):\n",
    "    for i in range(strip.numPixels()):\n",
    "        strip.setPixelColor(i, color)\n",
    "    strip.show()\n",
    "    \n",
    "ColorWipe(strip, Color(0,0,0))\n",
    "\n",
    "#node.linda_in([\"start\"])\n",
    "\n",
    "try: \n",
    "    while True:\n",
    "        lindaIn = node.linda_in([\"pos\", \"*\", \"*\"]) \n",
    "        row = int(lindaIn[1])\n",
    "        column = int(lindaIn[2])\n",
    "        pos = pos_generator(row, column)\n",
    "        ColorWipe(strip, wheel(pos))\n",
    "\n",
    "\n",
    "except KeyboardInterrupt:\n",
    "    ColorWipe(strip, Color(0, 0, 0))"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.2"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}