From 18b7ac83c141cceb1451d37ff56d21ed3c275d3c Mon Sep 17 00:00:00 2001
From: serso <se.solovyev@gmail.com>
Date: Sat, 23 Jan 2016 14:20:22 +0100
Subject: [PATCH] Focus on new parameter

---
 .../function/FunctionParamsView.java          | 22 ++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

diff --git a/app/src/main/java/org/solovyev/android/calculator/function/FunctionParamsView.java b/app/src/main/java/org/solovyev/android/calculator/function/FunctionParamsView.java
index d2cf2a29..69d25caf 100644
--- a/app/src/main/java/org/solovyev/android/calculator/function/FunctionParamsView.java
+++ b/app/src/main/java/org/solovyev/android/calculator/function/FunctionParamsView.java
@@ -96,7 +96,9 @@ public class FunctionParamsView extends LinearLayout {
         addButton.setOnClickListener(new View.OnClickListener() {
             @Override
             public void onClick(View v) {
-                addParam(null);
+                final LinearLayout rowView = addParam(null);
+                final EditText paramView = getParamView(rowView);
+                paramView.requestFocus();
             }
         });
         headerView.addView(addButton, makeButtonParams());
@@ -132,11 +134,13 @@ public class FunctionParamsView extends LinearLayout {
         }
     }
 
-    public void addParam(@Nullable String param) {
-        addParam(param, maxRowId++);
+    @NonNull
+    public LinearLayout addParam(@Nullable String param) {
+        return addParam(param, maxRowId++);
     }
 
-    private void addParam(@Nullable String param, final int id) {
+    @NonNull
+    private LinearLayout addParam(@Nullable String param, final int id) {
         final Context context = getContext();
         final LinearLayout rowView = makeRowView(context);
 
@@ -183,6 +187,7 @@ public class FunctionParamsView extends LinearLayout {
 
         // for row is added at 0 position, the consequent rows
         addView(rowView, Math.max(0, getChildCount() - 1), new ViewGroup.LayoutParams(MATCH_PARENT, WRAP_CONTENT));
+        return rowView;
     }
 
     @NonNull
@@ -243,7 +248,14 @@ public class FunctionParamsView extends LinearLayout {
     @Nonnull
     private EditText getParamView(@Nonnull ViewGroup row) {
         final TextInputLayout paramLabel = getParamLabel(row);
-        return (EditText) paramLabel.getChildAt(0);
+        for (int i = 0; i < paramLabel.getChildCount(); i++) {
+            final View child = paramLabel.getChildAt(i);
+            if (child instanceof EditText) {
+                return (EditText) child;
+            }
+        }
+        Check.shouldNotHappen();
+        return null;
     }
 
     @Nonnull