X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FAlbum.java;h=70478f34f3e112a55abe9337e5da70bd6a0cfd3b;hb=7df20dec7dcc9aa6d1aa657d24f273e1a82c60fc;hp=5375d983dabae801be1e3053e39bbcb93bdcbda6;hpb=6f019de1d4d9742981d851ac3c9097cca8bff58e;p=Sone.git
diff --git a/src/main/java/net/pterodactylus/sone/data/Album.java b/src/main/java/net/pterodactylus/sone/data/Album.java
index 5375d98..70478f3 100644
--- a/src/main/java/net/pterodactylus/sone/data/Album.java
+++ b/src/main/java/net/pterodactylus/sone/data/Album.java
@@ -20,16 +20,21 @@ package net.pterodactylus.sone.data;
import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Preconditions.checkState;
+import static java.util.Arrays.asList;
+import static java.util.Collections.emptyList;
import java.util.ArrayList;
+import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.UUID;
+import javax.annotation.Nonnull;
import com.google.common.base.Function;
import com.google.common.base.Optional;
+import com.google.common.base.Predicate;
import com.google.common.base.Predicates;
import com.google.common.collect.Collections2;
import com.google.common.collect.FluentIterable;
@@ -42,7 +47,7 @@ import com.google.common.hash.Hashing;
*
* @author David âBombeâ Roden
*/
-public class Album implements Fingerprintable {
+public class Album implements Identified, Fingerprintable {
/** Compares two {@link Album}s by {@link #getTitle()}. */
public static final Comparator TITLE_COMPARATOR = new Comparator() {
@@ -57,7 +62,11 @@ public class Album implements Fingerprintable {
public static final Function> FLATTENER = new Function>() {
@Override
+ @Nonnull
public List apply(Album album) {
+ if (album == null) {
+ return emptyList();
+ }
List albums = new ArrayList();
albums.add(album);
for (Album subAlbum : album.getAlbums()) {
@@ -67,6 +76,42 @@ public class Album implements Fingerprintable {
}
};
+ /** Function that transforms an album into the images it contains. */
+ public static final Function> IMAGES = new Function>() {
+
+ @Override
+ @Nonnull
+ public List apply(Album album) {
+ return (album != null) ? album.getImages() : Collections.emptyList();
+ }
+ };
+
+ /**
+ * Filter that removes all albums that do not have any images in any album
+ * below it.
+ */
+ public static final Predicate NOT_EMPTY = new Predicate() {
+
+ @Override
+ public boolean apply(Album album) {
+ /* so, we flatten all albums below the given one and check whether at least one album⦠*/
+ return FluentIterable.from(asList(album)).transformAndConcat(FLATTENER).anyMatch(new Predicate() {
+
+ @Override
+ public boolean apply(Album album) {
+ /* â¦contains any inserted images. */
+ return !album.getImages().isEmpty() && FluentIterable.from(album.getImages()).allMatch(new Predicate() {
+
+ @Override
+ public boolean apply(Image input) {
+ return input.isInserted();
+ }
+ });
+ }
+ });
+ }
+ };
+
/** The ID of this album. */
private final String id;
@@ -166,7 +211,6 @@ public class Album implements Fingerprintable {
public void addAlbum(Album album) {
checkNotNull(album, "album must not be null");
checkArgument(album.getSone().equals(sone), "album must belong to the same Sone as this album");
- checkState((this.parent == null) || (this.parent.equals(album.parent)), "album must not already be set to some other Sone");
album.setParent(this);
if (!albums.contains(album)) {
albums.add(album);
@@ -372,6 +416,16 @@ public class Album implements Fingerprintable {
}
/**
+ * Returns whether this album is an identitiyâs root album.
+ *
+ * @return {@code true} if this album is an identityâs root album, {@code
+ * false} otherwise
+ */
+ public boolean isRoot() {
+ return parent == null;
+ }
+
+ /**
* Returns the parent album of this album.
*
* @return The parent album of this album, or {@code null} if this album