Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 /* | 1 /* |
| 2 * Copyright 2015 Google Inc. | 2 * Copyright 2015 Google Inc. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license that can be | 4 * Use of this source code is governed by a BSD-style license that can be |
| 5 * found in the LICENSE file. | 5 * found in the LICENSE file. |
| 6 */ | 6 */ |
| 7 | 7 |
| 8 #include "Test.h" | 8 #include "Test.h" |
| 9 #include "SkCommonFlags.h" | |
|
Kimmo Kinnunen
2015/10/27 14:23:12
forgot to remove this..
| |
| 9 #include "SkOncePtr.h" | 10 #include "SkOncePtr.h" |
| 10 #include "SkTaskGroup.h" | 11 #include "SkTaskGroup.h" |
| 11 | 12 |
| 12 DEF_TEST(OncePtr, r) { | 13 DEF_TEST(OncePtr, r) { |
| 13 SkOncePtr<int> once; | 14 SkOncePtr<int> once; |
| 14 | 15 |
| 15 static SkAtomic<int> calls(0); | 16 static SkAtomic<int> calls(0); |
| 16 auto create = [&] { | 17 auto create = [&] { |
| 17 calls.fetch_add(1); | 18 calls.fetch_add(1); |
| 18 return new int(5); | 19 return new int(5); |
| 19 }; | 20 }; |
| 20 | 21 |
| 21 SkAtomic<int> force_a_race(sk_num_cores()); | 22 SkAtomic<int> force_a_race(sk_parallel_for_thread_count()); |
| 22 | 23 if (force_a_race < 1) { |
| 24 return; | |
| 25 } | |
| 23 sk_parallel_for(sk_num_cores()*4, [&](size_t) { | 26 sk_parallel_for(sk_num_cores()*4, [&](size_t) { |
| 24 force_a_race.fetch_add(-1); | 27 force_a_race.fetch_add(-1); |
| 25 while (force_a_race.load() > 0); | 28 while (force_a_race.load() > 0); |
| 26 | 29 |
| 27 int* n = once.get(create); | 30 int* n = once.get(create); |
| 28 REPORTER_ASSERT(r, *n == 5); | 31 REPORTER_ASSERT(r, *n == 5); |
| 29 }); | 32 }); |
| 30 REPORTER_ASSERT(r, calls.load() == 1); | 33 REPORTER_ASSERT(r, calls.load() == 1); |
| 31 } | 34 } |
| 32 | 35 |
| 33 /* TODO(mtklein): next CL | 36 /* TODO(mtklein): next CL |
| 34 | 37 |
| 35 SK_DECLARE_STATIC_ONCE(once_noptr); | 38 SK_DECLARE_STATIC_ONCE(once_noptr); |
| 36 DEF_TEST(OnceNoPtr, r) { | 39 DEF_TEST(OnceNoPtr, r) { |
| 37 static SkAtomic<int> calls(0); | 40 static SkAtomic<int> calls(0); |
| 38 | 41 |
| 39 SkAtomic<int> force_a_race(sk_num_cores()); | 42 SkAtomic<int> force_a_race(sk_num_cores()); |
| 40 sk_parallel_for(sk_num_cores()*4, [&](size_t) { | 43 sk_parallel_for(sk_num_cores()*4, [&](size_t) { |
| 41 force_a_race.fetch_add(-1); | 44 force_a_race.fetch_add(-1); |
| 42 while (force_a_race.load() > 0); | 45 while (force_a_race.load() > 0); |
| 43 | 46 |
| 44 SkOnce(&once_noptr, [&] { calls.fetch_add(1); }); | 47 SkOnce(&once_noptr, [&] { calls.fetch_add(1); }); |
| 45 }); | 48 }); |
| 46 REPORTER_ASSERT(r, calls.load() == 1); | 49 REPORTER_ASSERT(r, calls.load() == 1); |
| 47 } | 50 } |
| 48 */ | 51 */ |
| OLD | NEW |