lib: percpu_counter_sum_positive
[linux-2.6.git] / include / linux / percpu_counter.h
1 #ifndef _LINUX_PERCPU_COUNTER_H
2 #define _LINUX_PERCPU_COUNTER_H
3 /*
4  * A simple "approximate counter" for use in ext2 and ext3 superblocks.
5  *
6  * WARNING: these things are HUGE.  4 kbytes per counter on 32-way P4.
7  */
8
9 #include <linux/spinlock.h>
10 #include <linux/smp.h>
11 #include <linux/list.h>
12 #include <linux/threads.h>
13 #include <linux/percpu.h>
14 #include <linux/types.h>
15
16 #ifdef CONFIG_SMP
17
18 struct percpu_counter {
19         spinlock_t lock;
20         s64 count;
21 #ifdef CONFIG_HOTPLUG_CPU
22         struct list_head list;  /* All percpu_counters are on a list */
23 #endif
24         s32 *counters;
25 };
26
27 #if NR_CPUS >= 16
28 #define FBC_BATCH       (NR_CPUS*2)
29 #else
30 #define FBC_BATCH       (NR_CPUS*4)
31 #endif
32
33 void percpu_counter_init(struct percpu_counter *fbc, s64 amount);
34 void percpu_counter_destroy(struct percpu_counter *fbc);
35 void percpu_counter_set(struct percpu_counter *fbc, s64 amount);
36 void __percpu_counter_add(struct percpu_counter *fbc, s64 amount, s32 batch);
37 s64 percpu_counter_sum_positive(struct percpu_counter *fbc);
38
39 static inline void percpu_counter_add(struct percpu_counter *fbc, s64 amount)
40 {
41         __percpu_counter_add(fbc, amount, FBC_BATCH);
42 }
43
44 static inline s64 percpu_counter_read(struct percpu_counter *fbc)
45 {
46         return fbc->count;
47 }
48
49 /*
50  * It is possible for the percpu_counter_read() to return a small negative
51  * number for some counter which should never be negative.
52  *
53  */
54 static inline s64 percpu_counter_read_positive(struct percpu_counter *fbc)
55 {
56         s64 ret = fbc->count;
57
58         barrier();              /* Prevent reloads of fbc->count */
59         if (ret >= 0)
60                 return ret;
61         return 1;
62 }
63
64 #else
65
66 struct percpu_counter {
67         s64 count;
68 };
69
70 static inline void percpu_counter_init(struct percpu_counter *fbc, s64 amount)
71 {
72         fbc->count = amount;
73 }
74
75 static inline void percpu_counter_destroy(struct percpu_counter *fbc)
76 {
77 }
78
79 static inline void percpu_counter_set(struct percpu_counter *fbc, s64 amount)
80 {
81         fbc->count = amount;
82 }
83
84 #define __percpu_counter_add(fbc, amount, batch) \
85         percpu_counter_add(fbc, amount)
86
87 static inline void
88 percpu_counter_add(struct percpu_counter *fbc, s64 amount)
89 {
90         preempt_disable();
91         fbc->count += amount;
92         preempt_enable();
93 }
94
95 static inline s64 percpu_counter_read(struct percpu_counter *fbc)
96 {
97         return fbc->count;
98 }
99
100 static inline s64 percpu_counter_read_positive(struct percpu_counter *fbc)
101 {
102         return fbc->count;
103 }
104
105 static inline s64 percpu_counter_sum_positive(struct percpu_counter *fbc)
106 {
107         return percpu_counter_read_positive(fbc);
108 }
109
110 #endif  /* CONFIG_SMP */
111
112 static inline void percpu_counter_inc(struct percpu_counter *fbc)
113 {
114         percpu_counter_add(fbc, 1);
115 }
116
117 static inline void percpu_counter_dec(struct percpu_counter *fbc)
118 {
119         percpu_counter_add(fbc, -1);
120 }
121
122 static inline void percpu_counter_sub(struct percpu_counter *fbc, s64 amount)
123 {
124         percpu_counter_add(fbc, -amount);
125 }
126
127 #endif /* _LINUX_PERCPU_COUNTER_H */