/*
* 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-2013 the original author or authors.
*/
package org.fest.swing.monitor;
import static java.lang.Boolean.TRUE;
import static org.fest.swing.query.ComponentParentQuery.parentOf;
import static org.fest.util.Maps.newWeakHashMap;
import static org.fest.util.Sets.newHashSet;
import java.awt.Component;
import java.awt.EventQueue;
import java.awt.Toolkit;
import java.awt.Window;
import java.util.Collection;
import java.util.Map;
import java.util.Set;
import javax.annotation.Nonnull;
import org.fest.swing.annotation.RunsInCurrentThread;
/**
* A mapping of unique event queues to the set of root windows found on each queue.
*
* @author Alex Ruiz
* @author Yvonne Wang
*/
class WindowEventQueueMapping {
final Map<EventQueue, Map<Window, Boolean>> queueMap = newWeakHashMap();
void addQueueFor(@Nonnull Toolkit toolkit) {
Map<Window, Boolean> map = newWeakHashMap();
queueMap.put(toolkit.getSystemEventQueue(), map);
}
void addQueueFor(@Nonnull Component component) {
EventQueue queue = component.getToolkit().getSystemEventQueue();
Map<Window, Boolean> windowMapping = queueMap.get(queue);
if (windowMapping == null) {
windowMapping = createWindowMapping(queue);
}
if (!(component instanceof Window) || parentOf(component) != null) {
return;
}
windowMapping.put((Window)component, TRUE);
}
private @Nonnull Map<Window, Boolean> createWindowMapping(EventQueue queue) {
Map<Window, Boolean> windowMapping = newWeakHashMap();
queueMap.put(queue, windowMapping);
return windowMapping;
}
@RunsInCurrentThread
void removeMappingFor(@Nonnull Component component) {
EventQueue queue = component.getToolkit().getSystemEventQueue();
removeComponent(component, queue);
for (EventQueue q : queueMap.keySet()) {
removeComponent(component, q);
}
}
private void removeComponent(@Nonnull Component component, @Nonnull EventQueue queue) {
Map<Window, Boolean> windowMapping = queueMap.get(queue);
if (windowMapping != null) {
windowMapping.remove(component);
}
}
@Nonnull Collection<Window> windows() {
Set<Window> rootWindows = newHashSet();
for (EventQueue queue : queueMap.keySet()) {
rootWindows.addAll(queueMap.get(queue).keySet());
}
return rootWindows;
}
@Nonnull Collection<EventQueue> eventQueues() {
return queueMap.keySet();
}
}