def greedy_random_horizontal_make(self, pics):
slideshow = []
previous_slide = None
index = list(range(len(pics)))
random.shuffle(index)
for i in index:
pic = pics[i]
# First, only Horizontal
if pic.orientation == 1:
continue
current_slide = Slide(pic, None)
if current_slide.is_valid(): # Horizontal
if previous_slide is None:
slideshow.append(current_slide)
previous_slide = current_slide
else:
if transition_score(current_slide, previous_slide):
# Ok, current slide is valid, add it to slideshow
slideshow.append(current_slide)
previous_slide = current_slide
return slideshow
def _find_best_slide_transition(self, slides, previous_slide):
best_transition = -100000
best_transition_tag_count = 100000
best_slide_id = None
shuffled_slides = list(slides.keys())
random.shuffle(shuffled_slides)
for i in shuffled_slides:
max_target_score = 1000 # max(int(len(previous_slide.tags) / 2) + 10, 1) # Tweek here for better result
current_transition = transition_score(previous_slide, slides[i])
current_transition_tag_count = len(previous_slide.tags
| slides[i].tags)
if current_transition > best_transition or \
(current_transition == best_transition and \
current_transition_tag_count < best_transition_tag_count):
best_transition = current_transition
best_transition_tag_count = current_transition_tag_count
best_slide_id = i
if best_transition >= max_target_score: # and best_transition_tag_count <= (max_target_score * 4) + 5:
break
return best_slide_id
def greedy_random_vertical_make(self, pics):
slideshow = []
previous_slide = None
current_slide = None
pics_copy = copy.copy(pics)
keep_going = True
# print('Starting greedy slideshow maker...')
while keep_going:
current_slideshow = []
index = list(pics_copy.keys())
random.shuffle(index)
for i in index:
pic = pics_copy[i]
# First, only Vertical
if pic.orientation == 0:
continue
if current_slide is None:
current_slide = Slide(pic, None)
continue
else:
current_slide.add(pic)
if current_slide.is_valid():
if previous_slide is None:
slideshow.append(current_slide)
current_slideshow.append(current_slide)
previous_slide = current_slide
else:
if transition_score(current_slide, previous_slide):
# Ok, current slide is valid, add it to slideshow
slideshow.append(current_slide)
current_slideshow.append(current_slide)
previous_slide = current_slide
current_slide = None
if not current_slideshow:
# Stop when no more slideshow found
keep_going = False
else:
# print('Adding %s slides' % len(current_slideshow))
for slide in current_slideshow:
del pics_copy[slide.pic_a.id]
del pics_copy[slide.pic_b.id]
return slideshow
def greedy_vertical_make(self, pics):
slideshow = []
previous_slide = None
current_slide = None
pics_copy = copy.copy(pics)
keep_going = True
while keep_going:
current_slideshow = []
for i, pic in pics_copy.items():
# First, only Vertical
if pic.orientation == 0:
continue
if current_slide is None:
current_slide = Slide(pic, None)
continue
else:
current_slide.add(pic)
if current_slide.is_valid():
if previous_slide is None:
slideshow.append(current_slide)
current_slideshow.append(current_slide)
previous_slide = current_slide
else:
if transition_score(current_slide, previous_slide):
# Ok, current slide is valid, add it to slideshow
slideshow.append(current_slide)
current_slideshow.append(current_slide)
previous_slide = current_slide
current_slide = None
if not current_slideshow:
# Stop when no more slideshow found
keep_going = False
else:
# print('Adding %s slides' % len(current_slideshow))
for slide in current_slideshow:
del pics_copy[slide.pic_a.id]
del pics_copy[slide.pic_b.id]
return slideshow
def greedy_random_make(self, slides):
slideshow = []
previous_slide = None
slides = copy.copy(slides)
keep_going = True
while keep_going:
current_slideshow = []
index = list(list(slides.keys()))
random.shuffle(index)
for i in index:
slide = slides[i]
if previous_slide is None:
slideshow.append(slide)
current_slideshow.append(i)
previous_slide = slide
else:
if transition_score(slide, previous_slide):
# Ok, current slide is valid, add it to slideshow
slideshow.append(slide)
current_slideshow.append(i)
previous_slide = slide
if not current_slideshow:
# Stop when no more slideshow found
keep_going = False
else:
# print('Adding %s slides' % len(current_slideshow))
for slide_id in current_slideshow:
del slides[slide_id]
return slideshow
def greedy_horizontal_make(self, pics):
slideshow = []
previous_slide = None
for i, pic in pics.items():
# First, only Horizontal
if pic.orientation == 1:
continue
current_slide = Slide(pic, None)
if current_slide.is_valid(): # Horizontal
if previous_slide is None:
slideshow.append(current_slide)
previous_slide = current_slide
else:
if transition_score(current_slide, previous_slide):
# Ok, current slide is valid, add it to slideshow
slideshow.append(current_slide)
previous_slide = current_slide
return slideshow
def __init__(self, current_node, neighbour_node):
self.node = neighbour_node
self.weight = transition_score(current_node.get_slide(), neighbour_node.get_slide())
def greedy_on_pics(self, pics):
slideshow = []
pics = copy.copy(pics)
# Randomly pick first pick
index = list(pics.keys())
starting_index = random.choice(index)
current_slide = Slide(pics.pop(starting_index))
previous_slide = None
keep_going = True
counter = 0
while keep_going:
if counter % 10000 == 0 and counter != 0:
print('currently consumed %s pics...' % counter)
if current_slide.is_valid():
slideshow.append(current_slide)
previous_slide = current_slide
current_slide_id = self._find_best_slide_transition(
pics, previous_slide)
if current_slide_id is not None:
current_slide = Slide(pics[current_slide_id])
del pics[current_slide_id]
else:
break
else:
# Find another best pic and add it to current slide
second_pic_id = None
if previous_slide is not None:
pic_a = current_slide.pic_a
best_transition = -1
for i, pic_b in pics.items():
if pic_b.orientation == 1:
current_association = Slide(pic_a)
current_association.add(pic_b)
current_transition = transition_score(
previous_slide, current_association)
if current_transition > best_transition:
best_transition = current_transition
second_pic_id = i
else:
for i, pic in pics.items():
if pic.orientation == 1 and not set(
current_slide.tags) & set(pic.tags):
second_pic_id = i
break
if second_pic_id is not None:
current_slide.add(pics[second_pic_id])
del pics[second_pic_id]
else:
print('breaking at %s pics: cant find vertical match' %
counter)
keep_going = False
counter += 1
return slideshow