コード例 #1
0
ファイル: WinLineTest.py プロジェクト: zhafree/Ibex
    def on_btnExecPrecision_clicked(self, widget):
        folder = self.txtTestFolder.get_text()
        chain = self.txtFilterchain.get_text()
        if (self.validate_folder(folder) and self.validate_filterchain(chain)):

            self.test = LineTest(folder, self.chain)
            image_number = self.test.total_images()
            if self.validate_image_number(image_number):
                self.test.launch()
                self.fill_labels(self.test)
                self.fill_image_list(self.test.testable_images)
                self.imgGraphPrecision.set_from_pixbuf(
                    numpy_to_pixbuf(self.test.create_graphic()))
コード例 #2
0
ファイル: WinLineTest.py プロジェクト: zhafree/Ibex
class WinLineTest:
    def __init__(self):
        ui = get_ui(self, 'imageListStore', 'paramsListStore',
                    'paramvaluesListStore')
        self.window = ui.get_object(win_name(self))
        self.imageListStore = ui.get_object('imageListStore')
        self.txtFilterchain = ui.get_object('txtFilterchain')
        self.txtTestFolder = ui.get_object('txtTestFolder')
        self.lblPrecision = ui.get_object('lblPrecision')
        self.lblNoise = ui.get_object('lblNoise')
        self.lblNbImages = ui.get_object('lblNbImages')
        self.lstImage = ui.get_object('lstImage')
        self.imgOriginal = ui.get_object('imgOriginal')
        self.imgExample = ui.get_object('imgExample')
        self.imgGraphPrecision = ui.get_object('imgGraphPrecision')

        self.paramsListStore = ui.get_object('paramsListStore')
        self.paramvaluesListStore = ui.get_object('paramvaluesListStore')

        self.cboParams = ui.get_object('cboParams')
        self.spnFrom = ui.get_object('spnFrom')
        self.spnFrom.set_adjustment(self.create_adj(0))
        self.spnTo = ui.get_object('spnTo')
        self.spnTo.set_adjustment(self.create_adj(255))
        self.lblBestPrecision = ui.get_object('lblBestPrecision')
        self.lblBestNoise = ui.get_object('lblBestNoise')
        self.lblOverallBest = ui.get_object('lblOverallBest')
        self.imgGraphEval = ui.get_object('imgGraphEval')
        self.lblValue = ui.get_object('lblValue')

        self.test = None
        self.chain = None

    def init_window(self):
        pass

    def msg_on_no_filterchain(self):
        dialog = Gtk.MessageDialog(self.window, 0, Gtk.MessageType.ERROR,
                                   Gtk.ButtonsType.OK,
                                   "You must select a filterchain file.")
        result = dialog.run()
        dialog.destroy()
        return result

    def msg_on_no_test_folder(self):
        dialog = Gtk.MessageDialog(self.window, 0, Gtk.MessageType.ERROR,
                                   Gtk.ButtonsType.OK,
                                   "You must select a test folder.")
        result = dialog.run()
        dialog.destroy()
        return result

    def msg_on_no_test_images(self):
        dialog = Gtk.MessageDialog(
            self.window, 0, Gtk.MessageType.ERROR, Gtk.ButtonsType.OK,
            "There are no image map files in the test folder.")
        result = dialog.run()
        dialog.destroy()
        return result

    def create_adj(self, init_val):
        return Gtk.Adjustment(init_val, 0, 65535, 1, 10, 0)

    def fill_labels(self, test):
        self.lblNbImages.set_text(str(test.total_images()))
        noise = str(round(test.avg_noise() * 100.0, 2)) + '%'
        self.lblNoise.set_text(noise)
        precision = str(round(test.avg_precision() * 100.0, 2)) + '%'
        self.lblPrecision.set_text(precision)

    def fill_image_list(self, images):
        self.imageListStore.clear()
        for i in xrange(len(images)):
            self.imageListStore.append([images.keys()[i], i])
        if len(images) > 0:
            self.lstImage.set_cursor(0)

    def fill_filterchain_info_parameval(self, chain):
        self.paramsListStore.clear()
        for fname, params in filterchain.params_list(chain):
            for name, value in params:
                if filterchain.isnumeric(value) and not isinstance(
                        value, bool):
                    self.paramsListStore.append(
                        [fname, name, fname + ' - ' + name, value])
        self.cboParams.set_active(0)

    def fill_values_list(self, precisions, noises):
        self.paramvaluesListStore.clear()
        for i in xrange(int(self.spnFrom.get_value()),
                        int(self.spnTo.get_value())):
            idx = i - int(self.spnFrom.get_value())
            self.paramvaluesListStore.append(
                (i, str(round(noises[idx], 2)), str(round(precisions[idx],
                                                          2))))

    def validate_folder(self, folder):
        if folder == '':
            self.msg_on_no_test_folder()
            return False
        else:
            return True

    def validate_filterchain(self, chain):
        if chain == '':
            self.msg_on_no_filterchain()
            return False
        else:
            return True

    def validate_image_number(self, image_number):
        if image_number == 0:
            self.msg_on_no_test_images()
            return False
        else:
            return True

    def on_btnExecPrecision_clicked(self, widget):
        folder = self.txtTestFolder.get_text()
        chain = self.txtFilterchain.get_text()
        if (self.validate_folder(folder) and self.validate_filterchain(chain)):

            self.test = LineTest(folder, self.chain)
            image_number = self.test.total_images()
            if self.validate_image_number(image_number):
                self.test.launch()
                self.fill_labels(self.test)
                self.fill_image_list(self.test.testable_images)
                self.imgGraphPrecision.set_from_pixbuf(
                    numpy_to_pixbuf(self.test.create_graphic()))

    def on_btnExecParams_clicked(self, widget):
        folder = self.txtTestFolder.get_text()
        chain = self.txtFilterchain.get_text()
        if (self.validate_folder(folder) and self.validate_filterchain(chain)):
            index = self.cboParams.get_active()
            if index >= 0:
                fname, pname, _, _ = self.paramsListStore[index]
                peval = ParameterEvaluation(folder, self.chain, fname, pname,
                                            self.spnFrom.get_value(),
                                            self.spnTo.get_value())
                peval.launch()
                self.imgGraphEval.set_from_pixbuf(
                    numpy_to_pixbuf(peval.create_graphic()))
                self.fill_values_list(peval.precisions, peval.noises)

    def on_btnClear_clicked(self, widget):
        self.txtFilterchain.set_text('')
        self.txtTestFolder.set_text('')
        self.lblNoise.set_text('0%')
        self.lblPrecision.set_text('0%')
        self.lblNbImages.set_text('0')
        self.imageListStore.clear()
        self.test = None
        self.chain = None

    def on_btnOpenFilterchain_clicked(self, widget):
        dialog = Gtk.FileChooserDialog(
            "Choose a filterchain file", None, Gtk.FileChooserAction.OPEN,
            (Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL, Gtk.STOCK_OK,
             Gtk.ResponseType.OK))
        ff = Gtk.FileFilter()
        ff.set_name('Filterchain')
        ff.add_pattern('*.filterchain')

        dialog.set_filter(ff)
        response = dialog.run()
        if response == Gtk.ResponseType.OK:
            self.txtFilterchain.set_text(dialog.get_filename())
            self.chain = filterchain.read(self.txtFilterchain.get_text())
            self.fill_filterchain_info_parameval(self.chain)
        dialog.destroy()

    def on_btnOpenTestFolder_clicked(self, widget):
        dialog = Gtk.FileChooserDialog(
            "Choose an image folder", None,
            Gtk.FileChooserAction.SELECT_FOLDER,
            (Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL, Gtk.STOCK_OK,
             Gtk.ResponseType.OK))
        response = dialog.run()
        if response == Gtk.ResponseType.OK:
            self.txtTestFolder.set_text(dialog.get_filename())
        dialog.destroy()

    def on_btnSaveAsParams_clicked(self, widget):
        pass

    def on_btnSaveAsPrecision_clicked(self, widget):
        dialog = Gtk.FileChooserDialog(
            "Choose an image folder", None,
            Gtk.FileChooserAction.SELECT_FOLDER,
            (Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL, Gtk.STOCK_OK,
             Gtk.ResponseType.OK))
        response = dialog.run()
        if response == Gtk.ResponseType.OK:
            self.test.dump_images(dialog.get_filename())
        dialog.destroy()

    def on_cboParams_changed(self, widget):
        index = self.cboParams.get_active()
        if index >= 0:
            value = self.paramsListStore[index][3]
            self.lblValue.set_text(str(int(value)))

    def on_lstImage_cursor_changed(self, widget):
        index = tree_selected_index(self.lstImage)
        if index >= 0:
            file_name = self.imageListStore[index][0]
            self.imgExample.set_from_pixbuf(
                numpy_to_pixbuf(self.test.example_image(file_name)))
            self.imgOriginal.set_from_pixbuf(
                numpy_to_pixbuf(self.test.original_image(file_name)))