index.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 23,
   "id": "f879ccb7-2292-474a-a990-31ac0f474e4b",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "^C\n"
     ]
    }
   ],
   "source": [
    "#import sys\n",
    "#!conda install --yes -c conda-forge  pyvista\n",
    "#!conda install --yes  matplotlib\n",
    "#!conda install -c conda-forge pynrrd \n",
    "#!conda install -c conda-forge ipywidgets\n",
    "#!conda install -c anaconda scipy"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "fe1aac41",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "676ead7f",
   "metadata": {
    "scrolled": false
   },
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "dtype = np.dtype('B')\n",
    "display_data = None\n",
    "\n",
    "ORIG_SIZE = 128\n",
    "\n",
    "try:\n",
    "    ds = np.DataSource(None)\n",
    "    gfile = ds.open('http://cdn.klacansky.com/open-scivis-datasets/hydrogen_atom/hydrogen_atom_128x128x128_uint8.raw')\n",
    "    display_data = np.fromfile(gfile,dtype)\n",
    "        \n",
    "except IOError:\n",
    "    print(\"Error opening file\")\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "6e8fea78",
   "metadata": {},
   "outputs": [],
   "source": [
    "d2_raw = np.array(display_data).reshape((ORIG_SIZE,ORIG_SIZE,ORIG_SIZE))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "f440edc9",
   "metadata": {},
   "outputs": [],
   "source": [
    "max_val = 0;\n",
    "\n",
    "factor = 1\n",
    "SIZE = int(ORIG_SIZE/factor)\n",
    "\n",
    "d2 = np.zeros((SIZE,SIZE,SIZE))\n",
    "\n",
    "\n",
    "for i, plane in enumerate(d2):\n",
    "    if i%factor != 0:\n",
    "        continue\n",
    "    for j,row in enumerate(plane):\n",
    "        if j%factor != 0:\n",
    "            continue\n",
    "        for k, pix in enumerate(row):\n",
    "            if j%factor != 0:\n",
    "                continue\n",
    "            max_val = max(max_val,pix)\n",
    "            print(\"setting\", i//factor, j//factor, k//factor, \"to\", d2[i][j][k])\n",
    "            d2[i//factor][j//factor][k//factor] = d2[i][j][k]\n",
    "            \n",
    "print(max_val)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "659f0ff2-d35e-44ff-b948-2b615fb1c53c",
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "%matplotlib inline\n",
    "from matplotlib import pyplot as plt\n",
    "\n",
    "\n",
    "gamma_base = 0.2\n",
    "gamma_steps = 20\n",
    "gamma_step_size = 0.2\n",
    "\n",
    "# # gamma_base = 0.01\n",
    "# # gamma_steps = 15\n",
    "# # gamma_step_size = 0.05\n",
    "\n",
    "# # low_pass_value = 0\n",
    "# # high_pass_value = 255 \n",
    "# # gamma_base = 0.5\n",
    "# # gamma_steps = 20\n",
    "# # gamma_step_size = 1\n",
    "# gamma_lookup = np.zeros((gamma_steps,256))\n",
    "\n",
    "# raycast_display = np.zeros((gamma_steps,128,128))\n",
    "\n",
    "# #d2 = np.rot90(d2,axes=[0,2])\n",
    "\n",
    "# for i, plane in enumerate(d2):\n",
    "#     print(\"Plane: \",i)\n",
    "#     for j,row in enumerate(plane):\n",
    "#         for k, pix in enumerate(row):\n",
    "#             if (pix <= low_pass_value):\n",
    "#                 continue\n",
    "            \n",
    "#             if (pix >= high_pass_value):\n",
    "#                 continue\n",
    "\n",
    "#             scale = 1 # 255 / high_pass_value\n",
    "#             pix0 = min(255,int( pix * scale))\n",
    "            \n",
    "#             if(pix0 > 255):\n",
    "#                 print (\"Out of bounds\", pix, pix0, scale, high_pass_value)\n",
    "            \n",
    "#             for l in range(gamma_steps):\n",
    "               \n",
    "#                 if(raycast_display[l][j][k] > 0.99):\n",
    "#                     continue\n",
    "                    \n",
    "#                 g_pix = None\n",
    "\n",
    "#                 if g_pix == 0 or gamma_lookup[l][pix0] != 0:\n",
    "#                     g_pix = gamma_lookup[l][pix0]\n",
    "#                 else:\n",
    "#                     g_pix = (pix/255)**(gamma_base + l*gamma_step_size)\n",
    "#                     gamma_lookup[l][pix0] = g_pix\n",
    "\n",
    "#                 raycast_display[l][j][k] += g_pix*(1-raycast_display[l][j][k])\n",
    "# #                 raycast_display[l][j][k] += (pix/(255*(gamma_base + l*gamma_step_size)))*(1-raycast_display[l][j][k])\n",
    "\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "25d63789",
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "raster_display = np.zeros((SIZE,SIZE))\n",
    "def calculate_image(gamma,scale,low_pass_value,high_pass_value):\n",
    "    print(\"Gamma\",gamma,\"scale\",scale)\n",
    "\n",
    "    raster_display = np.zeros((SIZE,SIZE))\n",
    "    _gamma_lookup = np.zeros((256))\n",
    "\n",
    "    #d2 = np.rot90(d2,axes=[0,2])\n",
    "\n",
    "    for i, plane in enumerate(d2):\n",
    "        #print(\"Plane: \",i)\n",
    "        for j,row in enumerate(plane):\n",
    "            for k, pix in enumerate(row):\n",
    "                print(\"row\",row)\n",
    "                if (pix <= low_pass_value):\n",
    "                    continue\n",
    "\n",
    "                if (pix >= high_pass_value):\n",
    "                    continue\n",
    "\n",
    "                pix0 = min(255,int( pix * scale))\n",
    "\n",
    "                if(pix0 > 255):\n",
    "                    print (\"Out of bounds\", pix, pix0, scale, high_pass_value)\n",
    "\n",
    "                if(raster_display[j][k] > 0.995):\n",
    "                    print(\"max value at\",i,j)\n",
    "                    continue\n",
    "\n",
    "                g_pix = None\n",
    "\n",
    "                if g_pix == 0 or _gamma_lookup[pix0] != 0:\n",
    "                    g_pix = _gamma_lookup[pix0]\n",
    "                else:\n",
    "                    g_pix = (pix0/255)**(gamma)\n",
    "                    _gamma_lookup[pix0] = g_pix\n",
    "\n",
    "                raster_display[j][k] += g_pix*(1-raster_display[j][k])\n",
    "                \n",
    "                if j == SIZE-1 and k == SIZE-1:\n",
    "                    print(j,k,\" : \",  raster_display[j][k])\n",
    "                \n",
    "            \n",
    "    print(raster_display)\n",
    "#     plt.imshow(raster_display, interpolation='nearest')\n",
    "#     plt.show()  \n",
    "    \n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d8bcb6e1",
   "metadata": {},
   "outputs": [],
   "source": [
    "from __future__ import print_function\n",
    "from ipywidgets import interact, interactive, fixed, interact_manual\n",
    "import ipywidgets as widgets\n",
    "from IPython.display import display\n",
    "import matplotlib.animation as animation\n",
    "\n",
    "g_gamma = 1.0\n",
    "g_scale = 0.25\n",
    "lp_value = 0\n",
    "hp_value = 255\n",
    "# def f(x):\n",
    "#     plt.imshow(raycast_display[int((x-gamma_base)/gamma_step_size)], interpolation='nearest')\n",
    "#     plt.show()  \n",
    "\n",
    "# interact(f,  x=widgets.FloatSlider(min=gamma_base, step=gamma_step_size, max=gamma_base + (gamma_steps-1)*gamma_step_size, value =1))\n",
    "\n",
    "scale_widget = widgets.FloatSlider(min=0.05, step=0.05, max=1.0, value=g_scale)\n",
    "display(scale_widget)\n",
    "\n",
    "gamma_widget =widgets.FloatSlider(min=gamma_base, step=gamma_step_size, max=gamma_base + (gamma_steps-1)*gamma_step_size, value =1)\n",
    "display(gamma_widget)\n",
    "\n",
    "lp_widget =widgets.IntSlider(min=0, step=1, max=255, value=0)\n",
    "display(lp_widget)\n",
    "\n",
    "hp_widget =widgets.IntSlider(min=10, step=1, max=255, value=255)\n",
    "display(hp_widget)\n",
    "\n",
    "fig = plt.figure()\n",
    "im = plt.imshow(raster_display, animated=True)\n",
    "\n",
    "def updatefig(*args):\n",
    "    print(\"redrawing\")\n",
    "    im.set_data(raster_display)\n",
    "    return im,\n",
    "\n",
    "ani = animation.FuncAnimation(fig, updatefig, interval=5 )\n",
    "\n",
    "plt.show()\n",
    "\n",
    "def redraw(x):\n",
    "    print(gamma_widget.value, scale_widget.value,lp_widget.value,hp_widget.value)\n",
    "    calculate_image(gamma_widget.value,scale_widget.value,lp_widget.value,hp_widget.value)\n",
    "    im.set_data(raster_display)\n",
    "    plt.imshow(raster_display, interpolation='nearest')\n",
    "    plt.show()  \n",
    "\n",
    "\n",
    "#interact(f,  x=widgets.FloatSlider(min=gamma_base, step=gamma_step_size, max=gamma_base + (gamma_steps-1)*gamma_step_size, value =1))\n",
    "# text = widgets.Text(\n",
    "#        value='My Text',\n",
    "#        description='Title', )\n",
    "\n",
    "#interact(f,  x= widgets.Button(description='Generate'))\n",
    "button =  widgets.Button(description='Generate')\n",
    "button.on_click(redraw)\n",
    "display(button)\n",
    "                        "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d85b355c",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "5f92a161",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "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.8"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}