forked from libgdx/libgdx
/
Pool.java
117 lines (102 loc) · 4.53 KB
/
Pool.java
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
/*******************************************************************************
* Copyright 2011 See AUTHORS file.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
******************************************************************************/
package com.badlogic.gdx.utils;
/** A pool of objects that can be reused to avoid allocation.
* @see Pools
* @author Nathan Sweet */
abstract public class Pool<T> {
/** The maximum number of objects that will be pooled. */
public final int max;
/** The highest number of free objects. Can be reset any time. */
public int peak;
private final Array<T> freeObjects;
/** Creates a pool with an initial capacity of 16 and no maximum. */
public Pool () {
this(16, Integer.MAX_VALUE);
}
/** Creates a pool with the specified initial capacity and no maximum. */
public Pool (int initialCapacity) {
this(initialCapacity, Integer.MAX_VALUE);
}
/** @param initialCapacity The initial size of the array supporting the pool. No objects are created/pre-allocated.
* Use {@link #fill(int) after instantiation if needed.
* @param max The maximum number of free objects to store in this pool. */
public Pool (int initialCapacity, int max) {
freeObjects = new Array(false, initialCapacity);
this.max = max;
}
abstract protected T newObject ();
/** Returns an object from this pool. The object may be new (from {@link #newObject()}) or reused (previously
* {@link #free(Object) freed}). */
public T obtain () {
return freeObjects.size == 0 ? newObject() : freeObjects.pop();
}
/** Puts the specified object in the pool, making it eligible to be returned by {@link #obtain()}. If the pool already contains
* {@link #max} free objects, the specified object is reset but not added to the pool.
* <p>
* The pool does not check if an object is already freed, so the same object must not be freed multiple times. */
public void free (T object) {
if (object == null) throw new IllegalArgumentException("object cannot be null.");
if (freeObjects.size < max) {
freeObjects.add(object);
peak = Math.max(peak, freeObjects.size);
}
reset(object);
}
/** Adds the specified number of new free objects to the pool. Usually called early on as a pre-allocation mechanism but can be
* used at any time.
*
* @param size the number of objects to be added */
public void fill (int size) {
for (int i = 0; i < size; i++)
if (freeObjects.size < max) freeObjects.add(newObject());
peak = Math.max(peak, freeObjects.size);
}
/** Called when an object is freed to clear the state of the object for possible later reuse. The default implementation calls
* {@link Poolable#reset()} if the object is {@link Poolable}. */
protected void reset (T object) {
if (object instanceof Poolable) ((Poolable)object).reset();
}
/** Puts the specified objects in the pool. Null objects within the array are silently ignored.
* <p>
* The pool does not check if an object is already freed, so the same object must not be freed multiple times.
* @see #free(Object) */
public void freeAll (Array<T> objects) {
if (objects == null) throw new IllegalArgumentException("objects cannot be null.");
Array<T> freeObjects = this.freeObjects;
int max = this.max;
for (int i = 0; i < objects.size; i++) {
T object = objects.get(i);
if (object == null) continue;
if (freeObjects.size < max) freeObjects.add(object);
reset(object);
}
peak = Math.max(peak, freeObjects.size);
}
/** Removes all free objects from this pool. */
public void clear () {
freeObjects.clear();
}
/** The number of objects available to be obtained. */
public int getFree () {
return freeObjects.size;
}
/** Objects implementing this interface will have {@link #reset()} called when passed to {@link Pool#free(Object)}. */
static public interface Poolable {
/** Resets the object for reuse. Object references should be nulled and fields may be set to default values. */
public void reset ();
}
}