MainFragment.java

/*
 * This file provided by Facebook is for non-commercial testing and evaluation
 * purposes only.  Facebook reserves all rights not expressly granted.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
 * FACEBOOK BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */
package com.facebook.samples.scrollperf.fragments;

import android.Manifest;
import android.content.pm.PackageManager;
import android.net.Uri;
import android.os.Bundle;
import android.support.annotation.Nullable;
import android.support.v4.app.ActivityCompat;
import android.support.v4.app.Fragment;
import android.support.v7.widget.GridLayoutManager;
import android.support.v7.widget.LinearLayoutManager;
import android.support.v7.widget.RecyclerView;
import android.view.LayoutInflater;
import android.view.Menu;
import android.view.MenuInflater;
import android.view.View;
import android.view.ViewGroup;
import android.widget.ListAdapter;
import android.widget.ListView;

import com.facebook.samples.scrollperf.R;
import com.facebook.samples.scrollperf.conf.Config;
import com.facebook.samples.scrollperf.conf.Const;
import com.facebook.samples.scrollperf.data.SimpleAdapter;
import com.facebook.samples.scrollperf.data.impl.ContentProviderSimpleAdapter;
import com.facebook.samples.scrollperf.data.impl.DistinctUriDecorator;
import com.facebook.samples.scrollperf.data.impl.LocalResourceSimpleAdapter;
import com.facebook.samples.scrollperf.fragments.recycler.DraweeViewAdapter;
import com.facebook.samples.scrollperf.fragments.recycler.DraweeViewListAdapter;
import com.facebook.samples.scrollperf.instrumentation.PerfListener;
import com.facebook.samples.scrollperf.util.UI;

public class MainFragment extends Fragment {

  public static final String TAG = MainFragment.class.getSimpleName();

  private static final int REQUEST_READ_EXTERNAL_ID = 1;

  private RecyclerView mRecyclerView;

  private ListView mListView;

  private DraweeViewAdapter mDraweeViewAdapter;

  private ListAdapter mListAdapter;

  private SimpleAdapter<Uri> mSimpleAdapter;

  private Config mConfig;

  boolean mDistinctUriCompatible = true;

  private PerfListener mPerfListener;

  @Override
  public void onCreate(@Nullable Bundle savedInstanceState) {
    super.onCreate(savedInstanceState);
    setHasOptionsMenu(true);
    mPerfListener = new PerfListener();
  }

  @Override
  public View onCreateView(
      LayoutInflater inflater,
      ViewGroup container,
      Bundle savedInstanceState) {
    mConfig = Config.load(getContext());
    // Initialize the SimpleAdapter
    mSimpleAdapter = initializeSimpleAdapter(mConfig);
    // We use a different layout based on the type of output
    final View layout;
    switch (mConfig.recyclerLayoutType) {
      case Const.RECYCLER_VIEW_LAYOUT_VALUE:
        layout = inflater.inflate(R.layout.content_recyclerview, container, false);
        initializeRecyclerView(layout);
        break;
      case Const.LISTVIEW_LAYOUT_VALUE:
        layout = inflater.inflate(R.layout.content_listview, container, false);
        initializeListView(layout);
        break;
      case Const.GRID_RECYCLER_VIEW_LAYOUT_VALUE:
        layout = inflater.inflate(R.layout.content_recyclerview, container, false);
        initializeGridRecyclerView(layout);
        break;
      default:
        throw new IllegalStateException("Recycler Layout not supported");
    }
    updateAdapter();
    return layout;
  }

  @Override
  public void onCreateOptionsMenu(Menu menu, MenuInflater inflater) {
    inflater.inflate(R.menu.menu, menu);
    super.onCreateOptionsMenu(menu, inflater);
  }

  private void initializeRecyclerView(final View layout) {
    // Get RecyclerView
    mRecyclerView = UI.findViewById(layout, R.id.recycler_view);
    // Choose the LayoutManager
    LinearLayoutManager layoutManager = new LinearLayoutManager(getContext());
    layoutManager.setOrientation(LinearLayoutManager.VERTICAL);
    layoutManager.scrollToPosition(0);
    mRecyclerView.setLayoutManager(layoutManager);
  }

