import os
import pickle
import string
-import sys
import gettext
import logging
self._units_sort_direction = False
self._find_result = [] # empty find result list
- self._find_count = 0 # default to find result number zero
+ self._findIndex = 0 # default to find result number zero
self._selected_category = '' # preset to no selected category
self._selected_units = {} # empty dictionary for later use
self._categoryView = widgets.get_widget('categoryView')
self._unitsView = widgets.get_widget('unitsView')
+ self._unitsView.set_property('rules_hint', 1)
self._unitsView_selection = self._unitsView.get_selection()
self._unitName = widgets.get_widget('unitName')
self._unitSymbolColumn.connect("clicked", self._on_click_unit_column)
self._unitsView.append_column(self._unitSymbolColumn)
+ self._unitModel = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING, gobject.TYPE_STRING)
+ self._sortedUnitModel = gtk.TreeModelSort(self._unitModel)
+ columns = self._get_column_sort_stuff()
+ for columnIndex, (column, sortDirection, col_cmp) in enumerate(columns):
+ self._sortedUnitModel.set_sort_func(columnIndex, col_cmp)
+ self._unitsView.set_model(self._sortedUnitModel)
+
#Insert a column into the category list even though the heading will not be seen
renderer = gtk.CellRendererText()
self._categoryColumn = gtk.TreeViewColumn('Title', renderer)
"on_exit_menu_activate": self._on_user_exit,
"on_main_window_destroy": self._on_user_exit,
"on_categoryView_select_row": self._on_click_category,
- "on_unitsView__on_click_unit_column": self._on_click_unit_column,
"on_unitValue_changed": self._on_unit_value_changed,
"on_previousUnitValue_changed": self._on_previous_unit_value_changed,
"on_writeUnitsMenuItem_activate": self._on_user_write_units,
"on_messagebox_ok_clicked": self.messagebox_ok_clicked,
"on_clearSelectionMenuItem_activate": self._on_user_clear_selections,
"on_unitsView_cursor_changed": self._on_click_unit,
- "on_unitsView_button_released": self._on_button_released,
"on_shortlistcheck_toggled": self._on_shortlist_changed,
"on_toggleShortList_activate": self._on_edit_shortlist,
}
def _clear_find(self):
# switch to "new find" state
self._find_result = []
- self._find_count = 0
+ self._findIndex = 0
# Clear our user message
self._findLabel.set_text('')
def _find_first(self):
assert len(self._find_result) == 0
- assert self._find_count == 0
+ assert self._findIndex == 0
findString = string.lower(string.strip(self._findEntry.get_text()))
if not findString:
return
if not self._find_result:
return
- self._select_found_unit()
+ self._update_find_selection()
def _find_wrap_around(self):
assert 0 < len(self._find_result)
- assert self._find_count + 1 == len(self._find_result)
+ assert self._findIndex + 1 == len(self._find_result)
#select first result
- self._find_count = 0
- self._select_found_unit()
+ self._findIndex = 0
+ self._update_find_selection()
def _find_next(self):
assert 0 < len(self._find_result)
- assert self._find_count + 1 < len(self._find_result)
- self._find_count += 1
- self._select_found_unit()
+ assert self._findIndex + 1 < len(self._find_result)
+ self._findIndex += 1
+ self._update_find_selection()
- def _select_found_unit(self):
+ def _update_find_selection(self):
assert 0 < len(self._find_result)
#check if next find is in a new category (prevent category changes when unnecessary
- if self._selected_category != self._find_result[self._find_count][0]:
+ if self._selected_category != self._find_result[self._findIndex][0]:
self._categoryView.set_cursor(
- self._find_result[self._find_count][2], self._categoryColumn, False
+ self._find_result[self._findIndex][2], self._categoryColumn, False
)
self._unitsView.set_cursor(
- self._find_result[self._find_count][3], self._unitNameColumn, True
+ self._find_result[self._findIndex][3], self._unitNameColumn, True
)
- def _on_findEntry_changed(self, *args):
- """
- Clear out find results since the user wants to look for something new
- """
- try:
- self._clear_find()
- except Exception:
- _moduleLogger.exception()
-
def _on_shortlist_changed(self, *args):
try:
raise NotImplementedError("%s" % self._shortlistcheck.get_active())
except Exception:
_moduleLogger.exception()
+ def _on_findEntry_changed(self, *args):
+ """
+ Clear out find results since the user wants to look for something new
+ """
+ try:
+ self._clear_find()
+ except Exception:
+ _moduleLogger.exception()
+
def _on_find_activate(self, a):
"""
check if 'new find' or 'last find' or 'next-find'
new-find = run the find algorithm which also selects the first found unit
- = self._find_count = 0 and self._find_result = []
+ = self._findIndex = 0 and self._find_result = []
last-find = restart from top again
- = self._find_count = len(self._find_result)
+ = self._findIndex = len(self._find_result)
next-find = continue to next found location
- = self._find_count = 0 and len(self._find_result)>0
+ = self._findIndex = 0 and len(self._find_result)>0
"""
try:
if len(self._find_result) == 0:
self._find_first()
else:
- if self._find_count == len(self._find_result)-1:
+ if self._findIndex == len(self._find_result)-1:
self._find_wrap_around()
else:
self._find_next()
if not self._find_result:
self._findLabel.set_text('Text not found')
else:
- resultsLeft = len(self._find_result) - self._find_count - 1
+ resultsLeft = len(self._find_result) - self._findIndex - 1
self._findLabel.set_text(
'%s result(s) left' % (resultsLeft, )
)
except Exception:
_moduleLogger.exception()
+ def _unit_model_cmp(self, sortedModel, leftItr, rightItr):
+ leftUnitText = self._unitModel.get_value(leftItr, 0)
+ rightUnitText = self._unitModel.get_value(rightItr, 0)
+ return cmp(leftUnitText, rightUnitText)
+
+ def _symbol_model_cmp(self, sortedModel, leftItr, rightItr):
+ leftSymbolText = self._unitModel.get_value(leftItr, 2)
+ rightSymbolText = self._unitModel.get_value(rightItr, 2)
+ return cmp(leftSymbolText, rightSymbolText)
+
+ def _value_model_cmp(self, sortedModel, leftItr, rightItr):
+ #special sorting exceptions for ascii values (instead of float values)
+ if self._selected_category == "Computer Numbers":
+ leftValue = self._unitModel.get_value(leftItr, 1)
+ rightValue = self._unitModel.get_value(rightItr, 1)
+ else:
+ leftValueText = self._unitModel.get_value(leftItr, 1)
+ leftValue = float(leftValueText) if leftValueText else 0.0
+
+ rightValueText = self._unitModel.get_value(rightItr, 1)
+ rightValue = float(rightValueText) if rightValueText else 0.0
+ return cmp(leftValue, rightValue)
+
+ def _get_column_sort_stuff(self):
+ columns = (
+ (self._unitNameColumn, "_unit_sort_direction", self._unit_model_cmp),
+ (self._unitValueColumn, "_value_sort_direction", self._value_model_cmp),
+ (self._unitSymbolColumn, "_units_sort_direction", self._symbol_model_cmp),
+ )
+ return columns
+
def _on_click_unit_column(self, col):
"""
Sort the contents of the col when the user clicks on the title.
"""
- #Determine which column requires sorting
- if col is self._unitNameColumn:
- selectedUnitColumn = 0
- self._unitNameColumn.set_sort_indicator(True)
- self._unitValueColumn.set_sort_indicator(False)
- self._unitSymbolColumn.set_sort_indicator(False)
- self._unitNameColumn.set_sort_order(not self._unit_sort_direction)
- elif col is self._unitValueColumn:
- selectedUnitColumn = 1
- self._unitNameColumn.set_sort_indicator(False)
- self._unitValueColumn.set_sort_indicator(True)
- self._unitSymbolColumn.set_sort_indicator(False)
- self._unitValueColumn.set_sort_order(not self._value_sort_direction)
- elif col is self._unitSymbolColumn:
- selectedUnitColumn = 2
- self._unitNameColumn.set_sort_indicator(False)
- self._unitValueColumn.set_sort_indicator(False)
- self._unitSymbolColumn.set_sort_indicator(True)
- self._unitSymbolColumn.set_sort_order(not self._units_sort_direction)
- else:
- assert False, "Unknown column: %s" % (col.get_title(), )
-
- #declare a spot to hold the sorted list
- sorted_list = []
-
- #point to the first row
- iter = self._unitModel.get_iter_first()
- row = 0
-
- while iter:
- #grab all text from columns for sorting
-
- #get the text from each column
- unit_text = self._unitModel.get_value(iter, 0)
- units_text = self._unitModel.get_value(iter, 2)
-
- #do not bother sorting if the value column is empty
- if self._unitModel.get_value(iter, 1) == '' and selectedUnitColumn == 1:
- return
-
- #special sorting exceptions for ascii values (instead of float values)
- if self._selected_category == "Computer Numbers":
- value_text = self._unitModel.get_value(iter, 1)
- else:
- if self._unitModel.get_value(iter, 1) == None or self._unitModel.get_value(iter, 1) == '':
- value_text = ''
- else:
- value_text = float(self._unitModel.get_value(iter, 1))
-
- if selectedUnitColumn == 0:
- sorted_list.append((unit_text, value_text, units_text))
- elif selectedUnitColumn == 1:
- sorted_list.append((value_text, unit_text, units_text))
- else:
- sorted_list.append((units_text, value_text, unit_text))
-
- #point to the next row in the self._unitModel
- iter = self._unitModel.iter_next(iter)
- row = row+1
-
- #check if no calculations have been made yet (don't bother sorting)
- if row == 0:
- return
- else:
- if selectedUnitColumn == 0:
- if not self._unit_sort_direction:
- sorted_list.sort(lambda (x, xx, xxx), (y, yy, yyy): cmp(string.lower(x), string.lower(y)))
- self._unit_sort_direction = True
- else:
- sorted_list.sort(lambda (x, xx, xxx), (y, yy, yyy): cmp(string.lower(y), string.lower(x)))
- self._unit_sort_direction = False
- elif selectedUnitColumn == 1:
- sorted_list.sort()
- if not self._value_sort_direction:
- self._value_sort_direction = True
- else:
- sorted_list.reverse()
- self._value_sort_direction = False
- else:
- if not self._units_sort_direction:
- sorted_list.sort(lambda (x, xx, xxx), (y, yy, yyy): cmp(string.lower(x), string.lower(y)))
- self._units_sort_direction = True
- else:
- sorted_list.sort(lambda (x, xx, xxx), (y, yy, yyy): cmp(string.lower(y), string.lower(x)))
- self._units_sort_direction = False
-
- #Clear out the previous list of units
- self._unitModel = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING, gobject.TYPE_STRING)
- self._unitsView.set_model(self._unitModel)
-
- #colourize each row differently for easier reading
- self._unitsView.set_property('rules_hint', 1)
-
- #Clear out the description
- text_model = gtk.TextBuffer(None)
- self._unitDescription.set_buffer(text_model)
-
- if selectedUnitColumn == 0:
- for unit, value, units in sorted_list:
- iter = self._unitModel.append()
- self._unitModel.set(iter, 0, unit, 1, str(value), 2, units)
- elif selectedUnitColumn == 1:
- for value, unit, units in sorted_list:
- iter = self._unitModel.append()
- self._unitModel.set(iter, 0, unit, 1, str(value), 2, units)
+ try:
+ #Determine which column requires sorting
+ columns = self._get_column_sort_stuff()
+ for column, directionName, col_cmp in columns:
+ column.set_sort_indicator(False)
+ for columnIndex, (maybeCol, directionName, col_cmp) in enumerate(columns):
+ if col is maybeCol:
+ direction = getattr(self, directionName)
+ gtkDirection = gtk.SORT_ASCENDING if direction else gtk.SORT_DESCENDING
+
+ # cause a sort
+ self._sortedUnitModel.set_sort_column_id(columnIndex, gtkDirection)
+
+ # set the visual for sorting
+ col.set_sort_indicator(True)
+ col.set_sort_order(not direction)
+
+ setattr(self, directionName, not direction)
+ break
else:
- for units, value, unit in sorted_list:
- iter = self._unitModel.append()
- self._unitModel.set(iter, 0, unit, 1, str(value), 2, units)
- return
+ assert False, "Unknown column: %s" % (col.get_title(), )
+ except Exception:
+ _moduleLogger.exception()
def _on_click_category(self, row):
- #Clear out the previous list of units
- self._unitModel = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING, gobject.TYPE_STRING)
- self._unitsView.set_model(self._unitModel)
-
- #Colourize each row alternately for easier reading
- self._unitsView.set_property('rules_hint', 1)
-
#Clear out the description
text_model = gtk.TextBuffer(None)
self._unitDescription.set_buffer(text_model)
del keys[0] # do not display .base_unit description key
#Fill up the units descriptions and clear the value cells
+ self._unitModel.clear()
for key in keys:
iter = self._unitModel.append()
self._unitModel.set(iter, 0, key, 1, '', 2, self._unitDataInCategory[key][1])
+ self._sortedUnitModel.sort_column_changed()
self._unitName.set_text('')
self._unitValue.set_text('')
self._unitValue.grab_focus()
self._unitValue.select_region(0, -1)
- def _on_button_released(self, row, a):
- self._on_click_unit(row)
-
def _on_click_unit(self, row):
self._calcsuppress = True #suppress calculations