export-slr-images-from-earthtime.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "from urllib.parse import quote, urlparse\n",
    "import urllib.request\n",
    "import datetime, os, requests, sys\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def download_file(url, filename, timeout=3600):\n",
    "    if os.path.exists(filename):\n",
    "        sys.stdout.write('%s already downloaded\\n' % filename)\n",
    "        return True\n",
    "    else:\n",
    "        if not os.path.exists(os.path.dirname(filename)):\n",
    "            os.makedirs(os.path.dirname(filename))\n",
    "        sys.stdout.write('Downloading %s to %s\\n' % (url, filename))\n",
    "\n",
    "        try:\n",
    "            response = requests.Session().get(url, timeout=timeout)\n",
    "            if(response.status_code!=200):\n",
    "                print('Error response, code = %d, body = %s' % (response.status_code, response.text))\n",
    "                return False\n",
    "        except RequestException as e:\n",
    "            sys.stdout.write(\"Couldn't read %s because %s\" % (url, e))\n",
    "            return False\n",
    "\n",
    "        open(filename + '.tmp', \"wb\").write(response.content)\n",
    "        os.rename(filename + '.tmp', filename)\n",
    "        sys.stdout.write('Done, wrote %d bytes to %s\\n' % (len(response.content), filename))\n",
    "        return True\n",
    "\n",
    "            \n",
    "class RootArg:\n",
    "    arg_dict = {\n",
    "        'v': None,\n",
    "        't': 0,\n",
    "        'ps': 0,\n",
    "        'l': None,\n",
    "        'bt': None,\n",
    "        'et': None,\n",
    "        'startDwell': 0,\n",
    "        'endDwell': 0,\n",
    "        'fps': 30\n",
    "    }\n",
    "    def __init__(self, v, l, bt, et, **kwargs):\n",
    "        self.arg_dict['v'] = v\n",
    "        self.arg_dict['l'] = l\n",
    "        self.arg_dict['bt'] = bt\n",
    "        self.arg_dict['et'] = et\n",
    "        \n",
    "    def get(self):\n",
    "        arg_list = []\n",
    "        for (key,value) in self.arg_dict.items():\n",
    "            arg_list.append('%s=%s'%(key,value))\n",
    "        return 'https://headless.earthtime.org/#' + \"&\".join(arg_list)\n",
    "    \n",
    "class ThumbnailUrl:\n",
    "    arg_dict = {\n",
    "        'root': None,\n",
    "        'width': None,\n",
    "        'height': None,\n",
    "        'format': 'png',\n",
    "        'fps': 30,\n",
    "        'tileFormat': 'mp4',\n",
    "        'startDwell': 0,\n",
    "        'endDwell': 0,\n",
    "        'fromScreenshot': True\n",
    "    }\n",
    "    \n",
    "    def __init__(self, root, width, height):\n",
    "        self.arg_dict['root'] = root\n",
    "        self.arg_dict['width'] = width\n",
    "        self.arg_dict['height'] = height\n",
    "        \n",
    "\n",
    "    def get(self):\n",
    "        arg_list = []\n",
    "        for (key,value) in self.arg_dict.items():\n",
    "            arg_list.append('%s=%s'%(key,value))\n",
    "        return 'https://hal40-thumbnailworker.createlab.org/thumbnail?' + \"&\".join(arg_list)\n",
    "    \n",
    "\n",
    "def export_slr_images_from_earthtime(sharelink, odir=None, w='5120', h='2880'):\n",
    "    years = [x/100.0 for x in range(3,93, 10)]\n",
    "    dates = [str(x) for x in years]\n",
    "    celsius = [str(x/2).replace('.','_') for x in range(0,9)]    \n",
    "    u = urlparse(sharelink)\n",
    "    v = None\n",
    "    l = None\n",
    "    for fragment in u.fragment.split('&'):\n",
    "        key,val = fragment.split('=')\n",
    "        if key == 'v':\n",
    "            v = val\n",
    "        if key == 'l':\n",
    "            l = val\n",
    "    if v is None or l is None:\n",
    "        print('Malformed sharelink')\n",
    "        return False\n",
    "\n",
    "    slr4_present = l.find(\"slr4\")\n",
    "    if slr4_present < 0:\n",
    "        print('Missing layer slr4. Will be added')\n",
    "        l += ',slr4'\n",
    "        return False\n",
    "        \n",
    "    \n",
    "    if odir is None:\n",
    "        now = datetime.datetime.now()\n",
    "        timestamp = int(now.timestamp())\n",
    "        odir = '%s_%s' % (l.replace(',','_'), timestamp)\n",
    "\n",
    "    os.makedirs(odir, exist_ok=True)\n",
    "    thumbnails = []\n",
    "    i = 0\n",
    "    for date in dates:\n",
    "        bt = et = date\n",
    "        root_arg = RootArg(v,l,bt,et)\n",
    "        thumbnail_url = ThumbnailUrl(quote(root_arg.get(), safe=''), w, h)\n",
    "        fname = celsius[i] + '.png'\n",
    "        thumbnails.append((thumbnail_url.get(), os.path.join(odir,fname)))\n",
    "        i += 1\n",
    "    for t in thumbnails:\n",
    "        download_file(t[0], t[1])\n",
    "    return True\n",
    "    \n",
    "    \n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "#export_slr_images_from_earthtime('https://earthtime.org/explore#v=25.75098,-80.17346,9.056,latLng&t=0.03&ps=50&l=blsat,slr4&bt=0.03&et=0.83')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "export_slr_images_from_earthtime('https://earthtime.org/explore#v=25.75098,-80.17346,9.056,latLng&t=0.03&ps=50&l=blsat,slr4&bt=0.03&et=0.83')"
   ]
  }
 ],
 "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.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}