make_video.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "鳥の群れシミュレーション.ipynb",
      "provenance": []
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    }
  },
  "cells": [
    {
      "cell_type": "code",
      "metadata": {
        "id": "QJa0oPCKVA8a",
        "colab_type": "code",
        "colab": {}
      },
      "source": [
        "import numpy as np\n",
        "import matplotlib\n",
        "import matplotlib.pyplot as plt\n",
        "import matplotlib.animation as animation\n",
        "from tqdm import tqdm, trange\n",
        "FILE_PATH = 'drive/My Drive/'"
      ],
      "execution_count": 0,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "SJHnYvl9VJPx",
        "colab_type": "code",
        "colab": {}
      },
      "source": [
        "with open(FILE_PATH + 'result.txt') as f:\n",
        "    l_strip = [s.strip(' \\n│┌┐└┘') for s in f.readlines() if s.strip(' \\n│┌┐└┘') != '' ]"
      ],
      "execution_count": 0,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab_type": "code",
        "id": "aRRMk7C1l5Zg",
        "colab": {}
      },
      "source": [
        "larr = np.array(list(map(float, l_strip)))"
      ],
      "execution_count": 0,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "ngJ1yl3vXxyc",
        "colab_type": "code",
        "colab": {}
      },
      "source": [
        "l_arr = larr.reshape(-1, 40)"
      ],
      "execution_count": 0,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "kx2mDbeMXQCL",
        "colab_type": "code",
        "outputId": "eab5e103-bd7a-4fb9-c39e-89ee3dae8e3e",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 35
        }
      },
      "source": [
        "l_arr.shape"
      ],
      "execution_count": 32,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "(50000, 40)"
            ]
          },
          "metadata": {
            "tags": []
          },
          "execution_count": 32
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "bnra2i3bZ1Kb",
        "colab_type": "code",
        "outputId": "cf9c5a86-ec77-4fa4-8909-50ef5870423a",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 282
        }
      },
      "source": [
        "def create_movie(show_fig=True, save_mp4=True):\n",
        "    fig = plt.figure()\n",
        "    ax = []\n",
        "    ims = []\n",
        "    W = H = 1\n",
        "    N_BIRD = 10\n",
        "\n",
        "    for i in range(W * H):\n",
        "        ax.append(fig.add_subplot(W,H,i+1))\n",
        "    \n",
        "    for t in trange(500): #l_arr.shape[1]):\n",
        "        # try:\n",
        "        im = ax[0].scatter(l_arr[t, :N_BIRD*2:2], l_arr[t, 1:N_BIRD*2:2], c=\"b\")\n",
        "        # except:\n",
        "        #     print(t)\n",
        "        #     print(l_arr[t])\n",
        "        #     exit()\n",
        "        ims.append((im,))\n",
        "\n",
        "    ani = animation.ArtistAnimation(fig, ims, interval=100)\n",
        "    if show_fig:\n",
        "        plt.show()\n",
        "    if save_mp4:\n",
        "        # mp4での保存にはffpmegのパスが通っている必要がある\n",
        "        ani.save(FILE_PATH + \"plot.mp4\", writer=\"ffmpeg\", dpi=300)\n",
        "create_movie()"
      ],
      "execution_count": 33,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "100%|██████████| 500/500 [00:01<00:00, 250.94it/s]\n"
          ],
          "name": "stderr"
        },
        {
          "output_type": "display_data",
          "data": {
            "image/png": "iVBORw0KGgoAAAANSUhEUgAAAWoAAAD4CAYAAADFAawfAAAABHNCSVQICAgIfAhkiAAAAAlwSFlzAAALEgAACxIB0t1+/AAAADh0RVh0U29mdHdhcmUAbWF0cGxvdGxpYiB2ZXJzaW9uMy4yLjEsIGh0dHA6Ly9tYXRwbG90bGliLm9yZy+j8jraAAAJeElEQVR4nO3cX4ilB3nH8d/T3S1KFbzIoYgbOr0oQhBMyhCElEIXLGsUe6ugV8LeWIggSL30oreSm94sGixoFUGFEuifgCsi2NjZGCXJWhCJNCLsCSImN5bEpxczm3/OZM6EOfs+cT4fGPbMnpfDj5fslzdn3jnV3QFgrj9aegAAr0+oAYYTaoDhhBpgOKEGGO78Nl70jjvu6J2dnW28NMAfpOvXrz/b3avDnttKqHd2drK3t7eNlwb4g1RVPz/qOW99AAwn1ADDCTXAcEINMJxQAwwn1ADDCTXAcEINMJxQAwwn1ADDCTXAcEINMJxQAwwn1ADDCTXAcEINMJxQAwwn1ADDCTXAcEINMJxQAwwn1ADDCTXAcEINMJxQAwwn1ADDCTXAcEINMNz5TQ6qqqeTPJfkxSQvdPfuNkcB8LKNQn3gb7r72a0tAeBQ3voAGG7TUHeS/6yq61V15bADqupKVe1V1d56vT69hQBn3Kah/qvu/sskH0jyyar669ce0N1Xu3u3u3dXq9WpjgQ4yzYKdXf/4uDPm0m+leTebY4C4GXHhrqq/qSq3n7rcZK/TfLEtocBsG+Tuz7+NMm3qurW8f/S3f++1VUAvOTYUHf3z5K89zZsAeAQbs8DGE6oAYYTaoDhhBpgOKEGGE6oAYYTaoDhhBpgOKEGGE6oAYYTaoDhhBpgOKEGGE6oAYYTaoDhhBpgOKEGGE6oAYYTaoDhhBpgOKEGGE6oAYYTaoDhhBpgOKEGGE6oAYYTaoDhNg51VZ2rqh9W1cPbHATAq53kivqBJDe2NQSAw20U6qq6mOSDSb6w3TkAvNamV9QPJvlMkt8ddUBVXamqvaraW6/XpzIOgA1CXVUfSnKzu6+/3nHdfbW7d7t7d7VandpAgLNukyvq+5J8uKqeTvK1JJeq6stbXQXAS44NdXd/trsvdvdOko8k+XZ3f2zrywBI4j5qgPHOn+Tg7v5Oku9sZQkAh3JFDTCcUAMMJ9QAwwk1wHBCDTCcUAMMJ9QAwwk1wHBCDTCcUAMMJ9QAwwk1wHBCDTCcUAMMJ9QAwwk1wHBCDTCcUAMMJ9QAwwk1wHBCDTCcUAMMJ9QAwwk1wHBCDTCcUAMMJ9QAwx0b6qp6S1X9oKp+VFVPVtXnbscwAPad3+CY3ya51N3PV9WFJN+rqn/r7v/a8jYAskGou7uTPH/w7YWDr97mKABettF71FV1rqoeT3IzySPd/eh2ZwFwy0ah7u4Xu/vuJBeT3FtV73ntMVV1par2qmpvvV6f9k6AM+tEd31096+TXEty+ZDnrnb3bnfvrlar09oHcOZtctfHqqrecfD4rUnen+Qn2x4GwL5N7vp4Z5J/rqpz2Q/717v74e3OAuCWTe76+HGSe27DFgAO4TcTAYYTaoDhhBpgOKEGGE6oAYYTaoDhhBpgOKEGGE6oAYYTaoDhhBpgOKEGGE6oAYYTaoDhhBpgOKEGGE6oAYYTaoDhhBpgOKEGGE6oAYYTaoDhhBpgOKEGGE6oAYYTaoDhhBpgOKEGGO7YUFfVnVV1raqeqqonq+qB2zEMgH3nNzjmhSSf7u7HqurtSa5X1SPd/dSWtwGQDa6ou/uX3f3YwePnktxI8q5tDwNg34neo66qnST3JHn0kOeuVNVeVe2t1+vTWQfA5qGuqrcl+UaST3X3b177fHdf7e7d7t5drVanuRHgTNso1FV1IfuR/kp3f3O7kwB4pU3u+qgkX0xyo7s/v/1JALzSJlfU9yX5eJJLVfX4wdf9W94FwIFjb8/r7u8lqduwBYBD+M1EgOGEGmA4oQYYTqgBhhNqgOGEGmA4oQYYTqgBhhNqgOGEGmA4oQYYTqgBhhNqgOGEGmA4oQYYTqgBhhNqgOGEGmA4oQYYTqgBhhNqgOGEGmA4oQYYTqgBhhNqgOGEGmA4oQYY7thQV9VDVXWzqp64HYMAeLVNrqi/lOTylncAcIRjQ93d303yq9uwBYBDnNp71FV1par2qmpvvV6f1ssCnHmnFuruvtrdu929u1qtTutlAc48d30ADCfUAMNtcnveV5N8P8m7q+qZqvrE9mcBcMv54w7o7o/ejiEAHM5bHwDDCTXAcEINMJxQAwwn1ADDCTXAcEINMJxQAwwn1ADDCTXAcEINMJxQAwwn1ADDCTXAcEINMJxQAwwn1ADDCTXAcEINMJxQAwwn1ADDCTXAcEINMJxQAwwn1ADDCTXAcEINMNxGoa6qy1X1P1X106r6h22PAuBlx4a6qs4l+ackH0hyV5KPVtVd2x4GwL5NrqjvTfLT7v5Zd/9fkq8l+bvtzgLglk1C/a4k//uK7585+LtXqaorVbVXVXvr9fq09gGceaf2w8Tuvtrdu929u1qtTutlAc68TUL9iyR3vuL7iwd/B8BtsEmo/zvJX1TVn1fVHyf5SJJ/3e4sAG6p7j7+oKr7kzyY5FySh7r7H485fp3k5yfYcUeSZ09w/Fni3BzNuTmac3O0qefmz7r70PeNNwr1tlXVXnfvLr1jIufmaM7N0Zybo70Zz43fTAQYTqgBhpsS6qtLDxjMuTmac3M05+Zob7pzM+I9agCONuWKGoAjCDXAcIuH2keoHq6qHqqqm1X1xNJbpqmqO6vqWlU9VVVPVtUDS2+aoqreUlU/qKofHZybzy29aZKqOldVP6yqh5fechKLhtpHqL6uLyW5vPSIoV5I8unuvivJ+5J80n83L/ltkkvd/d4kdye5XFXvW3jTJA8kubH0iJNa+oraR6geobu/m+RXS++YqLt/2d2PHTx+Lvv/8H7vEx3Pot73/MG3Fw6+3DGQpKouJvlgki8sveWklg71Rh+hCkepqp0k9yR5dNklcxz87/3jSW4meaS7nZt9Dyb5TJLfLT3kpJYONbxhVfW2JN9I8qnu/s3Se6bo7he7++7sf9LlvVX1nqU3La2qPpTkZndfX3rLG7F0qH2EKm9IVV3IfqS/0t3fXHrPRN396yTX4mcdSXJfkg9X1dPZf4v1UlV9edlJm1s61D5ClROrqkryxSQ3uvvzS++ZpKpWVfWOg8dvTfL+JD9ZdtXyuvuz3X2xu3ey35lvd/fHFp61sUVD3d0vJPn7JP+R/R8Ifb27n1xy0xRV9dUk30/y7qp6pqo+sfSmQe5L8vHsXxU9fvB1/9KjhnhnkmtV9ePsXwg90t1vqlvR+H1+hRxguKXf+gDgGEINMJxQAwwn1ADDCTXAcEINMJxQAwz3/35xAJb1JTt+AAAAAElFTkSuQmCC\n",
            "text/plain": [
              "<Figure size 432x288 with 1 Axes>"
            ]
          },
          "metadata": {
            "tags": [],
            "needs_background": "light"
          }
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "BPhu8bAMZWaJ",
        "colab_type": "code",
        "colab": {}
      },
      "source": [
        ""
      ],
      "execution_count": 0,
      "outputs": []
    }
  ]
}