/*
* Created on Mar 22, 2008
*
* 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.
*
* Copyright @2008-2010 the original author or authors.
*/
package org.fest.swing.monitor;
import java.awt.Component;
import java.awt.EventQueue;
import java.lang.ref.WeakReference;
import java.util.*;
import org.fest.swing.annotation.RunsInCurrentThread;
/**
* Understands a mapping of <code>{@link Component}</code>s and their respective <code>{@link EventQueue}</code>.
*
* @author Yvonne Wang
* @author Alex Ruiz
*/
class EventQueueMapping {
final Map<Component, WeakReference<EventQueue>> queueMap = new WeakHashMap<Component, WeakReference<EventQueue>>();
@RunsInCurrentThread
void addQueueFor(Component c) {
EventQueue queue = c.getToolkit().getSystemEventQueue();
queueMap.put(c, new WeakReference<EventQueue>(queue));
}
@RunsInCurrentThread
EventQueue queueFor(Component c) {
EventQueue queue = storedQueueFor(c);
if (queue == null) return c.getToolkit().getSystemEventQueue();
return queue;
}
EventQueue storedQueueFor(Component c) {
return queueFrom(queueMap.get(c));
}
Collection<EventQueue> eventQueues() {
Set<EventQueue> eventQueues = new HashSet<EventQueue>();
for (WeakReference<EventQueue> reference : queueMap.values()) {
EventQueue queue = queueFrom(reference);
if (queue != null) eventQueues.add(queue);
}
return eventQueues;
}
private EventQueue queueFrom(WeakReference<EventQueue> reference) {
if (reference == null) return null;
return reference.get();
}
}