/**
* Sort the categories from the root.
*
* @param root The root element.
*/
private static void sortCategories(TreeElement root) {
List<CategoryElement> categories = new ArrayList<CategoryElement>(root.getChildCount());
for (int i = 0; i < root.getChildCount(); i++) {
categories.add((CategoryElement) root.getChild(i));
}
root.clear();
Collections.sort(
categories,
new Comparator<CategoryElement>() {
@Override
public int compare(CategoryElement o1, CategoryElement o2) {
return o1.getElementName().compareToIgnoreCase(o2.getElementName());
}
});
root.addAll(categories);
for (CategoryElement element : categories) {
if (!element.isLeaf()) {
sortCategories(element);
}
}
}
/**
* Add the categories to the model.
*
* @param root The root element of the model.
* @param categories The categories to add to the model.
* @param elements The elements collection to fill.
*/
private static void addCategories(
TreeElement root, Collection<Category> categories, Map<Category, CategoryElement> elements) {
while (!categories.isEmpty()) {
Iterator<Category> iterator = categories.iterator();
while (iterator.hasNext()) {
Category category = iterator.next();
if (category.getParent() == null) {
CategoryElement element = new CategoryElement(category.getElementName());
elements.put(category, element);
root.add(element);
iterator.remove();
} else if (elements.containsKey(category.getParent())) {
CategoryElement element = new CategoryElement(category.getElementName());
elements.put(category, element);
elements.get(category.getParent()).add(element);
iterator.remove();
}
}
}
}
/**
* Sort the specified tree model.
*
* @param model The tree model to sort.
*/
public void sort(JThequeTreeModel model) {
TreeElement root = model.getRoot();
root.clear();
Collection<Category> categories = categoriesService.getCategories();
Map<Category, CategoryElement> elements =
new HashMap<Category, CategoryElement>(categories.size());
addCategories(root, categories, elements);
sortCategories(root);
addMovies(root, elements);
model.setRootElement(root);
}
/**
* Add the movies to to the tree.
*
* @param root The root element of the tree.
* @param elements The category elements to add the movies to.
*/
private void addMovies(TreeElement root, Map<Category, CategoryElement> elements) {
for (Movie movie : moviesService.getMovies()) {
if (movie.hasCategories()) {
for (Category category : movie.getCategories()) {
elements.get(category).add(movie);
}
} else {
root.add(movie);
}
}
}