split_steps.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import vtk\n",
    "\n",
    "filename = \"/Volumes/richter/0104_0001/job_28532326/result_cycle_1.vtu\"\n",
    "\n",
    "target_folder = \"/Volumes/richter/0104_0001/job_28532326/split_time_data\"\n",
    "\n",
    "reader = vtk.vtkXMLUnstructuredGridReader()\n",
    "reader.SetFileName(filename)\n",
    "reader.Update()\n",
    "\n",
    "data = reader.GetOutput()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "0\n"
     ]
    }
   ],
   "source": [
    "geometry = vtk.vtkUnstructuredGrid()\n",
    "geometry.SetPoints(data.GetPoints())\n",
    "geometry.SetCells(data.GetCell(0).GetCellType(), data.GetCells())\n",
    "\n",
    "print(geometry.GetPointData().GetNumberOfArrays())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00000.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00010.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00020.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00030.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00040.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00050.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00060.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00070.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00080.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00090.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00100.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00110.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00120.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00130.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00140.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00150.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00160.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00170.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00180.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00190.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00200.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00210.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00220.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00230.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00240.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00250.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00260.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00270.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00280.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00290.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00300.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00310.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00320.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00330.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00340.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00350.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00360.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00370.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00380.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00390.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00400.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00410.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00420.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00430.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00440.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00450.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00460.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00470.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00480.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00490.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00500.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00510.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00520.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00530.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00540.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00550.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00560.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00570.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00580.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00590.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00600.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00610.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00620.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00630.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00640.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00650.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00660.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00670.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00680.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00690.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00700.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00710.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00720.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00730.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00740.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00750.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00760.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00770.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00780.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00790.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00800.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00810.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00820.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00830.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00840.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00850.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00860.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00870.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00880.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00890.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00900.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00910.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00920.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00930.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00940.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00950.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00960.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00970.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00980.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_00990.vtu\n",
      "/Volumes/richter/0104_0001/job_28532326/split_time_data/result_01000.vtu\n"
     ]
    }
   ],
   "source": [
    "import os\n",
    "\n",
    "for i in range(0, 1010, 10):\n",
    "    p_array_name = f\"pressure_{i:05}\"\n",
    "    q_array_name = f\"velocity_{i:05}\"\n",
    "\n",
    "    p_array = vtk.vtkDoubleArray()\n",
    "    p_array.DeepCopy(data.GetPointData().GetArray(p_array_name))\n",
    "    p_array.SetName(\"pressure\")\n",
    "\n",
    "    q_array = vtk.vtkDoubleArray()\n",
    "    q_array.DeepCopy(data.GetPointData().GetArray(q_array_name))\n",
    "    q_array.SetName(\"velocity\")\n",
    "\n",
    "    geometry.GetPointData().AddArray(p_array)\n",
    "    geometry.GetPointData().AddArray(q_array)\n",
    "\n",
    "    writer = vtk.vtkXMLUnstructuredGridWriter()\n",
    "    writer.SetInputData(geometry)\n",
    "    writer.SetFileName(os.path.join(target_folder, f\"result_{i:05}.vtu\"))\n",
    "    writer.Write()\n",
    "    print(writer.GetFileName())\n"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "estimator_new",
   "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.9.18"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}