コード例 #1
0
    def getArguments(self, parent=None):
        description = "The CSV export requires some information before it starts:"
        dialog = CustomDialog("CSV Export", description, parent)

        default_csv_output_path = self.getDataKWValue("CSV_OUTPUT_PATH",
                                                      default="output.csv")
        output_path_model = DefaultPathModel(default_csv_output_path)
        output_path_chooser = PathChooser(output_path_model,
                                          path_label="Output file path")

        design_matrix_default = self.getDataKWValue("DESIGN_MATRIX_PATH",
                                                    default="")
        design_matrix_path_model = DefaultPathModel(design_matrix_default,
                                                    is_required=False,
                                                    must_exist=True)
        design_matrix_path_chooser = PathChooser(
            design_matrix_path_model, path_label="Design Matrix path")

        all_case_list = self.getAllCaseList()
        list_edit = ListEditBox(all_case_list, "List of cases to export")

        infer_iteration_model = DefaultBooleanModel()
        infer_iteration_checkbox = CheckBox(infer_iteration_model,
                                            label="Infer iteration number",
                                            show_label=False)
        infer_iteration_checkbox.setToolTip(CSVExportJob.INFER_HELP)

        dialog.addOption(output_path_chooser)
        dialog.addOption(design_matrix_path_chooser)
        dialog.addOption(list_edit)
        dialog.addOption(infer_iteration_checkbox)

        dialog.addButtons()

        success = dialog.showAndTell()

        if success:
            design_matrix_path = design_matrix_path_model.getPath()
            if design_matrix_path.strip() == "":
                design_matrix_path = None

            case_list = ",".join(list_edit.getItems())

            return [
                output_path_model.getPath(), case_list, design_matrix_path,
                infer_iteration_model.isTrue()
            ]

        raise CancelPluginException("User cancelled!")
コード例 #2
0
ファイル: csv_export.py プロジェクト: edbru/ert
    def getArguments(self, parent=None):
        description = "The CSV export requires some information before it starts:"
        dialog = CustomDialog("CSV Export", description, parent)

        default_csv_output_path = self.getDataKWValue("CSV_OUTPUT_PATH", default="output.csv")
        output_path_model = DefaultPathModel(default_csv_output_path)
        output_path_chooser = PathChooser(output_path_model, path_label="Output file path")

        design_matrix_default = self.getDataKWValue("DESIGN_MATRIX_PATH", default="")
        design_matrix_path_model = DefaultPathModel(design_matrix_default, is_required=False, must_exist=True)
        design_matrix_path_chooser = PathChooser(design_matrix_path_model, path_label="Design Matrix path")

        all_case_list = self.getAllCaseList()
        list_edit = ListEditBox(all_case_list, "List of cases to export")

        infer_iteration_model = DefaultBooleanModel()
        infer_iteration_checkbox = CheckBox(infer_iteration_model, label="Infer iteration number", show_label=False)
        infer_iteration_checkbox.setToolTip(CSVExportJob.INFER_HELP)

        dialog.addOption(output_path_chooser)
        dialog.addOption(design_matrix_path_chooser)
        dialog.addOption(list_edit)
        dialog.addOption(infer_iteration_checkbox)

        dialog.addButtons()

        success = dialog.showAndTell()

        if success:
            design_matrix_path = design_matrix_path_model.getPath()
            if design_matrix_path.strip() == "":
                design_matrix_path = None

            case_list = ",".join(list_edit.getItems())

            return [output_path_model.getPath(), case_list, design_matrix_path, infer_iteration_model.isTrue()]

        raise CancelPluginException("User cancelled!")
コード例 #3
0
    def getArguments(self, parent=None):
        description = "The GEN_DATA RFT CSV export requires some information before it starts:"
        dialog = CustomDialog("Robust CSV Export", description, parent)
        
        output_path_model = DefaultPathModel("output.csv")
        output_path_chooser = PathChooser(output_path_model, path_label="Output file path")

        trajectory_model = DefaultPathModel("wellpath" , must_be_a_directory=True , must_be_a_file = False , must_exist = True)
        trajectory_chooser = PathChooser(trajectory_model, path_label="Trajectory file")

        fs_manager = self.ert().getEnkfFsManager()
        all_case_list = fs_manager.getCaseList()
        all_case_list = [case for case in all_case_list if fs_manager.caseHasData(case)]
        list_edit = ListEditBox(all_case_list, "List of cases to export")


        infer_iteration_model = DefaultBooleanModel()
        infer_iteration_checkbox = CheckBox(infer_iteration_model, label="Infer iteration number", show_label=False)
        infer_iteration_checkbox.setToolTip(GenDataRFTCSVExportJob.INFER_HELP)

        dialog.addOption(output_path_chooser)
        dialog.addOption(trajectory_chooser)
        dialog.addOption(list_edit)
        dialog.addOption(infer_iteration_checkbox)

        dialog.addButtons()

        success = dialog.showAndTell()

        if success:
            case_list = ",".join(list_edit.getItems())
            try:
                return [output_path_model.getPath(), trajectory_model.getPath() , case_list, infer_iteration_model.isTrue()]
            except ValueError:
                pass

        raise CancelPluginException("User cancelled!")
コード例 #4
0
        output_path_chooser = PathChooser(output_path_model,
                                          path_label="Output file path")

        trajectory_model = DefaultPathModel("wellpath",
                                            must_be_a_directory=True,
                                            must_be_a_file=False,
                                            must_exist=True)
        trajectory_chooser = PathChooser(trajectory_model,
                                         path_label="Trajectory file")

        fs_manager = self.ert().getEnkfFsManager()
        all_case_list = fs_manager.getCaseList()
        all_case_list = [
            case for case in all_case_list if fs_manager.caseHasData(case)
        ]
        list_edit = ListEditBox(all_case_list, "List of cases to export")

        infer_iteration_model = DefaultBooleanModel()
        infer_iteration_checkbox = CheckBox(infer_iteration_model,
                                            label="Infer iteration number",
                                            show_label=False)
        infer_iteration_checkbox.setToolTip(GenDataRFTCSVExportJob.INFER_HELP)

        dialog.addOption(output_path_chooser)
        dialog.addOption(trajectory_chooser)
        dialog.addOption(list_edit)
        dialog.addOption(infer_iteration_checkbox)

        dialog.addButtons()

        success = dialog.showAndTell()