Fix 5481444 Face clustering should use PWA profile shot instead of random photo

Change-Id: Ie4e3f9ae19431ab5b36d216cd7f4c626de503bbf

1 files changed, 10 insertions, 0 deletions

diff --git a/src/com/android/gallery3d/data/ClusterAlbum.java b/src/com/android/gallery3d/data/ClusterAlbum.java
index 32f902301..569e5a4b3 100644
--- a/src/com/android/gallery3d/data/ClusterAlbum.java
+++ b/src/com/android/gallery3d/data/ClusterAlbum.java
@@ -24,6 +24,7 @@ public class ClusterAlbum extends MediaSet implements ContentListener {
     private String mName = "";
     private DataManager mDataManager;
     private MediaSet mClusterAlbumSet;
+    private MediaItem mCover;
 
     public ClusterAlbum(Path path, DataManager dataManager,
             MediaSet clusterAlbumSet) {
@@ -33,6 +34,15 @@ public class ClusterAlbum extends MediaSet implements ContentListener {
         mClusterAlbumSet.addContentListener(this);
     }
 
+    public void setCoverMediaItem(MediaItem cover) {
+        mCover = cover;
+    }
+
+    @Override
+    public MediaItem getCoverMediaItem() {
+        return mCover != null ? mCover : super.getCoverMediaItem();
+    }
+
     void setMediaItems(ArrayList<Path> paths) {
         mPaths = paths;
     }