WallpaperPicker/src/com/android/launcher3/SavedWallpaperImages.java - platform/packages/apps/Launcher3 - Git at Google

 /*
  * Copyright (C) 2013 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.launcher3;

 import android.content.ContentValues;
 import android.content.Context;
 import android.database.Cursor;
 import android.database.sqlite.SQLiteDatabase;
 import android.database.sqlite.SQLiteOpenHelper;
 import android.graphics.Bitmap;
 import android.graphics.BitmapFactory;
 import android.graphics.drawable.BitmapDrawable;
 import android.graphics.drawable.Drawable;
 import android.net.Uri;
 import android.text.TextUtils;
 import android.util.Log;
 import android.util.Pair;
 import android.view.LayoutInflater;
 import android.view.View;
 import android.view.ViewGroup;
 import android.widget.BaseAdapter;
 import android.widget.ListAdapter;

 import com.android.launcher3.WallpaperCropActivity.CropViewScaleAndOffsetProvider;
 import com.android.photos.views.TiledImageRenderer.TileSource;

 import java.io.ByteArrayInputStream;
 import java.io.File;
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.io.InputStream;
 import java.util.ArrayList;


 public class SavedWallpaperImages extends BaseAdapter implements ListAdapter {
     private static String TAG = "Launcher3.SavedWallpaperImages";
     private ImageDb mDb;
     ArrayList<SavedWallpaperTile> mImages;
     Context mContext;
     LayoutInflater mLayoutInflater;

     public static class SavedWallpaperTile extends WallpaperPickerActivity.FileWallpaperInfo {
         private int mDbId;

         // three floats representing scale, centerX and centerY of the crop view in order.
         private Float[] mExtras;
         public SavedWallpaperTile(int dbId, File target, Drawable thumb, Float[] extras) {
             super(target, thumb);
             mDbId = dbId;
             mExtras = extras != null && extras.length == 3 ? extras : null;
         }

         @Override
         public void onDelete(WallpaperPickerActivity a) {
             a.getSavedImages().deleteImage(mDbId);
         }

         @Override
         protected CropViewScaleAndOffsetProvider getCropViewScaleAndOffsetProvider() {
             if (mExtras != null) {
                 return new CropViewScaleAndOffsetProvider() {
                     @Override
                     public void updateCropView(WallpaperCropActivity a, TileSource src) {
                         a.mCropView.setScaleAndCenter(mExtras[0], mExtras[1], mExtras[2]);
                     }
                 };
             }
             return null;
         }

         @Override
         public void onSave(final WallpaperPickerActivity a) {
             if (mExtras == null) {
                 super.onSave(a);
             } else {
                 boolean shouldFadeOutOnFinish = a.getWallpaperParallaxOffset() == 0f;
                 a.cropImageAndSetWallpaper(Uri.fromFile(mFile), null, true, shouldFadeOutOnFinish);
             }
         }
     }

     public SavedWallpaperImages(Context context) {
         // We used to store the saved images in the cache directory, but that meant they'd get
         // deleted sometimes-- move them to the data directory
         ImageDb.moveFromCacheDirectoryIfNecessary(context);
         mDb = new ImageDb(context);
         mContext = context;
         mLayoutInflater = LayoutInflater.from(context);
     }

     public void loadThumbnailsAndImageIdList() {
         mImages = new ArrayList<SavedWallpaperTile>();
         SQLiteDatabase db = mDb.getReadableDatabase();
         Cursor result = db.query(ImageDb.TABLE_NAME,
                 new String[] { ImageDb.COLUMN_ID,
                     ImageDb.COLUMN_IMAGE_THUMBNAIL_FILENAME,
                     ImageDb.COLUMN_IMAGE_FILENAME,
                     ImageDb.COLUMN_EXTRAS}, // cols to return
                 null, // select query
                 null, // args to select query
                 null,
                 null,
                 ImageDb.COLUMN_ID + " DESC",
                 null);

         while (result.moveToNext()) {
             String filename = result.getString(1);
             File file = new File(mContext.getFilesDir(), filename);

             Bitmap thumb = BitmapFactory.decodeFile(file.getAbsolutePath());
             if (thumb != null) {

                 Float[] extras = null;
                 String extraStr = result.getString(3);
                 if (extraStr != null) {
                     String[] parts = extraStr.split(",");
                     extras = new Float[parts.length];
                     for (int i = 0; i < parts.length; i++) {
                         try {
                             extras[i] = Float.parseFloat(parts[i]);
                         } catch (Exception e) {
                             extras = null;
                             break;
                         }
                     }
                 }

                 mImages.add(new SavedWallpaperTile(result.getInt(0),
                         new File(mContext.getFilesDir(), result.getString(2)),
                         new BitmapDrawable(thumb), extras));
             }
         }
         result.close();
     }

     public int getCount() {
         return mImages.size();
     }

     public SavedWallpaperTile getItem(int position) {
         return mImages.get(position);
     }

     public long getItemId(int position) {
         return position;
     }

     public View getView(int position, View convertView, ViewGroup parent) {
         Drawable thumbDrawable = mImages.get(position).mThumb;
         if (thumbDrawable == null) {
             Log.e(TAG, "Error decoding thumbnail for wallpaper #" + position);
         }
         return WallpaperPickerActivity.createImageTileView(
                 mLayoutInflater, convertView, parent, thumbDrawable);
     }

     private Pair<String, String> getImageFilenames(int id) {
         SQLiteDatabase db = mDb.getReadableDatabase();
         Cursor result = db.query(ImageDb.TABLE_NAME,
                 new String[] { ImageDb.COLUMN_IMAGE_THUMBNAIL_FILENAME,
                     ImageDb.COLUMN_IMAGE_FILENAME }, // cols to return
                 ImageDb.COLUMN_ID + " = ?", // select query
                 new String[] { Integer.toString(id) }, // args to select query
                 null,
                 null,
                 null,
                 null);
         if (result.getCount() > 0) {
             result.moveToFirst();
             String thumbFilename = result.getString(0);
             String imageFilename = result.getString(1);
             result.close();
             return new Pair<String, String>(thumbFilename, imageFilename);
         } else {
             return null;
         }
     }

     public void deleteImage(int id) {
         Pair<String, String> filenames = getImageFilenames(id);
         File imageFile = new File(mContext.getFilesDir(), filenames.first);
         imageFile.delete();
         File thumbFile = new File(mContext.getFilesDir(), filenames.second);
         thumbFile.delete();
         SQLiteDatabase db = mDb.getWritableDatabase();
         db.delete(ImageDb.TABLE_NAME,
                 ImageDb.COLUMN_ID + " = ?", // SELECT query
                 new String[] {
                     Integer.toString(id) // args to SELECT query
                 });
     }

     public void writeImage(Bitmap thumbnail, byte[] imageBytes) {
         try {
             writeImage(thumbnail, new ByteArrayInputStream(imageBytes), null);
         } catch (IOException e) {
             Log.e(TAG, "Failed writing images to storage " + e);
         }
     }

     public void writeImage(Bitmap thumbnail, Uri uri, Float[] extras) {
         try {
             writeImage(thumbnail, mContext.getContentResolver().openInputStream(uri), extras);
         } catch (IOException e) {
             Log.e(TAG, "Failed writing images to storage " + e);
         }
     }

     private void writeImage(Bitmap thumbnail, InputStream in, Float[] extras) throws IOException {
         File imageFile = File.createTempFile("wallpaper", "", mContext.getFilesDir());
         FileOutputStream imageFileStream =
                 mContext.openFileOutput(imageFile.getName(), Context.MODE_PRIVATE);
         byte[] buf = new byte[4096];    // 4k
         int len;
         while ((len = in.read(buf)) > 0) {
             imageFileStream.write(buf, 0, len);
         }
         imageFileStream.close();
         in.close();

         File thumbFile = File.createTempFile("wallpaperthumb", "", mContext.getFilesDir());
         FileOutputStream thumbFileStream =
                 mContext.openFileOutput(thumbFile.getName(), Context.MODE_PRIVATE);
         thumbnail.compress(Bitmap.CompressFormat.JPEG, 95, thumbFileStream);
         thumbFileStream.close();

         SQLiteDatabase db = mDb.getWritableDatabase();
         ContentValues values = new ContentValues();
         values.put(ImageDb.COLUMN_IMAGE_THUMBNAIL_FILENAME, thumbFile.getName());
         values.put(ImageDb.COLUMN_IMAGE_FILENAME, imageFile.getName());
         if (extras != null) {
             values.put(ImageDb.COLUMN_EXTRAS, TextUtils.join(",", extras));
         }
         db.insert(ImageDb.TABLE_NAME, null, values);
     }

     static class ImageDb extends SQLiteOpenHelper {
         final static int DB_VERSION = 2;
         final static String TABLE_NAME = "saved_wallpaper_images";
         final static String COLUMN_ID = "id";
         final static String COLUMN_IMAGE_THUMBNAIL_FILENAME = "image_thumbnail";
         final static String COLUMN_IMAGE_FILENAME = "image";
         final static String COLUMN_EXTRAS = "extras";

         Context mContext;

         public ImageDb(Context context) {
             super(context, context.getDatabasePath(LauncherFiles.WALLPAPER_IMAGES_DB).getPath(),
                     null, DB_VERSION);
             // Store the context for later use
             mContext = context;
         }

         public static void moveFromCacheDirectoryIfNecessary(Context context) {
             // We used to store the saved images in the cache directory, but that meant they'd get
             // deleted sometimes-- move them to the data directory
             File oldSavedImagesFile = new File(context.getCacheDir(),
                     LauncherFiles.WALLPAPER_IMAGES_DB);
             File savedImagesFile = context.getDatabasePath(LauncherFiles.WALLPAPER_IMAGES_DB);
             if (oldSavedImagesFile.exists()) {
                 oldSavedImagesFile.renameTo(savedImagesFile);
             }
         }
         @Override
         public void onCreate(SQLiteDatabase database) {
             database.execSQL("CREATE TABLE IF NOT EXISTS " + TABLE_NAME + " (" +
                     COLUMN_ID + " INTEGER NOT NULL, " +
                     COLUMN_IMAGE_THUMBNAIL_FILENAME + " TEXT NOT NULL, " +
                     COLUMN_IMAGE_FILENAME + " TEXT NOT NULL, " +
                     COLUMN_EXTRAS + " TEXT, " +
                     "PRIMARY KEY (" + COLUMN_ID + " ASC) " +
                     ");");
         }

         @Override
         public void onUpgrade(SQLiteDatabase db, int oldVersion, int newVersion) {
             if (oldVersion == 1) {
                 // Add extras column
                 db.execSQL("ALTER TABLE " + TABLE_NAME + " ADD COLUMN " + COLUMN_EXTRAS + " TEXT;");
             } else if (oldVersion != newVersion) {
                 // Delete all the records; they'll be repopulated as this is a cache
                 db.execSQL("DELETE FROM " + TABLE_NAME);
                 onCreate(db);
             }
         }
     }
 }
	/*
	* Copyright (C) 2013 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.launcher3;

	import android.content.ContentValues;
	import android.content.Context;
	import android.database.Cursor;
	import android.database.sqlite.SQLiteDatabase;
	import android.database.sqlite.SQLiteOpenHelper;
	import android.graphics.Bitmap;
	import android.graphics.BitmapFactory;
	import android.graphics.drawable.BitmapDrawable;
	import android.graphics.drawable.Drawable;
	import android.net.Uri;
	import android.text.TextUtils;
	import android.util.Log;
	import android.util.Pair;
	import android.view.LayoutInflater;
	import android.view.View;
	import android.view.ViewGroup;
	import android.widget.BaseAdapter;
	import android.widget.ListAdapter;

	import com.android.launcher3.WallpaperCropActivity.CropViewScaleAndOffsetProvider;
	import com.android.photos.views.TiledImageRenderer.TileSource;

	import java.io.ByteArrayInputStream;
	import java.io.File;
	import java.io.FileOutputStream;
	import java.io.IOException;
	import java.io.InputStream;
	import java.util.ArrayList;


	public class SavedWallpaperImages extends BaseAdapter implements ListAdapter {
	private static String TAG = "Launcher3.SavedWallpaperImages";
	private ImageDb mDb;
	ArrayList<SavedWallpaperTile> mImages;
	Context mContext;
	LayoutInflater mLayoutInflater;

	public static class SavedWallpaperTile extends WallpaperPickerActivity.FileWallpaperInfo {
	private int mDbId;

	// three floats representing scale, centerX and centerY of the crop view in order.
	private Float[] mExtras;
	public SavedWallpaperTile(int dbId, File target, Drawable thumb, Float[] extras) {
	super(target, thumb);
	mDbId = dbId;
	mExtras = extras != null && extras.length == 3 ? extras : null;
	}

	@Override
	public void onDelete(WallpaperPickerActivity a) {
	a.getSavedImages().deleteImage(mDbId);
	}

	@Override
	protected CropViewScaleAndOffsetProvider getCropViewScaleAndOffsetProvider() {
	if (mExtras != null) {
	return new CropViewScaleAndOffsetProvider() {
	@Override
	public void updateCropView(WallpaperCropActivity a, TileSource src) {
	a.mCropView.setScaleAndCenter(mExtras[0], mExtras[1], mExtras[2]);
	}
	};
	}
	return null;
	}

	@Override
	public void onSave(final WallpaperPickerActivity a) {
	if (mExtras == null) {
	super.onSave(a);
	} else {
	boolean shouldFadeOutOnFinish = a.getWallpaperParallaxOffset() == 0f;
	a.cropImageAndSetWallpaper(Uri.fromFile(mFile), null, true, shouldFadeOutOnFinish);
	}
	}
	}

	public SavedWallpaperImages(Context context) {
	// We used to store the saved images in the cache directory, but that meant they'd get
	// deleted sometimes-- move them to the data directory
	ImageDb.moveFromCacheDirectoryIfNecessary(context);
	mDb = new ImageDb(context);
	mContext = context;
	mLayoutInflater = LayoutInflater.from(context);
	}

	public void loadThumbnailsAndImageIdList() {
	mImages = new ArrayList<SavedWallpaperTile>();
	SQLiteDatabase db = mDb.getReadableDatabase();
	Cursor result = db.query(ImageDb.TABLE_NAME,
	new String[] { ImageDb.COLUMN_ID,
	ImageDb.COLUMN_IMAGE_THUMBNAIL_FILENAME,
	ImageDb.COLUMN_IMAGE_FILENAME,
	ImageDb.COLUMN_EXTRAS}, // cols to return
	null, // select query
	null, // args to select query
	null,
	null,
	ImageDb.COLUMN_ID + " DESC",
	null);

	while (result.moveToNext()) {
	String filename = result.getString(1);
	File file = new File(mContext.getFilesDir(), filename);

	Bitmap thumb = BitmapFactory.decodeFile(file.getAbsolutePath());
	if (thumb != null) {

	Float[] extras = null;
	String extraStr = result.getString(3);
	if (extraStr != null) {
	String[] parts = extraStr.split(",");
	extras = new Float[parts.length];
	for (int i = 0; i < parts.length; i++) {
	try {
	extras[i] = Float.parseFloat(parts[i]);
	} catch (Exception e) {
	extras = null;
	break;
	}
	}
	}

	mImages.add(new SavedWallpaperTile(result.getInt(0),
	new File(mContext.getFilesDir(), result.getString(2)),
	new BitmapDrawable(thumb), extras));
	}
	}
	result.close();
	}

	public int getCount() {
	return mImages.size();
	}

	public SavedWallpaperTile getItem(int position) {
	return mImages.get(position);
	}

	public long getItemId(int position) {
	return position;
	}

	public View getView(int position, View convertView, ViewGroup parent) {
	Drawable thumbDrawable = mImages.get(position).mThumb;
	if (thumbDrawable == null) {
	Log.e(TAG, "Error decoding thumbnail for wallpaper #" + position);
	}
	return WallpaperPickerActivity.createImageTileView(
	mLayoutInflater, convertView, parent, thumbDrawable);
	}

	private Pair<String, String> getImageFilenames(int id) {
	SQLiteDatabase db = mDb.getReadableDatabase();
	Cursor result = db.query(ImageDb.TABLE_NAME,
	new String[] { ImageDb.COLUMN_IMAGE_THUMBNAIL_FILENAME,
	ImageDb.COLUMN_IMAGE_FILENAME }, // cols to return
	ImageDb.COLUMN_ID + " = ?", // select query
	new String[] { Integer.toString(id) }, // args to select query
	null,
	null,
	null,
	null);
	if (result.getCount() > 0) {
	result.moveToFirst();
	String thumbFilename = result.getString(0);
	String imageFilename = result.getString(1);
	result.close();
	return new Pair<String, String>(thumbFilename, imageFilename);
	} else {
	return null;
	}
	}

	public void deleteImage(int id) {
	Pair<String, String> filenames = getImageFilenames(id);
	File imageFile = new File(mContext.getFilesDir(), filenames.first);
	imageFile.delete();
	File thumbFile = new File(mContext.getFilesDir(), filenames.second);
	thumbFile.delete();
	SQLiteDatabase db = mDb.getWritableDatabase();
	db.delete(ImageDb.TABLE_NAME,
	ImageDb.COLUMN_ID + " = ?", // SELECT query
	new String[] {
	Integer.toString(id) // args to SELECT query
	});
	}

	public void writeImage(Bitmap thumbnail, byte[] imageBytes) {
	try {
	writeImage(thumbnail, new ByteArrayInputStream(imageBytes), null);
	} catch (IOException e) {
	Log.e(TAG, "Failed writing images to storage " + e);
	}
	}

	public void writeImage(Bitmap thumbnail, Uri uri, Float[] extras) {
	try {
	writeImage(thumbnail, mContext.getContentResolver().openInputStream(uri), extras);
	} catch (IOException e) {
	Log.e(TAG, "Failed writing images to storage " + e);
	}
	}

	private void writeImage(Bitmap thumbnail, InputStream in, Float[] extras) throws IOException {
	File imageFile = File.createTempFile("wallpaper", "", mContext.getFilesDir());
	FileOutputStream imageFileStream =
	mContext.openFileOutput(imageFile.getName(), Context.MODE_PRIVATE);
	byte[] buf = new byte[4096]; // 4k
	int len;
	while ((len = in.read(buf)) > 0) {
	imageFileStream.write(buf, 0, len);
	}
	imageFileStream.close();
	in.close();

	File thumbFile = File.createTempFile("wallpaperthumb", "", mContext.getFilesDir());
	FileOutputStream thumbFileStream =
	mContext.openFileOutput(thumbFile.getName(), Context.MODE_PRIVATE);
	thumbnail.compress(Bitmap.CompressFormat.JPEG, 95, thumbFileStream);
	thumbFileStream.close();

	SQLiteDatabase db = mDb.getWritableDatabase();
	ContentValues values = new ContentValues();
	values.put(ImageDb.COLUMN_IMAGE_THUMBNAIL_FILENAME, thumbFile.getName());
	values.put(ImageDb.COLUMN_IMAGE_FILENAME, imageFile.getName());
	if (extras != null) {
	values.put(ImageDb.COLUMN_EXTRAS, TextUtils.join(",", extras));
	}
	db.insert(ImageDb.TABLE_NAME, null, values);
	}

	static class ImageDb extends SQLiteOpenHelper {
	final static int DB_VERSION = 2;
	final static String TABLE_NAME = "saved_wallpaper_images";
	final static String COLUMN_ID = "id";
	final static String COLUMN_IMAGE_THUMBNAIL_FILENAME = "image_thumbnail";
	final static String COLUMN_IMAGE_FILENAME = "image";
	final static String COLUMN_EXTRAS = "extras";

	Context mContext;

	public ImageDb(Context context) {
	super(context, context.getDatabasePath(LauncherFiles.WALLPAPER_IMAGES_DB).getPath(),
	null, DB_VERSION);
	// Store the context for later use
	mContext = context;
	}

	public static void moveFromCacheDirectoryIfNecessary(Context context) {
	// We used to store the saved images in the cache directory, but that meant they'd get
	// deleted sometimes-- move them to the data directory
	File oldSavedImagesFile = new File(context.getCacheDir(),
	LauncherFiles.WALLPAPER_IMAGES_DB);
	File savedImagesFile = context.getDatabasePath(LauncherFiles.WALLPAPER_IMAGES_DB);
	if (oldSavedImagesFile.exists()) {
	oldSavedImagesFile.renameTo(savedImagesFile);
	}
	}
	@Override
	public void onCreate(SQLiteDatabase database) {
	database.execSQL("CREATE TABLE IF NOT EXISTS " + TABLE_NAME + " (" +
	COLUMN_ID + " INTEGER NOT NULL, " +
	COLUMN_IMAGE_THUMBNAIL_FILENAME + " TEXT NOT NULL, " +
	COLUMN_IMAGE_FILENAME + " TEXT NOT NULL, " +
	COLUMN_EXTRAS + " TEXT, " +
	"PRIMARY KEY (" + COLUMN_ID + " ASC) " +
	");");
	}

	@Override
	public void onUpgrade(SQLiteDatabase db, int oldVersion, int newVersion) {
	if (oldVersion == 1) {
	// Add extras column
	db.execSQL("ALTER TABLE " + TABLE_NAME + " ADD COLUMN " + COLUMN_EXTRAS + " TEXT;");
	} else if (oldVersion != newVersion) {
	// Delete all the records; they'll be repopulated as this is a cache
	db.execSQL("DELETE FROM " + TABLE_NAME);
	onCreate(db);
	}
	}
	}
	}