  private void initializeGridRecyclerView(final View layout) {
    // Get RecyclerView
    mRecyclerView = UI.findViewById(layout, R.id.recycler_view);
    // Choose the LayoutManager
    GridLayoutManager layoutManager = new GridLayoutManager(getContext(), mConfig.gridSpanCount);
    layoutManager.setOrientation(LinearLayoutManager.VERTICAL);
    layoutManager.scrollToPosition(0);
    mRecyclerView.setLayoutManager(layoutManager);
  }

  private void initializeListView(final View layout) {
    // get the ListView
    mListView = UI.findViewById(layout, R.id.list_view);
  }

  private SimpleAdapter<Uri> initializeSimpleAdapter(final Config config) {
    SimpleAdapter<Uri> simpleAdapter = null;
    switch (config.dataSourceType) {
      case Const.LOCAL_RESOURCE_URIS:
        simpleAdapter = LocalResourceSimpleAdapter
                .getEagerAdapter(getContext(), R.array.example_uris);
        break;
      case Const.LOCAL_RESOURCE_WEBP_URIS:
        simpleAdapter = LocalResourceSimpleAdapter
            .getEagerAdapter(getContext(), R.array.example_webp_uris);
        break;
      case Const.LOCAL_RESOURCE_PNG_URIS:
        simpleAdapter = LocalResourceSimpleAdapter
            .getEagerAdapter(getContext(), R.array.example_png_uris);
        break;
      case Const.LOCAL_INTERNAL_PHOTO_URIS:
        simpleAdapter = ContentProviderSimpleAdapter.getInternalPhotoSimpleAdapter(getActivity());
        mDistinctUriCompatible = false;
        break;
      case Const.LOCAL_EXTERNAL_PHOTO_URIS:
        simpleAdapter = getExternalPhotoSimpleAdapter();
        mDistinctUriCompatible = false;
        break;
    }
    return simpleAdapter;
  }

  private void updateAdapter() {
    if (mSimpleAdapter == null) {
      return;
    }
    if (mConfig.infiniteDataSource) {
      mSimpleAdapter = SimpleAdapter.Util.makeItInfinite(mSimpleAdapter);
      if (mDistinctUriCompatible && mConfig.distinctUriDataSource) {
        mSimpleAdapter = SimpleAdapter.Util
            .decorate(mSimpleAdapter, DistinctUriDecorator.SINGLETON);
      }
    }
    switch (mConfig.recyclerLayoutType) {
      case Const.RECYCLER_VIEW_LAYOUT_VALUE:
      case Const.GRID_RECYCLER_VIEW_LAYOUT_VALUE:
        // Create the Adapter
        mDraweeViewAdapter = new DraweeViewAdapter(
            getContext(),
            mSimpleAdapter,
            mConfig,
            mPerfListener);
        mRecyclerView.setAdapter(mDraweeViewAdapter);
        break;
      case Const.LISTVIEW_LAYOUT_VALUE:
        // Create the Adapter
        mListAdapter = new DraweeViewListAdapter(
            getContext(),
            mSimpleAdapter,
            mConfig,
            mPerfListener);
        // Set the adapter
        mListView.setAdapter(mListAdapter);
        break;
      default:
        throw new IllegalStateException("Recycler Layout not supported");
    }
  }

  private SimpleAdapter<Uri> getExternalPhotoSimpleAdapter() {
    if (ActivityCompat.checkSelfPermission(getContext(), Manifest.permission.READ_EXTERNAL_STORAGE)
        == PackageManager.PERMISSION_GRANTED) {
      return ContentProviderSimpleAdapter.getExternalPhotoSimpleAdapter(getActivity());
    } else {
      requestPermissions(
          new String[] {Manifest.permission.READ_EXTERNAL_STORAGE},
          REQUEST_READ_EXTERNAL_ID);
    }
    return SimpleAdapter.Util.EMPTY_ADAPTER;
  }

  @Override
  public void onRequestPermissionsResult(
      int requestCode, String[] permissions, int[] grantResults) {
    if (requestCode == REQUEST_READ_EXTERNAL_ID &&
        grantResults[0] == PackageManager.PERMISSION_GRANTED) {
      mSimpleAdapter = ContentProviderSimpleAdapter.getExternalPhotoSimpleAdapter(getActivity());
      updateAdapter();
    }
    super.onRequestPermissionsResult(requestCode, permissions, grantResults);
  }
}