Home | History | Annotate | Download | only in toolbar 
      1 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #ifndef CHROME_BROWSER_UI_TOOLBAR_WRENCH_ICON_PAINTER_H_
      6 #define CHROME_BROWSER_UI_TOOLBAR_WRENCH_ICON_PAINTER_H_
      7 
      8 #include "base/compiler_specific.h"
      9 #include "base/gtest_prod_util.h"
     10 #include "base/memory/scoped_ptr.h"
     11 #include "chrome/browser/ui/global_error/global_error.h"
     12 #include "chrome/browser/upgrade_detector.h"
     13 #include "ui/base/animation/animation_delegate.h"
     14 #include "ui/gfx/image/image_skia.h"
     15 
     16 namespace gfx {
     17 class Canvas;
     18 class Rect;
     19 }
     20 namespace ui {
     21 class MultiAnimation;
     22 class ThemeProvider;
     23 }
     24 
     25 // This class is used to draw the wrench icon. It can signify severity levels
     26 // by changing the wrench icon to different colors.
     27 class WrenchIconPainter : ui::AnimationDelegate {
     28  public:
     29   enum BezelType {
     30     BEZEL_NONE,
     31     BEZEL_HOVER,
     32     BEZEL_PRESSED,
     33   };
     34 
     35   enum Severity {
     36     SEVERITY_NONE,
     37     SEVERITY_LOW,
     38     SEVERITY_MEDIUM,
     39     SEVERITY_HIGH,
     40   };
     41 
     42   class Delegate {
     43    public:
     44     virtual void ScheduleWrenchIconPaint() = 0;
     45    protected:
     46     virtual ~Delegate() {}
     47   };
     48 
     49   // Map an upgrade level to a severity level.
     50   static Severity SeverityFromUpgradeLevel(
     51       UpgradeDetector::UpgradeNotificationAnnoyanceLevel level);
     52 
     53   // Checks if the wrench icon should be animated for the given upgrade level.
     54   static bool ShouldAnimateUpgradeLevel(
     55       UpgradeDetector::UpgradeNotificationAnnoyanceLevel level);
     56 
     57   // Get the severity level for global errors.
     58   static Severity GlobalErrorSeverity();
     59 
     60   explicit WrenchIconPainter(Delegate* delegate);
     61   virtual ~WrenchIconPainter();
     62 
     63   // If |severity| is not |SEVERITY_NONE| then the wrench icon is colored to
     64   // match the severity level.
     65   void SetSeverity(Severity severity, bool animate);
     66 
     67   // A badge drawn on the top left.
     68   void set_badge(const gfx::ImageSkia& badge) { badge_ = badge; }
     69 
     70   void Paint(gfx::Canvas* canvas,
     71              ui::ThemeProvider* theme_provider,
     72              const gfx::Rect& rect,
     73              BezelType bezel_type);
     74 
     75  private:
     76   FRIEND_TEST_ALL_PREFIXES(WrenchIconPainterTest, PaintCallback);
     77 
     78   // AnimationDelegate:
     79   virtual void AnimationProgressed(const ui::Animation* animation) OVERRIDE;
     80 
     81   // Gets the image ID used to draw bars for the current severity level.
     82   int GetCurrentSeverityImageID() const;
     83 
     84   Delegate* delegate_;
     85   Severity severity_;
     86   gfx::ImageSkia badge_;
     87   scoped_ptr<ui::MultiAnimation> animation_;
     88 
     89   DISALLOW_COPY_AND_ASSIGN(WrenchIconPainter);
     90 };
     91 
     92 #endif  // CHROME_BROWSER_UI_TOOLBAR_WRENCH_ICON_PAINTER_H_
     93