Implement IntArrayView.copyToBuffer().
Bug: 14425059 Change-Id: Ic1b7de72701eae0c0b1bb28d12514099446a66a9
This commit is contained in:
parent
4634b75d34
commit
a3b0eb1685
2 changed files with 30 additions and 1 deletions
|
@ -17,8 +17,9 @@
|
||||||
#ifndef LATINIME_INT_ARRAY_VIEW_H
|
#ifndef LATINIME_INT_ARRAY_VIEW_H
|
||||||
#define LATINIME_INT_ARRAY_VIEW_H
|
#define LATINIME_INT_ARRAY_VIEW_H
|
||||||
|
|
||||||
|
#include <array>
|
||||||
#include <cstdint>
|
#include <cstdint>
|
||||||
#include <cstdlib>
|
#include <cstring>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
#include "defines.h"
|
#include "defines.h"
|
||||||
|
@ -103,6 +104,12 @@ class IntArrayView {
|
||||||
return IntArrayView(mPtr + n, mSize - n);
|
return IntArrayView(mPtr + n, mSize - n);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
template <size_t N>
|
||||||
|
void copyToArray(std::array<int, N> *const buffer, const size_t offset) const {
|
||||||
|
ASSERT(mSize + offset <= N);
|
||||||
|
memmove(buffer->data() + offset, mPtr, sizeof(int) * mSize);
|
||||||
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
DISALLOW_ASSIGNMENT_OPERATOR(IntArrayView);
|
DISALLOW_ASSIGNMENT_OPERATOR(IntArrayView);
|
||||||
|
|
||||||
|
|
|
@ -18,6 +18,7 @@
|
||||||
|
|
||||||
#include <gtest/gtest.h>
|
#include <gtest/gtest.h>
|
||||||
|
|
||||||
|
#include <array>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
namespace latinime {
|
namespace latinime {
|
||||||
|
@ -89,5 +90,26 @@ TEST(IntArrayViewTest, TestSkip) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
TEST(IntArrayViewTest, TestCopyToArray) {
|
||||||
|
// "{{" to suppress warning.
|
||||||
|
std::array<int, 7> buffer = {{10, 20, 30, 40, 50, 60, 70}};
|
||||||
|
const std::vector<int> intVector = {3, 2, 1, 0, -1, -2};
|
||||||
|
IntArrayView intArrayView(intVector);
|
||||||
|
intArrayView.limit(0).copyToArray(&buffer, 0);
|
||||||
|
EXPECT_EQ(10, buffer[0]);
|
||||||
|
EXPECT_EQ(20, buffer[1]);
|
||||||
|
intArrayView.limit(1).copyToArray(&buffer, 0);
|
||||||
|
EXPECT_EQ(intVector[0], buffer[0]);
|
||||||
|
EXPECT_EQ(20, buffer[1]);
|
||||||
|
intArrayView.limit(1).copyToArray(&buffer, 1);
|
||||||
|
EXPECT_EQ(intVector[0], buffer[0]);
|
||||||
|
EXPECT_EQ(intVector[0], buffer[1]);
|
||||||
|
intArrayView.copyToArray(&buffer, 0);
|
||||||
|
for (size_t i = 0; i < intArrayView.size(); ++i) {
|
||||||
|
EXPECT_EQ(intVector[i], buffer[i]);
|
||||||
|
}
|
||||||
|
EXPECT_EQ(70, buffer[6]);
|
||||||
|
}
|
||||||
|
|
||||||
} // namespace
|
} // namespace
|
||||||
} // namespace latinime
|
} // namespace latinime
|
||||||
|
|
Loading…
Reference in a new issue