1
- # MIT License
2
- # Leland McInnes, Sergey Alexandrov
3
-
1
+ # glasbey vendored from https://github.com/lmcinnes/glasbey.
2
+ # see __init__ and LICENSE_GLASBEY
4
3
import numpy as np
5
4
5
+ # "numbafication" is done at the bottom of the module
6
6
7
7
def get_next_color (distances , colors , new_color ):
8
8
argmax = - 1
@@ -104,42 +104,6 @@ def generate_palette_cam02ucs_and_other(
104
104
return result
105
105
106
106
107
- # def generate_next_color_cam02ucs(colors, current_distances, new_palette_block):
108
- # for i in range(new_palette_block.shape[0]):
109
- # _ = get_next_color(current_distances, colors, new_palette_block[i])
110
- # return get_next_color(current_distances, colors, new_palette_block[0])
111
-
112
-
113
- # def generate_next_color_cam02ucs_and_other(
114
- # colors1,
115
- # colors2,
116
- # current_distances,
117
- # new_palette_block1,
118
- # new_palette_block2,
119
- # alpha=0.0,
120
- # ):
121
- # for i in range(new_palette_block1.shape[0]):
122
- # _ = two_space_get_next_color(
123
- # current_distances,
124
- # colors1,
125
- # colors2,
126
- # new_palette_block1[i],
127
- # new_palette_block2[i],
128
- # alpha=alpha,
129
- # )
130
-
131
- # result, _ = two_space_get_next_color(
132
- # current_distances,
133
- # colors1,
134
- # colors2,
135
- # new_palette_block1[0],
136
- # new_palette_block2[0],
137
- # alpha=alpha,
138
- # )
139
-
140
- # return result
141
-
142
-
143
107
try :
144
108
import numba
145
109
except ImportError :
@@ -168,6 +132,7 @@ def generate_palette_cam02ucs_and_other(
168
132
cache = True ,
169
133
)
170
134
135
+ # get_next_color(distances, colors, new_color)
171
136
get_next_color = jit1 (get_next_color )
172
137
173
138
jit2 = numba .njit (
@@ -187,25 +152,9 @@ def generate_palette_cam02ucs_and_other(
187
152
cache = True ,
188
153
)
189
154
155
+ # generate_palette_cam02ucs(colors, initial_palette, size)
190
156
generate_palette_cam02ucs = jit2 (generate_palette_cam02ucs )
191
157
192
- # jit3 = numba.njit(
193
- # [
194
- # # 'f4[::1](f4[:,::1], f4[::1], f4[:,::1])',
195
- # numba.types.Array(numba.float32, 1, "C", readonly=True)(
196
- # numba.types.Array(numba.float32, 2, "C", readonly=True),
197
- # numba.types.Array(numba.float32, 1, "C"),
198
- # numba.types.Array(numba.float32, 2, "C", readonly=True),
199
- # )
200
- # ],
201
- # locals={
202
- # "i": numba.uint16,
203
- # },
204
- # cache=True,
205
- # )
206
-
207
- # generate_next_color_cam02ucs = jit3(generate_next_color_cam02ucs)
208
-
209
158
jit4 = numba .njit (
210
159
[
211
160
"UniTuple(f4[::1], 2)(f4[::1], f4[:, ::1], f4[:, ::1], f4[::1], f4[::1], f4)" ,
@@ -257,23 +206,3 @@ def generate_palette_cam02ucs_and_other(
257
206
)
258
207
259
208
generate_palette_cam02ucs_and_other = jit5 (generate_palette_cam02ucs_and_other )
260
-
261
- # jit6 = numba.njit(
262
- # [
263
- # numba.types.Array(numba.float32, 1, "C", readonly=True)(
264
- # numba.types.Array(numba.float32, 2, "C", readonly=True),
265
- # numba.types.Array(numba.float32, 2, "C", readonly=True),
266
- # numba.types.Array(numba.float32, 1, "C"),
267
- # numba.types.Array(numba.float32, 2, "C", readonly=True),
268
- # numba.types.Array(numba.float32, 2, "C", readonly=True),
269
- # numba.float32,
270
- # )
271
- # ],
272
- # locals={
273
- # "i": numba.uint16,
274
- # },
275
- # cache=True,
276
- # )
277
- # generate_next_color_cam02ucs_and_other = jit6(
278
- # generate_next_color_cam02ucs_and_other
279
- # )
0 commit comments