{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "89f93b0e-e855-4405-8493-ee537f7e9680",
   "metadata": {},
   "source": [
    "# Export Data in separated variables (PXDMF Format) "
   ]
  },
  {
   "cell_type": "markdown",
   "id": "21b5c5cb-3264-4063-a014-47ac4422cbc6",
   "metadata": {},
   "source": [
    "## Inports"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "295b8b6b-0c89-44a6-9d2c-633022b1ee96",
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import pyvista\n",
    "pyvista.global_theme._jupyter_backend = 'panel' # remove this line to get interactive 3D plots\n",
    "from Muscat.IO.XdmfWriter import XdmfWriter\n",
    "from Muscat.MeshTools.MeshCreationTools import  CreateUniformMeshOfBars, CreateConstantRectilinearMesh\n",
    "from Muscat.Helpers.IO.TemporaryDirectory import TemporaryDirectory\n"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "49a2a8b1-d8f6-4b10-8c82-8ef459acf9c2",
   "metadata": {},
   "source": [
    "## Mesh and data Creation"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "c18edbb8-79d0-4ef4-9d07-2444a64f06fe",
   "metadata": {},
   "outputs": [],
   "source": [
    "mesh1DTime = CreateUniformMeshOfBars(2,5,10)\n",
    "mesh1DTime.props['ParafacDims'] = 1\n",
    "mesh1DTime.props['ParafacDim0'] = \"T\"\n",
    "\n",
    "\n",
    "mesh2DParameters = CreateConstantRectilinearMesh(dimensions=[4,4], origin=[0,0], spacing=[1,1])\n",
    "mesh2DParameters.props['ParafacDims'] = 2\n",
    "mesh2DParameters.props['ParafacDim0'] = \"Px\"\n",
    "mesh2DParameters.props['ParafacDim1'] = \"Py\"\n",
    "\n",
    "mesh3DSpace = CreateConstantRectilinearMesh(dimensions=[8,8,8], origin=[0,0,0], spacing=[1,1,1])\n"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "9609511f-5e9d-428a-8404-e8d3db32e014",
   "metadata": {},
   "source": [
    "## Plot meshes for each dimension"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "81683b51-a3a6-4f07-9cd3-2c47d788e54b",
   "metadata": {},
   "outputs": [],
   "source": [
    "import pyvista as pv\n",
    "from Muscat.Bridges.PyVistaBridge import MeshToPyVista\n",
    "\n",
    "def GetAxes(names):\n",
    "    axes = pv.Axes()\n",
    "    actor = axes.axes_actor\n",
    "    for name, coordName in zip(names, (\"X\",\"Y\",\"Z\")):\n",
    "        if len(name):\n",
    "            getattr(actor, \"Set\"+coordName+\"AxisLabelText\")(name)\n",
    "        else:\n",
    "            getattr(actor, \"Set\"+coordName+\"AxisLabelText\")(\"\")\n",
    "            getattr(actor, \"Get\"+coordName+\"AxisTipProperty\")().SetOpacity(0)\n",
    "            getattr(actor, \"Get\"+coordName+\"AxisShaftProperty\")().SetOpacity(0)\n",
    "\n",
    "    return axes\n",
    "\n",
    "plotter = pv.Plotter(shape=(1, 3))\n",
    "\n",
    "plotter.subplot(0, 0)\n",
    "plotter.add_text(\"1D Time Domain\", font_size=10)\n",
    "plotter.add_mesh(MeshToPyVista(mesh1DTime), show_edges=True, show_scalar_bar=False, edge_color=\"grey\", cmap=\"rainbow_r\")\n",
    "plotter.add_actor(GetAxes((\"T\",\"\",\"\")).axes_actor)\n",
    "\n",
    "plotter.subplot(0, 1)\n",
    "plotter.add_text(\"2D Parametric Domain\", font_size=10)\n",
    "plotter.add_mesh(MeshToPyVista(mesh2DParameters), show_edges=True, show_scalar_bar=False, edge_color=\"grey\", cmap=\"rainbow_r\")\n",
    "plotter.add_actor(GetAxes((\"Px\",\"Py\",\"\")).axes_actor)\n",
    "\n",
    "plotter.subplot(0, 2)\n",
    "plotter.add_text(\"3D Physical Domain\", font_size=10)\n",
    "plotter.add_mesh(MeshToPyVista(mesh3DSpace), show_edges=True, show_scalar_bar=False, edge_color=\"grey\", cmap=\"rainbow_r\")\n",
    "plotter.add_actor(GetAxes((\"x\",\"y\",\"z\")).axes_actor)\n",
    "\n",
    "plotter.show()\n"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "8ce81934-8b2f-480e-8126-3691e3437d1f",
   "metadata": {},
   "source": [
    "## Export "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "1f3a5421-2d6f-470f-9fe3-2bc55b6ec7a6",
   "metadata": {},
   "outputs": [],
   "source": [
    "writer = XdmfWriter()\n",
    "writer.SetBinary(True)\n",
    "writer.SetHdf5(False)\n",
    "writer.SetParafac(True)\n",
    "writer.Open(filename=TemporaryDirectory.GetTempPath() + 'parafac.pxdmf')\n",
    "\n",
    "writer.Write(mesh2DParameters, CellFields= [np.arange(mesh2DParameters.GetNumberOfElements())+0.1 ],CellFieldsNames=[\"Sigma_0\"])\n",
    "writer.Write(mesh1DTime, CellFields = [np.arange(mesh1DTime.GetNumberOfElements())+0.1 ], CellFieldsNames=[\"Sigma_0\"])\n",
    "writer.Write(mesh3DSpace, CellFields = [np.arange(mesh3DSpace.GetNumberOfElements())+0.1 ], CellFieldsNames=[\"Sigma_0\"])\n",
    "\n",
    "writer.Close()\n"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.11.10"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}