|
Packit Service |
7f3b24 |
// SPDX-License-Identifier: BSD-3-Clause
|
|
Packit Service |
7f3b24 |
/* Copyright (C) 2018 - 2020 Intel Corporation. */
|
|
Packit |
345191 |
|
|
Packit |
345191 |
#include "pmem_allocator.h"
|
|
Packit |
345191 |
|
|
Packit |
345191 |
#include <sys/stat.h>
|
|
Packit |
345191 |
#include <iostream>
|
|
Packit |
345191 |
#include <vector>
|
|
Packit |
345191 |
#include <list>
|
|
Packit |
345191 |
#include <map>
|
|
Packit |
345191 |
#include <string>
|
|
Packit |
345191 |
#include <scoped_allocator>
|
|
Packit |
345191 |
#include <cassert>
|
|
Packit |
345191 |
|
|
Packit |
345191 |
#define STL_VECTOR_TEST
|
|
Packit |
345191 |
#define STL_LIST_TEST
|
|
Packit |
345191 |
#if _GLIBCXX_USE_CXX11_ABI
|
|
Packit |
345191 |
#define STL_VEC_STRING_TEST
|
|
Packit |
345191 |
#define STL_MAP_INT_STRING_TEST
|
|
Packit |
345191 |
#endif
|
|
Packit |
345191 |
|
|
Packit |
345191 |
void cpp_allocator_test(const char *pmem_directory)
|
|
Packit |
345191 |
{
|
|
Packit |
345191 |
std::cout << "TEST SCOPE: HELLO" << std::endl;
|
|
Packit |
345191 |
|
|
Packit |
345191 |
size_t pmem_max_size = 1024*1024*1024;
|
|
Packit |
345191 |
|
|
Packit |
345191 |
#ifdef STL_VECTOR_TEST
|
|
Packit |
345191 |
{
|
|
Packit |
345191 |
std::cout << "VECTOR OPEN" << std::endl;
|
|
Packit |
345191 |
libmemkind::pmem::allocator<int> alc{ pmem_directory, pmem_max_size };
|
|
Packit |
345191 |
std::vector<int, libmemkind::pmem::allocator<int>> vector{ alc };
|
|
Packit |
345191 |
|
|
Packit |
345191 |
for (int i = 0; i < 20; ++i) {
|
|
Packit |
345191 |
vector.push_back(0xDEAD + i);
|
|
Packit |
345191 |
assert(vector.back() == 0xDEAD + i);
|
|
Packit |
345191 |
}
|
|
Packit |
345191 |
|
|
Packit |
345191 |
std::cout << "VECTOR CLOSE" << std::endl;
|
|
Packit |
345191 |
}
|
|
Packit |
345191 |
#endif
|
|
Packit |
345191 |
|
|
Packit |
345191 |
#ifdef STL_LIST_TEST
|
|
Packit |
345191 |
{
|
|
Packit |
345191 |
std::cout << "LIST OPEN" << std::endl;
|
|
Packit |
345191 |
libmemkind::pmem::allocator<int> alc{ pmem_directory, pmem_max_size };
|
|
Packit |
345191 |
std::list<int, libmemkind::pmem::allocator<int>> list{ alc };
|
|
Packit |
345191 |
|
|
Packit |
345191 |
const int nx2 = 4;
|
|
Packit |
345191 |
for (int i = 0; i < nx2; ++i) {
|
|
Packit |
345191 |
list.emplace_back(0xBEAC011 + i);
|
|
Packit |
345191 |
assert(list.back() == 0xBEAC011 + i);
|
|
Packit |
345191 |
}
|
|
Packit |
345191 |
|
|
Packit |
345191 |
for (int i = 0; i < nx2; ++i) {
|
|
Packit |
345191 |
list.pop_back();
|
|
Packit |
345191 |
}
|
|
Packit |
345191 |
|
|
Packit |
345191 |
std::cout << "LIST CLOSE" << std::endl;
|
|
Packit |
345191 |
}
|
|
Packit |
345191 |
#endif
|
|
Packit |
345191 |
|
|
Packit |
345191 |
#ifdef STL_VEC_STRING_TEST
|
|
Packit |
345191 |
{
|
|
Packit |
345191 |
std::cout << "STRINGED VECTOR OPEN" << std::endl;
|
|
Packit |
345191 |
typedef libmemkind::pmem::allocator<char> str_alloc_t;
|
|
Packit |
345191 |
typedef std::basic_string<char, std::char_traits<char>, str_alloc_t>
|
|
Packit |
345191 |
pmem_string;
|
|
Packit |
345191 |
typedef libmemkind::pmem::allocator<pmem_string> vec_alloc_t;
|
|
Packit |
345191 |
|
|
Packit |
345191 |
vec_alloc_t vec_alloc{ pmem_directory, pmem_max_size };
|
|
Packit |
345191 |
str_alloc_t str_alloc{ pmem_directory, pmem_max_size };
|
|
Packit |
345191 |
|
|
Packit |
345191 |
std::vector<pmem_string, std::scoped_allocator_adaptor<vec_alloc_t> >
|
|
Packit |
345191 |
vec{ std::scoped_allocator_adaptor<vec_alloc_t>(vec_alloc) };
|
|
Packit |
345191 |
|
|
Packit |
345191 |
pmem_string arg{ "Very very loooong striiiing", str_alloc };
|
|
Packit |
345191 |
|
|
Packit |
345191 |
vec.push_back(arg);
|
|
Packit |
345191 |
assert(vec.back() == arg);
|
|
Packit |
345191 |
|
|
Packit |
345191 |
std::cout << "STRINGED VECTOR CLOSE" << std::endl;
|
|
Packit |
345191 |
}
|
|
Packit |
345191 |
|
|
Packit |
345191 |
#endif
|
|
Packit |
345191 |
|
|
Packit |
345191 |
#ifdef STL_MAP_INT_STRING_TEST
|
|
Packit |
345191 |
{
|
|
Packit |
345191 |
std::cout << "INT_STRING MAP OPEN" << std::endl;
|
|
Packit |
345191 |
typedef std::basic_string<char, std::char_traits<char>, libmemkind::pmem::allocator<char>>
|
|
Packit |
345191 |
pmem_string;
|
|
Packit |
345191 |
typedef int key_t;
|
|
Packit |
345191 |
typedef pmem_string value_t;
|
|
Packit |
345191 |
typedef libmemkind::pmem::allocator<std::pair<const key_t, value_t>>
|
|
Packit |
345191 |
allocator_t;
|
|
Packit |
345191 |
typedef std::map<key_t, value_t, std::less<key_t>, std::scoped_allocator_adaptor<allocator_t>>
|
|
Packit |
345191 |
map_t;
|
|
Packit |
345191 |
|
|
Packit |
345191 |
allocator_t allocator( pmem_directory, pmem_max_size );
|
|
Packit |
345191 |
|
|
Packit |
345191 |
value_t source_str1("Lorem ipsum dolor ", allocator);
|
|
Packit |
345191 |
value_t source_str2("sit amet consectetuer adipiscing elit", allocator );
|
|
Packit |
345191 |
|
|
Packit |
345191 |
map_t target_map{ std::scoped_allocator_adaptor<allocator_t>(allocator) };
|
|
Packit |
345191 |
|
|
Packit |
345191 |
target_map[key_t(165)] = source_str1;
|
|
Packit |
345191 |
assert(target_map[key_t(165)] == source_str1);
|
|
Packit |
345191 |
target_map[key_t(165)] = source_str2;
|
|
Packit |
345191 |
assert(target_map[key_t(165)] == source_str2);
|
|
Packit |
345191 |
|
|
Packit |
345191 |
std::cout << "INT_STRING MAP CLOSE" << std::endl;
|
|
Packit |
345191 |
}
|
|
Packit |
345191 |
#endif
|
|
Packit |
345191 |
std::cout << "TEST SCOPE: GOODBYE" << std::endl;
|
|
Packit |
345191 |
}
|
|
Packit |
345191 |
|
|
Packit |
345191 |
int main(int argc, char *argv[])
|
|
Packit |
345191 |
{
|
|
Packit |
345191 |
const char *pmem_directory = "/tmp/";
|
|
Packit |
345191 |
|
|
Packit |
345191 |
if (argc > 2) {
|
|
Packit |
345191 |
std::cerr << "Usage: pmem_cpp_allocator [directory path]\n"
|
|
Packit |
345191 |
<< "\t[directory path] - directory where temporary file is created (default = \"/tmp/\")"
|
|
Packit |
345191 |
<< std::endl;
|
|
Packit |
345191 |
return 0;
|
|
Packit |
345191 |
} else if (argc == 2) {
|
|
Packit |
345191 |
struct stat st;
|
|
Packit |
345191 |
if (stat(argv[1], &st) != 0 || !S_ISDIR(st.st_mode)) {
|
|
Packit |
345191 |
fprintf(stderr,"%s : Invalid path to pmem kind directory\n", argv[1]);
|
|
Packit |
345191 |
return 1;
|
|
Packit |
345191 |
}
|
|
Packit |
345191 |
pmem_directory = argv[1];
|
|
Packit |
345191 |
}
|
|
Packit |
345191 |
|
|
Packit |
345191 |
cpp_allocator_test(pmem_directory);
|
|
Packit |
345191 |
return 0;
|
|
Packit |
345191 |
}
|