Merge remote-tracking branch 'origin/master' into numpy-array-view
authorAlexander Marchuk <Alexander.Marchuk@jetbrains.com>
Thu, 4 Sep 2014 15:35:25 +0000 (19:35 +0400)
committerAlexander Marchuk <Alexander.Marchuk@jetbrains.com>
Thu, 4 Sep 2014 15:35:25 +0000 (19:35 +0400)
14 files changed:
platform/platform-resources-en/src/messages/ActionsBundle.properties
python/helpers/pydev/pydevd_resolver.py
python/helpers/pydev/pydevd_vars.py
python/pydevSrc/com/jetbrains/python/debugger/PyDebugValue.java
python/pydevSrc/com/jetbrains/python/debugger/PyTypeHandler.java
python/src/META-INF/python-core.xml
python/src/com/jetbrains/python/actions/view/array/ArrayTableCellEditor.java [new file with mode: 0644]
python/src/com/jetbrains/python/actions/view/array/ArrayTableCellRenderer.java [new file with mode: 0644]
python/src/com/jetbrains/python/actions/view/array/ArrayTableComponent.java [new file with mode: 0644]
python/src/com/jetbrains/python/actions/view/array/ArrayValueProvider.java [new file with mode: 0644]
python/src/com/jetbrains/python/actions/view/array/NumpyArrayValueProvider.java [new file with mode: 0644]
python/src/com/jetbrains/python/actions/view/array/PyViewArrayAction.java [new file with mode: 0644]
python/src/com/jetbrains/python/actions/view/array/RowNumberTable.java [new file with mode: 0644]
python/src/com/jetbrains/python/actions/view/array/XDebuggerTreeTableListener.java [new file with mode: 0644]

index 05f2cd2f8f700fc5da4e41a99ff1e5bf9c48a52b..43488acbeae8da2a387f7c1706157334ec5036a9 100644 (file)
@@ -964,6 +964,7 @@ action.Debugger.AddToWatch.text=Add to Watches
 action.Debugger.EvaluateInConsole.text=Evaluate in Console
 action.Debugger.UnmuteOnStop.text=Unmute breakpoints on session finish
 action.Debugger.AutoRenderer.text=Auto
+action.PyDebugger.ViewArray.text = View as array
 group.EditorPopupMenu.text=Editor Popup Menu
 group.EditorPopupMenu.description=Editor Popup Menu
 action.Add\ to\ migration\ roots.text=_Add to migration roots
index ad49bd881ba074a96ffda3f162cbf3c8b8726ad7..acb611dc4528e8188e06b363426b0d5251eb9f31 100644 (file)
@@ -409,6 +409,18 @@ class NdArrayResolver:
             return obj.dtype
         if attribute == 'size':
             return obj.size
+        if attribute.startswith('['):
+            l = len(obj)
+            container = NdArrayItemsContainer()
+            if l > MAX_ITEMS_TO_HANDLE:
+                setattr(container, TOO_LARGE_ATTR, TOO_LARGE_MSG)
+            else:
+                i = 0
+                format_str = '%0' + str(int(len(str(l)))) + 'd'
+                for item in obj:
+                    setattr(container, format_str % i, item)
+                    i += 1
+            return container
         return None
 
     def getDictionary(self, obj):
@@ -427,9 +439,10 @@ class NdArrayResolver:
         ret['shape'] = obj.shape
         ret['dtype'] = obj.dtype
         ret['size'] = obj.size
+        ret['[0:%s]' % (len(obj))] = list(obj)
         return ret
 
-
+class NdArrayItemsContainer: pass
 #=======================================================================================================================
 # FrameResolver
 #=======================================================================================================================
index 3baea5b61b99f97f6a1b45889fae788d624458c5..2922dc2fe33fa4181e0d079936362167f2b913c1 100644 (file)
@@ -363,7 +363,7 @@ def changeAttrExpression(thread_id, frame_id, attr, expression):
         if isinstance(frame, DjangoTemplateFrame):
             result = eval(expression, frame.f_globals, frame.f_locals)
             frame.changeVariable(attr, result)
-            return
+            return result
 
         if attr[:7] == "Globals":
             attr = attr[8:]
@@ -374,7 +374,7 @@ def changeAttrExpression(thread_id, frame_id, attr, expression):
             if pydevd_save_locals.is_save_locals_available():
                 frame.f_locals[attr] = eval(expression, frame.f_globals, frame.f_locals)
                 pydevd_save_locals.save_locals(frame)
