skia2/tests/AtomicTest.cpp
2015-09-29 11:47:45 -07:00

67 lines
1.7 KiB
C++

/*
* Copyright 2012 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include "SkAtomics.h"
#include "SkThreadUtils.h"
#include "SkTypes.h"
#include "Test.h"
struct AddInfo {
int32_t valueToAdd;
int timesToAdd;
unsigned int processorAffinity;
};
static int32_t base = 0;
static AddInfo gAdds[] = {
{ 3, 100, 23 },
{ 2, 200, 2 },
{ 7, 150, 17 },
};
static void addABunchOfTimes(void* data) {
AddInfo* addInfo = static_cast<AddInfo*>(data);
for (int i = 0; i < addInfo->timesToAdd; i++) {
sk_atomic_add(&base, addInfo->valueToAdd);
}
}
DEF_TEST(Atomic, reporter) {
int32_t total = base;
SkThread* threads[SK_ARRAY_COUNT(gAdds)];
for (size_t i = 0; i < SK_ARRAY_COUNT(gAdds); i++) {
total += gAdds[i].valueToAdd * gAdds[i].timesToAdd;
}
// Start the threads
for (size_t i = 0; i < SK_ARRAY_COUNT(gAdds); i++) {
threads[i] = new SkThread(addABunchOfTimes, &gAdds[i]);
threads[i]->setProcessorAffinity(gAdds[i].processorAffinity);
threads[i]->start();
}
// Now end the threads
for (size_t i = 0; i < SK_ARRAY_COUNT(gAdds); i++) {
threads[i]->join();
delete threads[i];
}
REPORTER_ASSERT(reporter, total == base);
// Ensure that the returned value from sk_atomic_add is correct.
int32_t valueToModify = 3;
const int32_t originalValue = valueToModify;
REPORTER_ASSERT(reporter, originalValue == sk_atomic_add(&valueToModify, 7));
{
SkAtomic<int> v {0};
REPORTER_ASSERT(reporter, 0 == v.load());
v = 10;
REPORTER_ASSERT(reporter, 10 == v.load());
int q = v;
REPORTER_ASSERT(reporter, 10 == q);
}
}