Added possibility of any DataCell's types using for columns of the output table

Author: rssdev10

RubyScript/rb/knime.rb

@@ -30,20 +30,23 @@ def add_cell(cell)
       [:double, DoubleCell],
       [:stringCell, StringCell] # workaround for RowKey class
     ].each do |name, cls|
-      Object.send(:define_method, name) do |val| 
+      Object.send(:define_method, name) do |val|
         add_cell cls.new(val)
-        self 
+        self
       end
-    end    
+    end
 
     # generate an appropriate methods for any types annotated in the output model
-    $outColumnTypes.each do |name|      
-      cls = const_get name
-      Object.send(:define_method, name) do |val|
-        add_cell cls.new(val)
+    # name should be a fully qualified Java class name!
+    $outColumnTypes.each do |name|
+      cls = Java::JavaClass.for_name( name )
+      import cls.to_s
+      rb_cls = cls.ruby_class
+      Object.send(:define_method, cls.simple_name) do |val|
+        add_cell rb_cls.new(val)
         self
       end
-    end    
+    end
   end
 
   class Cells
@@ -117,7 +120,7 @@ def [](idx)
 end
 
 # Extended knime class
-class BufferedDataTable
+class Java::OrgKnimeCoreNode::BufferedDataTable
   # Add Ruby specific methods
   def length
     getRowCount

RubyScript/src/org/knime/ext/jruby/RubyScriptNodeDialog.java

@@ -116,9 +116,11 @@ public void actionPerformed(final ActionEvent e) {
 
         TableColumn typeColumn = table.getColumnModel().getColumn(1);
         JComboBox<String> typeSelector = new JComboBox<String>();
-        typeSelector.addItem("StringCell");
-        typeSelector.addItem("IntegerCell");
-        typeSelector.addItem("DoubleCell");
+        typeSelector.addItem("String");
+        typeSelector.addItem("Integer");
+        typeSelector.addItem("Double");
+        typeSelector.setEditable(true);
+
         typeColumn.setCellEditor(new DefaultCellEditor(typeSelector));
 
         // construct the panel for script loading/authoring

RubyScript/src/org/knime/ext/jruby/RubyScriptNodeModel.java

@@ -301,25 +301,35 @@ protected final DataTableSpec[] configure(final DataTableSpec[] inSpecs)
             DataType type = StringCell.TYPE;
             String columnType = columnTypes[i];
 
+            // convert short classes names
             if ("String".equals(columnType)) {
-                type = StringCell.TYPE;
+                columnType = StringCell.class.getName();
             } else if ("Integer".equals(columnType)) {
-                type = IntCell.TYPE;
+                columnType = IntCell.class.getName();
             } else if ("Double".equals(columnType)) {
-                type = DoubleCell.TYPE;
-            } else {
-                try {
-                    Class<DataCell> cls = (Class<DataCell>) Class.forName(columnType);
-                    if (cls != null)
-                        type = DataType.getType(cls);
-                    else
-                        columnType = "StringCell";
-
-                } catch (ClassNotFoundException e) {
-                    // e.printStackTrace();
+                columnType = DoubleCell.class.getName();
+            }
+
+            try {
+                Class<DataCell> cls = (Class<DataCell>) Class
+                        .forName(columnType);
+                if (cls != null)
+                    type = DataType.getType(cls);
+                else
                     columnType = "StringCell";
-                }
+
+            } catch (ClassNotFoundException e) {
+                // e.printStackTrace();
+                throw new InvalidSettingsException (columnType
+                        + " is an incorrect Java class name. "
+                        + "Please check it and specify a fully qualified class name.");
+
+                //columnType = "StringCell";
             }
+
+            if (!columnTypes[i].equals(columnType))
+                columnTypes[i] = columnType;
+
             DataColumnSpec newColumn = new DataColumnSpecCreator(
                     columnNames[i], type).createSpec();