TreeGnomeAgility.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "%matplotlib inline\n",
    "from IPython.display import clear_output\n",
    "from pytesseract import image_to_string\n",
    "\n",
    "from srbot import *"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "#Tree gnome village agility\n",
    "\n",
    "target()\n",
    "total_trips = 0\n",
    "last_run = mark_time()\n",
    "logins = 0\n",
    "while True:\n",
    "    client = get_client()\n",
    "    if len(find_bitmap(loginscreen,client)) > 0:\n",
    "        logins = logins + 1\n",
    "        if logins > 10:\n",
    "            raise RuntimeError('Too many logins! Bailing out before we get fishy!')\n",
    "        login()\n",
    "        continue\n",
    "    if mark_time()-last_run > 10*60:\n",
    "        raise RuntimeError('Took more than 10min to complete course, something is wrong.')\n",
    "    \n",
    "    minimap = get_minimap()\n",
    "    floor1 = find_colors([54,43,8],minimap,tol=(0.05,0.08,0.1),mode='hsl')\n",
    "    floor2 = find_colors([91,56,31],minimap,tol=(0.05,0.08,0.1),mode='hsl')\n",
    "    water = find_colors([124,143,184],minimap,tol=(0.1,0.1,0.1),mode='hsl')\n",
    "    floor1 = filter_far(floor1,water,20)\n",
    "    floor2 = filter_far(floor2,water,20)\n",
    "    floors = []\n",
    "    if len(floor1) > 0:\n",
    "        floors.extend(floor1)\n",
    "    if len(floor2) > 0:\n",
    "        floors.extend(floor2)\n",
    "    floors = np.asarray(floors)\n",
    "    floors = floors-[mmxc-mmxs,mmyc-mmys] if len(floors) else floors\n",
    "    clusters,counts = cluster(floors,10)\n",
    "    mask = counts > 30\n",
    "    clusters,counts = clusters[mask],counts[mask]\n",
    "    clusters = clusters[np.argsort(-counts)]\n",
    "    print('clusters',len(counts))\n",
    "    if len(counts) == 1: #middle platform\n",
    "        print('climb limb')\n",
    "        mainscreen = get_mainscreen()\n",
    "        limb = find_colors([111,85,26],mainscreen,tol=0.045,mode='hsl')\n",
    "        floor = find_colors([80,56,42],mainscreen,tol=0.045,mode='hsl')\n",
    "        climbp = filter_near(limb,np.mean(floor.T,axis=1),45)\n",
    "        if confirm_click(climbp+[msxs,msys]):\n",
    "            sleep(3.5)\n",
    "    elif len(counts) > 1:\n",
    "        if np.random.random() < 0.05:\n",
    "            #polish_minimap() //FIXME\n",
    "            run_on(restore_tab='stats')\n",
    "        centers = np.asarray([np.mean(cluster.T,axis=1) for cluster in clusters])\n",
    "        dists = np.sqrt(np.sum(np.square(centers),axis=1))\n",
    "        print(dists)\n",
    "        if dists[1] > 45 and np.min(dists) > 40: #just climbed down\n",
    "            print('walking to rope net two')\n",
    "            click_mouse(mmxc,mmyc-15)\n",
    "            flag_wait(init=0.5,post=0.7,imax=10)\n",
    "        else:\n",
    "            near = np.argmin(dists)\n",
    "            print('nearest:',near)\n",
    "            if near == 0: #at log ropes\n",
    "                if centers[near,1] < 0:\n",
    "                    print('walk to rope net')\n",
    "                    click_mouse(*(centers[2]+[mmxc,mmyc-15]))\n",
    "                    flag_wait(init=0.5,post=0.7,imax=10)\n",
    "                else:\n",
    "                    print('do log balance')\n",
    "                    mainscreen = get_mainscreen()\n",
    "                    log = find_colors([99,59,8],mainscreen,mode='hsl',tol=0.1)\n",
    "                    clusters,counts = cluster(log)\n",
    "                    log = clusters[np.argmax(counts)]\n",
    "                    grass = find_colors([139,145,35],mainscreen,mode='hsl',tol=0.1)\n",
    "                    dirt = find_colors([102,81,35],mainscreen,mode='hsl',tol=0.1)\n",
    "                    balance = filter_near(filter_near(log,grass,20),dirt,20)\n",
    "                    if confirm_click(balance+[msxs,msys]):\n",
    "                        sleep(5.0)\n",
    "            elif near == 1: #going to pipes\n",
    "                if centers[near,1] > 0:\n",
    "                    print('walk to log balance')\n",
    "                    click_mouse(*(centers[0]+[mmxc,mmyc-15]))\n",
    "                    flag_wait(init=1.0,post=0.7,imax=30)\n",
    "                else:\n",
    "                    if np.min(dists) > 30: \n",
    "                        print('climb rope net two')\n",
    "                        mainscreen = get_mainscreen()\n",
    "                        net = find_colors([120,96,70],mainscreen,mode='hsl',tol=0.05)\n",
    "                        border = find_colors([111,68,13],mainscreen,mode='hsl',tol=0.05)\n",
    "                        clusters,counts = cluster(border,10)\n",
    "                        border = clusters[np.argmax(counts)]\n",
    "                        climbp = filter_near(net,border,10)\n",
    "                        if confirm_click(climbp+[msxs,msys]):\n",
    "                            flag_wait(init=0.5,post=0.7)\n",
    "                            sleep(2.0)\n",
    "                    else:\n",
    "                        print('go through pipes')\n",
    "                        last_run = mark_time()\n",
    "                        mainscreen = get_mainscreen()\n",
    "                        pipes = find_colors([122,113,113],mainscreen,mode='hsl',tol=0.05)\n",
    "                        clusters,counts = cluster(pipes)\n",
    "                        if len(counts) > 0:\n",
    "                            pipes = np.concatenate(clusters[counts > 50])\n",
    "                            if confirm_click(pipes+[msxs,msys]):\n",
    "                                sleep(10.0)\n",
    "                                clear_output()\n",
    "                                total_trips += 1\n",
    "                                print('completed trip %i'%total_trips)\n",
    "            elif near == 2: #going to climb up\n",
    "                print('climb rope net')\n",
    "                mainscreen = get_mainscreen()\n",
    "                net = find_colors([120,96,70],mainscreen,mode='hsl',tol=0.05)\n",
    "                border = find_colors([111,68,13],mainscreen,mode='hsl',tol=0.05)\n",
    "                clusters,counts = cluster(border,10)\n",
    "                border = clusters[np.argmax(counts)]\n",
    "                climbp = filter_near(net,border,10)\n",
    "                if confirm_click(climbp+[msxs,msys]):\n",
    "                    flag_wait(init=0.5,post=0.7,imax=10)\n",
    "                    sleep(2.0)\n",
    "    else:\n",
    "        walls = find_colors([238,238,238],minimap,tol=0.05)-[mmxc-mmxs,mmyc-mmys]\n",
    "        delta = np.mean(walls[:,0])\n",
    "        print(delta)\n",
    "        if delta > 0:\n",
    "            print('top left')\n",
    "            mainscreen = get_mainscreen()\n",
    "            grass = find_colors([107,118,28],mainscreen,mode='hsl',tol=0.01)\n",
    "            rope = find_colors([166,134,93],mainscreen,mode='hsl',tol=0.05)\n",
    "            walk = filter_near(rope,grass,4)+[msxs,msys]\n",
    "            walk = walk + np.random.normal(0,5,size=walk.shape)\n",
    "            dist = np.sum(np.square(walk-[msxc,msyc]),axis=1)\n",
    "            if confirm_click(walk,order=np.argsort(dist)):\n",
    "                flag_wait(init=0.5,post=0.7,imax=10)\n",
    "                sleep(5.0)\n",
    "        else:\n",
    "            print('top right')        \n",
    "            mainscreen = get_mainscreen()\n",
    "            tree = find_colors([103,78,26],mainscreen,mode='hsl',tol=(0.01,0.02,0.02))\n",
    "            knot = find_colors([131,105,73],mainscreen,mode='hsl',tol=(0.02,0.1,0.1))\n",
    "            clusters,counts = cluster(knot,radius=1)\n",
    "            mask = np.logical_and(counts > 8, counts < 30)\n",
    "            clusters,counts = clusters[mask],counts[mask]\n",
    "            click = filter_near(tree,np.concatenate(clusters),15)\n",
    "            order = np.argsort(click[:,0]+np.random.normal(0,10,len(click)))\n",
    "            if confirm_click(click+[msxs,msys],order=order):\n",
    "                flag_wait(init=0.5,post=0.7)\n",
    "                sleep(2.0)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "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.6.9"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}