// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "ui/views/examples/native_theme_button_example.h"

#include <string>

#include "base/logging.h"
#include "base/stringprintf.h"
#include "base/utf_string_conversions.h"
#include "ui/base/animation/throb_animation.h"
#include "ui/base/models/combobox_model.h"
#include "ui/gfx/canvas.h"
#include "ui/views/controls/combobox/combobox.h"
#include "ui/views/controls/label.h"
#include "ui/views/examples/example_combobox_model.h"
#include "ui/views/layout/grid_layout.h"
#include "ui/views/native_theme_painter.h"

namespace {

const char* kParts[] = {
    "PushButton",
    "RadioButton",
    "Checkbox",
};

const char* kStates[] = {
    "Disabled",
    "Normal",
    "Hot",
    "Pressed",
    "<Dynamic>",
};

}  // namespace

namespace views {
namespace examples {

ExampleNativeThemeButton::ExampleNativeThemeButton(
    ButtonListener* listener,
    Combobox* cb_part,
    Combobox* cb_state)
    : CustomButton(listener),
      cb_part_(cb_part),
      cb_state_(cb_state),
      count_(0),
      is_checked_(false),
      is_indeterminate_(false) {
  cb_part_->set_listener(this);
  cb_state_->set_listener(this);

  painter_.reset(new NativeThemePainter(this));
  set_background(Background::CreateBackgroundPainter(
      false, painter_.get()));
}

ExampleNativeThemeButton::~ExampleNativeThemeButton() {
}

std::string ExampleNativeThemeButton::MessWithState() {
  const char* message = NULL;
  switch (GetThemePart()) {
    case gfx::NativeTheme::kPushButton:
      message = "Pressed! count:%d";
      break;
    case gfx::NativeTheme::kRadio:
      is_checked_ = !is_checked_;
      message = is_checked_ ? "Checked! count:%d" : "Unchecked! count:%d";
      break;
    case gfx::NativeTheme::kCheckbox:
      if (is_indeterminate_) {
        is_checked_ = false;
        is_indeterminate_ = false;
      } else if (!is_checked_) {
        is_checked_ = true;
      } else {
        is_checked_ = false;
        is_indeterminate_ = true;
      }
      if (is_checked_)
        message = "Checked! count:%d";
      else if (is_indeterminate_)
        message = "Indeterminate! count:%d";
      else
        message = "Unchecked! count:%d";
      break;
    default:
      NOTREACHED();
  }

  return base::StringPrintf(message, ++count_);
}

void ExampleNativeThemeButton::OnSelectedIndexChanged(Combobox* combobox) {
  SchedulePaint();
}

gfx::NativeTheme::Part ExampleNativeThemeButton::GetThemePart() const {
  switch (cb_part_->selected_index()) {
    case 0:
      return gfx::NativeTheme::kPushButton;
    case 1:
      return gfx::NativeTheme::kRadio;
    case 2:
      return gfx::NativeTheme::kCheckbox;
    default:
      NOTREACHED();
  }
  return gfx::NativeTheme::kPushButton;
}

gfx::Rect ExampleNativeThemeButton::GetThemePaintRect() const {
  gfx::NativeTheme::ExtraParams extra;
  gfx::NativeTheme::State state = GetThemeState(&extra);
  gfx::Size size(gfx::NativeTheme::instance()->GetPartSize(GetThemePart(),
                                                           state,
                                                           extra));
  gfx::Rect rect(size);
  rect.set_x(GetMirroredXForRect(rect));
  return rect;
}

gfx::NativeTheme::State ExampleNativeThemeButton::GetThemeState(
    gfx::NativeTheme::ExtraParams* params) const {
  GetExtraParams(params);

  int selected = cb_state_->selected_index();
  if (selected > 3) {
    switch (state()) {
      case BS_DISABLED:
        return gfx::NativeTheme::kDisabled;
      case BS_NORMAL:
        return gfx::NativeTheme::kNormal;
      case BS_HOT:
        return gfx::NativeTheme::kHovered;
      case BS_PUSHED:
        return gfx::NativeTheme::kPressed;
      default:
        NOTREACHED();
    }
  }

  switch (selected) {
    case 0:
      return gfx::NativeTheme::kDisabled;
    case 1:
      return gfx::NativeTheme::kNormal;
    case 2:
      return gfx::NativeTheme::kHovered;
    case 3:
      return gfx::NativeTheme::kPressed;
    default:
      NOTREACHED();
  }
  return gfx::NativeTheme::kNormal;
}

void ExampleNativeThemeButton::GetExtraParams(
    gfx::NativeTheme::ExtraParams* params) const {

  params->button.checked = is_checked_;
  params->button.indeterminate = is_indeterminate_;
  params->button.is_default = false;
  params->button.has_border = false;
  params->button.classic_state = 0;
  params->button.background_color = SkColorSetARGB(0, 0, 0, 0);
}

const ui::Animation* ExampleNativeThemeButton::GetThemeAnimation() const {
  return cb_state_->selected_index() <= 3 ? NULL : hover_animation_.get();
}

gfx::NativeTheme::State ExampleNativeThemeButton::GetBackgroundThemeState(
    gfx::NativeTheme::ExtraParams* params) const {
  GetExtraParams(params);
  return gfx::NativeTheme::kNormal;
}

gfx::NativeTheme::State ExampleNativeThemeButton::GetForegroundThemeState(
    gfx::NativeTheme::ExtraParams* params) const {
  GetExtraParams(params);
  return gfx::NativeTheme::kHovered;
}

gfx::Size ExampleNativeThemeButton::GetPreferredSize() {
  return painter_.get() == NULL ? gfx::Size() : painter_->GetPreferredSize();
}

void ExampleNativeThemeButton::OnPaintBackground(gfx::Canvas* canvas) {
  // Fill the background with a known colour so that we know where the bounds
  // of the View are.
  canvas->FillRect(GetLocalBounds(), SkColorSetRGB(255, 128, 128));
  CustomButton::OnPaintBackground(canvas);
}

////////////////////////////////////////////////////////////////////////////////

NativeThemeButtonExample::NativeThemeButtonExample()
    : ExampleBase("Native Theme Button") {
}

NativeThemeButtonExample::~NativeThemeButtonExample() {
}

void NativeThemeButtonExample::CreateExampleView(View* container) {
  GridLayout* layout = new GridLayout(container);
  container->SetLayoutManager(layout);

  layout->AddPaddingRow(0, 8);

  ColumnSet* column_set = layout->AddColumnSet(0);
  column_set->AddPaddingColumn(0, 8);
  column_set->AddColumn(GridLayout::LEADING, GridLayout::FILL,
                        0.1f, GridLayout::USE_PREF, 0, 0);
  column_set->AddColumn(GridLayout::FILL, GridLayout::FILL,
                        0.9f, GridLayout::USE_PREF, 0, 0);
  column_set->AddPaddingColumn(0, 8);

  layout->StartRow(0, 0);
  layout->AddView(new Label(ASCIIToUTF16("Part:")));
  combobox_model_part_.reset(
      new ExampleComboboxModel(kParts, arraysize(kParts)));
  Combobox* combobox_part = new Combobox(combobox_model_part_.get());
  combobox_part->SetSelectedIndex(0);
  layout->AddView(combobox_part);

  layout->StartRow(0, 0);
  layout->AddView(new Label(ASCIIToUTF16("State:")));
  combobox_model_state_.reset(
      new ExampleComboboxModel(kStates, arraysize(kStates)));
  Combobox* combobox_state = new Combobox(combobox_model_state_.get());
  combobox_state->SetSelectedIndex(0);
  layout->AddView(combobox_state);

  layout->AddPaddingRow(0, 32);

  button_ = new ExampleNativeThemeButton(this, combobox_part, combobox_state);

  column_set = layout->AddColumnSet(1);
  column_set->AddPaddingColumn(0, 16);
  column_set->AddColumn(GridLayout::FILL, GridLayout::FILL,
                        1, GridLayout::USE_PREF, 0, 0);
  column_set->AddPaddingColumn(0, 16);
  layout->StartRow(1, 1);
  layout->AddView(button_);

  layout->AddPaddingRow(0, 8);
}

void NativeThemeButtonExample::ButtonPressed(Button* sender,
                                             const Event& event) {
  PrintStatus(button_->MessWithState().c_str());
}

}  // namespace examples
}  // namespace views