|
7 | 7 |
|
8 | 8 | namespace axom |
9 | 9 | { |
10 | | - std::map<int, std::shared_ptr<ConduitMemory>> ConduitMemory::s_axomToInstance; |
11 | | - std::map<conduit::index_t, std::shared_ptr<ConduitMemory>> ConduitMemory::s_conduitToInstance; |
| 10 | +std::map<int, std::shared_ptr<ConduitMemory>> ConduitMemory::s_axomToInstance; |
| 11 | +std::map<conduit::index_t, std::shared_ptr<ConduitMemory>> |
| 12 | + ConduitMemory::s_conduitToInstance; |
12 | 13 |
|
13 | | - const ConduitMemory& ConduitMemory::instanceForAxomId(int axomAllocId) |
14 | | - { |
15 | | - // This method IS NOT thread safe. |
16 | | - |
17 | | - if(s_axomToInstance.empty()) |
18 | | - { |
19 | | - // Required one-time actions |
20 | | - static auto axomMemcopy = [](void* dst, const void* src, size_t byteCount) { |
21 | | - axom::copy(dst, src, byteCount); |
22 | | - }; |
23 | | - static auto axomMemset = [](void* ptr, int value, size_t count) { |
24 | | - if(axom::getAllocatorIDFromPointer(ptr) == axom::DYNAMIC_ALLOCATOR_ID) |
25 | | - { |
26 | | - std::memset(ptr, value, count); |
27 | | - } |
28 | | - else |
29 | | - { |
30 | | - umpire::ResourceManager& rm = umpire::ResourceManager::getInstance(); |
31 | | - rm.memset(ptr, value, count); |
32 | | - } |
33 | | - }; |
34 | | - conduit::utils::set_memcpy_handler(axomMemcopy); |
35 | | - conduit::utils::set_memset_handler(axomMemset); |
36 | | - } |
| 14 | +const ConduitMemory& ConduitMemory::instanceForAxomId(int axomAllocId) |
| 15 | +{ |
| 16 | + // This method IS NOT thread safe. |
37 | 17 |
|
38 | | - auto it = s_axomToInstance.find(axomAllocId); |
39 | | - if(it == s_axomToInstance.end()) |
40 | | - { |
41 | | - std::shared_ptr<ConduitMemory> newInstance(new ConduitMemory(axomAllocId)); |
42 | | - s_axomToInstance[axomAllocId] = newInstance; |
43 | | - it = s_axomToInstance.insert({axomAllocId, newInstance}).first; |
| 18 | + if(s_axomToInstance.empty()) |
| 19 | + { |
| 20 | + // Required one-time actions |
| 21 | + static auto axomMemcopy = [](void* dst, const void* src, size_t byteCount) { |
| 22 | + axom::copy(dst, src, byteCount); |
| 23 | + }; |
| 24 | + static auto axomMemset = [](void* ptr, int value, size_t count) { |
| 25 | + if(axom::getAllocatorIDFromPointer(ptr) == axom::DYNAMIC_ALLOCATOR_ID) |
| 26 | + { |
| 27 | + std::memset(ptr, value, count); |
| 28 | + } |
| 29 | + else |
| 30 | + { |
| 31 | + umpire::ResourceManager& rm = umpire::ResourceManager::getInstance(); |
| 32 | + rm.memset(ptr, value, count); |
| 33 | + } |
| 34 | + }; |
| 35 | + conduit::utils::set_memcpy_handler(axomMemcopy); |
| 36 | + conduit::utils::set_memset_handler(axomMemset); |
| 37 | + } |
44 | 38 |
|
45 | | - auto conduitAllocId = newInstance->m_conduitId; |
46 | | - assert(s_conduitToInstance.find(conduitAllocId) == s_conduitToInstance.end()); |
47 | | - s_conduitToInstance[conduitAllocId] = newInstance; |
48 | | - } |
49 | | - assert(it->first == axomAllocId); |
| 39 | + auto it = s_axomToInstance.find(axomAllocId); |
| 40 | + if(it == s_axomToInstance.end()) |
| 41 | + { |
| 42 | + std::shared_ptr<ConduitMemory> newInstance(new ConduitMemory(axomAllocId)); |
| 43 | + s_axomToInstance[axomAllocId] = newInstance; |
| 44 | + it = s_axomToInstance.insert({axomAllocId, newInstance}).first; |
50 | 45 |
|
51 | | - return *it->second; |
| 46 | + auto conduitAllocId = newInstance->m_conduitId; |
| 47 | + assert(s_conduitToInstance.find(conduitAllocId) == s_conduitToInstance.end()); |
| 48 | + s_conduitToInstance[conduitAllocId] = newInstance; |
52 | 49 | } |
| 50 | + assert(it->first == axomAllocId); |
53 | 51 |
|
54 | | - const ConduitMemory& ConduitMemory::instanceForConduitId(conduit::index_t conduitAllocId) |
55 | | - { |
56 | | - // This method IS thread safe. |
| 52 | + return *it->second; |
| 53 | +} |
57 | 54 |
|
58 | | - auto it = s_conduitToInstance.find(conduitAllocId); |
59 | | - if(it == s_conduitToInstance.end()) |
60 | | - { |
61 | | - std::cerr << "*** Error: Axom allocator for Conduit allocator id " << conduitAllocId << " doesn't exist. You have to register the Axom allocator first, using ConduitMemory::getAxomInstance()." << std::endl; |
62 | | - axom::utilities::processAbort(); |
63 | | - } |
64 | | - assert(it->first == conduitAllocId); |
| 55 | +const ConduitMemory& ConduitMemory::instanceForConduitId( |
| 56 | + conduit::index_t conduitAllocId) |
| 57 | +{ |
| 58 | + // This method IS thread safe. |
65 | 59 |
|
66 | | - return *it->second; |
| 60 | + auto it = s_conduitToInstance.find(conduitAllocId); |
| 61 | + if(it == s_conduitToInstance.end()) |
| 62 | + { |
| 63 | + std::cerr << "*** Error: Axom allocator for Conduit allocator id " |
| 64 | + << conduitAllocId << " doesn't exist. You have to register the Axom allocator first, using ConduitMemory::getAxomInstance()." |
| 65 | + << std::endl; |
| 66 | + axom::utilities::processAbort(); |
67 | 67 | } |
| 68 | + assert(it->first == conduitAllocId); |
| 69 | + |
| 70 | + return *it->second; |
| 71 | +} |
68 | 72 | } // end namespace axom |
0 commit comments