Untitled9.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "<Context <Color {'mode': 'rgba', 'value': ((0, 0, 0, 1),)}> <Color {'mode': 'rgba', 'value': ((1, 1, 1, 1),)}>>\n",
      "<Context <Color {'mode': 'rgba', 'value': (0, 0, 1, 1)}> <Color {'mode': 'rgba', 'value': (1, 1, 0, 1)}>>\n",
      "<Context <Color {'mode': 'rgba', 'value': ((0, 0, 0, 1),)}> <Color {'mode': 'rgba', 'value': ((1, 1, 1, 1),)}>>\n"
     ]
    }
   ],
   "source": [
    "class Color:\n",
    "    def __init__(self, context, *args, cm_restore=None):\n",
    "        self._state = {\"mode\": \"rgba\", \"value\": args}\n",
    "        self._cm_restore = cm_restore\n",
    "        self._context = context\n",
    "       \n",
    "    def __enter__(self):\n",
    "        return self._context\n",
    "    \n",
    "    def __exit__(self, *args):\n",
    "        if self._cm_restore:\n",
    "            attr, value = self._cm_restore\n",
    "            setattr(self._context, attr, value)\n",
    "    \n",
    "    def __repr__(self):\n",
    "        return f\"<{self.__class__.__name__} {self._state}>\"\n",
    "\n",
    "import math\n",
    "\n",
    "IDENITY_MATRIX = [\n",
    "    1, 0, 0,\n",
    "    0, 1, 0,\n",
    "    0, 0, 0,\n",
    "]\n",
    "\n",
    "class TransformStack:\n",
    "    def __init__(self, context):\n",
    "        self._context = context\n",
    "        self._stack = []\n",
    "        self._matrix = IDENITY_MATRIX\n",
    "\n",
    "    def push(self, matrix):\n",
    "        self._stack[-1] = self._matrix\n",
    "        self._stack.append(matrix)\n",
    "        \n",
    "    def pop(self):\n",
    "        del self._stack[-1]\n",
    "        self._transform = self._stack[-1]\n",
    "    \n",
    "    def skew(self):\n",
    "        pass\n",
    "    \n",
    "    def translate(self, x, y):\n",
    "        pass\n",
    "    \n",
    "    def rotate(self, degrees=None, radians=None):\n",
    "        if degrees is not None:\n",
    "            radians = degrees * math.tau\n",
    "            \n",
    "        cos_theta = math.cos(radians)\n",
    "        sin_theta = math.sin(radians)\n",
    "        \n",
    "        return [\n",
    "            cos_theta, sin_theta, 0,\n",
    "            sin_theta, -cos_theta, 0,\n",
    "            0, 0, 1,\n",
    "        ]\n",
    "\n",
    "\n",
    "class Context:\n",
    "    _state_vars = [\"_fill\", \"_stroke, _transform\"]\n",
    "    def __init__(self):\n",
    "        self._fill = Color(self, (1, 1, 1, 1))\n",
    "        self._stroke = Color(self, (0, 0, 0, 1))\n",
    "        self._transform = TransformStack(self)\n",
    "        self._position = (0, 0)\n",
    "\n",
    "    def move_to(self, *args):\n",
    "        if self._path is None:\n",
    "            self._transform.move_to(*args)\n",
    "        else:\n",
    "            self._path.move_to(*args)\n",
    "    \n",
    "    def rect(self, width, height):\n",
    "        pass\n",
    "    \n",
    "    def fill(self, *args):\n",
    "        if not len(args):\n",
    "            return self._fill\n",
    "        self._fill = Color(self, *args, cm_restore=(\"_fill\", self._fill))\n",
    "        return self._fill\n",
    "    \n",
    "    def stroke(self, *args):\n",
    "        if not len(args):\n",
    "            return self._stroke\n",
    "        self._stroke = Color(self, *args, cm_restore=(\"_stroke\", self._stroke))\n",
    "        return self._stroke\n",
    "    \n",
    "    def __repr__(self):\n",
    "        return f\"<{self.__class__.__name__} {self._stroke} {self._fill}>\"\n",
    "\n",
    "    \n",
    "ctx = Context()\n",
    "print(ctx)\n",
    "\n",
    "with ctx.fill(1, 1, 0, 1), ctx.stroke(0, 0, 1, 1) as c:\n",
    "    print(ctx)\n",
    "    \n",
    "print(ctx)"
   ]
  }
 ],
 "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
}