tasks/app/src/main/java/com/todoroo/astrid/ui/NNumberPickerDialog.java

/**
 * Copyright (c) 2012 Todoroo Inc
 *
 * <p>See the file "LICENSE" for the full license governing this code.
 */
package com.todoroo.astrid.ui;

import android.content.Context;
import android.content.DialogInterface;
import android.content.DialogInterface.OnClickListener;
import androidx.appcompat.app.AlertDialog;
import android.view.Gravity;
import android.view.LayoutInflater;
import android.view.View;
import android.widget.FrameLayout.LayoutParams;
import android.widget.LinearLayout;
import android.widget.TextView;
import java.util.LinkedList;
import java.util.List;
import org.tasks.R;

/** Dialog box with an arbitrary number of number pickers */
class NNumberPickerDialog extends AlertDialog implements OnClickListener {

  private final List<NumberPicker> pickers = new LinkedList<>();
  private final OnNNumberPickedListener mCallback;

  /**
   * Instantiate the dialog box.
   *
   * @param callBack callback function to get the numbers you requested
   * @param title title of the dialog box
   * @param initialValue initial picker values array
   * @param incrementBy picker increment by array
   * @param start picker range start array
   * @param end picker range end array
   * @param separators text separating the spinners. whole array, or individual elements can be null
   */
  public NNumberPickerDialog(
      Context context,
      OnNNumberPickedListener callBack,
      String title,
      int[] initialValue,
      int[] incrementBy,
      int[] start,
      int[] end,
      String[] separators) {
    super(context);
    mCallback = callBack;

    setButton(DialogInterface.BUTTON_POSITIVE, context.getText(android.R.string.ok), this);
    setButton(
        DialogInterface.BUTTON_NEGATIVE,
        context.getText(android.R.string.cancel),
        (OnClickListener) null);

    LayoutInflater inflater =
        (LayoutInflater) context.getSystemService(Context.LAYOUT_INFLATER_SERVICE);
    View view = inflater.inflate(R.layout.n_number_picker_dialog, null);
    setView(view);
    LinearLayout container = (LinearLayout) view;

    setTitle(title);
    LayoutParams npLayout = new LayoutParams(LayoutParams.WRAP_CONTENT, LayoutParams.MATCH_PARENT);
    npLayout.gravity = 1;
    LayoutParams sepLayout = new LayoutParams(LayoutParams.WRAP_CONTENT, LayoutParams.MATCH_PARENT);
    for (int i = 0; i < incrementBy.length; i++) {
      NumberPicker np = new NumberPicker(context, null);
      np.setIncrementBy(incrementBy[i]);
      np.setLayoutParams(npLayout);
      np.setRange(start[i], end[i]);
      np.setCurrent(initialValue[i]);

      container.addView(np);
      pickers.add(np);

      if (separators != null && separators[i] != null) {
        TextView text = new TextView(context);
        text.setText(separators[i]);
        if (separators[i].length() < 3) {
          text.setTextSize(48);
        } else {
          text.setTextSize(20);
        }
        text.setGravity(Gravity.CENTER);
        text.setLayoutParams(sepLayout);
        container.addView(text);
      }
    }
  }

  public NumberPicker getPicker(int index) {
    return pickers.get(index);
  }

  public void setInitialValues(int[] values) {
    for (int i = 0; i < pickers.size(); i++) {
      pickers.get(i).setCurrent(values[i]);
    }
  }

  @Override
  public void onClick(DialogInterface dialog, int which) {
    if (mCallback != null) {
      int[] values = new int[pickers.size()];
      for (int i = 0; i < pickers.size(); i++) {
        pickers.get(i).clearFocus();
        values[i] = pickers.get(i).getCurrent();
      }
      mCallback.onNumbersPicked(values);
    }
  }

  public interface OnNNumberPickedListener {

    void onNumbersPicked(int[] number);
  }
}