2925cf0128a4c750b34ecc680b3d1d7c2c4aa22d
[libcds.git] / 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     template <typename Set>
293     static inline void print_stat( cds_test::property_stream&, Set const& /*s*/ )
294     {}
295
296
297     //*******************************************************
298     // additional_check
299     //*******************************************************
300
301     template <typename Set>
302     static inline void additional_check( Set& /*set*/ )
303     {}
304
305     template <typename Set>
306     static inline void additional_cleanup( Set& /*set*/ )
307     {}
308
309     //*******************************************************
310     // check_before_clear
311     //*******************************************************
312
313     template <typename Set>
314     static inline void check_before_clear( Set& /*s*/ )
315     {}
316
317 } // namespace set
318
319
320 #endif // ifndef CDSUNIT_SET_TYPE_H