test/stress/set/set_type.h

   1 /*
   2     This file is a part of libcds - Concurrent Data Structures library
   3
   4     (C) Copyright Maxim Khizhinsky (libcds.dev@gmail.com) 2006-2016
   5
   6     Source code repo: http://github.com/khizmax/libcds/
   7     Download: http://sourceforge.net/projects/libcds/files/
   8
   9     Redistribution and use in source and binary forms, with or without
  10     modification, are permitted provided that the following conditions are met:
  11
  12     * Redistributions of source code must retain the above copyright notice, this
  13       list of conditions and the following disclaimer.
  14
  15     * Redistributions in binary form must reproduce the above copyright notice,
  16       this list of conditions and the following disclaimer in the documentation
  17       and/or other materials provided with the distribution.
  18
  19     THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  20     AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  21     IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  22     DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
  23     FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  24     DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  25     SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  26     CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  27     OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  28     OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29 */
  30
  31 #ifndef CDSUNIT_SET_TYPE_H
  32 #define CDSUNIT_SET_TYPE_H
  33
  34 #include <cds/urcu/general_instant.h>
  35 #include <cds/urcu/general_buffered.h>
  36 #include <cds/urcu/general_threaded.h>
  37 #include <cds/urcu/signal_buffered.h>
  38 #include <cds/urcu/signal_threaded.h>
  39
  40 #include <cds/opt/hash.h>
  41 #include <cds/sync/spinlock.h>
  42
  43 #include <cds_test/stress_test.h>
  44 #include "framework/michael_alloc.h"
  45
  46 namespace set {
  47     namespace cc = cds::container;
  48     namespace co = cds::opt;
  49
  50     typedef cds::urcu::gc< cds::urcu::general_instant_stripped >   rcu_gpi;
  51     typedef cds::urcu::gc< cds::urcu::general_buffered_stripped >  rcu_gpb;
  52     typedef cds::urcu::gc< cds::urcu::general_threaded_stripped >  rcu_gpt;
  53 #ifdef CDS_URCU_SIGNAL_HANDLING_ENABLED
  54     typedef cds::urcu::gc< cds::urcu::signal_buffered_stripped >  rcu_shb;
  55     typedef cds::urcu::gc< cds::urcu::signal_threaded_stripped >  rcu_sht;
  56 #endif
  57
  58     template <typename Key>
  59     struct less;
  60
  61     template <typename Key>
  62     struct cmp
  63     {
  64         int operator ()(Key const& k1, Key const& k2) const
  65         {
  66             if ( less<Key>( k1, k2 ) )
  67                 return -1;
  68             return less<Key>( k2, k1 ) ? 1 : 0;
  69         }
  70     };
  71
  72     template <typename Key>
  73     struct hash;
  74
  75 #define CDSUNIT_INT_COMPARE(t)  template <> struct cmp<t> { int operator()( t k1, t k2 ){ return (int)(k1 - k2); } }
  76     CDSUNIT_INT_COMPARE(char);
  77     CDSUNIT_INT_COMPARE(unsigned char);
  78     CDSUNIT_INT_COMPARE(int);
  79     CDSUNIT_INT_COMPARE(unsigned int);
  80     CDSUNIT_INT_COMPARE(long);
  81     CDSUNIT_INT_COMPARE(unsigned long);
  82     CDSUNIT_INT_COMPARE(long long);
  83     CDSUNIT_INT_COMPARE(unsigned long long);
  84 #undef CDSUNIT_INT_COMPARE
  85
  86 #define CDSUNIT_INT_LESS(t)  template <> struct less<t> { bool operator()( t k1, t k2 ){ return k1 < k2; } }
  87     CDSUNIT_INT_LESS( char );
  88     CDSUNIT_INT_LESS( unsigned char );
  89     CDSUNIT_INT_LESS( int );
  90     CDSUNIT_INT_LESS( unsigned int );
  91     CDSUNIT_INT_LESS( long );
  92     CDSUNIT_INT_LESS( unsigned long );
  93     CDSUNIT_INT_LESS( long long );
  94     CDSUNIT_INT_LESS( unsigned long long );
  95 #undef CDSUNIT_INT_LESS
  96
  97     template <>
  98     struct cmp<std::string>
  99     {
 100         int operator()(std::string const& s1, std::string const& s2)
 101         {
 102             return s1.compare( s2 );
 103         }
 104         int operator()(std::string const& s1, char const * s2)
 105         {
 106             return s1.compare( s2 );
 107         }
 108         int operator()(char const * s1, std::string const& s2)
 109         {
 110             return -s2.compare( s1 );
 111         }
 112     };
 113
 114     template <>
 115     struct less<std::string>
 116     {
 117         bool operator ()( std::string const& k1, std::string const& k2 ) const
 118         {
 119             return cmp<std::string>()( k1, k2 ) < 0;
 120         }
 121         bool operator ()( std::string const& k1, char const* k2 ) const
 122         {
 123             return cmp<std::string>()( k1, k2 ) < 0;
 124         }
 125         bool operator ()( char const* k1, std::string const& k2 ) const
 126         {
 127             return cmp<std::string>()( k1, k2 ) < 0;
 128         }
 129     };
 130
 131     template <typename T>
 132     struct hash
 133     {
 134         typedef size_t result_type;
 135         typedef T      argument_type;
 136
 137         size_t operator()( T const& k ) const
 138         {
 139             return std::hash<size_t>()(k.nKey);
 140         }
 141
 142         size_t operator()( size_t k ) const
 143         {
 144             return std::hash<size_t>()(k);
 145         }
 146     };
 147
 148     template <>
 149     struct hash<size_t>
 150     {
 151         typedef size_t result_type;
 152         typedef size_t argument_type;
 153
 154         size_t operator()( size_t k ) const
 155         {
 156             return std::hash<size_t>()(k);
 157         }
 158     };
 159
 160     template <>
 161     struct hash<std::string>
 162     {
 163         typedef size_t result_type;
 164         typedef std::string argument_type;
 165
 166         size_t operator()( std::string const& k ) const
 167         {
 168             return std::hash<std::string>()(k);
 169         }
 170     };
 171
 172     // forward
 173     template <typename ImplSelector, typename Key, typename Value>
 174     struct set_type;
 175
 176     template <typename Key, typename Value>
 177     struct set_type_base
 178     {
 179         typedef Key     key_type;
 180         typedef Value   value_type;
 181
 182         struct key_val {
 183             key_type    key;
 184             value_type  val;
 185
 186             /*explicit*/ key_val( key_type const& k ): key(k), val() {}
 187             key_val( key_type const& k, value_type const& v ): key(k), val(v) {}
 188
 189             template <typename K>
 190             explicit key_val( K const& k ): key(k) {}
 191
 192             template <typename K, typename T>
 193             key_val( K const& k, T const& v ): key(k), val(v) {}
 194         };
 195
 196         typedef set::hash<key_type>   key_hash;
 197         typedef set::less<key_type>   key_less;
 198         typedef set::cmp<key_type>    key_compare;
 199
 200         struct less {
 201             bool operator()( key_val const& k1, key_val const& k2 ) const
 202             {
 203                 return key_less()( k1.key, k2.key );
 204             }
 205             bool operator()( key_type const& k1, key_val const& k2 ) const
 206             {
 207                 return key_less()( k1, k2.key );
 208             }
 209             bool operator()( key_val const& k1, key_type const& k2 ) const
 210             {
 211                 return key_less()( k1.key, k2 );
 212             }
 213         };
 214
 215         struct compare {
 216             int operator()( key_val const& k1, key_val const& k2 ) const
 217             {
 218                 return key_compare()( k1.key, k2.key );
 219             }
 220             int operator()( key_type const& k1, key_val const& k2 ) const
 221             {
 222                 return key_compare()( k1, k2.key );
 223             }
 224             int operator()( key_val const& k1, key_type const& k2 ) const
 225             {
 226                 return key_compare()( k1.key, k2 );
 227             }
 228         };
 229
 230         struct equal_to {
 231             bool operator()( key_val const& k1, key_val const& k2 ) const
 232             {
 233                 return key_compare()( k1.key, k2.key ) == 0;
 234             }
 235             bool operator()( key_type const& k1, key_val const& k2 ) const
 236             {
 237                 return key_compare()( k1, k2.key ) == 0;
 238             }
 239             bool operator()( key_val const& k1, key_type const& k2 ) const
 240             {
 241                 return key_compare()( k1.key, k2 ) == 0;
 242             }
 243         };
 244
 245
 246         struct hash: public key_hash
 247         {
 248             size_t operator()( key_val const& v ) const
 249             {
 250                 return key_hash::operator()( v.key );
 251             }
 252             size_t operator()( key_type const& key ) const
 253             {
 254                 return key_hash::operator()( key );
 255             }
 256             template <typename Q>
 257             size_t operator()( Q const& k ) const
 258             {
 259                 return key_hash::operator()( k );
 260             }
 261         };
 262
 263         struct hash2: public hash
 264         {
 265             size_t operator()( key_val const& k ) const
 266             {
 267                 size_t h = hash::operator ()( k.key );
 268                 size_t seed = ~h;
 269                 seed ^= h + 0x9e3779b9 + (seed << 6) + (seed >> 2);
 270                 return seed;
 271             }
 272             size_t operator()( key_type const& k ) const
 273             {
 274                 size_t h = hash::operator ()( k );
 275                 size_t seed = ~h;
 276                 seed ^= h + 0x9e3779b9 + (seed << 6) + (seed >> 2);
 277                 return seed;
 278             }
 279             template <typename Q>
 280             size_t operator()( Q const& k ) const
 281             {
 282                 return key_hash::operator()( k );
 283             }
 284         };
 285     };
 286
 287
 288     // *************************************************
 289     // print_stat
 290     // *************************************************
 291
 292     struct empty_stat {};
 293     static inline cds_test::property_stream& operator <<( cds_test::property_stream& o, empty_stat const& )
 294     {
 295         return o;
 296     }
 297
 298     template <typename Set>
 299     static inline void print_stat( cds_test::property_stream& o, Set const& s )
 300     {
 301         o << s.statistics();
 302     }
 303
 304
 305     //*******************************************************
 306     // additional_check
 307     //*******************************************************
 308
 309     template <typename Set>
 310     static inline void additional_check( Set& /*set*/ )
 311     {}
 312
 313     template <typename Set>
 314     static inline void additional_cleanup( Set& /*set*/ )
 315     {}
 316
 317     //*******************************************************
 318     // check_before_clear
 319     //*******************************************************
 320
 321     template <typename Set>
 322     static inline void check_before_clear( Set& /*s*/ )
 323     {}
 324
 325 } // namespace set
 326
 327
 328 #endif // ifndef CDSUNIT_SET_TYPE_H