def add_point(self, y, x, image_file_name):
image = PImage.from_file(image_file_name)
point = SpatialPoint(image_file_name, y, x, image.height(),
image.width())
if image_file_name in self.points:
raise Exception("duplicate key")
self.points[image_file_name] = point
#print self.x_list
#print self.y_list
self.y_list.add(point)
self.x_list.add(point)
self.is_sorted = False
def get_file_names(file_names_in, depth):
file_names = list()
first_parts = set()
second_parts = set()
for file_name_in in file_names_in:
if os.path.isfile(file_name_in):
if PImage.is_image_filename(file_name_in):
file_names.append(file_name_in)
elif os.path.isdir(file_name_in):
if depth:
for file_name in os.listdir(file_name_in):
file_names.append(
get_file_names(
os.path.join(file_name_in, file_name),
depth - 1))
return file_names
def get_image(self):
if self.image is None:
self.image = PImage.from_file(self.get_name())
return self.image
def calc_il_dim(il):
name = il.get_name()
pimage = PImage.from_file(name)
il.set_width(pimage.width())
il.set_height(pimage.height())
def control_points_by_subimage(self, pair, image_fn_pair):
'''Stitch two images together by cropping to restrict overlap'''
# subimage_factor: (y, x) overlap percent tuple or none for default
# pair: pair of row/col or coordinate positions (used to determine relative positions)
# (0, 0) at upper left
# image_fn_pair: pair of image file names
print 'Preparing subimage stitch on %s:%s' % (image_fn_pair[0],
image_fn_pair[1])
'''
Just work on the overlap section, maybe even less
'''
images = [
PImage.from_file(image_file_name)
for image_file_name in image_fn_pair
]
'''
image_0 used as reference
4 basic situations: left, right, up right
8 extended: 4 basic + corners
Pairs should be sorted, which simplifies the logic
'''
sub_image_0_x_delta = 0
sub_image_0_y_delta = 0
sub_image_1_x_end = images[1].width()
sub_image_1_y_end = images[1].height()
# Add some backlash margin
# "more overlap" means will try a slightly larger area
#margin = 0.05
x_overlap = self.x_overlap
y_overlap = self.y_overlap
# image 0 left of image 1?
if pair.first.col < pair.second.col:
# Keep image 0 right, image 1 left
sub_image_0_x_delta = int(images[0].width() * x_overlap)
sub_image_1_x_end = int(
round(images[1].width() * (1.0 - x_overlap)))
# image 0 above image 1?
if pair.first.row < pair.second.row:
# Keep image 0 top, image 1 bottom
sub_image_0_y_delta = int(images[0].height() * y_overlap)
sub_image_1_y_end = int(
round(images[1].height() * (1.0 - y_overlap)))
'''
print 'image 0 x delta: %d, y delta: %d' % (sub_image_0_x_delta, sub_image_0_y_delta)
Note y starts at top in PIL
'''
sub_image_0 = images[0].subimage(sub_image_0_x_delta, None,
sub_image_0_y_delta, None)
sub_image_1 = images[1].subimage(None, sub_image_1_x_end, None,
sub_image_1_y_end)
sub_image_0_file = ManagedTempFile.get(None, '.jpg')
sub_image_1_file = ManagedTempFile.get(None, '.jpg')
print 'sub image 0: width=%d, height=%d, name=%s' % (sub_image_0.width(
), sub_image_0.height(), sub_image_0_file.file_name)
print 'sub image 1: width=%d, height=%d, name=%s' % (sub_image_1.width(
), sub_image_1.height(), sub_image_1_file.file_name)
#sys.exit(1)
sub_image_0.image.save(sub_image_0_file.file_name)
sub_image_1.image.save(sub_image_1_file.file_name)
sub_image_fn_pair = (sub_image_0_file.file_name,
sub_image_1_file.file_name)
# subimage file name symbolic link to subimage file name
# this should be taken care of inside of control point actually
#sub_link_to_sub = dict()
# subimage to the image it came from
sub_to_real = dict()
sub_to_real[sub_image_0_file.file_name] = image_fn_pair[0]
sub_to_real[sub_image_1_file.file_name] = image_fn_pair[1]
# Returns a pto project object
pair_project = self.control_point_gen.generate_core(sub_image_fn_pair)
if pair_project is None:
print 'WARNING: failed to gen control points @ %s' % repr(pair)
return None
# all we need to do is adjust xy positions
# afaik above is way overcomplicated
final_pair_project = pto_unsub(
pair_project, (sub_image_0_file, sub_image_1_file),
(sub_image_0_x_delta, sub_image_0_y_delta), sub_to_real)
# Filenames become absolute
#sys.exit(1)
return final_pair_project