diff --git a/src/main.py b/src/main.py index 4528b4f..209140d 100644 --- a/src/main.py +++ b/src/main.py @@ -15,6 +15,8 @@ import matplotlib.pyplot as plt #from PIL import Image import cv2 as cv +from stl import mesh + # Import custom image filter library import filters as flt @@ -25,7 +27,7 @@ class apply_filters: self.parse_arguments() self.params = {} self.filters = [] - + # Parse configuration from json file if self.args.config: self.config_file = self.args.config[0] @@ -36,14 +38,20 @@ class apply_filters: # If no config file given, expect filters in command line else: + if not self.args.filters: + print("No filters given, saving original image") + print("No config file given, using command line arguments") i = 0 + for filter in self.args.filters: if filter.find('=') == -1: + # if no '=' in filter, it is a new filter self.filters.append(filter) i += 1 self.params[i] = {} else: + # else it's a parameter for current filter key, value = filter.split('=') self.params[i][key] = value self.parse_params(self.params[i]) @@ -51,44 +59,44 @@ class apply_filters: self.input_file = self.args.input_file self.output_file = self.args.output_file self.dpi = self.args.dpi - self.mirror = self.args.mirror if self.args.mirror else 0 + self.run() - #read as numpy.array + def run(self): + # read as numpy.array self.img = cv.imread(self.input_file, cv.IMREAD_GRAYSCALE) - + self.width = self.img.shape[1] self.height = self.img.shape[0] - #print(self.width, self.height) - - fig = plt.figure(figsize = (self.width, self.height), - frameon = False, dpi = self.dpi / 100) # dpi is in cm - + print(self.width, self.height) + + fig = plt.figure(figsize=(self.width, self.height), + frameon=False, dpi=self.dpi / 100) # dpi is in cm + ax = plt.Axes(fig, [0., 0., 1., 1.]) ax.set_axis_off() fig.add_axes(ax) - - if self.mirror: + + if self.args.mirror: self.mirror_image() # Apply all filters and save image self.apply_filter() self.save_image(fig, ax) - - - def check_param(self, params, key): - try: - params[key] = params[key] - except KeyError: - params[key] = None + plt.close() + if self.args.stl: + self.make_lithophane() def parse_params(self, params): - possible_params = {"h", "searchWindowSize","templateWindowSize", - "ksize", "kernel", "sigmaX" , "sigmaY", - "sigmaColor", "sigmaSpace", "d", "anchor", "iterations", "op"} + possible_params = {"h", "searchWindowSize", "templateWindowSize", + "ksize", "kernel", "sigmaX", "sigmaY", + "sigmaColor", "sigmaSpace", "d", "anchor", "iterations", + "op", "strength"} for key in possible_params: - self.check_param(params, key) - + try: + params[key] = params[key] + except KeyError: + params[key] = None def parse_conf(self): # Parse configuration file if given. @@ -103,60 +111,68 @@ class apply_filters: self.parse_params(self.params[i]) except(KeyError): - print("Preset not found", file = sys.stderr) - + print("Preset not found", file=sys.stderr) def parse_arguments(self): # Parse arguments - parser = ap.ArgumentParser(prog = 'main.py', - description = 'Program for processing a 2D image into 3D fingerprint.', + parser = ap.ArgumentParser(prog='main.py', + description='Program for processing a 2D image into 3D fingerprint.', usage='%(prog)s [-h] [-m | --mirror | --no-mirror] input_file output_file dpi ([-c config_file preset | --config config_file preset] | [filters ...])') # positional arguments - parser.add_argument("input_file", type = str, help = "location with input file") - parser.add_argument("output_file", type = str, help = "output file location") - parser.add_argument("dpi", type = int, help = "scanner dpi") + parser.add_argument("input_file", type=str, + help="location with input file") + parser.add_argument("output_file", type=str, + help="output file location") + parser.add_argument("dpi", type=int, help="scanner dpi") # boolean switch - parser.add_argument('-m', "--mirror", help = "mirror input image", - type = bool, action = ap.BooleanOptionalAction) + parser.add_argument('-m', "--mirror", help="mirror input image", + type=bool, action=ap.BooleanOptionalAction) + + parser.add_argument('-s', '--stl', help="make stl model from processed image", + type=bool, action=ap.BooleanOptionalAction) # file with configuration containing presets, new preset name # pair argument - give both or none - parser.add_argument('-c','--config', nargs = 2, metavar = ('config_file', 'preset'), - help = 'pair: name of the config file with presets, name of the preset') + parser.add_argument('-c', '--config', nargs=2, metavar=('config_file', 'preset'), + help='pair: name of the config file with presets, name of the preset') # array of unknown length, all filter names saved inside - parser.add_argument('filters', type = str, nargs = '*', + parser.add_argument('filters', type=str, nargs='*', help="list of filter names and their parameters in form [filter_name1 param1=value1 param2=value2 filter_name2 param1=value1...]") self.args = parser.parse_args() - def filter_factory(self, filter_name): - # selects filter method of filters library - # better this than a 100 if/else - print("Applying " + filter_name + " filter", file = sys.stderr) - return getattr(flt, filter_name) + ''' Selects filter method of filters library. + ''' + print("Applying " + filter_name + " filter", file=sys.stderr) + return getattr(flt, filter_name) def resize_image(self): - print("Resize image", file = sys.stderr) - self.img = self.img.resize((np.array(self.width, self.height)).astype(int)) - + print("Resize image", file=sys.stderr) + self.img = self.img.resize( + (np.array(self.width, self.height)).astype(int)) def mirror_image(self): + ''' Mirror image when mirroring is needed, + should be used only if we want a positive model + ''' - # mirror image when mirroring is needed - # should be used only if we want a positive form + #TODO make this automatic for positive STL print("Mirroring image", file=sys.stderr) self.img = cv.flip(self.img, 1) # 1 for vertical mirror - def apply_filter(self): - + ''' Apply filters to image. + + Applies the filters one by one, if no filters were given, just save original image output. + ''' + if len(self.filters) == 0: # No filter given, just save the image pass @@ -165,18 +181,22 @@ class apply_filters: for i, filter_name in enumerate(self.filters): filter = self.filter_factory(filter_name) filter.apply(self, self.params[i+1]) - def print_size(self, size): - print("Width: " + str(size[0]), file = sys.stderr) - print("Height: " + str(size[1]), file = sys.stderr) - + print("Width: " + str(size[0]), file=sys.stderr) + print("Height: " + str(size[1]), file=sys.stderr) def save_image(self, fig, ax): - # Save processed image. - print("Saving image", file = sys.stderr) + ''' Save processed image. + Colormap set to grayscale to avoid color mismatch. + ''' + + print("Saving image", file=sys.stderr) ax.imshow(self.img, cmap="gray") fig.savefig(fname=self.output_file) + def make_lithophane(self): + pass + -app = apply_filters() +image = apply_filters()