src/com/android/gallery3d/data/LocalAlbumSet.java - fp2-dev/platform/packages/apps/Camera2 - Gitiles

 /*
  * Copyright (C) 2010 The Android Open Source Project
  *
  * 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 com.android.gallery3d.data;

 import com.android.gallery3d.R;
 import com.android.gallery3d.app.GalleryApp;
 import com.android.gallery3d.common.Utils;
 import com.android.gallery3d.util.GalleryUtils;
 import com.android.gallery3d.util.MediaSetUtils;

 import android.content.ContentResolver;
 import android.database.Cursor;
 import android.net.Uri;
 import android.provider.MediaStore.Files;
 import android.provider.MediaStore.Files.FileColumns;
 import android.provider.MediaStore.Images;
 import android.provider.MediaStore.Images.ImageColumns;
 import android.provider.MediaStore.Video;

 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Comparator;
 import java.util.HashSet;

 // LocalAlbumSet lists all image or video albums in the local storage.
 // The path should be "/local/image", "local/video" or "/local/all"
 public class LocalAlbumSet extends MediaSet {
     public static final Path PATH_ALL = Path.fromString("/local/all");
     public static final Path PATH_IMAGE = Path.fromString("/local/image");
     public static final Path PATH_VIDEO = Path.fromString("/local/video");

     private static final String TAG = "LocalAlbumSet";
     private static final String EXTERNAL_MEDIA = "external";

     // The indices should match the following projections.
     private static final int INDEX_BUCKET_ID = 0;
     private static final int INDEX_MEDIA_TYPE = 1;
     private static final int INDEX_BUCKET_NAME = 2;

     private static final Uri mBaseUri = Files.getContentUri(EXTERNAL_MEDIA);
     private static final Uri mWatchUriImage = Images.Media.EXTERNAL_CONTENT_URI;
     private static final Uri mWatchUriVideo = Video.Media.EXTERNAL_CONTENT_URI;

     // The order is import it must match to the index in MediaStore.
     private static final String[] PROJECTION_BUCKET = {
             ImageColumns.BUCKET_ID,
             FileColumns.MEDIA_TYPE,
             ImageColumns.BUCKET_DISPLAY_NAME };

     private final GalleryApp mApplication;
     private final int mType;
     private ArrayList<MediaSet> mAlbums = new ArrayList<MediaSet>();
     private final ChangeNotifier mNotifierImage;
     private final ChangeNotifier mNotifierVideo;
     private final String mName;

     public LocalAlbumSet(Path path, GalleryApp application) {
         super(path, nextVersionNumber());
         mApplication = application;
         mType = getTypeFromPath(path);
         mNotifierImage = new ChangeNotifier(this, mWatchUriImage, application);
         mNotifierVideo = new ChangeNotifier(this, mWatchUriVideo, application);
         mName = application.getResources().getString(
                 R.string.set_label_local_albums);
     }

     private static int getTypeFromPath(Path path) {
         String name[] = path.split();
         if (name.length < 2) {
             throw new IllegalArgumentException(path.toString());
         }
         if ("all".equals(name[1])) return MEDIA_TYPE_ALL;
         if ("image".equals(name[1])) return MEDIA_TYPE_IMAGE;
         if ("video".equals(name[1])) return MEDIA_TYPE_VIDEO;
         throw new IllegalArgumentException(path.toString());
     }

     @Override
     public MediaSet getSubMediaSet(int index) {
         return mAlbums.get(index);
     }

     @Override
     public int getSubMediaSetCount() {
         return mAlbums.size();
     }

     @Override
     public String getName() {
         return mName;
     }

     private BucketEntry[] loadBucketEntries(Cursor cursor) {
         HashSet<BucketEntry> buffer = new HashSet<BucketEntry>();
         int typeBits = 0;
         if ((mType & MEDIA_TYPE_IMAGE) != 0) {
             typeBits |= (1 << FileColumns.MEDIA_TYPE_IMAGE);
         }
         if ((mType & MEDIA_TYPE_VIDEO) != 0) {
             typeBits |= (1 << FileColumns.MEDIA_TYPE_VIDEO);
         }
         try {
             while (cursor.moveToNext()) {
                 if ((typeBits & (1 << cursor.getInt(INDEX_MEDIA_TYPE))) != 0) {
                     buffer.add(new BucketEntry(
                             cursor.getInt(INDEX_BUCKET_ID),
                             cursor.getString(INDEX_BUCKET_NAME)));
                 }
             }
         } finally {
             cursor.close();
         }
         return buffer.toArray(new BucketEntry[buffer.size()]);
     }


     private static int findBucket(BucketEntry entries[], int bucketId) {
         for (int i = 0, n = entries.length; i < n ; ++i) {
             if (entries[i].bucketId == bucketId) return i;
         }
         return -1;
     }

     @SuppressWarnings("unchecked")
     protected ArrayList<MediaSet> loadSubMediaSets() {
         // Note: it will be faster if we only select media_type and bucket_id.
         //       need to test the performance if that is worth

         Uri uri = mBaseUri.buildUpon().
                 appendQueryParameter("distinct", "true").build();
         GalleryUtils.assertNotInRenderThread();
         Cursor cursor = mApplication.getContentResolver().query(
                 uri, PROJECTION_BUCKET, null, null, null);
         if (cursor == null) {
             Log.w(TAG, "cannot open local database: " + uri);
             return new ArrayList<MediaSet>();
         }
         BucketEntry[] entries = loadBucketEntries(cursor);
         int offset = 0;

         int index = findBucket(entries, MediaSetUtils.CAMERA_BUCKET_ID);
         if (index != -1) {
             Utils.swap(entries, index, offset++);
         }
         index = findBucket(entries, MediaSetUtils.DOWNLOAD_BUCKET_ID);
         if (index != -1) {
             Utils.swap(entries, index, offset++);
         }

         Arrays.sort(entries, offset, entries.length, new Comparator<BucketEntry>() {
             @Override
             public int compare(BucketEntry a, BucketEntry b) {
                 int result = a.bucketName.compareTo(b.bucketName);
                 return result != 0
                         ? result
                         : Utils.compare(a.bucketId, b.bucketId);
             }
         });
         ArrayList<MediaSet> albums = new ArrayList<MediaSet>();
         DataManager dataManager = mApplication.getDataManager();
         for (BucketEntry entry : entries) {
             albums.add(getLocalAlbum(dataManager,
                     mType, mPath, entry.bucketId, entry.bucketName));
         }
         for (int i = 0, n = albums.size(); i < n; ++i) {
             albums.get(i).reload();
         }
         return albums;
     }

     private MediaSet getLocalAlbum(
             DataManager manager, int type, Path parent, int id, String name) {
         Path path = parent.getChild(id);
         MediaObject object = manager.peekMediaObject(path);
         if (object != null) return (MediaSet) object;
         switch (type) {
             case MEDIA_TYPE_IMAGE:
                 return new LocalAlbum(path, mApplication, id, true, name);
             case MEDIA_TYPE_VIDEO:
                 return new LocalAlbum(path, mApplication, id, false, name);
             case MEDIA_TYPE_ALL:
                 Comparator<MediaItem> comp = DataManager.sDateTakenComparator;
                 return new LocalMergeAlbum(path, comp, new MediaSet[] {
                         getLocalAlbum(manager, MEDIA_TYPE_IMAGE, PATH_IMAGE, id, name),
                         getLocalAlbum(manager, MEDIA_TYPE_VIDEO, PATH_VIDEO, id, name)});
         }
         throw new IllegalArgumentException(String.valueOf(type));
     }

     public static String getBucketName(ContentResolver resolver, int bucketId) {
         Uri uri = mBaseUri.buildUpon()
                 .appendQueryParameter("limit", "1")
                 .build();

         Cursor cursor = resolver.query(
                 uri, PROJECTION_BUCKET, "bucket_id = ?",
                 new String[]{String.valueOf(bucketId)}, null);

         if (cursor == null) {
             Log.w(TAG, "query fail: " + uri);
             return "";
         }
         try {
             return cursor.moveToNext()
                     ? cursor.getString(INDEX_BUCKET_NAME)
                     : "";
         } finally {
             cursor.close();
         }
     }

     @Override
     public long reload() {
         // "|" is used instead of "||" because we want to clear both flags.
         if (mNotifierImage.isDirty() | mNotifierVideo.isDirty()) {
             mDataVersion = nextVersionNumber();
             mAlbums = loadSubMediaSets();
         }
         return mDataVersion;
     }

     // For debug only. Fake there is a ContentObserver.onChange() event.
     void fakeChange() {
         mNotifierImage.fakeChange();
         mNotifierVideo.fakeChange();
     }

     private static class BucketEntry {
         public String bucketName;
         public int bucketId;

         public BucketEntry(int id, String name) {
             bucketId = id;
             bucketName = Utils.ensureNotNull(name);
         }

         @Override
         public int hashCode() {
             return bucketId;
         }

         @Override
         public boolean equals(Object object) {
             if (!(object instanceof BucketEntry)) return false;
             BucketEntry entry = (BucketEntry) object;
             return bucketId == entry.bucketId;
         }
     }
 }
	/*
	* Copyright (C) 2010 The Android Open Source Project
	*
	* 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 com.android.gallery3d.data;

	import com.android.gallery3d.R;
	import com.android.gallery3d.app.GalleryApp;
	import com.android.gallery3d.common.Utils;
	import com.android.gallery3d.util.GalleryUtils;
	import com.android.gallery3d.util.MediaSetUtils;

	import android.content.ContentResolver;
	import android.database.Cursor;
	import android.net.Uri;
	import android.provider.MediaStore.Files;
	import android.provider.MediaStore.Files.FileColumns;
	import android.provider.MediaStore.Images;
	import android.provider.MediaStore.Images.ImageColumns;
	import android.provider.MediaStore.Video;

	import java.util.ArrayList;
	import java.util.Arrays;
	import java.util.Comparator;
	import java.util.HashSet;

	// LocalAlbumSet lists all image or video albums in the local storage.
	// The path should be "/local/image", "local/video" or "/local/all"
	public class LocalAlbumSet extends MediaSet {
	public static final Path PATH_ALL = Path.fromString("/local/all");
	public static final Path PATH_IMAGE = Path.fromString("/local/image");
	public static final Path PATH_VIDEO = Path.fromString("/local/video");

	private static final String TAG = "LocalAlbumSet";
	private static final String EXTERNAL_MEDIA = "external";

	// The indices should match the following projections.
	private static final int INDEX_BUCKET_ID = 0;
	private static final int INDEX_MEDIA_TYPE = 1;
	private static final int INDEX_BUCKET_NAME = 2;

	private static final Uri mBaseUri = Files.getContentUri(EXTERNAL_MEDIA);
	private static final Uri mWatchUriImage = Images.Media.EXTERNAL_CONTENT_URI;
	private static final Uri mWatchUriVideo = Video.Media.EXTERNAL_CONTENT_URI;

	// The order is import it must match to the index in MediaStore.
	private static final String[] PROJECTION_BUCKET = {
	ImageColumns.BUCKET_ID,
	FileColumns.MEDIA_TYPE,
	ImageColumns.BUCKET_DISPLAY_NAME };

	private final GalleryApp mApplication;
	private final int mType;
	private ArrayList<MediaSet> mAlbums = new ArrayList<MediaSet>();
	private final ChangeNotifier mNotifierImage;
	private final ChangeNotifier mNotifierVideo;
	private final String mName;

	public LocalAlbumSet(Path path, GalleryApp application) {
	super(path, nextVersionNumber());
	mApplication = application;
	mType = getTypeFromPath(path);
	mNotifierImage = new ChangeNotifier(this, mWatchUriImage, application);
	mNotifierVideo = new ChangeNotifier(this, mWatchUriVideo, application);
	mName = application.getResources().getString(
	R.string.set_label_local_albums);
	}

	private static int getTypeFromPath(Path path) {
	String name[] = path.split();
	if (name.length < 2) {
	throw new IllegalArgumentException(path.toString());
	}
	if ("all".equals(name[1])) return MEDIA_TYPE_ALL;
	if ("image".equals(name[1])) return MEDIA_TYPE_IMAGE;
	if ("video".equals(name[1])) return MEDIA_TYPE_VIDEO;
	throw new IllegalArgumentException(path.toString());
	}

	@Override
	public MediaSet getSubMediaSet(int index) {
	return mAlbums.get(index);
	}

	@Override
	public int getSubMediaSetCount() {
	return mAlbums.size();
	}

	@Override
	public String getName() {
	return mName;
	}

	private BucketEntry[] loadBucketEntries(Cursor cursor) {
	HashSet<BucketEntry> buffer = new HashSet<BucketEntry>();
	int typeBits = 0;
	if ((mType & MEDIA_TYPE_IMAGE) != 0) {
	typeBits \|= (1 << FileColumns.MEDIA_TYPE_IMAGE);
	}
	if ((mType & MEDIA_TYPE_VIDEO) != 0) {
	typeBits \|= (1 << FileColumns.MEDIA_TYPE_VIDEO);
	}
	try {
	while (cursor.moveToNext()) {
	if ((typeBits & (1 << cursor.getInt(INDEX_MEDIA_TYPE))) != 0) {
	buffer.add(new BucketEntry(
	cursor.getInt(INDEX_BUCKET_ID),
	cursor.getString(INDEX_BUCKET_NAME)));
	}
	}
	} finally {
	cursor.close();
	}
	return buffer.toArray(new BucketEntry[buffer.size()]);
	}


	private static int findBucket(BucketEntry entries[], int bucketId) {
	for (int i = 0, n = entries.length; i < n ; ++i) {
	if (entries[i].bucketId == bucketId) return i;
	}
	return -1;
	}

	@SuppressWarnings("unchecked")
	protected ArrayList<MediaSet> loadSubMediaSets() {
	// Note: it will be faster if we only select media_type and bucket_id.
	// need to test the performance if that is worth

	Uri uri = mBaseUri.buildUpon().
	appendQueryParameter("distinct", "true").build();
	GalleryUtils.assertNotInRenderThread();
	Cursor cursor = mApplication.getContentResolver().query(
	uri, PROJECTION_BUCKET, null, null, null);
	if (cursor == null) {
	Log.w(TAG, "cannot open local database: " + uri);
	return new ArrayList<MediaSet>();
	}
	BucketEntry[] entries = loadBucketEntries(cursor);
	int offset = 0;

	int index = findBucket(entries, MediaSetUtils.CAMERA_BUCKET_ID);
	if (index != -1) {
	Utils.swap(entries, index, offset++);
	}
	index = findBucket(entries, MediaSetUtils.DOWNLOAD_BUCKET_ID);
	if (index != -1) {
	Utils.swap(entries, index, offset++);
	}

	Arrays.sort(entries, offset, entries.length, new Comparator<BucketEntry>() {
	@Override
	public int compare(BucketEntry a, BucketEntry b) {
	int result = a.bucketName.compareTo(b.bucketName);
	return result != 0
	? result
	: Utils.compare(a.bucketId, b.bucketId);
	}
	});
	ArrayList<MediaSet> albums = new ArrayList<MediaSet>();
	DataManager dataManager = mApplication.getDataManager();
	for (BucketEntry entry : entries) {
	albums.add(getLocalAlbum(dataManager,
	mType, mPath, entry.bucketId, entry.bucketName));
	}
	for (int i = 0, n = albums.size(); i < n; ++i) {
	albums.get(i).reload();
	}
	return albums;
	}

	private MediaSet getLocalAlbum(
	DataManager manager, int type, Path parent, int id, String name) {
	Path path = parent.getChild(id);
	MediaObject object = manager.peekMediaObject(path);
	if (object != null) return (MediaSet) object;
	switch (type) {
	case MEDIA_TYPE_IMAGE:
	return new LocalAlbum(path, mApplication, id, true, name);
	case MEDIA_TYPE_VIDEO:
	return new LocalAlbum(path, mApplication, id, false, name);
	case MEDIA_TYPE_ALL:
	Comparator<MediaItem> comp = DataManager.sDateTakenComparator;
	return new LocalMergeAlbum(path, comp, new MediaSet[] {
	getLocalAlbum(manager, MEDIA_TYPE_IMAGE, PATH_IMAGE, id, name),
	getLocalAlbum(manager, MEDIA_TYPE_VIDEO, PATH_VIDEO, id, name)});
	}
	throw new IllegalArgumentException(String.valueOf(type));
	}

	public static String getBucketName(ContentResolver resolver, int bucketId) {
	Uri uri = mBaseUri.buildUpon()
	.appendQueryParameter("limit", "1")
	.build();

	Cursor cursor = resolver.query(
	uri, PROJECTION_BUCKET, "bucket_id = ?",
	new String[]{String.valueOf(bucketId)}, null);

	if (cursor == null) {
	Log.w(TAG, "query fail: " + uri);
	return "";
	}
	try {
	return cursor.moveToNext()
	? cursor.getString(INDEX_BUCKET_NAME)
	: "";
	} finally {
	cursor.close();
	}
	}

	@Override
	public long reload() {
	// "\|" is used instead of "\|\|" because we want to clear both flags.
	if (mNotifierImage.isDirty() \| mNotifierVideo.isDirty()) {
	mDataVersion = nextVersionNumber();
	mAlbums = loadSubMediaSets();
	}
	return mDataVersion;
	}

	// For debug only. Fake there is a ContentObserver.onChange() event.
	void fakeChange() {
	mNotifierImage.fakeChange();
	mNotifierVideo.fakeChange();
	}

	private static class BucketEntry {
	public String bucketName;
	public int bucketId;

	public BucketEntry(int id, String name) {
	bucketId = id;
	bucketName = Utils.ensureNotNull(name);
	}

	@Override
	public int hashCode() {
	return bucketId;
	}

	@Override
	public boolean equals(Object object) {
	if (!(object instanceof BucketEntry)) return false;
	BucketEntry entry = (BucketEntry) object;
	return bucketId == entry.bucketId;
	}
	}
	}