/*
* Copyright 2016 Anton Tananaev (anton@traccar.org)
*
* 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 org.traccar.database;
import org.traccar.model.Device;
import org.traccar.model.Group;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
public class GroupTree {
private static class TreeNode {
private Group group;
private Device device;
private Collection<TreeNode> children = new HashSet<>();
TreeNode(Group group) {
this.group = group;
}
TreeNode(Device device) {
this.device = device;
}
@Override
public int hashCode() {
if (group != null) {
return (int) group.getId();
} else {
return (int) device.getId();
}
}
@Override
public boolean equals(Object obj) {
if (!(obj instanceof TreeNode)) {
return false;
}
TreeNode other = (TreeNode) obj;
if (other == this) {
return true;
}
if (group != null && other.group != null) {
return group.getId() == other.group.getId();
} else if (device != null && other.device != null) {
return device.getId() == other.device.getId();
}
return false;
}
public Group getGroup() {
return group;
}
public Device getDevice() {
return device;
}
public void setParent(TreeNode parent) {
if (parent != null) {
parent.children.add(this);
}
}
public Collection<TreeNode> getChildren() {
return children;
}
}
private final Map<Long, TreeNode> groupMap = new HashMap<>();
public GroupTree(Collection<Group> groups, Collection<Device> devices) {
for (Group group : groups) {
groupMap.put(group.getId(), new TreeNode(group));
}
for (TreeNode node : groupMap.values()) {
if (node.getGroup().getGroupId() != 0) {
node.setParent(groupMap.get(node.getGroup().getGroupId()));
}
}
Map<Long, TreeNode> deviceMap = new HashMap<>();
for (Device device : devices) {
deviceMap.put(device.getId(), new TreeNode(device));
}
for (TreeNode node : deviceMap.values()) {
if (node.getDevice().getGroupId() != 0) {
node.setParent(groupMap.get(node.getDevice().getGroupId()));
}
}
}
public Collection<Group> getGroups(long groupId) {
Set<TreeNode> results = new HashSet<>();
getNodes(results, groupMap.get(groupId));
Collection<Group> groups = new ArrayList<>();
for (TreeNode node : results) {
if (node.getGroup() != null) {
groups.add(node.getGroup());
}
}
return groups;
}
public Collection<Device> getDevices(long groupId) {
Set<TreeNode> results = new HashSet<>();
getNodes(results, groupMap.get(groupId));
Collection<Device> devices = new ArrayList<>();
for (TreeNode node : results) {
if (node.getDevice() != null) {
devices.add(node.getDevice());
}
}
return devices;
}
private void getNodes(Set<TreeNode> results, TreeNode node) {
if (node != null) {
for (TreeNode child : node.getChildren()) {
results.add(child);
getNodes(results, child);
}
}
}
}