category.cc 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  1. // Copyright 2016 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. #include "components/ntp_snippets/category.h"
  5. #include "base/check_op.h"
  6. namespace ntp_snippets {
  7. // static
  8. Category Category::FromKnownCategory(KnownCategories known_category) {
  9. return FromIDValue(static_cast<int>(known_category));
  10. }
  11. // static
  12. Category Category::FromRemoteCategory(int remote_category) {
  13. DCHECK_GT(remote_category, 0);
  14. return Category(static_cast<int>(KnownCategories::REMOTE_CATEGORIES_OFFSET) +
  15. remote_category);
  16. }
  17. // static
  18. Category Category::FromIDValue(int id) {
  19. DCHECK(IsValidIDValue(id)) << id
  20. << " is not a valid category ID. This may have "
  21. "been caused by removal of a local "
  22. "KnownCategory.";
  23. return Category(id);
  24. }
  25. // static
  26. bool Category::IsValidIDValue(int id) {
  27. return (id >= 0) &&
  28. ((id < static_cast<int>(KnownCategories::LOCAL_CATEGORIES_COUNT) ||
  29. id > static_cast<int>(KnownCategories::REMOTE_CATEGORIES_OFFSET)));
  30. }
  31. Category::Category(int id) : id_(id) {}
  32. int Category::remote_id() const {
  33. DCHECK_GT(id_, static_cast<int>(KnownCategories::REMOTE_CATEGORIES_OFFSET));
  34. return id_ - static_cast<int>(KnownCategories::REMOTE_CATEGORIES_OFFSET);
  35. }
  36. bool Category::IsKnownCategory(KnownCategories known_category) const {
  37. DCHECK_NE(known_category, KnownCategories::LOCAL_CATEGORIES_COUNT);
  38. DCHECK_NE(known_category, KnownCategories::REMOTE_CATEGORIES_OFFSET);
  39. return id_ == static_cast<int>(known_category);
  40. }
  41. bool operator==(const Category& left, const Category& right) {
  42. return left.id() == right.id();
  43. }
  44. bool operator!=(const Category& left, const Category& right) {
  45. return !(left == right);
  46. }
  47. bool Category::CompareByID::operator()(const Category& left,
  48. const Category& right) const {
  49. return left.id() < right.id();
  50. }
  51. std::ostream& operator<<(std::ostream& os, const Category& obj) {
  52. os << obj.id();
  53. return os;
  54. }
  55. } // namespace ntp_snippets