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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
# coding: UTF8
# Copyright 2009 Thomas Jourdan
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
import random
import cairo
import ka_extensionpoint
import ka_random
import model_constraintpool
import model_allele
EPSILON = 0.00001
COLOR_CONSTRAINT = 'colorconstraint'
class Color(model_allele.Allele):
"""Color
inv: len(self.rgba) == 4
inv: 0.0 <= self.rgba[0] <= 1.0
inv: 0.0 <= self.rgba[1] <= 1.0
inv: 0.0 <= self.rgba[2] <= 1.0
inv: 0.0 <= self.rgba[3] <= 1.0
"""
cdef = [{'bind' : COLOR_CONSTRAINT,
'name' : 'Color constraint',
'domain': model_constraintpool.STRING_1_OF_N,
'enum' : ka_extensionpoint.list_extensions(COLOR_CONSTRAINT)
},
]
def __init__(self, trunk, red, green, blue, alpha):
"""Color constructor
pre: 0.0 <= red <= 1.0
pre: 0.0 <= green <= 1.0
pre: 0.0 <= blue <= 1.0
pre: 0.0 <= alpha <= 1.0
"""
super(Color, self).__init__(trunk)
cpool = model_constraintpool.ConstraintPool.get_pool()
constraint_name = cpool.get(self, COLOR_CONSTRAINT)[0]
self.constraint = ka_extensionpoint.create(constraint_name, self.path)
self.rgba = self.constraint.filter((red, green, blue, alpha))
def __eq__(self, other):
"""Equality based on color components."""
return isinstance(other, Color) \
and abs(self.rgba[0] - other.rgba[0]) < EPSILON \
and abs(self.rgba[1] - other.rgba[1]) < EPSILON \
and abs(self.rgba[2] - other.rgba[2]) < EPSILON \
and abs(self.rgba[3] - other.rgba[3]) < EPSILON
def randomize(self):
"""Set red, green, blue and alpha to random values."""
cpool = model_constraintpool.ConstraintPool.get_pool()
constraints = cpool.get(self, COLOR_CONSTRAINT)
self.constraint = ka_extensionpoint.create(random.choice(constraints), self.path)
self.rgba = self.constraint.randomize()
def mutate(self):
"""Make small random changes in hue, lightness, saturation."""
if ka_random.is_mutating():
if ka_random.is_mutating():
cpool = model_constraintpool.ConstraintPool.get_pool()
constraints = cpool.get(self, COLOR_CONSTRAINT)
self.constraint = ka_extensionpoint.create(random.choice(constraints), self.path)
self.rgba = self.constraint.mutate(self.rgba)
def shuffle(self):
"""Not implemented."""
pass
def crossingover(self, other):
"""Returns either a copy of self or a copy of other.
pre: isinstance(other, Color)
# check for distinct references, needs to copy content, not references
post: __return__ is not self
post: __return__ is not other
"""
randseq = ka_random.crossing_sequence(1)
return self.copy() if randseq[0] else other.copy()
def copy(self):
"""A color copy constructor
post: isinstance(__return__, Color)
# check for distinct references, needs to copy content, not references
post: __return__ is not self
"""
new_one = Color(self.get_trunk(), 0.0, 0.0, 0.0, 0.0)
new_one.rgba = (self.rgba[0], self.rgba[1], self.rgba[2], self.rgba[3])
new_one.constraint = self.constraint
return new_one
def explain(self, alfa=True):
return self.constraint.explain(self.rgba, alfa)
@staticmethod
def make_icon(rgba, alfa, width, height):
surface = cairo.ImageSurface(cairo.FORMAT_ARGB32, width, height)
ctx = cairo.Context(surface)
if alfa:
# paint checker board background
steps = 4
delta = (1.0 * width) / (1.0 * steps)
ctx.set_operator(cairo.OPERATOR_SOURCE)
for row in range(steps):
for col in range(steps):
ctx.rectangle(col * delta, row * delta, delta, delta)
if (col + row) % 2 == 0:
ctx.set_source_rgb(0.4, 0.4, 0.4)
else:
ctx.set_source_rgb(0.6, 0.6, 0.6)
ctx.fill()
# paint color and alfa
ctx.set_operator(cairo.OPERATOR_OVER)
ctx.set_source_rgba(rgba[0], rgba[1], rgba[2], rgba[3])
else:
# paint color
ctx.set_operator(cairo.OPERATOR_OVER)
ctx.set_source_rgb(rgba[0], rgba[1], rgba[2])
ctx.paint()
return surface
|
