-
Notifications
You must be signed in to change notification settings - Fork 72
/
figure-2.py
75 lines (64 loc) · 2.93 KB
/
figure-2.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
# -*- coding: utf-8 -*-
# -----------------------------------------------------------------------------
# Copyright INRIA
# Contributors: Wahiba Taouali ([email protected])
# Nicolas P. Rougier ([email protected])
#
# This software is governed by the CeCILL license under French law and abiding
# by the rules of distribution of free software. You can use, modify and/ or
# redistribute the software under the terms of the CeCILL license as circulated
# by CEA, CNRS and INRIA at the following URL
# http://www.cecill.info/index.en.html.
#
# As a counterpart to the access to the source code and rights to copy, modify
# and redistribute granted by the license, users are provided only with a
# limited warranty and the software's author, the holder of the economic
# rights, and the successive licensors have only limited liability.
#
# In this respect, the user's attention is drawn to the risks associated with
# loading, using, modifying and/or developing or reproducing the software by
# the user in light of its specific status of free software, that may mean that
# it is complicated to manipulate, and that also therefore means that it is
# reserved for developers and experienced professionals having in-depth
# computer knowledge. Users are therefore encouraged to load and test the
# software's suitability as regards their requirements in conditions enabling
# the security of their systems and/or data to be ensured and, more generally,
# to use and operate it in the same conditions as regards security.
#
# The fact that you are presently reading this means that you have had
# knowledge of the CeCILL license and that you accept its terms.
# -----------------------------------------------------------------------------
import os
import numpy as np
from helper import *
from graphics import *
from projections import *
if __name__ == '__main__':
import matplotlib.pyplot as plt
from matplotlib.patches import Polygon
from mpl_toolkits.axes_grid1 import ImageGrid
from mpl_toolkits.axes_grid1.inset_locator import zoomed_inset_axes
from mpl_toolkits.axes_grid1.inset_locator import mark_inset
P = retina_projection()
# Checkerboard pattern for retina
grid = 2*32
even = grid / 2 * [0, 1]
odd = grid / 2 * [1, 0]
R = np.row_stack(grid / 2 * (even, odd))
R = R.repeat(grid, axis=0).repeat(grid, axis=1)
# Mask with a disc
R = R* disc((retina_shape[0],retina_shape[0]),
(retina_shape[0]//2,retina_shape[0]//2),
retina_shape[0]//2)
# Take half-retina
R = R[:,retina_shape[1]:]
# Project to colliculus
SC = R[P[...,0], P[...,1]]
fig = plt.figure(figsize=(10,8), facecolor='w')
ax1, ax2 = ImageGrid(fig, 111, nrows_ncols=(1,2), axes_pad=0.5)
polar_frame(ax1, legend=True)
polar_imshow(ax1, R, vmin=0, vmax=5)
logpolar_frame(ax2, legend=True)
logpolar_imshow(ax2, SC, vmin=0, vmax=5)
plt.savefig("figure-2.pdf")
plt.show()