summaryrefslogtreecommitdiffstats
path: root/chrome/common/extensions/value_counter.h
blob: 84fe06c91c3bfc7d46b653f56522179af54fc808 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_COMMON_EXTENSIONS_VALUE_COUNTER_H_
#define CHROME_COMMON_EXTENSIONS_VALUE_COUNTER_H_

#include "base/memory/linked_ptr.h"

#include <vector>

namespace base {
class Value;
}

namespace extensions {

// Keeps a running count of Values, like map<Value, int>. Adding / removing
// values increments / decrements the count associated with a given Value.
//
// Add() and Remove() are linear in the number of Values in the ValueCounter,
// because there is no operator<() defined on Value, so we must iterate to find
// whether a Value is equal to an existing one.
class ValueCounter {
 public:
  ValueCounter();
  ~ValueCounter();

  // Adds |value| to the set and returns how many equal values are in the set
  // after. Does not take ownership of |value|. In the case where a Value equal
  // to |value| doesn't already exist in this map, this function makes a
  // DeepCopy() of |value|.
  int Add(const base::Value& value);

  // Removes |value| from the set and returns how many equal values are in
  // the set after.
  int Remove(const base::Value& value);

  // Same as Add() but only performs the add if the value isn't present.
  int AddIfMissing(const base::Value& value);

 private:
  class Entry {
   public:
    explicit Entry(const base::Value& value);
    ~Entry();

    int Increment();
    int Decrement();

    const base::Value* value() const { return value_.get(); }
    int count() const { return count_; }

   private:
    linked_ptr<base::Value> value_;
    int count_;

    DISALLOW_COPY_AND_ASSIGN(Entry);
  };
  typedef std::vector<linked_ptr<Entry> > EntryList;

  int AddImpl(const base::Value& value, bool increment);

  EntryList entries_;

  DISALLOW_COPY_AND_ASSIGN(ValueCounter);
};

}  // namespace extensions

#endif  // CHROME_COMMON_EXTENSIONS_VALUE_COUNTER_H_