parachute.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "dying-benjamin",
   "metadata": {},
   "outputs": [],
   "source": [
    "from IPython.display import Image, display\n",
    "from math import pi\n",
    "import math\n",
    "from io import BytesIO\n",
    "import cairo\n",
    "\n",
    "# Specify values for the colours\n",
    "WHITE = (1,1,1)\n",
    "BLACK = (0,0,0)\n",
    "ONES = (183/256,110/256,35/256)\n",
    "ZEROS = (201/256,204/256,183/256)\n",
    "PADDING = (180/256,82/256,35/256)"
   ]
  },
  {
   "source": [
    "def draw_parachute(width=600, height=600, message=(\"DARE\", \"MIGHTY\", \"THINGS\", [\"N\",118,10,31,\"W\",34,11,58])):\n",
    "    if (isinstance(message, tuple) and len(message) == 4):\n",
    "        surface = cairo.ImageSurface(cairo.FORMAT_ARGB32, width, height)\n",
    "        ctx = cairo.Context(surface)\n",
    "\n",
    "        ctx.scale(width, height)\n",
    "        ctx.set_line_width(0.001)\n",
    "        \n",
    "        finished_clock = 0\n",
    "        max_clock = 80\n",
    "\n",
    "        for index, r in enumerate([(0.5, message[3]), (0.45,None), (0.40, message[2]), (0.35, message[1]), (0.2, message[0])]):\n",
    "            \n",
    "            if index == 1:\n",
    "                ctx.set_source_rgb(*BLACK)\n",
    "                ctx.arc(0.5,0.5,r[0],0, 2*math.pi)\n",
    "                ctx.fill_preserve()\n",
    "                ctx.set_source_rgb(*BLACK)\n",
    "                ctx.stroke()\n",
    "            else:\n",
    "        \n",
    "                starting_clock = finished_clock\n",
    "                ending_clock = (starting_clock + 80)%80\n",
    "                current_clock=finished_clock\n",
    "\n",
    "                def draw_slice(colour):\n",
    "                    ctx.set_source_rgb(*colour)\n",
    "                    ctx.move_to(0.5, 0.5)\n",
    "                    ctx.arc(0.5,0.5,r[0],((2*math.pi)/80)*current_clock,((2*math.pi)/80)*(current_clock+1))\n",
    "                    ctx.fill_preserve()\n",
    "                    ctx.set_source_rgb(*BLACK)\n",
    "                    ctx.stroke()\n",
    "\n",
    "                def draw_gap():\n",
    "                    nonlocal current_clock\n",
    "                    for i in range(3):\n",
    "                        draw_slice(WHITE)\n",
    "                        current_clock += 1\n",
    "                        current_clock = current_clock % 80\n",
    "\n",
    "                def fill_circle():\n",
    "                    nonlocal current_clock\n",
    "                    nonlocal max_clock\n",
    "                    while (current_clock+3)%80 != ending_clock:\n",
    "                        draw_slice(PADDING)\n",
    "                        current_clock+= 1\n",
    "                        current_clock = current_clock % 80\n",
    "                    draw_gap()\n",
    "\n",
    "\n",
    "\n",
    "                for c in r[1]:\n",
    "                    if isinstance(c, str):\n",
    "                        binary = int.from_bytes(c.encode(), 'big')\n",
    "                        binary -= 64\n",
    "                    elif isinstance(c, int):\n",
    "                        binary = c\n",
    "                    binary_str = f\"{binary:07b}\"\n",
    "                    for i in binary_str:\n",
    "                        if i == '0':\n",
    "                            draw_slice(ZEROS)\n",
    "                        elif i == '1':\n",
    "                            draw_slice(ONES)\n",
    "                        else:\n",
    "                            print(i)\n",
    "\n",
    "                        current_clock += 1\n",
    "                        current_clock = current_clock % 80\n",
    "\n",
    "                    draw_gap()\n",
    "                finished_clock = current_clock\n",
    "                if current_clock != ending_clock:\n",
    "                    fill_circle()\n",
    "\n",
    "        with BytesIO() as fileobj:\n",
    "            surface.write_to_png(fileobj)\n",
    "            display(Image(fileobj.getvalue(), width=width+100, height=height+100))\n"
   ],
   "cell_type": "code",
   "metadata": {},
   "execution_count": null,
   "outputs": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "draw_parachute(message=(\"test\", \"test\", \"test\", [9,9,9,9,9,9,9,9]))\n",
    "draw_parachute()"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "env",
   "language": "python",
   "name": "env"
  },
  "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.5-final"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}