OSDN Git Service

marshal database generator & accessor
[amulettoolsmh4/main.git] / amulettool.py
index 5a5c9c4..8fe7159 100644 (file)
@@ -47,6 +47,7 @@ class AmuletToolController(wx.App):
         self.frame_view.DisableNoteBook()
 
         self._init_database()
+        self._init_notebook_seed1()
         self._init_notebook_seed2()
         self._init_notebook_skill2()
         self._init_notebook_setting()
@@ -101,13 +102,21 @@ class AmuletToolController(wx.App):
         except sqlite3.Error as e:
             self._show_error_dialog(u"データベースが壊れています")
 
-        # get dictionaries
-        (self._skill_id2name_dict, 
-            self._skill_name2id_dict,
-            self._amulet_id2name_dict, 
-            self._amulet_name2id_dict,
-            self._amulet_id2skill1_id_dict,
-            self._amulet_id2skill2_id_dict) = self.db_accessor.get_dicts()
+        self._minmax_dict = self.db_accessor.get_minmax_dict()
+        self._amulet_names = self.db_accessor.get_id_sorted_amulet_names()
+        self._skill_names = self.db_accessor.get_id_sorted_skill_names()
+
+    def _init_notebook_seed1(self):
+        u""" Seed1特定タブのviewの初期化 """
+        self.notebook_seed1_view = self.frame_view.notebook_seed1_view
+        self.notebook_seed1_view.bind_button_search(self.OnClickSeed1Search)
+        self.notebook_seed1_view.bind_button_clear(self.OnClickSeed1Clear)
+
+        u" 各種お守りの第1スキル選択のセット "
+        name2skill1s = {}
+        for name in self._amulet_names:
+            name2skill1s[name] = tuple([x for x in self._skill_names if x in self._minmax_dict[name][0]])
+        self.notebook_seed1_view.set_amuletname_skillnames_dict(self._amulet_names, name2skill1s)
 
     def _init_notebook_seed2(self):
         u""" Seed2特定タブのviewの初期化 """
@@ -117,11 +126,10 @@ class AmuletToolController(wx.App):
         self.notebook_seed2_view.bind_button_skill2(self.OnClickSkill2SearchFromSeed2)
 
         u" 各種お守りの第2スキル選択 "
