evolveRobots/imgTools.py at master · zaneck/evolveRobots · GitHub

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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
# coding: utf8
#############################################################################
#
# This file is part of evolveRobot.
#
# Contributors:
#	- created by Valentin Owczarek
#       - damien.marchal@univ.lille1.fr
#############################################################################
from PIL import Image, ImageDraw

import math
import sys

def matriceToImage(m, w, h, destinationPath):
    """ Save a matrix into an RGB image of size w,h with a given name
        w is width and h height.
        destinationPath is the name of the destination file. No directory is created in case a path is given
        Example of use:
                matriceToImage(aMatrix, 16, 16, "myFile.png")
    """
    img = Image.new("RGB", (w+2, h+2))
    draw = ImageDraw.Draw(img)

    color=[]

    # Draw a gray border line around the real image data to improve clarity.
    for i in range(1,w+1):
        draw.point((i,0), fill=(127,127,127))
        draw.point((i,h+1), fill=(127,127,127))

    for j in range(1,h+1):
        draw.point((0,j), fill=(127,127,127))
        draw.point((w+1,j), fill=(127,127,127))

    #This generate a kind of optical illusion :)
    #draw.point((0,0), fill=(255,255,255))
    #draw.point((0,h+1), fill=(255,255,255))
    #draw.point((w+1,h+1), fill=(255,255,255))
    #draw.point((w+1,0), fill=(255,255,255))

    # Draw the candidate data into the image
    for i in range(1,w+1):
        for j in range(1,h+1):
            if m[i-1][j-1] == 1:
                color.append((i,j))

    draw.point(color, fill=(255,0,0))
    img.save(destinationPath)

def simpleColorMap(c):
      if not isinstance(c, list):
         c = [c]
      dist=c[0]
      if dist > -0.01 and dist < 0.01:
        return (255,0,0)
      else:
        g=int(dist*255)
        return (g,g,g)

def matriceToGrayImage(m, destinationPath, getColor=simpleColorMap):
    """ Save a scalar [0,1] matrix into an RGB image of size w,h with a given name
        destinationPath is the name of the destination file. No directory is created in case a path is given
        Example of use:
                matriceToImage(aMatrix, 16, 16, "myFile.png")
    """
    w = len(m)
    h = len(m[0])
    img = Image.new("RGB", (w+2, h+2))
    draw = ImageDraw.Draw(img)

    color=[]

    # Draw a gray border line around the real image data to improve clarity.
    for i in range(1,w+1):
        draw.point((i,0), fill=(127,127,127))
        draw.point((i,h+1), fill=(127,127,127))

    for j in range(1,h+1):
        draw.point((0,j), fill=(127,127,127))
        draw.point((w+1,j), fill=(127,127,127))

    #This generate a kind of optical illusion :)
    #draw.point((0,0), fill=(255,255,255))
    #draw.point((0,h+1), fill=(255,255,255))
    #draw.point((w+1,h+1), fill=(255,255,255))
    #draw.point((w+1,0), fill=(255,255,255))

    # Draw the candidate data into the image
    for i in range(1,w+1):
        for j in range(1,h+1):
                draw.point((i,j), getColor(m[i-1][j-1]))


    img.save(destinationPath)