{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "# imports\n",
    "import pathlib\n",
    "import scipy.io"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "coda_dir = pathlib.Path().resolve().parent.parent\n",
    "matlab_dir = coda_dir / \"registration_testing_matlab\"\n",
    "python_dir = coda_dir / \"registration_testing_python\"\n",
    "matlab_save_warp_dir = matlab_dir / \"registered\" / \"elastic registration\" / \"save_warps\"\n",
    "python_save_warp_dir = python_dir / \"registered\" / \"elastic registration\" / \"save_warps\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "im_stems = sorted([fp.stem for fp in matlab_save_warp_dir.glob(\"*.mat\") if \"003\" not in fp.stem])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "GLOBAL_KEYS_MATLAB = (\"szz\", \"padall\", \"cent\", \"f\", \"krf\", \"None\")\n",
    "GLOBAL_KEYS_PYTHON = (\"max_size\", \"pad_all\", \"center\", \"flipped\", \"ref_index\", \"global_transform\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "------------- lungs_001 -------------\n",
      "szz/max_size:\n",
      "\tmatlab: [[1057  933]]\n",
      "\tpython: [[1057  933]]\n",
      "padall/pad_all:\n",
      "\tmatlab: [[250]]\n",
      "\tpython: [[250]]\n",
      "cent/center:\n",
      "\tmatlab: [[717 779]]\n",
      "\tpython: [[717. 779.]]\n",
      "f/flipped:\n",
      "\tmatlab: [[0]]\n",
      "\tpython: [[0]]\n",
      "krf/ref_index:\n",
      "\tmatlab: [[2]]\n",
      "\tpython: [[1]]\n",
      "python transform:\n",
      "[[ 9.9995965e-01 -8.9814849e-03 -1.7066542e+01]\n",
      " [ 8.9814849e-03  9.9995965e-01 -1.8601867e+00]]\n",
      "------------- lungs_005 -------------\n",
      "szz/max_size:\n",
      "\tmatlab: [[1057  933]]\n",
      "\tpython: [[1057  933]]\n",
      "padall/pad_all:\n",
      "\tmatlab: [[250]]\n",
      "\tpython: [[250]]\n",
      "cent/center:\n",
      "\tmatlab: [[717 779]]\n",
      "\tpython: [[717. 779.]]\n",
      "f/flipped:\n",
      "\tmatlab: [[0]]\n",
      "\tpython: [[0]]\n",
      "krf/ref_index:\n",
      "\tmatlab: [[2]]\n",
      "\tpython: [[1]]\n",
      "python transform:\n",
      "[[ 0.99995995 -0.00895176 -3.7279801 ]\n",
      " [ 0.00895176  0.99995995 -1.1443613 ]]\n"
     ]
    }
   ],
   "source": [
    "for im_stem in im_stems:\n",
    "    matlab_data = scipy.io.loadmat(matlab_save_warp_dir / f\"{im_stem}.mat\")\n",
    "    python_data = scipy.io.loadmat(python_save_warp_dir / f\"{im_stem}.mat\")\n",
    "    print(f\"------------- {im_stem} -------------\")\n",
    "    for key_m, key_p in zip(GLOBAL_KEYS_MATLAB, GLOBAL_KEYS_PYTHON):\n",
    "        if key_m == \"None\":\n",
    "            print(\"python transform:\")\n",
    "            print(python_data[key_p])\n",
    "        else:\n",
    "            print(f\"{key_m}/{key_p}:\\n\\tmatlab: {matlab_data[key_m]}\\n\\tpython: {python_data[key_p]}\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "coda_scratch",
   "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.10.15"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}