Alt State Stack-3.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "<bound method Context.fill of <__main__.Context object at 0x7fabf4521850>>\n",
      "<Color(ColorState: ())> <Color(ColorState: ())>\n"
     ]
    },
    {
     "ename": "AttributeError",
     "evalue": "'Context' object has no attribute 'skew'",
     "output_type": "error",
     "traceback": [
      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[0;31mAttributeError\u001b[0m                            Traceback (most recent call last)",
      "\u001b[0;32m<ipython-input-7-13f0edf559ad>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m\u001b[0m\n\u001b[1;32m    148\u001b[0m \u001b[0;32mwith\u001b[0m \u001b[0mctx\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mfill\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;36m1\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;36m0\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;36m0\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;36m1\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mctx\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mstroke\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;36m0\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    149\u001b[0m     \u001b[0mprint\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mctx\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mfill\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mctx\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mstroke\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 150\u001b[0;31m     \u001b[0;32mwith\u001b[0m \u001b[0mctx\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mskew\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;36m45\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    151\u001b[0m         \u001b[0mprint\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mctx\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mtransform\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    152\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;31mAttributeError\u001b[0m: 'Context' object has no attribute 'skew'"
     ]
    }
   ],
   "source": [
    "from affine import Affine\n",
    "\n",
    "class RestorableState:\n",
    "    def __init__(self, context, state, cm_raises_exception):\n",
    "        self.state = state\n",
    "        self.restore_state = None\n",
    "        self.context = context\n",
    "        self.cm_raises_exception = cm_raises_exception\n",
    "        \n",
    "    def __enter__(self):\n",
    "        if self.cm_raises_exception:\n",
    "            ex, *args = self.cm_raises_exception\n",
    "            raise ex(*args)\n",
    "        return self.context\n",
    "\n",
    "    def __exit__(self, *args):\n",
    "        self.state = self.restore_state\n",
    "        return False\n",
    "\n",
    "#     def __call__(self, *args):\n",
    "#         self.restore_state = self.state\n",
    "#         self.state = args\n",
    "#         return self #.color\n",
    "\n",
    "    def __repr__(self):\n",
    "        return f\"{self.__class__.__name__}: {self.state}\"\n",
    "\n",
    "    \n",
    "def normalize_color(args):\n",
    "    if args is None:\n",
    "        return None\n",
    "    \n",
    "    if len(args) == 1:\n",
    "        c = args[0]\n",
    "        return (c, c, c, 1)\n",
    "    elif len(args) == 2:\n",
    "        return (args[0], args[1], 1, 1)\n",
    "    elif len(args) == 3:\n",
    "        return (*args, 1)\n",
    "    else:\n",
    "        return args\n",
    "\n",
    "    \n",
    "# class ColorState(RestorableState):\n",
    "#     def __init__(self, context, color, cm_raises_exception=None):\n",
    "#         RestorableState.__init__(self, context, color, cm_raises_exception=cm_raises_exception)\n",
    "        \n",
    "#     def __call__(self, *args):\n",
    "#         color = normalize_color(args)\n",
    "#         return super().__call__(color)\n",
    "\n",
    "\n",
    "class ColorState(RestorableState):\n",
    "    def __init__(self, context, color, cm_raises_exception=None):\n",
    "        RestorableState.__init__(self, context, color, cm_raises_exception=cm_raises_exception)\n",
    "        \n",
    "#     def __call__(self, *args):\n",
    "#         color = normalize_color(args)\n",
    "#         return super().__call__(*color)\n",
    "    \n",
    "\n",
    "class Grob:\n",
    "    pass\n",
    "\n",
    "class Color(Grob):\n",
    "    def __init__(self, context, *args):\n",
    "        self._state = ColorState(context, normalize_color(args))\n",
    "        \n",
    "    def __enter__(self):\n",
    "        return Color(self._state.state)\n",
    "    \n",
    "    def __exit__(self, *args):\n",
    "        self._state = self._state.__exit__()\n",
    "        return False\n",
    "    \n",
    "    def __call__(self, *args):\n",
    "        self.restore_state = self.state\n",
    "        self.state = args\n",
    "        return self #.color\n",
    "    \n",
    "    def __repr__(self):\n",
    "        return f\"<Color({self._state})>\"\n",
    "\n",
    "    \n",
    "class Transform(RestorableState):\n",
    "    def __init__(self, context, matrix=None):\n",
    "        if matrix is None:\n",
    "            matrix = Affine.identity()\n",
    "        else:\n",
    "            matrix = Affine(*matrix)\n",
    "        RestorableState.__init__(self, context, matrix, cm_raises_exception=False)\n",
    "        \n",
    "    def skew(self, x=0, y=0):\n",
    "        if (x, y) == (0, 0):\n",
    "            return self\n",
    "        \n",
    "        self.state *= Affine.skew(x, y)\n",
    "        return self\n",
    "    \n",
    "    def rotate(self, degrees):\n",
    "        self.state *= Affine.rotation(degrees)\n",
    "        return self\n",
    "    \n",
    "    def push(self, t):\n",
    "        self.__call__(t)\n",
    "        return self.__enter__()\n",
    "    \n",
    "    def pop(self):\n",
    "        return self.__exit__()\n",
    "\n",
    "class Context:\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 = Transform(self)\n",
    "        \n",
    "    def fill(self, *args):\n",
    "        return Color(self, *args)\n",
    "    \n",
    "    def stroke(self, *args):\n",
    "        return Color(self, *args)\n",
    "    \n",
    "        \n",
    "#     def move(self, x, y):\n",
    "#         self.transform.move_to(x, y)\n",
    "    \n",
    "#     def push(self):\n",
    "#         return self.transform.push()\n",
    "    \n",
    "#     def pop(self):\n",
    "#         return self.transform.pop()\n",
    "    \n",
    "#     def skew(self, *args):\n",
    "#         return self.transform.skew(*args)\n",
    "        \n",
    "        \n",
    "ctx = Context()\n",
    "#ctx.fill()\n",
    "\n",
    "# print(dir(ctx))\n",
    "# with ctx.fill(0):\n",
    "#     pass\n",
    "#print(dir(ctx.fill()))\n",
    "\n",
    "ctx.fill(.1, .1, .1, 1)\n",
    "\n",
    "print(ctx.fill)\n",
    "with ctx.fill(1, 0, 0, 1), ctx.stroke(0):\n",
    "    print(ctx.fill(), ctx.stroke())\n",
    "    with ctx.skew(45):\n",
    "        print(ctx.transform)\n",
    "    \n",
    "print(ctx.fill)\n",
    "\n",
    "ctx.fill()"
   ]
  }
 ],
 "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": 4
}