Add option to bitmap_viewer to save all snapshots at once.

scripts/newbitmaps/lib/pixcontrol.py

 #!/usr/bin/python -tt
 # Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """Edit buttons for bmpblock object"""
 
 import wx
 
 class Frame(wx.Frame):
 
   def __init__(self, bmpblock=None, title=None):
     wx.Frame.__init__(self, None, wx.ID_ANY, title, size=(200,400))
     menuFile = wx.Menu()
     m_about = menuFile.Append(wx.ID_ANY, "About...\tCtrl+A")
     menuFile.AppendSeparator()
     m_reload = menuFile.Append(wx.ID_ANY, "Reload\tCtrl+R")
     m_snapshot = menuFile.Append(wx.ID_ANY, "Save snapshot")
+    m_snapshotall = menuFile.Append(wx.ID_ANY, "Save snapshot of all screens")
     m_quit = menuFile.Append(wx.ID_ANY, "Quit\tCtrl+Q")
     menuBar = wx.MenuBar()
     menuBar.Append(menuFile, "&File")
     self.SetMenuBar(menuBar)
     self.CreateStatusBar()
     self.Bind(wx.EVT_MENU, self.OnAbout, m_about)
     self.Bind(wx.EVT_MENU, self.OnReload, m_reload)
     self.Bind(wx.EVT_MENU, self.OnSaveit, m_snapshot)
+    self.Bind(wx.EVT_MENU, self.OnSaveAll, m_snapshotall)
     self.Bind(wx.EVT_MENU, self.OnQuit, m_quit)
     self.Bind(wx.EVT_CLOSE, self.OnQuit)
 
     acctbl = wx.AcceleratorTable([
       (wx.ACCEL_CTRL, ord('A'), m_about.GetId()),
       (wx.ACCEL_CTRL, ord('R'), m_reload.GetId()),
       (wx.ACCEL_CTRL, ord('Q'), m_quit.GetId())
       ])
 
     self.SetAcceleratorTable(acctbl)
@@ skipping 62 matching lines @@
     current = self.bmpblock.current_screen
     self.screenlist.SetStringSelection(current)
     self.SetStatusText(self.bmpblock.filename)
 
   def OnIdle(self, event=None):
     """What to do, what to do..."""
     if self.do_update:
       # FIXME: The model should know when to do this itself, right?
       self.bmpblock.Redisplay()
       self.do_update = False
+
+  def OnSaveAll(self, event=None):
+    """Save snapshots of all screens"""
+    start = self.bmpblock.current_screen
+    thinglist = self.screenlist.GetItems()
+    for thing in thinglist:
+      self.bmpblock.current_screen = thing
+      self.bmpblock.Redisplay()
+      self.bmpblock.Saveit()
+    self.bmpblock.current_screen = start
+    self.do_update = True