Matrices with Affine.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "%reload_ext cairo_jupyter"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAIAAAABACAIAAABdtOgoAAAABmJLR0QA/wD/AP+gvaeTAAAC30lEQVR4nO3aPyi8cRzA8ffXv/xJ/qRIVwZFyvLUDQZXDllMFoMsFoMymIyMFossFosMFouB8nCRhKIsdFeKmGQgJI7n+Q3n1/WT/56fT89zn1c33HT3+X7eV/dc9+C6/Pyxt0dFBUBLS8v19bWrPs2D7aceBwdUVwNEIpGrqyvpc/mGZwFcl3icUAggHA5fXFxIH80fvAzguhwfU1sLYFnW+fm59Ol8wLgu3jo9pb2dRILGxsbl5eWqqiqP3yBYsqQHyHQeBzg5IRolkcCyrFgsph//D3kZIB6nuZmjI8LhsG3bFakrU/UuzwIcHtLaytkZkUhkdXW1vLzcq1cOOE8ufnZ3n3+IRaNR/SH2JR5sf2eH1Me9s7Pz7u5O+kQ+89Ptr61RXGyA7u7uh4cH6eP4z4+2v7hIQQFAT09PMpmUPosvfX/7Cwvk5wP09/c/PT1JH8Svvrn9uTlycwEGBgYcx5E+hY9956JodpacHIDh4WHp+X3vywGmpsjKAhgdHZUePghIP/tEgMlJjMEYMz4+Ljh0kHwhwNgYgDFmYmJCcOKA+WyAkRGA7Ozs6elpwXGD5+MAjsPQ0PP2Z2ZmBGcNpA8COA6DgwB5eXnz8/OCgwbVewEeH+nrAygsLFhaWhKcMsDeDJBM0tsLUFRUaNu24IjB9nqA+3u6ugBKSko2NzcF5wu8VwLc3tLRAVBWVra9vS04XCZ4GeDmhrY2gMrKyv39fcHJMoRx/67eGAM0NbG1RSgUsm27vr7+rf/RlFdeBgBqampWVlZqUzdYqf/s5Z/ydXV1Gxsbuv1f80+AhoaGWCwWSt3gqX5FOoBlWevr69WpW5zVb0l/B1xeXpaWlspOk4HSAZQIvTlXmAYQpgGEaQBhGkCYBhCmAYRpAGEaQJgGEKYBhGkAYRpAmAYQpgGEaQBhGkCYBhCmAYRpAGEaQJgGEKYBhGkAYRpAmAYQpgGEaQBhGkCYBhD2B4OX0KUc3eqiAAAAAElFTkSuQmCC\n",
      "text/plain": [
       "<cairo.ImageSurface at 0x7fa00483e370>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAIAAAABACAIAAABdtOgoAAAABmJLR0QA/wD/AP+gvaeTAAAC+klEQVR4nO3az0oyURjH8Z/D62YITNyLtWontJE2LYQSIopIFxlkQlB4Ad2Nt9CiNiL0b1ZJBt1AUBcQZOmik/m8iwHT0il15jyHej60S5o5z3fOaRaCzNBsNhcXFwHE47i7A1EgP60WANi2zb3cDxYM0Gq1VlZWHMeJx3F+jtnZoC709gYA4XA4qAuM7h/3DfRN/+ICMzMBXsvAAMw7QOf0IQE+0Tx9SIBe+qcPCdDFMn1IANfz8/PS0pLjOIkEHMfn6SuFnR3U64N/a2AAaH7tbTQaCwsLABIJ3N/7/Jr/+oq1NQCYnsb19YAP3NwAwPz8vOZVe9Aa4OXlxZ1+PB7g9C3LGtbg6goAUqmUzlV703oE2bY9NzcH4PEx9PDg519WCrkcjo8Ri8VqtVo2m316wvJy6NNZJEcQdTqdg4MDALaNszPfnv31dQCIRqP1ep2I2u12Pp8HEImgVvv45OkpAKTTac2r9qA7APnd4Ov0XQMbVCoAkMlk9K96GIYA5F+DYdN3fW1wcgIAq6urLKseiCcA+dHAe/qudru9tbXVbXB0BAAbGxuaF+uBLQARvb+/7+7uApiaCl1ejvnOE4vFbm9vPa6ilMpmswCi0dDhIQDkcjlta/wWZwAadx/85Nnv1d0Hru3tbQ1L+yHmADT6Pvj5s9+ruw8AFAqFIBc0Gv4A1NcA3g3Gm75LKbW5uQlgb28voIWMwYgARNTpdPb3973PolFPnq/cs6hUKvl+/2MzJQB9dxZN8uz3UkpVq1Ufb3tCBgWg4ftg8mffWGYFoEENfvH0ycAA1N+gUvnN0yczA1DP/wPXhOe+yQwNQD0NfvH0iYj/e0HDWJZVLpcjkUixWEwmk9y3E5QQEXHfw59mxFcT/zIJwEwCMJMAzCQAMwnATAIwkwDMJAAzCcBMAjCTAMwkADMJwEwCMJMAzCQAMwnATAIwkwDMJAAzCcBMAjCTAMwkADMJwEwCMJMAzCQAMwnATAIwkwDMJAAzCcBMAjCTAMz+A7sXj5Nc2HbPAAAAAElFTkSuQmCC\n",
      "text/plain": [
       "<cairo.ImageSurface at 0x7fa00483e370>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "import cairo\n",
    "import math\n",
    "\n",
    "from affine import Affine\n",
    "from collections import namedtuple\n",
    "\n",
    "surface = cairo.ImageSurface(cairo.FORMAT_ARGB32, 128, 64)\n",
    "context = cairo.Context(surface)\n",
    "\n",
    "\n",
    "def class_repr(self):\n",
    "    \"\"\"\n",
    "    Default class repr that outputs the content of a class.\n",
    "    \"\"\"\n",
    "    members = ', '.join({f'{k}={v}'\n",
    "                         for k, v in \n",
    "                         self.__dict__.items()})\n",
    "    return f'{self.__class__.__name__}({members})'\n",
    "\n",
    "\n",
    "class Path:\n",
    "    def __init__(self):\n",
    "        self.elements = []\n",
    "        self.closed = False\n",
    "        \n",
    "    def line_to(self, *args):\n",
    "        assert len(args) == 2\n",
    "        \n",
    "        self.elements.append((\"line_to\", args))\n",
    "        \n",
    "    def move_to(self, *args):\n",
    "        assert len(args) == 2\n",
    "        \n",
    "        self.elements.append((\"move_to\", args))\n",
    "        \n",
    "    def close(self):\n",
    "        self.closed = True\n",
    "        \n",
    "    def transform(self, m):\n",
    "        p = Path()\n",
    "        for t, pos in self.elements:\n",
    "            new_pos = m * pos\n",
    "            if t == \"line_to\":\n",
    "                p.line_to(*new_pos)\n",
    "            elif t == \"move_to\":\n",
    "                p.move_to(*new_pos)\n",
    "            else:\n",
    "                raise NotImplemented(t)\n",
    "            if self.closed:\n",
    "                p.close()\n",
    "        return p\n",
    "            \n",
    "    def draw(self, ctx):\n",
    "        for t, pos in self.elements:\n",
    "            if t == \"line_to\":\n",
    "                ctx.line_to(*pos)\n",
    "            elif t == \"move_to\":\n",
    "                ctx.move_to(*pos)\n",
    "            else:\n",
    "                raise NotImplemented(t)\n",
    "        if self.closed:\n",
    "            ctx.close_path()\n",
    "            \n",
    "            \n",
    "    __repr__ = class_repr\n",
    "\n",
    "\n",
    "def arrow(x, y, width, height, a, b):\n",
    "    p = Path()\n",
    "    p.move_to(x, y + b)\n",
    "    p.line_to(x, y + height - b)\n",
    "    p.line_to(x + a, y + height - b)\n",
    "    p.line_to(x + a, y + height)\n",
    "    p.line_to(x + width, y + height/2)\n",
    "    p.line_to(x + a, y)\n",
    "    p.line_to(x + a, y + b)\n",
    "    p.close()\n",
    "    return p\n",
    " \n",
    "def draw(path):\n",
    "    context.set_source_rgb(1, 1, 1)\n",
    "    #context.scale(10, 10)\n",
    "    context.paint()\n",
    "    context.set_source_rgba(1, 1, 0, 1)\n",
    "    path.draw(context)\n",
    "    context.fill_preserve()\n",
    "    context.set_source_rgba(0, 0, 0, 1)\n",
    "    context.stroke()\n",
    "    \n",
    "    display(surface)\n",
    "\n",
    "\n",
    "path = arrow(100, 10, 50, 50, 24, 14)\n",
    "\n",
    "\n",
    "path = arrow(-20, -20, 50, 50, 24, 14)\n",
    "draw(path)\n",
    "\n",
    "path2 = path.transform(Affine.rotation(-45) * Affine.translation(40, 40))\n",
    "\n",
    "#print(path2)\n",
    "\n",
    "draw(path2)\n",
    "\n",
    "#Matrix2D.Rotate(.1)"
   ]
  }
 ],
 "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.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}