-                return
+                return frame.f_locals[attr]
 
             #default way (only works for changing it in the topmost frame)
             result = eval(expression, frame.f_globals, frame.f_locals)
index bdcf5e41a4cfa49472d2aa34e2003c120db2a95e..ddec6797ea8dcc6019b066473815ac8fa523efea 100644 (file)
@@ -97,7 +97,7 @@ public class PyDebugValue extends XNamedValue {
       myParent.buildExpression(result);
       if (("dict".equals(myParent.getType()) || "list".equals(myParent.getType()) || "tuple".equals(myParent.getType())) &&
           !isLen(myName)) {
-        result.append('[').append(removeId(myName)).append(']');
+        result.append('[').append(removeLeadingZeros(removeId(myName))).append(']');
       }
       else if (("set".equals(myParent.getType())) && !isLen(myName)) {
         //set doesn't support indexing
@@ -105,6 +105,9 @@ public class PyDebugValue extends XNamedValue {
       else if (isLen(myName)) {
         result.append('.').append(myName).append("()");
       }
+      else if (("ndarray".equals(myParent.getType()) || "matrix".equals(myParent.getType())) && myName.startsWith("[")) {
+        result.append(removeLeadingZeros(myName));
+      }
       else {
         result.append('.').append(myName);
       }
@@ -119,6 +122,11 @@ public class PyDebugValue extends XNamedValue {
     return name;
   }
 
+  private static String removeLeadingZeros(@NotNull String name) {
+    //bugs.python.org/issue15254: "0" prefix for octal
+    return name.replaceFirst("^0+(?!$)", "");
+  }
+
   private static boolean isLen(String name) {
     return "__len__".equals(name);
   }
index 6db612488d4b8647d41e790c10d090d74cb24180..f882aae4d0e751e47316d76d48602013ce230498 100644 (file)
@@ -37,6 +37,9 @@ public class PyTypeHandler {
     FORMATTERS = new HashMap<String, Formatter>();
     FORMATTERS.put("str", STR_FORMATTER);
     FORMATTERS.put("unicode", UNI_FORMATTER);
+    //numpy types
+    FORMATTERS.put("string_", STR_FORMATTER);
+    FORMATTERS.put("unicode_", UNI_FORMATTER);
   }
 
   private PyTypeHandler() { }
index 264b29f002500eb332f0d7443106ee43f187e081..1d10206886a90767497de820d0a06a9d46bf61d4 100644 (file)
     <action id="PyInvertBooleanAction" class="com.jetbrains.python.refactoring.invertBoolean.PyInvertBooleanAction" text="Invert Boolean">
       <add-to-group group-id="RefactoringMenu" anchor="last" />
     </action>
+
+    <action id="PyDebugger.ViewArray" class="com.jetbrains.python.actions.view.array.PyViewArrayAction">
+      <add-to-group group-id="XDebugger.ValueGroup" anchor="after" relative-to-action="Debugger.Tree.AddToWatches"/>
+    </action>
+
   </actions>
 
   <extensions defaultExtensionNs="com.intellij.spellchecker">
diff --git a/python/src/com/jetbrains/python/actions/view/array/ArrayTableCellEditor.java b/python/src/com/jetbrains/python/actions/view/array/ArrayTableCellEditor.java
new file mode 100644 (file)
index 0000000..9ac0c07
--- /dev/null
@@ -0,0 +1,90 @@
+/*
+ * Copyright 2000-2014 JetBrains s.r.o.
+ *
+ * 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.jetbrains.python.actions.view.array;
+
+import com.intellij.openapi.editor.Editor;
+import com.intellij.openapi.editor.impl.DocumentImpl;
+import com.intellij.openapi.editor.impl.EditorFactoryImpl;
+import com.intellij.openapi.project.Project;
+import com.jetbrains.python.PythonFileType;
+
+import javax.swing.*;
+import javax.swing.table.TableCellEditor;
+import java.awt.*;
+import java.awt.event.ActionEvent;
+import java.awt.event.KeyEvent;
+
+/**
+* @author amarch
+*/
+class ArrayTableCellEditor extends AbstractCellEditor implements TableCellEditor {
+  Editor myEditor;
+  Project myProject;
+
+  public ArrayTableCellEditor(Project project) {
+    super();
+    myProject = project;
+  }
+
+  public Component getTableCellEditorComponent(JTable table, Object value, boolean isSelected,
+                                               int rowIndex, int vColIndex) {
+
+
+    //PyExpressionCodeFragmentImpl fragment = new PyExpressionCodeFragmentImpl(myProject, "array_view.py", value.toString(), true);
+    //
+    //myEditor = EditorFactoryImpl.getInstance().
+    //  createEditor(PsiDocumentManager.getInstance(myProject).getDocument(fragment), myProject);
+
+
+    myEditor =
+      EditorFactoryImpl.getInstance().createEditor(new DocumentImpl(value.toString()), myProject, PythonFileType.INSTANCE, false);
+
+
+    JComponent editorComponent = myEditor.getContentComponent();
+
+    editorComponent.getInputMap(JComponent.WHEN_ANCESTOR_OF_FOCUSED_COMPONENT)
+      .put(KeyStroke.getKeyStroke(KeyEvent.VK_ENTER, 0), "enterStroke");
+    editorComponent.getActionMap().put("enterStroke", new AbstractAction() {
+      @Override
+      public void actionPerformed(ActionEvent e) {
+        doOKAction();
+      }
+    });
+    editorComponent.getInputMap(JComponent.WHEN_ANCESTOR_OF_FOCUSED_COMPONENT)
+      .put(KeyStroke.getKeyStroke(KeyEvent.VK_ESCAPE, 0), "escapeStroke");
+    editorComponent.getActionMap().put("escapeStroke", new AbstractAction() {
+      @Override
+      public void actionPerformed(ActionEvent e) {
+        cancelEditing();
+      }
+    });
+
+    return editorComponent;
+  }
+
+  public Object getCellEditorValue() {
+    return myEditor.getDocument().getText();
+  }
+
+  public void doOKAction() {
+    //todo: not performed
+    System.out.println("ok");
+  }
+
+  public void cancelEditing() {
+    System.out.println("esc");
+  }
+}
diff --git a/python/src/com/jetbrains/python/actions/view/array/ArrayTableCellRenderer.java b/python/src/com/jetbrains/python/actions/view/array/ArrayTableCellRenderer.java
new file mode 100644 (file)
index 0000000..0a9ae77
--- /dev/null
@@ -0,0 +1,77 @@
+/*
+ * Copyright 2000-2014 JetBrains s.r.o.
+ *
+ * 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.jetbrains.python.actions.view.array;
+
+import javax.swing.*;
+import javax.swing.table.DefaultTableCellRenderer;
+import java.awt.*;
+
+/**
+* @author amarch
+*/
+class ArrayTableCellRenderer extends DefaultTableCellRenderer {
+
+  double min;
+  double max;
+  Color minColor;
+  Color maxColor;
+  boolean colored = true;
+
+  public ArrayTableCellRenderer(double min, double max) {
+    this.min = min;
+    this.max = max;
+    minColor = new Color(100, 0, 0, 200);
+    maxColor = new Color(254, 0, 0, 200);
+  }
+
+  public void setColored(boolean colored) {
+    this.colored = colored;
+  }
+
+  public Component getTableCellRendererComponent(JTable table, Object value,
+                                                 boolean isSelected, boolean hasFocus, int rowIndex, int vColIndex) {
+    if (isSelected) {
+      // cell (and perhaps other cells) are selected
+    }
+
+    if (hasFocus) {
+      // this cell is the anchor and the table has the focus
+    }
+
+    if (value != null) {
+      setText(value.toString());
+    }
+
+
+    if (max != min) {
+      if (colored) {
+        try {
+          double med = Double.parseDouble(value.toString());
+          int r = (int)(minColor.getRed() + Math.round((maxColor.getRed() - minColor.getRed()) / (max - min) * (med - min)));
+          this.setBackground(new Color(r % 256, 0, 0, 200));
+        }
+        catch (NumberFormatException e) {
+        }
+      }
+      else {
+        this.setBackground(new Color(255, 255, 255));
+      }
+    }
+
+
+    return this;
+  }
+}
diff --git a/python/src/com/jetbrains/python/actions/view/array/ArrayTableComponent.java b/python/src/com/jetbrains/python/actions/view/array/ArrayTableComponent.java
new file mode 100644 (file)
index 0000000..b8d739f
--- /dev/null
@@ -0,0 +1,106 @@
+/*
+ * Copyright 2000-2014 JetBrains s.r.o.
+ *
+ * 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.jetbrains.python.actions.view.array;
+
+import com.intellij.ui.components.JBScrollPane;
+import com.intellij.ui.table.JBTable;
+
+import javax.swing.*;
+import javax.swing.table.DefaultTableModel;
+import java.awt.*;
+import java.awt.event.ItemEvent;
+import java.awt.event.ItemListener;
+
+/**
+* @author amarch
+*/
+class ArrayTableComponent extends JPanel {
+  private JScrollPane myScrollPane;
+  private JTextField myTextField;
+  private JBTable myTable;
+  private JCheckBox myCheckBox;
+
+  private static final String DATA_LOADING_IN_PROCESS = "Please wait, load array data.";
+
+  public ArrayTableComponent() {
+    super(new GridBagLayout());
+
+    myTextField = new JTextField();
+    myTextField.setToolTipText("Format");
+
+    myTable = new JBTable() {
+      public boolean getScrollableTracksViewportWidth() {
+        return getPreferredSize().width < getParent().getWidth();
+      }
+    };
+    myTable.setAutoResizeMode(JTable.AUTO_RESIZE_SUBSEQUENT_COLUMNS);
+
+    myCheckBox = new JCheckBox();
+    myCheckBox.setText("Colored");
+    myCheckBox.setSelected(true);
+    myCheckBox.addItemListener(new ItemListener() {
+      @Override
+      public void itemStateChanged(ItemEvent e) {
+        if (e.getSource() == myCheckBox) {
+          if (myTable.getColumnCount() > 0 && myTable.getCellRenderer(0, 0) instanceof ArrayTableCellRenderer) {
+            ArrayTableCellRenderer renderer = (ArrayTableCellRenderer)myTable.getCellRenderer(0, 0);
+            if (myCheckBox.isSelected()) {
+              renderer.setColored(true);
+            }
+            else {
+              renderer.setColored(false);
+            }
+          }
+          myScrollPane.repaint();
+        }
+      }
+    });
+
+    myScrollPane = new JBScrollPane(myTable);
+    myScrollPane.setHorizontalScrollBarPolicy(JBScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED);
+    myScrollPane.setVerticalScrollBarPolicy(JBScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED);
+
+    JTable rowTable = new RowNumberTable(myTable);
+    myScrollPane.setRowHeaderView(rowTable);
+    myScrollPane.setCorner(JScrollPane.UPPER_LEFT_CORNER,
+                           rowTable.getTableHeader());
+
+    add(myScrollPane,
+        new GridBagConstraints(0, 0, 2, 1, 1, 0, GridBagConstraints.WEST, GridBagConstraints.HORIZONTAL, new Insets(0, 0, 0, 0), 0, 0));
+    add(myTextField,
+        new GridBagConstraints(0, 1, 1, 1, 1, 0, GridBagConstraints.SOUTH, GridBagConstraints.HORIZONTAL, new Insets(0, 0, 0, 0), 0, 0));
+    add(myCheckBox,
+        new GridBagConstraints(1, 1, 1, 1, 1, 0, GridBagConstraints.SOUTH, GridBagConstraints.HORIZONTAL, new Insets(0, 0, 0, 0), 0, 0));
+  }
+
+  public JTextField getTextField() {
+    return myTextField;
+  }
+
+  public JBTable getTable() {
+    return myTable;
+  }
+
+  public JCheckBox getColored() {
+    return myCheckBox;
+  }
+
+  public void setDefaultSpinnerText() {
+    DefaultTableModel model = new DefaultTableModel(1, 1);
+    myTable.setModel(model);
+    myTable.setValueAt(DATA_LOADING_IN_PROCESS, 0, 0);
+  }
+}
diff --git a/python/src/com/jetbrains/python/actions/view/array/ArrayValueProvider.java b/python/src/com/jetbrains/python/actions/view/array/ArrayValueProvider.java
new file mode 100644 (file)
index 0000000..895c6e4
--- /dev/null
@@ -0,0 +1,24 @@
+/*
+ * Copyright 2000-2014 JetBrains s.r.o.
+ *
+ * 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.jetbrains.python.actions.view.array;
+
+/**
+* @author amarch
+*/
+abstract class ArrayValueProvider {
+
+  public abstract Object[][] parseValues(String rawValue);
+}
diff --git a/python/src/com/jetbrains/python/actions/view/array/NumpyArrayValueProvider.java b/python/src/com/jetbrains/python/actions/view/array/NumpyArrayValueProvider.java
new file mode 100644 (file)
index 0000000..aa21e11
--- /dev/null
@@ -0,0 +1,31 @@
+/*
+ * Copyright 2000-2014 JetBrains s.r.o.
+ *
+ * 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.jetbrains.python.actions.view.array;
+
+/**
+* @author amarch
+*/
+class NumpyArrayValueProvider extends ArrayValueProvider {
+
+  @Override
+  public Object[][] parseValues(String rawValues) {
+    return null;
+  }
+
+  private boolean isNumeric(String value) {
+    return true;
+  }
+}
diff --git a/python/src/com/jetbrains/python/actions/view/array/PyViewArrayAction.java b/python/src/com/jetbrains/python/actions/view/array/PyViewArrayAction.java
new file mode 100644 (file)
index 0000000..247f508
--- /dev/null
@@ -0,0 +1,138 @@
+/*
+ * Copyright 2000-2014 JetBrains s.r.o.
+ *
+ * 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.jetbrains.python.actions.view.array;
+
+import com.intellij.openapi.actionSystem.AnActionEvent;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.ui.DialogWrapper;
+import com.intellij.xdebugger.frame.XFullValueEvaluator;
+import com.intellij.xdebugger.impl.ui.tree.actions.XDebuggerTreeActionBase;
+import com.intellij.xdebugger.impl.ui.tree.nodes.XValueNodeImpl;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import javax.swing.*;
+import java.awt.*;
+
+/**
+ * @author amarch
+ */
+
+public class PyViewArrayAction extends XDebuggerTreeActionBase {
+
+  @Override
+  protected void perform(XValueNodeImpl node, @NotNull String nodeName, AnActionEvent e) {
+    final MyDialog dialog = new MyDialog(e.getProject(), node, nodeName);
+    dialog.setTitle("View Array");
+    dialog.setValue(node);
+    dialog.show();
+  }
+
+
+  private class MyDialog extends DialogWrapper {
+    public JTable myTable;
+    private XValueNodeImpl myNode;
+    private String myNodeName;
+    private Project myProject;
+    private ArrayTableComponent myComponent;
+
+    private MyDialog(Project project, XValueNodeImpl node, @NotNull String nodeName) {
+      super(project, false);
+      setModal(false);
+      setCancelButtonText("Close");
+      setCrossClosesWindow(true);
+
+      myNode = node;
+      myNodeName = nodeName;
+      myProject = project;
+
+      myComponent = new ArrayTableComponent();
+      myTable = myComponent.getTable();
+
+      init();
+    }
+
+
+    public void setValue(XValueNodeImpl node) {
+      final ArrayValueProvider valueProvider;
+
+      if (node.getValuePresentation() != null &&
+          node.getValuePresentation().getType() != null &&
+          node.getValuePresentation().getType().equals("ndarray")) {
+        valueProvider = new NumpyArrayValueProvider();
+
+        //myComponent.setDefaultSpinnerText();
+
+        XDebuggerTreeTableListener tableUpdater = new XDebuggerTreeTableListener(node, myTable, myComponent, myProject);
+
+        node.getTree().addTreeListener(tableUpdater);
+
+        node.startComputingChildren();
+      }
+    }
+
+    private String evaluateFullValue(XValueNodeImpl node) {
+      final String[] result = new String[1];
+
+      XFullValueEvaluator.XFullValueEvaluationCallback valueEvaluationCallback = new XFullValueEvaluator.XFullValueEvaluationCallback() {
+        @Override
+        public void evaluated(@NotNull String fullValue) {
+          result[0] = fullValue;
+        }
+
+        @Override
+        public void evaluated(@NotNull String fullValue, @Nullable Font font) {
+          result[0] = fullValue;
+        }
+
+        @Override
+        public void errorOccurred(@NotNull String errorMessage) {
+          result[0] = errorMessage;
+        }
+
+        @Override
+        public boolean isObsolete() {
+          return false;
+        }
+      };
+
+      if (node.getFullValueEvaluator() != null) {
+        node.getFullValueEvaluator().startEvaluation(valueEvaluationCallback);
+      }
+      else {
+        return node.getRawValue();
+      }
+
+      return result[0];
+    }
+
+    @Override
+    @NotNull
+    protected Action[] createActions() {
+      return new Action[]{getCancelAction()};
+    }
+
+    @Override
+    protected String getDimensionServiceKey() {
+      return "#com.jetbrains.python.actions.view.array.PyViewArrayAction";
+    }
+
+    @Override
+    protected JComponent createCenterPanel() {
+      return myComponent;
+    }
+  }
+}
diff --git a/python/src/com/jetbrains/python/actions/view/array/RowNumberTable.java b/python/src/com/jetbrains/python/actions/view/array/RowNumberTable.java
new file mode 100644 (file)
index 0000000..50c2e55
--- /dev/null
@@ -0,0 +1,184 @@
+/*
+ * Copyright 2000-2014 JetBrains s.r.o.
+ *
+ * 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.jetbrains.python.actions.view.array;
+
+import javax.swing.*;
+import javax.swing.event.ChangeEvent;
+import javax.swing.event.ChangeListener;
+import javax.swing.event.TableModelEvent;
+import javax.swing.event.TableModelListener;
+import javax.swing.table.DefaultTableCellRenderer;
+import javax.swing.table.JTableHeader;
+import javax.swing.table.TableColumn;
+import java.awt.*;
+import java.beans.PropertyChangeEvent;
+import java.beans.PropertyChangeListener;
+
+/*
+*      Use a JTable as a renderer for row numbers of a given main table.
+*  This table must be added to the row header of the scrollpane that
+*  contains the main table.
+*/
+public class RowNumberTable extends JTable
+  implements ChangeListener, PropertyChangeListener, TableModelListener {
+  private JTable main;
+
+  public RowNumberTable(JTable table) {
+    main = table;
+    main.addPropertyChangeListener(this);
+    main.getModel().addTableModelListener(this);
+
+    setFocusable(false);
+    setAutoCreateColumnsFromModel(false);
+    setSelectionModel(main.getSelectionModel());
+
+
+    TableColumn column = new TableColumn();
+    column.setHeaderValue(" ");
+    addColumn(column);
+    column.setCellRenderer(new RowNumberRenderer());
+
+    getColumnModel().getColumn(0).setPreferredWidth(50);
+    setPreferredScrollableViewportSize(getPreferredSize());
+  }
+
+  @Override
+  public void addNotify() {
+    super.addNotify();
+
+    Component c = getParent();
+
+    //  Keep scrolling of the row table in sync with the main table.
+
+    if (c instanceof JViewport) {
+      JViewport viewport = (JViewport)c;
+      viewport.addChangeListener(this);
+    }
+  }
+
+  /*
+   *  Delegate method to main table
+   */
+  @Override
+  public int getRowCount() {
+    return main.getRowCount();
+  }
+
+  @Override
+  public int getRowHeight(int row) {
+    int rowHeight = main.getRowHeight(row);
+
+    if (rowHeight != super.getRowHeight(row)) {
+      super.setRowHeight(row, rowHeight);
+    }
+
+    return rowHeight;
+  }
+
+  /*
+   *  No model is being used for this table so just use the row number
+   *  as the value of the cell.
+   */
+  @Override
+  public Object getValueAt(int row, int column) {
+    return Integer.toString(row + 1);
+  }
+
+  /*
+   *  Don't edit data in the main TableModel by mistake
+   */
+  @Override
+  public boolean isCellEditable(int row, int column) {
+    return false;
+  }
+
+  /*
+   *  Do nothing since the table ignores the model
+   */
+  @Override
+  public void setValueAt(Object value, int row, int column) {
+  }
+
+  //
+  //  Implement the ChangeListener
+  //
+  public void stateChanged(ChangeEvent e) {
+    //  Keep the scrolling of the row table in sync with main table
+
+    JViewport viewport = (JViewport)e.getSource();
+    JScrollPane scrollPane = (JScrollPane)viewport.getParent();
+    scrollPane.getVerticalScrollBar().setValue(viewport.getViewPosition().y);
+  }
+
+  //
+  //  Implement the PropertyChangeListener
+  //
+  public void propertyChange(PropertyChangeEvent e) {
+    //  Keep the row table in sync with the main table
+
+    if ("selectionModel".equals(e.getPropertyName())) {
+      setSelectionModel(main.getSelectionModel());
+    }
+
+    if ("rowHeight".equals(e.getPropertyName())) {
+      repaint();
+    }
+
+    if ("model".equals(e.getPropertyName())) {
+      main.getModel().addTableModelListener(this);
+      revalidate();
+    }
+  }
+
+  //
+  //  Implement the TableModelListener
+  //
+  @Override
+  public void tableChanged(TableModelEvent e) {
+    revalidate();
+  }
+
+  /*
+   *  Attempt to mimic the table header renderer
+   */
+  private class RowNumberRenderer extends DefaultTableCellRenderer {
+    public RowNumberRenderer() {
+      setHorizontalAlignment(JLabel.CENTER);
+    }
+
+    public Component getTableCellRendererComponent(
+      JTable table, Object value, boolean isSelected, boolean hasFocus, int row, int column) {
+      if (table != null) {
+        JTableHeader header = table.getTableHeader();
+
+        if (header != null) {
+          setForeground(header.getForeground());
+          setBackground(header.getBackground());
+          setFont(header.getFont());
+        }
+      }
+
+      if (isSelected) {
+        setFont(getFont().deriveFont(Font.BOLD));
+      }
+
+      setText((value == null) ? "" : value.toString());
+      setBorder(UIManager.getBorder("TableHeader.cellBorder"));
+
+      return this;
+    }
+  }
+}
diff --git a/python/src/com/jetbrains/python/actions/view/array/XDebuggerTreeTableListener.java b/python/src/com/jetbrains/python/actions/view/array/XDebuggerTreeTableListener.java
new file mode 100644 (file)
index 0000000..d368158
--- /dev/null
@@ -0,0 +1,272 @@
+/*
+ * Copyright 2000-2014 JetBrains s.r.o.
+ *
+ * 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.jetbrains.python.actions.view.array;
+
+import com.intellij.openapi.project.Project;
+import com.intellij.util.containers.HashSet;
+import com.intellij.xdebugger.impl.ui.tree.XDebuggerTreeListener;
+import com.intellij.xdebugger.impl.ui.tree.nodes.RestorableStateNode;
+import com.intellij.xdebugger.impl.ui.tree.nodes.XDebuggerTreeNode;
+import com.intellij.xdebugger.impl.ui.tree.nodes.XValueContainerNode;
+import com.intellij.xdebugger.impl.ui.tree.nodes.XValueNodeImpl;
+import org.jetbrains.annotations.NotNull;
+
+import javax.swing.*;
+import javax.swing.table.DefaultTableModel;
+import java.util.List;
+
+/**
+* @author amarch
+*/
+public class XDebuggerTreeTableListener implements XDebuggerTreeListener {
+
+  XValueNodeImpl baseNode;
+
+  XValueContainerNode innerNdarray;
+
+  XValueContainerNode innerItems;
+
+  ArrayTableComponent myComponent;
+
+  Project myProject;
+
+  JTable myTable;
+
+  int[] shape;
+
+  int depth = 0;
+
+  int loadedRows = 0;
+
+  HashSet<String> unloadedRowNumbers = new HashSet<String>();
+
+  boolean baseChildrenLoaded = false;
+
+  boolean numeric = false;
+
+  boolean dataLoaded = false;
+
+  Object[][] data;
+
+  public XDebuggerTreeTableListener(XValueNodeImpl node, JTable table, ArrayTableComponent component, Project project) {
+    super();
+    baseNode = baseNode;
+    myTable = table;
+    myComponent = component;
+    myProject= project;
+  }
+
+  @Override
+  public void nodeLoaded(@NotNull RestorableStateNode node, String name) {
+    System.out.printf(name + " node loaded\n");
+
+    if (!baseChildrenLoaded &&
+        (shape == null || name.equals("dtype")) &&
+        ((XValueNodeImpl)node.getParent()).getName().equals(baseNode.getName())) {
+      if (name.equals("shape")) {
+        String rawValue = node.getRawValue();
+        String[] shapes = rawValue.substring(1, rawValue.length() - 1).split(",");
+        shape = new int[shapes.length];
+        for (int i = 0; i < shapes.length; i++) {
+          shape[i] = Integer.parseInt(shapes[i].trim());
+        }
+        depth = Math.max(shape.length - 2, 0);
+      }
+
+      if (name.equals("dtype")) {
+        String rawValue = node.getRawValue();
+        if ("biufc".contains(rawValue.substring(0, 1))) {
+          numeric = true;
+        }
+      }
+    }
+  }
+
+  @Override
+  public void childrenLoaded(@NotNull XDebuggerTreeNode node, @NotNull List<XValueContainerNode<?>> children, boolean last) {
+    System.out.printf(children + "children loaded\n");
+
+    if (dataLoaded) {
+      return;
+    }
+
+    //todo: not compute children if they yet computed
+
+    if (!baseChildrenLoaded && node.equals(baseNode)) {
+      baseChildrenLoaded = true;
+      innerNdarray = (XValueContainerNode)node;
+      if (shape != null) {
+        if (shape.length >= 2) {
+          data = new Object[shape[shape.length - 2]][shape[shape.length - 1]];
+        }
+        else {
+          data = new Object[1][shape[0]];
+        }
+      }
+    }
+
+    //go deeper
+    if (depth > 0) {
+      if (innerNdarray != null && innerNdarray.equals(node)) {
+        innerNdarray = null;
+        innerItems = findItems(node);
+        innerItems.startComputingChildren();
+      }
+
+      if (innerItems != null && innerItems.equals(node)) {
+        innerNdarray = (XValueContainerNode)node.getChildAt(1);
+        innerItems = null;
+        innerNdarray.startComputingChildren();
+        depth -= 1;
+      }
+
+      return;
+    }
+
+    //find ndarray slice to display
+    if (depth == 0) {
+      innerItems = findItems(node);
+      innerItems.startComputingChildren();
+      depth -= 1;
+      return;
+    }
+
+    if (depth == -1 && node.equals(innerItems)) {
+      if (shape != null && shape.length == 1) {
+        for (int i = 0; i < node.getChildCount() - 1; i++) {
+          data[0][i] = ((XValueNodeImpl)node.getChildAt(i + 1)).getRawValue();
+        }
+        loadData();
+        loadedRows = 1;
+      }
+      else {
+        for (int i = 0; i < node.getChildCount() - 1; i++) {
+          ((XValueNodeImpl)node.getChildAt(i + 1)).startComputingChildren();
+          unloadedRowNumbers.add(((XValueNodeImpl)node.getChildAt(i + 1)).getName());
+        }
+        depth -= 1;
+      }
+      return;
+    }
+
+
+    if (depth == -2) {
+      String name = ((XValueNodeImpl)node).getName();
+      // ndarrray children not computed yet
+      if (unloadedRowNumbers.contains(name)) {
+        unloadedRowNumbers.remove(name);
+        findItems(node).startComputingChildren();
+        return;
+      }
+
+      if (name.startsWith("[")) {
+        int row = Integer.parseInt((((XValueNodeImpl)node.getParent()).getName()));
+        if (data[row][0] == null) {
+          for (int i = 0; i < node.getChildCount() - 1; i++) {
+            data[row][i] = ((XValueNodeImpl)node.getChildAt(i + 1)).getRawValue();
+          }
+          loadedRows += 1;
+        }
+      }
+    }
+
+    if (loadedRows == shape[shape.length - 2]) {
+      loadData();
+    }
+  }
+
+  XValueContainerNode findItems(@NotNull XDebuggerTreeNode node) {
+    for (int i = 0; i < node.getChildCount(); i++) {
+      if (node.getChildAt(i).toString().startsWith("[")) {
+        return (XValueContainerNode)node.getChildAt(i);
+      }
+    }
+    return null;
+  }
+
+  private String[] range(int min, int max) {
+    String[] array = new String[max - min + 1];
+    for (int i = min; i <= max; i++) {
+      array[i] = Integer.toString(i);
+    }
+    return array;
+  }
+
+  private void loadData() {
+
+    DefaultTableModel model = new DefaultTableModel(data, range(0, data[0].length - 1));
+
+    myTable.setModel(model);
+    myTable.setDefaultEditor(myTable.getColumnClass(0), new ArrayTableCellEditor(myProject));
+
+    if (numeric) {
+      double min = Double.MAX_VALUE;
+      double max = Double.MIN_VALUE;
+      if (data.length > 0) {
+        try {
+          for (int i = 0; i < data.length; i++) {
+            for (int j = 0; j < data[0].length; j++) {
+              double d = Double.parseDouble(data[i][j].toString());
+              min = min > d ? d : min;
+              max = max < d ? d : max;
+            }
+          }
+        }
+        catch (NumberFormatException e) {
+          min = 0;
+          max = 0;
+        }
+      }
+      else {
+        min = 0;
+        max = 0;
+      }
+
+      myTable.setDefaultRenderer(myTable.getColumnClass(0), new ArrayTableCellRenderer(min, max));
+    }
+    else {
+      myComponent.getColored().setSelected(false);
+      myComponent.getColored().setVisible(false);
+    }
+
+    myComponent.getTextField().setText(getDefaultSliceRepresentation());
+    dataLoaded = true;
+    innerItems = null;
+    innerNdarray = null;
+  }
+
+  public String getDefaultSliceRepresentation() {
+    String representation = "";
+
+    if (baseNode != null) {
+      representation += baseNode.getName();
+      if (shape != null && shape.length > 0) {
+        for (int i = 0; i < shape.length - 2; i++) {
+          representation += "[0]";
+        }
+        if (shape.length == 1) {
+          representation += "[0:" + shape[0] + "]";
+        }
+        else {
+          representation += "[0:" + shape[shape.length - 2] + "][0:" + shape[shape.length - 1] + "]";
+        }
+      }
+    }
+
+    return representation;
+  }
+
+}