session-android/src/org/thoughtcrime/securesms/components/emoji/EmojiProvider.java

package org.thoughtcrime.securesms.components.emoji;

import android.content.Context;
import android.graphics.Bitmap;
import android.graphics.Canvas;
import android.graphics.ColorFilter;
import android.graphics.Paint;
import android.graphics.PixelFormat;
import android.graphics.Rect;
import android.graphics.drawable.Drawable;
import android.text.Spannable;
import android.text.SpannableStringBuilder;
import android.text.style.ImageSpan;
import android.util.Log;
import android.util.SparseArray;
import android.view.View;

import org.thoughtcrime.securesms.R;
import org.thoughtcrime.securesms.util.BitmapDecodingException;
import org.thoughtcrime.securesms.util.BitmapUtil;
import org.thoughtcrime.securesms.util.ResUtil;
import org.thoughtcrime.securesms.util.Util;

import java.io.IOException;
import java.io.InputStream;
import java.lang.ref.SoftReference;
import java.util.Arrays;
import java.util.concurrent.ExecutorService;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class EmojiProvider {
  private static final    String                             TAG      = EmojiProvider.class.getSimpleName();
  private static final    ExecutorService                    executor = Util.newSingleThreadedLifoExecutor();
  private static volatile EmojiProvider                      instance = null;
  private static final    SparseArray<SoftReference<Bitmap>> bitmaps  = new SparseArray<>();

  private final SparseArray<DrawInfo> offsets = new SparseArray<>();

  @SuppressWarnings("MalformedRegex")
  //                                                            0x20a0-0x32ff          0x1f00-0x1fff              0xfe4e5-0xfe4ee
  //                                                           |==== misc ====||======== emoticons ========||========= flags ==========|
  private static final Pattern EMOJI_RANGE = Pattern.compile("[\\u20a0-\\u32ff\\ud83c\\udc00-\\ud83d\\udeff\\udbb9\\udce5-\\udbb9\\udcee]");

  public static final double EMOJI_HUGE     = 1.00;
  public static final double EMOJI_LARGE    = 0.75;
  public static final double EMOJI_SMALL    = 0.60;
  public static final int    EMOJI_RAW_SIZE = 128;
  public static final int    EMOJI_PER_ROW  = 16;

  private final Context context;
  private final int     bigDrawSize;
  private final int[]   pages;

  public static EmojiProvider getInstance(Context context) {
    if (instance == null) {
      synchronized (EmojiProvider.class) {
        if (instance == null) {
          instance = new EmojiProvider(context);
        }
      }
    }
    return instance;
  }

  private EmojiProvider(Context context) {
    this.context = context.getApplicationContext();
    this.bigDrawSize = context.getResources().getDimensionPixelSize(R.dimen.emoji_drawer_size);
    this.pages = ResUtil.getResourceIds(context, R.array.emoji_categories);
    for (int i = 0; i < pages.length; i++) {
      final int[] page = context.getResources().getIntArray(pages[i]);
      for (int j = 0; j < page.length; j++) {
        offsets.put(page[j], new DrawInfo(i, j));
      }
    }
  }

  private void preloadPage(final int page, final PageLoadedListener pageLoadListener) {
    executor.submit(new Runnable() {
      @Override
      public void run() {
        try {
          loadPage(page);
          if (pageLoadListener != null) {
            Log.w(TAG, "onPageLoaded("+page+")");
            pageLoadListener.onPageLoaded();
          }
        } catch (IOException ioe) {
          Log.w(TAG, ioe);
        }
      }
    });
  }

  private void loadPage(int page) throws IOException {
    if (page < 0 || page >= pages.length) {
      throw new IndexOutOfBoundsException("can't load page that doesn't exist");
    }

    if (bitmaps.get(page) != null && bitmaps.get(page).get() != null) return;

    try {
      final String file = "emoji_" + page + "_wrapped.png";
      final InputStream measureStream = context.getAssets().open(file);
      final InputStream bitmapStream  = context.getAssets().open(file);
      final Bitmap      bitmap        = BitmapUtil.createScaledBitmap(measureStream, bitmapStream, (float) bigDrawSize / (float) EMOJI_RAW_SIZE);
      bitmaps.put(page, new SoftReference<>(bitmap));
    } catch (IOException ioe) {
      Log.w(TAG, ioe);
      throw ioe;
    } catch (BitmapDecodingException bde) {
      Log.w(TAG, bde);
      throw new AssertionError("emoji sprite asset is corrupted or android decoding is broken");
    }
  }

  public CharSequence emojify(CharSequence text, PageLoadedListener pageLoadedListener) {
    return emojify(text, EMOJI_LARGE, pageLoadedListener);
  }

  public CharSequence emojify(CharSequence text, double size, PageLoadedListener pageLoadedListener) {
    Matcher                matches = EMOJI_RANGE.matcher(text);
    SpannableStringBuilder builder = new SpannableStringBuilder(text);

    while (matches.find()) {
      int codePoint = matches.group().codePointAt(0);
      Drawable drawable = getEmojiDrawable(codePoint, size, pageLoadedListener);
      if (drawable != null) {
        ImageSpan imageSpan = new ImageSpan(drawable, ImageSpan.ALIGN_BOTTOM);
        char[] chars = new char[matches.end() - matches.start()];
        Arrays.fill(chars, ' ');
        builder.setSpan(imageSpan, matches.start(), matches.end(),
                        Spannable.SPAN_EXCLUSIVE_EXCLUSIVE);
      }
    }

    return builder;
  }

  public Drawable getEmojiDrawable(int emojiCode, double size, PageLoadedListener pageLoadedListener) {
    return getEmojiDrawable(offsets.get(emojiCode), size, pageLoadedListener);
  }

  private Drawable getEmojiDrawable(DrawInfo drawInfo, double size, PageLoadedListener pageLoadedListener) {
    if (drawInfo == null) {
      return null;
    }
    final Drawable drawable = new EmojiDrawable(drawInfo, bigDrawSize);
    drawable.setBounds(0, 0, (int)((double)bigDrawSize * size), (int)((double)bigDrawSize * size));
    if (bitmaps.get(drawInfo.page) == null || bitmaps.get(drawInfo.page).get() == null) {
      preloadPage(drawInfo.page, pageLoadedListener);
    }
    return drawable;
  }

  public static class EmojiDrawable extends Drawable {
    private final        int    index;
    private final        int    page;
    private final        int    emojiSize;
    private static final Paint  paint;
    private              Bitmap bmp;

    static {
      paint = new Paint();
      paint.setFilterBitmap(true);
    }

    public EmojiDrawable(DrawInfo info, int emojiSize) {
      this.index     = info.index;
      this.page      = info.page;
      this.emojiSize = emojiSize;
    }

    @Override
    public void draw(Canvas canvas) {
      if (bitmaps.get(page) == null || bitmaps.get(page).get() == null) {
        Log.w(TAG, "bitmap for this page was null");
        return;
      }
      if (bmp == null) {
        bmp = bitmaps.get(page).get();
      }

      Rect b = copyBounds();

      final int row = index / EMOJI_PER_ROW;
      final int row_index = index % EMOJI_PER_ROW;

      canvas.drawBitmap(bmp,
                        new Rect(row_index * emojiSize,
                                 row * emojiSize,
                                 (row_index + 1) * emojiSize,
                                 (row + 1) * emojiSize),
                        b,
                        paint);
    }

    @Override
    public int getOpacity() {
      return PixelFormat.TRANSLUCENT;
    }

    @Override
    public void setAlpha(int alpha) { }

    @Override
    public void setColorFilter(ColorFilter cf) { }

    @Override
    public String toString() {
      return "EmojiDrawable{" +
          "page=" + page +
          ", index=" + index +
          '}';
    }
  }

  public static class InvalidatingPageLoadedListener implements PageLoadedListener {
    private final View view;

    public InvalidatingPageLoadedListener(final View view) {
      this.view = view;
    }

    @Override
    public void onPageLoaded() {
      view.postInvalidate();
    }

    @Override
    public String toString() {
      return "InvalidatingPageLoadedListener{}";
    }
  }

  class DrawInfo {
    int page;
    int index;

    public DrawInfo(final int page, final int index) {
      this.page = page;
      this.index = index;
    }

    @Override
    public String toString() {
      return "DrawInfo{" +
          "page=" + page +
          ", index=" + index +
          '}';
    }
  }

  interface PageLoadedListener {
    void onPageLoaded();
  }
}