1 files changed, 226 insertions, 0 deletions

diff --git a/bitbake/lib/bb/ui/crumbs/packageselectionpage.py b/bitbake/lib/bb/ui/crumbs/packageselectionpage.py
new file mode 100755
index 0000000000..8a8ab7585e
--- /dev/null
+++ b/bitbake/lib/bb/ui/crumbs/packageselectionpage.py
@@ -0,0 +1,226 @@
+#!/usr/bin/env python
+#
+# BitBake Graphical GTK User Interface
+#
+# Copyright (C) 2012        Intel Corporation
+#
+# Authored by Dongxiao Xu <dongxiao.xu@intel.com>
+# Authored by Shane Wang <shane.wang@intel.com>
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License version 2 as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+
+import gtk
+import glib
+from bb.ui.crumbs.hobcolor import HobColors
+from bb.ui.crumbs.hobwidget import HobViewBar, HobViewTable
+from bb.ui.crumbs.hoblistmodel import PackageListModel
+from bb.ui.crumbs.hobpages import HobPage
+
+#
+# PackageSelectionPage
+#
+class PackageSelectionPage (HobPage):
+
+    pages = [
+        {
+         'name'    : 'All packages',
+         'filter'  : {},
+         'columns' : [{
+                       'col_name' : 'Name',
+                       'col_id'   : PackageListModel.COL_NAME,
+                       'col_style': 'text',
+                       'col_min'  : 100,
+                       'col_max'  : 400
+                      }, {
+                       'col_name' : 'size',
+                       'col_id'   : PackageListModel.COL_SIZE,
+                       'col_style': 'text',
+                       'col_min'  : 100,
+                       'col_max'  : 500
+                      }, {
+                       'col_name' : 'Included',
+                       'col_id'   : PackageListModel.COL_INC,
+                       'col_style': 'toggle',
+                       'col_min'  : 50,
+                       'col_max'  : 50
+                      }]
+        }, {
+         'name'    : 'Included',
+         'filter'  : { PackageListModel.COL_INC : [True] },
+         'columns' : [{
+                       'col_name' : 'Name',
+                       'col_id'   : PackageListModel.COL_NAME,
+                       'col_style': 'text',
+                       'col_min'  : 100,
+                       'col_max'  : 300
+                      }, {
+                       'col_name' : 'Brought by',
+                       'col_id'   : PackageListModel.COL_BINB,
+                       'col_style': 'text',
+                       'col_min'  : 100,
+                       'col_max'  : 350
+                      }, {
+                       'col_name' : 'size',
+                       'col_id'   : PackageListModel.COL_SIZE,
+                       'col_style': 'text',
+                       'col_min'  : 100,
+                       'col_max'  : 300
+                      }, {
+                       'col_name' : 'Included',
+                       'col_id'   : PackageListModel.COL_INC,
+                       'col_style': 'toggle',
+                       'col_min'  : 50,
+                       'col_max'  : 50
+                     }]
+        }
+    ]
+
+    def __init__(self, builder):
+        super(PackageSelectionPage, self).__init__(builder, "Package Selection")
+
+        # set invisiable members
+        self.package_model = self.builder.package_model
+
+        # create visual elements
+        self.create_visual_elements()
+
+    def create_visual_elements(self):
+        self.label = gtk.Label("Packages included: 0\nSelected packages size: 0 MB")
+        self.eventbox = self.add_onto_top_bar(self.label, 73)
+        self.pack_start(self.eventbox, expand=False, fill=False)
+        self.pack_start(self.group_align, expand=True, fill=True)
+
+        # set visiable members
+        self.grid = gtk.Table(10, 1, True)
+        self.grid.set_col_spacings(3)
+
+        self.ins = gtk.Notebook()
+        self.ins.set_show_tabs(False)
+        self.tables = [] # we need to modify table when the dialog is shown
+        # append the tab
+        for i in range(len(self.pages)):
+            columns = self.pages[i]['columns']
+            tab = HobViewTable(columns, self.reset_clicked_cb, self.table_toggled_cb)
+            filter = self.pages[i]['filter']
+            tab.table_tree.set_model(self.package_model.tree_model(filter))
+            label = gtk.Label(self.pages[i]['name'])
+            self.ins.append_page(tab, label)
+            self.tables.append(tab)
+
+        self.grid.attach(self.ins, 0, 1, 1, 10, gtk.FILL | gtk.EXPAND, gtk.FILL | gtk.EXPAND, 1, 1)
+        # a black bar associated with the notebook
+        self.topbar = HobViewBar(self.ins) 
+        self.grid.attach(self.topbar, 0, 1, 0, 1, gtk.FILL | gtk.EXPAND, gtk.FILL | gtk.EXPAND, 1, 1)
+        # set the search entry for each table
+        for tab in self.tables:
+            tab.table_tree.set_search_entry(self.topbar.search)
+
+        inctab_tree_view = self.tables[len(self.pages)-1].table_tree
+        inctab_tree_selection = inctab_tree_view.get_selection()
+        inctab_tree_selection.connect("changed", self.tree_selection_cb, inctab_tree_view)
+
+        # add all into the dialog
+        self.box_group_area.add(self.grid)
+
+        button_box = gtk.HBox(False, 5)
+        self.box_group_area.pack_start(button_box, expand=False, fill=False)
+
+        self.build_image_button = gtk.Button()
+        label = gtk.Label()
+        mark = "<span %s>Build image</span>" % self.span_tag('24px', 'bold')
+        label.set_markup(mark)
+        self.build_image_button.set_image(label)
+        self.build_image_button.set_size_request(205, 49)
+        self.build_image_button.modify_bg(gtk.STATE_NORMAL, gtk.gdk.Color(HobColors.ORANGE))
+        self.build_image_button.modify_bg(gtk.STATE_PRELIGHT, gtk.gdk.Color(HobColors.ORANGE))
+        self.build_image_button.modify_bg(gtk.STATE_SELECTED, gtk.gdk.Color(HobColors.ORANGE))
+        self.build_image_button.set_tooltip_text("Build image to get your target image")
+        self.build_image_button.set_flags(gtk.CAN_DEFAULT)
+        self.build_image_button.grab_default()
+        self.build_image_button.connect("clicked", self.build_image_clicked_cb)
+        button_box.pack_end(self.build_image_button, expand=False, fill=False)
+
+        self.back_button = gtk.LinkButton("Go back to Image Configuration screen", "<< Back to image configuration")
+        self.back_button.connect("clicked", self.back_button_clicked_cb)
+        button_box.pack_start(self.back_button, expand=False, fill=False)
+
+    def tree_selection_cb(self, tree_selection, tree_view):
+        tree_model = tree_view.get_model()
+        path, column = tree_view.get_cursor()
+        if not path or column == tree_view.get_column(2):
+            return
+
+        it = tree_model.get_iter(path)
+        binb = tree_model.get_value(it, PackageListModel.COL_BINB)
+        if binb:
+            self.builder.show_binb_dialog(binb)
+
+    def build_image_clicked_cb(self, button):
+        self.builder.build_image()
+
+    def back_button_clicked_cb(self, button):
+        self.builder.show_configuration()
+
+    def _expand_all(self):
+        for tab in self.tables:
+            tab.table_tree.expand_all()
+
+    def refresh_selection(self):
+        self._expand_all()
+
+        self.builder.configuration.selected_packages = self.package_model.get_selected_packages()
+        selected_packages_num = len(self.builder.configuration.selected_packages)
+        selected_packages_size = float(self.package_model.get_packages_size())
+        selected_packages_size_str = self._size_to_string(selected_packages_size)
+
+        image_overhead_factor = self.builder.configuration.image_overhead_factor
+        image_rootfs_size = self.builder.configuration.image_rootfs_size
+        image_extra_size = self.builder.configuration.image_extra_size
+        base_size = image_overhead_factor * selected_packages_size
+        image_total_size = max(base_size, image_rootfs_size) + image_extra_size
+        image_total_size_str = self._size_to_string(image_total_size)
+
+        self.label.set_text("Packages included: %s\nSelected packages size: %s\nTotal image size: %s" %
+                            (selected_packages_num, selected_packages_size_str, image_total_size_str))
+
+    """
+    Helper function to convert the package size to string format.
+    The unit of size is KB
+    """
+    def _size_to_string(self, size):
+        if len(str(int(size))) > 3:
+            size_str = '%.1f' % (size*1.0/1024) + ' MB'
+        else:
+            size_str = str(size) + ' KB'
+        return size_str
+
+    # Callback functions
+    def reset_clicked_cb(self, button):
+        self.package_model.reset()
+        self.builder.reset_package_model()
+
+    def toggle_item_idle_cb(self, path):
+        if not self.package_model.path_included(path):
+            self.package_model.include_item(item_path=path, binb="User Selected")
+        else:
+            self.package_model.exclude_item(item_path=path)
+
+        self.builder.window_sensitive(True)
+
+    def table_toggled_cb(self, cell, view_path, view_tree):
+        # Click to include a package
+        self.builder.window_sensitive(False)
+        view_model = view_tree.get_model()
+        path = self.package_model.convert_vpath_to_path(view_model, view_path)
+        glib.idle_add(self.toggle_item_idle_cb, path)