- /** Nested albums. */
- private final List<Album> albums = new ArrayList<Album>();
-
- /** The images in this album. */
- private final List<Image> images = new ArrayList<Image>();
-
- /** The parent album. */
- private Album parent;
+ @Override
+ public int compare(Album leftAlbum, Album rightAlbum) {
+ return leftAlbum.getTitle().compareToIgnoreCase(rightAlbum.getTitle());
+ }
+ };
+
+ /** Function that flattens the given album and all albums beneath it. */
+ Function<Album, List<Album>> FLATTENER = new Function<Album, List<Album>>() {
+
+ @Override
+ @Nonnull
+ public List<Album> apply(Album album) {
+ if (album == null) {
+ return emptyList();
+ }
+ List<Album> albums = new ArrayList<Album>();
+ albums.add(album);
+ for (Album subAlbum : album.getAlbums()) {
+ albums.addAll(FluentIterable.from(ImmutableList.of(subAlbum)).transformAndConcat(FLATTENER).toList());
+ }
+ return albums;
+ }
+ };