1a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org/* 2a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org * Copyright 2014 Google, Inc 3a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org * 4a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org * Use of this source code is governed by a BSD-style license that can be 5a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org * found in the LICENSE file. 6a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org */ 7a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org 8a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org#include "SkSmallAllocator.h" 9a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org#include "SkTypes.h" 10a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org#include "Test.h" 11a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org 12a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.orgclass CountingClass { 13a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.orgpublic: 14a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org CountingClass() { 15a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org kCount++; 16a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org } 17a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org 18a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org ~CountingClass() { 19a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org kCount--; 20a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org } 21a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org 22a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org static int GetCount() { return kCount; } 23a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org 24a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.orgprivate: 25a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org static int kCount; 26a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org}; 27a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org 28a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.orgint CountingClass::kCount; 29a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org 30a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.orgtemplate<uint32_t kMaxObjects, size_t kBytes> void test_allocator(skiatest::Reporter* reporter) { 31a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org { 32a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org SkSmallAllocator<kMaxObjects, kBytes> alloc; 33a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org for (uint32_t i = 0; i < kMaxObjects; ++i) { 34a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org CountingClass* c = alloc.template createT<CountingClass>(); 35a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org REPORTER_ASSERT(reporter, c != NULL); 36a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org REPORTER_ASSERT(reporter, CountingClass::GetCount() == static_cast<int>(i+1)); 37a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org } 38a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org } 39a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org REPORTER_ASSERT(reporter, CountingClass::GetCount() == 0); 40a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org} 41a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org 42a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org// Tests that ensure that the destructor is called, whether the objects 43a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org// were created in fStorage or on the heap. 44a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.orgDEF_TEST(SmallAllocator_destructor, reporter) { 45a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org // Four times as many bytes as objects will never require any heap 46a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org // allocations (since SkAlign4(sizeof(CountingClass)) == 4 and the allocator 47a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org // will stop once it reaches kMaxObjects). 48a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org test_allocator<5, 20>(reporter); 49a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org test_allocator<10, 40>(reporter); 50a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org test_allocator<20, 80>(reporter); 51a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org 52a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org#ifndef SK_DEBUG 53a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org // Allowing less bytes than objects means some will be allocated on the 54a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org // heap. Don't run these in debug where we assert. 55a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org test_allocator<50, 20>(reporter); 56a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org test_allocator<100, 20>(reporter); 57a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org#endif 58a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org} 59a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org 60a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.orgclass Dummy { 61a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org}; 62a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org 63a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.orgclass DummyContainer { 64a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.orgpublic: 65a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org explicit DummyContainer(Dummy* d) 66a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org :fDummy(d) 67a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org {} 68a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org 69a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org Dummy* getDummy() const { return fDummy; } 70a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org 71a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.orgprivate: 72a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org Dummy* fDummy; 73a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org}; 74a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org 75a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org// Test that using a createT with a constructor taking a pointer as a 76a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org// parameter works as expected. 77a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.orgDEF_TEST(SmallAllocator_pointer, reporter) { 78a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org SkSmallAllocator<1, 8> alloc; 79a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org Dummy d; 80a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org DummyContainer* container = alloc.createT<DummyContainer>(&d); 81a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org REPORTER_ASSERT(reporter, container != NULL); 82a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org REPORTER_ASSERT(reporter, container->getDummy() == &d); 83a5572e5bb2a2bbeeb59de0741c2527869d365a0ccommit-bot@chromium.org} 84