diff options
Diffstat (limited to 'partner_support/samples/src/com/example/partnersupportsampletvinput/LineupSelectionFragment.java')
-rw-r--r-- | partner_support/samples/src/com/example/partnersupportsampletvinput/LineupSelectionFragment.java | 278 |
1 files changed, 278 insertions, 0 deletions
diff --git a/partner_support/samples/src/com/example/partnersupportsampletvinput/LineupSelectionFragment.java b/partner_support/samples/src/com/example/partnersupportsampletvinput/LineupSelectionFragment.java new file mode 100644 index 00000000..8c3ca77a --- /dev/null +++ b/partner_support/samples/src/com/example/partnersupportsampletvinput/LineupSelectionFragment.java @@ -0,0 +1,278 @@ +/* + * Copyright (C) 2017 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.example.partnersupportsampletvinput; + +import android.app.Activity; +import android.app.FragmentManager; +import android.content.ContentResolver; +import android.content.ContentValues; +import android.os.AsyncTask; +import android.os.Bundle; +import android.support.annotation.NonNull; +import android.support.annotation.Nullable; +import android.support.v17.leanback.app.GuidedStepFragment; +import android.support.v17.leanback.widget.GuidanceStylist; +import android.support.v17.leanback.widget.GuidanceStylist.Guidance; +import android.support.v17.leanback.widget.GuidedAction; +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 com.google.android.tv.partner.support.EpgContract; +import com.google.android.tv.partner.support.EpgInput; +import com.google.android.tv.partner.support.EpgInputs; +import com.google.android.tv.partner.support.Lineup; +import com.google.android.tv.partner.support.Lineups; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.Collections; +import java.util.Comparator; +import java.util.HashMap; +import java.util.List; +import java.util.Map; +import java.util.regex.Pattern; + +/** Lineup Selection Fragment for users to select a lineup */ +public class LineupSelectionFragment extends GuidedStepFragment { + public static final boolean DEBUG = false; + public static final String TAG = "LineupSelectionFragment"; + + public static final String KEY_TITLE = "title"; + public static final String KEY_DESCRIPTION = "description"; + private static final long ACTION_ID_STATUS = 1; + private static final long ACTION_ID_LINEUPS = 2; + private static final Pattern CHANNEL_NUMBER_DELIMITER = Pattern.compile("([ .-])"); + private Activity mActivity; + private GuidedAction mStatusAction; + private String mPostcode; + private List<String> mChannelNumbers; + private Map<GuidedAction, Lineup> mActionLineupMap = new HashMap<>(); + + private AsyncTask<Void, Void, List<Pair<Lineup, Integer>>> mFetchLineupTask; + + @Override + public void onCreate(Bundle savedInstanceState) { + mActivity = getActivity(); + mStatusAction = + new GuidedAction.Builder(mActivity) + .id(ACTION_ID_STATUS) + .title("Loading lineups") + .description("please wait") + .focusable(false) + .build(); + + mFetchLineupTask = + new AsyncTask<Void, Void, List<Pair<Lineup, Integer>>>() { + @Override + protected List<Pair<Lineup, Integer>> doInBackground(Void... voids) { + return lineupChannelMatchCount(getLineups(), getChannelNumbers()); + } + + @Override + protected void onPostExecute(List<Pair<Lineup, Integer>> result) { + List<GuidedAction> actions = new ArrayList<>(); + if (result.isEmpty()) { + mStatusAction.setTitle("No lineup found for postcode: " + mPostcode); + mStatusAction.setDescription(""); + mStatusAction.setFocusable(true); + notifyActionChanged(findActionPositionById(ACTION_ID_STATUS)); + return; + } + mActionLineupMap = new HashMap<>(); + for (Pair<Lineup, Integer> pair : result) { + Lineup lineup = pair.first; + String title = + TextUtils.isEmpty(lineup.getName()) + ? lineup.getId() + : lineup.getName(); + GuidedAction action = + new GuidedAction.Builder(getActivity()) + .id(ACTION_ID_LINEUPS) + .title(title) + .description(pair.second + " channels match") + .build(); + mActionLineupMap.put(action, lineup); + actions.add(action); + } + setActions(actions); + } + }; + + super.onCreate(savedInstanceState); + } + + @NonNull + @Override + public GuidanceStylist.Guidance onCreateGuidance(Bundle savedInstanceState) { + return new Guidance( + "LineupSelectionFragment", "LineupSelectionFragment Description", null, null); + } + + @Override + public View onCreateView( + LayoutInflater inflater, ViewGroup container, Bundle savedInstanceState) { + View v = super.onCreateView(inflater, container, savedInstanceState); + if (!mFetchLineupTask.isCancelled()) { + mPostcode = getArguments().getString(ChannelScanFragment.KEY_POSTCODE); + mChannelNumbers = + getArguments().getStringArrayList(ChannelScanFragment.KEY_CHANNEL_NUMBERS); + mFetchLineupTask.execute(); + } + return v; + } + + @Override + public void onDestroyView() { + mFetchLineupTask.cancel(true); + super.onDestroyView(); + } + + @Override + public void onCreateActions(@NonNull List<GuidedAction> actions, Bundle savedInstanceState) { + actions.add(mStatusAction); + } + + @Override + public void onGuidedActionClicked(GuidedAction action) { + if (action == null) { + return; + } + switch ((int) action.getId()) { + case (int) ACTION_ID_STATUS: + finishGuidedStepFragments(); + break; + case (int) ACTION_ID_LINEUPS: + FragmentManager fm = getFragmentManager(); + super.onGuidedActionClicked(action); + onLineupSelected(mActionLineupMap.get(action)); + ResultFragment fragment = new ResultFragment(); + Bundle args = new Bundle(); + args.putString(KEY_TITLE, action.getTitle().toString()); + args.putString(KEY_DESCRIPTION, action.getDescription().toString()); + fragment.setArguments(args); + GuidedStepFragment.add(fm, fragment); + break; + default: + } + } + + private List<Pair<Lineup, Integer>> lineupChannelMatchCount( + List<Lineup> lineups, List<String> localChannels) { + List<Pair<Lineup, Integer>> result = new ArrayList<>(); + for (Lineup lineup : lineups) { + result.add(new Pair<>(lineup, getMatchCount(lineup.getChannels(), localChannels))); + } + // sort in decreasing order + Collections.sort( + result, + new Comparator<Pair<Lineup, Integer>>() { + @Override + public int compare(Pair<Lineup, Integer> pair, Pair<Lineup, Integer> other) { + return Integer.compare(other.second, pair.second); + } + }); + return result; + } + + private static int getMatchCount(List<String> lineupChannels, List<String> localChannels) { + int count = 0; + for (String lineupChannel : lineupChannels) { + if (TextUtils.isEmpty(lineupChannel)) { + continue; + } + List<String> parsedNumbers = parseChannelNumber(lineupChannel); + for (String channel : localChannels) { + if (TextUtils.isEmpty(channel)) { + continue; + } + if (matchChannelNumber(parsedNumbers, parseChannelNumber(channel))) { + if (DEBUG) { + Log.d(TAG, lineupChannel + " matches " + channel); + } + count++; + break; + } + } + } + return count; + } + + private List<Lineup> getLineups() { + return new ArrayList<>(Lineups.query(mActivity.getContentResolver(), mPostcode)); + } + + private List<String> getChannelNumbers() { + return mChannelNumbers; + } + + private void onLineupSelected(@Nullable Lineup lineup) { + if (lineup == null) { + return; + } + ContentValues values = new ContentValues(); + values.put(EpgContract.EpgInputs.COLUMN_INPUT_ID, SampleTvInputService.INPUT_ID); + values.put(EpgContract.EpgInputs.COLUMN_LINEUP_ID, lineup.getId()); + + ContentResolver contentResolver = getActivity().getContentResolver(); + EpgInput epgInput = EpgInputs.queryEpgInput(contentResolver, SampleTvInputService.INPUT_ID); + if (epgInput == null) { + contentResolver.insert(EpgContract.EpgInputs.CONTENT_URI, values); + } else { + values.put(EpgContract.EpgInputs.COLUMN_ID, epgInput.getId()); + EpgInputs.update(contentResolver, EpgInput.createEpgChannel(values)); + } + } + + /** + * Parses the channel number string to a list of numbers (major number, minor number, etc.). + * + * @param channelNumber the display number of the channel + * @return a list of numbers + */ + private static List<String> parseChannelNumber(String channelNumber) { + // TODO: add tests for this method + List<String> numbers = + new ArrayList<>( + Arrays.asList(TextUtils.split(channelNumber, CHANNEL_NUMBER_DELIMITER))); + numbers.removeAll(Collections.singleton("")); + if (numbers.size() < 1 || numbers.size() > 2) { + Log.w(TAG, "unsupported channel number format: " + channelNumber); + return new ArrayList<>(); + } + return numbers; + } + + /** + * Checks whether two lists of channel numbers match or not. If the sizes are different, + * additional elements are ignore. + */ + private static boolean matchChannelNumber(List<String> numbers, List<String> other) { + if (numbers.isEmpty() || other.isEmpty()) { + return false; + } + int i = 0; + int j = 0; + while (i < numbers.size() && j < other.size()) { + if (!numbers.get(i++).equals(other.get(j++))) { + return false; + } + } + return true; + } +} |