@@ -277,44 +277,44 @@ def _buildGraph(self, transposed):
277
277
278
278
279
279
280
- def _edgeTodistTo (self , v , transposed , edgeL = False , edgeR = False ):
280
+ def _edgeTodistTo (self , pixel , transposed , edgeL = False , edgeR = False ):
281
281
# returns pixel connected to v with min energy
282
282
if edgeL :
283
283
# left edge
284
- vC = v - self ._width
285
- vRD = v - self ._width + 1
286
- vLU = vC
284
+ up_pixel = pixel - self ._width
285
+ right_up_diagonal_pixel = pixel - self ._width + 1
286
+ left_up_diagonal_pixel = up_pixel
287
287
elif edgeR :
288
288
# right edge
289
- vLU = v - self ._width - 1
290
- vC = v - self ._width
291
- vRD = vC
289
+ left_up_diagonal_pixel = pixel - self ._width - 1
290
+ up_pixel = pixel - self ._width
291
+ right_up_diagonal_pixel = up_pixel
292
292
else :
293
- # pixels connect to v
294
- vLU = v - self ._width - 1
295
- vC = v - self ._width
296
- vRD = v - self ._width + 1
293
+ # pixels connect to pixel
294
+ left_up_diagonal_pixel = pixel - self ._width - 1
295
+ up_pixel = pixel - self ._width
296
+ right_up_diagonal_pixel = pixel - self ._width + 1
297
297
298
- # energy of pixels connected to v
298
+ # energy of pixels connected to pixel
299
299
if transposed :
300
- (colU , rowU ) = self ._toGrid (vLU )
301
- (colC , rowC ) = self ._toGrid (vC )
302
- (colD , rowD ) = self ._toGrid (vRD )
300
+ (colU , rowU ) = self ._toGrid (left_up_diagonal_pixel )
301
+ (colC , rowC ) = self ._toGrid (up_pixel )
302
+ (colD , rowD ) = self ._toGrid (right_up_diagonal_pixel )
303
303
# read energy
304
304
eLU = self ._energy [self ._height * colU + rowU ]
305
305
eC = self ._energy [self ._height * colC + rowC ]
306
306
eRD = self ._energy [self ._height * colD + rowD ]
307
307
else :
308
308
# read energy directly from energy array
309
- eLU = self ._energy [vLU ]
310
- eC = self ._energy [vC ]
311
- eRD = self ._energy [vRD ]
312
- #print (eLU, vLU ), (eC, vC ), (eRD, vRD )
309
+ eLU = self ._energy [left_up_diagonal_pixel ]
310
+ eC = self ._energy [up_pixel ]
311
+ eRD = self ._energy [right_up_diagonal_pixel ]
312
+ #print (eLU, left_up_diagonal_pixel ), (eC, up_pixel ), (eRD, right_up_diagonal_pixel )
313
313
# find min distance and its associated vertex
314
- dist , from_vertex = min ((self ._distTo [vLU ] + eLU , vLU ), (self ._distTo [vC ] + eC , vC ), (self ._distTo [vRD ] + eRD , vRD ))
315
- #e, vertex = min([(eC, vC ), (eLU, vLU ), (eRD, vRD )])
316
- self ._edgeTo [v ] = from_vertex
317
- self ._distTo [v ] = dist
314
+ dist , from_vertex = min ((self ._distTo [left_up_diagonal_pixel ] + eLU , left_up_diagonal_pixel ), (self ._distTo [up_pixel ] + eC , up_pixel ), (self ._distTo [right_up_diagonal_pixel ] + eRD , right_up_diagonal_pixel ))
315
+ #e, vertex = min([(eC, up_pixel ), (eLU, left_up_diagonal_pixel ), (eRD, right_up_diagonal_pixel )])
316
+ self ._edgeTo [pixel ] = from_vertex
317
+ self ._distTo [pixel ] = dist
318
318
319
319
320
320
0 commit comments