ACE/tests/Map_Test.h

   1 /* -*- C++ -*- */
   2
   3 //=============================================================================
   4 /**
   5  *  @file    Map_Test.h
   6  *
   7  * This file has the class definitions needed for template generation in
   8  * Map_Test.cpp.  They have to be in a separate file so AIX xlC can
   9  * find them at auto-instantiate time.
  10  *
  11  *  @author Irfan Pyarali <irfan@cs.wustl.edu>
  12  */
  13 //=============================================================================
  14
  15
  16 #ifndef ACE_TESTS_MAP_TEST_H
  17 #define ACE_TESTS_MAP_TEST_H
  18
  19 #include "ace/OS_NS_string.h"
  20
  21 #if !defined (ACE_LACKS_PRAGMA_ONCE)
  22 # pragma once
  23 #endif /* ACE_LACKS_PRAGMA_ONCE */
  24
  25 #include "ace/Active_Map_Manager.h"
  26 #include "ace/Containers.h"
  27
  28 // Key data type.
  29 typedef ACE_Array<char> KEY;
  30
  31 /**
  32  * @class Key_Generator
  33  *
  34  * @brief Defines a key generator.
  35  *
  36  * This class is used in adapters of maps that do not produce keys.
  37  */
  38 class Key_Generator
  39 {
  40 public:
  41
  42   Key_Generator (void)
  43     : counter_ (0)
  44     {
  45     }
  46
  47   int operator() (KEY &key)
  48     {
  49       // Keep original information in the key intact.
  50       size_t original_size = key.size ();
  51
  52       // Size of this counter key.
  53       const size_t counter_key_size = sizeof this->counter_;
  54
  55       // Resize to accommodate both the original data and the new key.
  56       key.size (counter_key_size + original_size);
  57
  58       // Add new key data.
  59       ACE_OS::memcpy (&key[original_size],
  60                       &++this->counter_,
  61                       counter_key_size);
  62
  63       // Success.
  64       return 0;
  65     }
  66
  67 private:
  68   u_long counter_;
  69 };
  70
  71 class Hash_Key
  72 {
  73 public:
  74   u_long operator () (const KEY &key) const
  75     {
  76       // Recover system generated part of key.
  77       u_long value;
  78       size_t counter_key_size = sizeof (u_long);
  79
  80       // Copy system key from user key.
  81       ACE_OS::memcpy (&value,
  82                       &key[key.size () - counter_key_size],
  83                       sizeof value);
  84
  85       // Return the system key value as the hash value.
  86       return value;
  87     }
  88 };
  89
  90 class Key_Adapter
  91 {
  92 public:
  93
  94   int encode (const KEY &original_key,
  95               const ACE_Active_Map_Manager_Key &active_key,
  96               KEY &modified_key)
  97     {
  98       // Keep original information in the key intact.
  99       modified_key = original_key;
 100       size_t original_size = modified_key.size ();
 101
 102       // Size of active key.
 103       size_t active_key_size = active_key.size ();
 104
 105       // Resize to accommodate both the original data and the new active key.
 106       modified_key.size (active_key_size + original_size);
 107
 108       // Copy active key data into user key.
 109       active_key.encode (&modified_key[original_size]);
 110
 111       // Success.
 112       return 0;
 113     }
 114
 115   int decode (const KEY &modified_key,
 116               ACE_Active_Map_Manager_Key &active_key)
 117     {
 118       // Read the active key data from the back of the key.
 119       size_t active_key_size = active_key.size ();
 120       size_t original_size = modified_key.size () - active_key_size;
 121
 122       // Read off value of index and generation.
 123       active_key.decode (&modified_key[original_size]);
 124
 125       // Success.
 126       return 0;
 127     }
 128
 129   int decode (const KEY &modified_key,
 130               KEY &original_key)
 131     {
 132       // Read the original user key data from the front of the
 133       // modified key.
 134       size_t active_key_size = ACE_Active_Map_Manager_Key::size ();
 135
 136       // Copy all the data.
 137       original_key = modified_key;
 138
 139       // Resize to ignore active key data.
 140       original_key.size (original_key.size () - active_key_size);
 141
 142       // Success.
 143       return 0;
 144     }
 145 };
 146
 147 #endif /* ACE_TESTS_MAP_TEST_H */