-        for amu_key, amu_name in zip([view.KEY_AMULET1, view.KEY_AMULET2, view.KEY_AMULET3], 
+        for amu_key, amu_name in zip([view.KEY_AMULET1, view.KEY_AMULET2, view.KEY_AMULET3],
                 [view.NAME_AMULET1, view.NAME_AMULET2, view.NAME_AMULET3]):
-            amu_id = self._amulet_name2id_dict[amu_name]
-            skill_ids = self._amulet_id2skill2_id_dict[amu_id]
-            skill_names = [view.VAL_NO_SKILL] + [self._skill_id2name_dict[x] for x in skill_ids]
+            skill_names = [view.VAL_NO_SKILL] + [x for x in self._skill_names 
+                    if x in self._minmax_dict[amu_name][1]]
             self.notebook_seed2_view.set_skill_names(amu_key, skill_names)
         self.notebook_seed2_view.set_skill_selected_idx(0)
 
@@ -135,88 +143,98 @@ class AmuletToolController(wx.App):
         u""" お守り検索タブのviewの更新
         amulet_id が Noneでない場合、お守りの種類に応じてComboboxを再設定する
         また、選択中のスキルに応じて最大最小を設定する"""
-        #min1, max1, min2, max2 = self.db_accessor.get_skill_minmax()
-        #self.notebook_amulet_view.set_skill_minmax(min1, max1, min2, max2)
-
-        if amulet_id is not None:
-            for i, skill_dict in enumerate([self._amulet_id2skill1_id_dict, self._amulet_id2skill2_id_dict]):
-                skill_ids = skill_dict[amulet_id]
-                skill_names = [self._skill_id2name_dict[_id] for _id in skill_ids]
-                if i == 0:
-                    self.notebook_amulet_view.set_items_skill1_combobox(skill_names)
-                    skill1_selected = skill_ids[0]
-                    self.notebook_amulet_view.set_selection_skill1_combobox(0)
-                elif i == 1:
-                    self.notebook_amulet_view.set_items_skill2_combobox([view.VAL_NO_SKILL] + skill_names)
-                    self.notebook_amulet_view.set_selection_skill2_combobox(0)  # select No Skill
-        else:
-            amulet_name = self.notebook_amulet_view.get_selected_radio_name()
-            amulet_id = self._amulet_name2id_dict[amulet_name]
-
-        # settting min max
-        skill1_str = self.notebook_amulet_view.get_string_selection_skill1_combobox()
-        skill2_str = self.notebook_amulet_view.get_string_selection_skill2_combobox()
-        skill1_id = self._skill_name2id_dict[skill1_str] if skill1_str in self._skill_name2id_dict else None
-        skill2_id = self._skill_name2id_dict[skill2_str] if skill2_str in self._skill_name2id_dict else None
-
-        if skill2_id is None:
-            minmax_dict = self.db_accessor.select_minmax_from_skill_ids(amulet_id, [skill1_id])
-            min1, max1 = minmax_dict[skill1_id][0], minmax_dict[skill1_id][1]
-            min2, max2 = 0, 0
-        else:
-            minmax_dict = self.db_accessor.select_minmax_from_skill_ids(amulet_id, [skill1_id, skill2_id])
-            min1, max1 = minmax_dict[skill1_id][0], minmax_dict[skill1_id][1]
-            min2, max2 = minmax_dict[skill2_id][2], minmax_dict[skill2_id][3]
-        self.notebook_amulet_view.set_skill_minmax(min1, max1, min2, max2)
+        pass
+
+
+    def _update_notebook_amulet_seed2s(self):
+        u"""お守り検索タブのSeed2リストを更新する"""
+        pass
 
     def _init_notebook_amulet(self):
         u""" お守り検索タブの初期設定 """
         self.notebook_amulet_view = self.frame_view.notebook_amulet_view
-        self.notebook_amulet_view.bind_radio_button(self.OnClickAmuletRadio)
+        self.notebook_amulet_view.bind_radiobox_amulet(self.OnClickAmuletRadioAmulet)
+        self.notebook_amulet_view.bind_radiobox_threshold_type(self.OnClickAmuletRadioThresholdType)
         self.notebook_amulet_view.bind_combobox(self.OnClickAmuletCombo)
         self.notebook_amulet_view.bind_button_search(self.OnClickAmuletSearch)
         self.notebook_amulet_view.bind_button_clear(self.OnClickAmuletClear)
         self.notebook_amulet_view.bind_button_skill(self.OnClickSkillSearchFromAmulet)
 
-        amu_id = self._amulet_name2id_dict[view.NAME_AMULET1]
         self.notebook_amulet_view.set_radio_value(True, view.NAME_AMULET1)
-        self._update_notebook_amulet(amu_id)
+        self._update_notebook_amulet()
 
     def _init_notebook_setting(self):
         u""" 設定タブの初期設定 """
         self.notebook_setting_view = self.frame_view.notebook_setting_view
         self.notebook_setting_view.bind_button_ok(self.OnClickSettingOK)
         self.notebook_setting_view.bind_button_clear(self.OnClickSettingClear)
-        self.notebook_setting_view.set_skill_strings(self._skill_name2id_dict.keys())
+        self.notebook_setting_view.set_skill_strings(self._skill_names)
         self._update_notebook_setting()
 
     def _update_notebook_setting(self):
         u"""設定タブの値更新"""
-        self.notebook_setting_view.set_threshold(self._highlight_threshold1, 
+        self.notebook_setting_view.set_threshold(self._highlight_threshold1,
                 self._highlight_threshold2)
-        self.notebook_setting_view.set_checked_strings([self._skill_id2name_dict[x] for x in self._highlight_skills])
+        self.notebook_setting_view.set_checked_strings(self._highlight_skills)
 
     def _update_highlight(self):
         u""" update highlight cells """
-        skill_names = [self._skill_id2name_dict[x] for x in self._highlight_skills]
         self.notebook_skill2_view.set_skill2_highlight(
-                skill_names, self._highlight_threshold1, self._highlight_threshold2)
+                self._highlight_skills, self._highlight_threshold1, self._highlight_threshold2)
+        self.notebook_seed1_view.update_highlight(self._highlight_skills)
+
+    u""" Seed1 view's event"""
+    def OnClickSeed1Search(self, evt):
+        u""" search seed1s from selected skill1s """
+        alchemy_type = self.notebook_seed1_view.get_tenun_radio_key()
+        amu_skill_name_list = self.notebook_seed1_view.get_selected_amulets_and_names()
+        if alchemy_type == view.KEY_TENUN555:
+            alchemy_type = model.KEY_TENUN555
+        elif alchemy_type == view.KEY_TENUN888:
+            alchemy_type = model.KEY_TENUN888
+        else:
+            raise KeyError(u"key '{0}' not found in alchemy type".format(alchemy_type))
+        seed1s = self.db_accessor.select_seed1s(amu_skill_name_list, alchemy_type)
+        if len(seed1s) == 0:
+            self.notebook_seed1_view.set_text_result(u"条件に一致するSeed1は見つかりません")
+            self.notebook_seed1_view.clear_seed1_grid()
+        elif len(seed1s) == 1:
+            seed1 = [x for x in seed1s][0]
+            no, table_no, result_num = self.db_accessor.select_table_nos_from_seed1(seed1, alchemy_type)
+            self.notebook_seed1_view.set_text_result(
+                    u"Seed1: {0}, テーブルNo: {1}, 通し番号: {2}".format(seed1, table_no, no))
+            # list of (no, seed1)
+            near_num = view.NEAR_SEED1_NUMBERS
+            no_seed1_dict = self.db_accessor.select_near_seed1s_from_table_no(
+                    no, table_no, near_num, near_num, alchemy_type)
+
+            # no -> (seed1, result_num, amu_names, skill_names)
+            self.notebook_seed1_view.set_no2seed1s_dict(no_seed1_dict, no)
+            self.notebook_seed1_view.update_highlight(self._highlight_skills)
+        else:
+            self.notebook_seed1_view.set_text_result(
+                    u"Seed1は{0}件あります。条件を絞ってください".format(len(seed1s)))
+            self.notebook_seed1_view.clear_seed1_grid()
+
+    def OnClickSeed1Clear(self, evt):
+        u""" clear seed1s from selected skill1s """
+        self.notebook_seed1_view.clear_combobox()
+        self.notebook_seed1_view.clear_seed1_grid()
+        self.notebook_seed1_view.clear_text_result()
 
+    u""" Seed2 view's event """
     def OnClickSeed2Search(self, evt):
         u""" search seed2s from selected skill2s """
-        amu_id2skill_id_list_dict = {}
+        amu2skills_dict = {}
         for amu_key, amu_name in zip([view.KEY_AMULET1, view.KEY_AMULET2, view.KEY_AMULET3], 
                 [view.NAME_AMULET1, view.NAME_AMULET2, view.NAME_AMULET3]):
-            amu_id = self._amulet_name2id_dict[amu_name]
             names = self.notebook_seed2_view.get_selected_skill_names(amu_key)
-            ls = [self._skill_name2id_dict[name] if name in self._skill_name2id_dict else None 
-                    for name in names]
-            amu_id2skill_id_list_dict[amu_id] = ls
-        seed_sets = self.db_accessor.select_seeds(amu_id2skill_id_list_dict)
+            amu2skills_dict[amu_name] = [name if name in self._skill_names else None for name in names]
+        seed_sets = self.db_accessor.select_seed2s(amu2skills_dict)
         self.notebook_seed2_view.set_result_text(u"""Seedの候補は{0}個です。""".format(len(seed_sets)))
 
         if len(seed_sets) > 0:
-            self.notebook_seed2_view.set_seed_lists([u"{0}".format(seed) for seed in seed_sets])
+            self.notebook_seed2_view.set_seed_lists([u"{0}".format(seed) for seed in sorted(seed_sets)])
             self.notebook_seed2_view.set_skill2_button_enable(True)
         else:
             self.notebook_seed2_view.clear_seed_list()
@@ -237,19 +255,27 @@ class AmuletToolController(wx.App):
             self.frame_view.note_book.SetSelection(view.SKILL_SEARCH_PAGE)
             self.OnClickSkillSearch(evt)
 
+    u""" Skill2 search from Seed2's event """
     def OnClickSkillSearch(self, evt):
-        u""" skill search from seed"""
-        seed = self.notebook_skill2_view.get_seed2_value()
-        if seed is not None:
-            skill_dict, threshold1_dict = self.db_accessor.select_skills_from_seeds([seed])
+        u""" skill search from seed2"""
+        seed2 = self.notebook_skill2_view.get_seed2_value()
+        if seed2 is not None:
+            table_no, no, skill_dict, th1s, th2s = self.db_accessor.select_names_from_seed2(seed2)
             try:
                 for amu_key, amu_name in zip([view.KEY_AMULET1, view.KEY_AMULET2, view.KEY_AMULET3], 
                         [view.NAME_AMULET1, view.NAME_AMULET2, view.NAME_AMULET3]):
-                    amu_id = self._amulet_name2id_dict[amu_name]
-                    skill_names = [self._skill_id2name_dict[x] for x in skill_dict[amu_id][seed]]
+                    skill_names = skill_dict[amu_name]
                     self.notebook_skill2_view.set_skill2_by_col_key(amu_key, skill_names)
-                th_vals = [u"{0}".format(x) for x in threshold1_dict[seed]]
+                th_vals = [u"{0}".format(x) for x in th1s]
                 self.notebook_skill2_view.set_skill2_by_col_key(view.KEY_THRESHOLD1, th_vals)
+                th_vals = [u"{0}".format(x) for x in th2s]
+                self.notebook_skill2_view.set_skill2_by_col_key(view.KEY_THRESHOLD2, th_vals)
+                #inishie
+                skill_name, th1, th2 = self.db_accessor.select_inishie_skill2_from_seed2(seed2)
+                self.notebook_skill2_view.set_inishie(skill_name, th1, th2)
+                # explanation
+                self.notebook_skill2_view.set_result_text(
+                        u"SEED2: {2}, 通し番号: {1}, テーブルNo: {0}".format(table_no, no, seed2))
             except KeyError, e:
                 self._show_message_dialog(message=u"指定されたSeed値は存在しません")
             finally:
@@ -263,16 +289,23 @@ class AmuletToolController(wx.App):
         self.notebook_skill2_view.clear_skill2_grid()
         self.notebook_skill2_view.clear_skill2_highlight()
         self.notebook_skill2_view.clear_items_amulet_prospects()
+        self.notebook_skill2_view.clear_result_text()
+        self.notebook_skill2_view.clear_inishie()
 
-    def OnClickAmuletRadio(self, evt):
+    u""" amulet search event """
+    def OnClickAmuletRadioAmulet(self, evt):
         u""" switch skill lists by amulet id """
         #btn_id = evt.GetId()
-        amulet_name = self.notebook_amulet_view.get_selected_radio_name()
+        amulet_name = self.notebook_amulet_view.get_selected_amulet()
         amu_id = self._amulet_name2id_dict[amulet_name]
         self._update_notebook_amulet(amu_id)
         self.notebook_amulet_view.set_result_text_ctrl_value(u"")
         self.notebook_amulet_view.set_skill_button_enable(False)
-        self.notebook_amulet_view.clear_listbox_items()
+        self.notebook_amulet_view.clear_grid()
+
+    def OnClickAmuletRadioThresholdType(self, evt):
+        u""" switch seed lists by threshold_type """
+        self._update_notebook_amulet_seed2s()
 
     def OnClickAmuletCombo(self, evt):
         u""" switch skill minmax by amulet id and skill_id"""
@@ -280,70 +313,27 @@ class AmuletToolController(wx.App):
 
     def OnClickAmuletSearch(self, evt):
         u""" search seeds from amulet condition """
-        skill1_name, skill2_name = self.notebook_amulet_view.get_string_selection_skill_names()
-        skill1_val, skill2_val = self.notebook_amulet_view.get_skill_values()
-        slot_val = self.notebook_amulet_view.get_slot_value()
-        amulet_name = self.notebook_amulet_view.get_selected_radio_name()
-        amu_id = self._amulet_name2id_dict[amulet_name]
-
-        if skill1_name == skill2_name:
-            self._show_message_dialog(message=u"異なるスキルを選択してください")
-        elif skill1_val == 0:
-            self._show_message_dialog(message=u"第1スキルの値には0以外を指定してください")
-        else:
-            skill1_id = self._skill_name2id_dict[skill1_name]
-            if skill2_name in self._skill_name2id_dict:
-                skill2_id = self._skill_name2id_dict[skill2_name]
-            else:
-                skill2_id = None # for skill2 is 0
-            suff_dict = {}
-            tup = self.db_accessor.get_sufficient_value(
-                amu_id, skill1_id, skill2_id, skill1_val, skill2_val)
-            if tup is not None:
-                suff_val = tup[0]
-                seeds_set = self.db_accessor.select_seeds_from_sufficient_val(amu_id, suff_val, slot_val, skill2_id)
-                if len(seeds_set) > 0:
-                    self.notebook_amulet_view.set_result_text_ctrl_value(
-                        u"{0}個のSeedで出現するお守りです".format(len(seeds_set)))
-                    self.notebook_amulet_view.set_listbox_items(
-                            [u"{0}".format(seed) for seed in seeds_set])
-                    self.notebook_amulet_view.set_selection_listbox(0)
-                    self.notebook_amulet_view.set_skill_button_enable(True)
-                else:
-                    self.notebook_amulet_view.set_result_text_ctrl_value(
-                        u"指定されたお守りは見つかりませんでした")
-                    self.notebook_amulet_view.set_skill_button_enable(False)
-                    self.notebook_amulet_view.clear_listbox_items()
-            else:
-                self.notebook_amulet_view.set_result_text_ctrl_value(
-                    u"エラー。充足値が計算できません")
-                self.notebook_amulet_view.set_skill_button_enable(False)
-                self.notebook_amulet_view.clear_listbox_items()
+        pass
 
     def OnClickAmuletClear(self, evt):
         u""" clear amulet conditions """
-        self.notebook_amulet_view.set_radio_value(True, view.NAME_AMULET1)
-        amu_id = self._amulet_name2id_dict[view.NAME_AMULET1]
-        self._update_notebook_amulet(amu_id)
-        self.notebook_amulet_view.set_skill_button_enable(False)
-        self.notebook_amulet_view.set_result_text_ctrl_value(u"")
-        self.notebook_amulet_view.clear_listbox_items()
+        pass
 
     def OnClickSkillSearchFromAmulet(self, evt):
         u""" change page to skill search from amulet"""
-        seed = self.notebook_amulet_view.get_string_selection_listbox()
-        if seed.isdigit():
+        seed = self.notebook_amulet_view.get_grid_selected_seed2()
+        if seed is not None:
             self.notebook_skill2_view.set_seed2_value(seed)
             self.frame_view.note_book.SetSelection(view.SKILL_SEARCH_PAGE)
             self.OnClickSkillSearch(evt)
 
+    u""" settings' event """
     def OnClickSettingOK(self, evt):
         u""" get settings of setting tab """
         (self._highlight_threshold1, 
                 self._highlight_threshold2) = self.frame_view.notebook_setting_view.get_threshold()
-        self._highlight_skills = set([self._skill_name2id_dict[x] for x in 
-                self.frame_view.notebook_setting_view.get_checked_strings()
-                if x in self._skill_name2id_dict])
+        self._highlight_skills = set([x for x in self.frame_view.notebook_setting_view.get_checked_strings()
+                if x in self._skill_names])
         self._update_highlight()
 
     def OnClickSettingClear(self, evt):
@@ -354,7 +344,6 @@ class AmuletToolController(wx.App):
         self._update_notebook_setting()
         self._update_highlight()
 
-
     def _show_error_dialog(self, message=u"予期せぬエラーが発生しました", caption=u"エラー"):
         u""" エラーダイアログを表示し、
         OKボタンが押されたらアプリケーションを終了する
@@ -365,7 +354,7 @@ class AmuletToolController(wx.App):
         dlg.ShowModal()
         dlg.Destroy()
         wx.Exit()
-    
+
     def _show_message_dialog(self, message, caption=u"メッセージ"):
         u""" メッセージダイアログを表示する
         """