YUVCacheTest.cpp 2.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  1. /*
  2. * Copyright 2014 Google Inc.
  3. *
  4. * Use of this source code is governed by a BSD-style license that can be
  5. * found in the LICENSE file.
  6. */
  7. #include "src/core/SkCachedData.h"
  8. #include "src/core/SkResourceCache.h"
  9. #include "src/core/SkYUVPlanesCache.h"
  10. #include "tests/Test.h"
  11. enum LockedState {
  12. kUnlocked,
  13. kLocked,
  14. };
  15. enum CachedState {
  16. kNotInCache,
  17. kInCache,
  18. };
  19. static void check_data(skiatest::Reporter* reporter, SkCachedData* data,
  20. int refcnt, CachedState cacheState, LockedState lockedState) {
  21. REPORTER_ASSERT(reporter, data->testing_only_getRefCnt() == refcnt);
  22. REPORTER_ASSERT(reporter, data->testing_only_isInCache() == (kInCache == cacheState));
  23. bool isLocked = (data->data() != nullptr);
  24. REPORTER_ASSERT(reporter, isLocked == (lockedState == kLocked));
  25. }
  26. DEF_TEST(YUVPlanesCache, reporter) {
  27. SkResourceCache cache(1024);
  28. SkYUVPlanesCache::Info yuvInfo;
  29. for (int i = 0; i < SkYUVASizeInfo::kMaxCount; i++) {
  30. yuvInfo.fSizeInfo.fSizes[i].fWidth = 20 * (i + 1);
  31. yuvInfo.fSizeInfo.fSizes[i].fHeight = 10 * (i + 1);
  32. yuvInfo.fSizeInfo.fWidthBytes[i] = 80 * (i + 1);
  33. }
  34. for (int i = 0; i < SkYUVAIndex::kIndexCount; ++i) {
  35. yuvInfo.fYUVAIndices[i].fIndex = -1;
  36. yuvInfo.fYUVAIndices[i].fChannel = SkColorChannel::kR;
  37. }
  38. yuvInfo.fColorSpace = kRec601_SkYUVColorSpace;
  39. const uint32_t genID = 12345678;
  40. SkCachedData* data = SkYUVPlanesCache::FindAndRef(genID, &yuvInfo, &cache);
  41. REPORTER_ASSERT(reporter, nullptr == data);
  42. size_t size = 256;
  43. data = cache.newCachedData(size);
  44. memset(data->writable_data(), 0xff, size);
  45. SkYUVPlanesCache::Add(genID, data, &yuvInfo, &cache);
  46. check_data(reporter, data, 2, kInCache, kLocked);
  47. data->unref();
  48. check_data(reporter, data, 1, kInCache, kUnlocked);
  49. SkYUVPlanesCache::Info yuvInfoRead;
  50. data = SkYUVPlanesCache::FindAndRef(genID, &yuvInfoRead, &cache);
  51. REPORTER_ASSERT(reporter, data);
  52. REPORTER_ASSERT(reporter, data->size() == size);
  53. REPORTER_ASSERT(reporter, yuvInfo.fSizeInfo == yuvInfoRead.fSizeInfo);
  54. for (int i = 0; i < SkYUVAIndex::kIndexCount; ++i) {
  55. REPORTER_ASSERT(reporter, yuvInfo.fYUVAIndices[i] == yuvInfoRead.fYUVAIndices[i]);
  56. }
  57. REPORTER_ASSERT(reporter, yuvInfo.fColorSpace == yuvInfoRead.fColorSpace);
  58. check_data(reporter, data, 2, kInCache, kLocked);
  59. cache.purgeAll();
  60. check_data(reporter, data, 1, kNotInCache, kLocked);
  61. data->unref();
  62. }