Convert RDD[Vector]/RDD[LabeledPoint] to HomogenTable

mllib-dal/src/main/java/com/intel/oneapi/dal/table/HomogenTable.java

@@ -3,6 +3,11 @@
 public class HomogenTable extends Table {
     HomogenTableImpl impl;
 
+    public HomogenTable(Common.ComputeDevice device){
+        super();
+        impl = new HomogenTableImpl(device);
+    }
+
     public HomogenTable(long cTable){
         super();
         impl = new HomogenTableImpl(cTable);
@@ -138,6 +143,10 @@ public  double[] getDoubleData() {
         return  impl.getDoubleData();
     }
 
+    public void addHomogenTable(long homogenTableAddr ) {
+        impl.addHomogenTable(homogenTableAddr);
+    }
+
     public long getcObejct() {
         return impl.getcObject();
     }

mllib-dal/src/main/java/com/intel/oneapi/dal/table/HomogenTableImpl.java

@@ -16,8 +16,10 @@ public class HomogenTableImpl implements HomogenTableIface {
     private TableMetadata metadata;
     private Common.ComputeDevice device;
 
-    protected HomogenTableImpl() {
+    protected HomogenTableImpl(Common.ComputeDevice computeDevice) {
         super();
+        this.device = computeDevice;
+        this.cObject = this.cEmptyTableInit();
     }
 
     public HomogenTableImpl(long cTable) {
@@ -138,6 +140,9 @@ public long getcObject(){
         return this.cObject;
     }
 
+    public void addHomogenTable(long homogenTableAddr ) {
+        this.cObject = cAddHomogenTable(this.cObject, homogenTableAddr, this.device.ordinal());
+    }
     private native long iInit(long rowCount,
                               long colCount,
                               int[] data,
@@ -173,4 +178,8 @@ private native long lInit(long rowCount,
     private native long[] cGetLongData(long cObject);
     private native float[] cGetFloatData(long cObject);
     private native double[] cGetDoubleData(long cObject);
+    private native long cAddHomogenTable(long cObject,
+                                         long homogenTableAddr,
+                                         int computeDeviceIndex);
+    private native long cEmptyTableInit();
 }

mllib-dal/src/main/java/com/intel/oneapi/dal/table/RowAccessor.java

@@ -3,10 +3,9 @@
 public class RowAccessor {
     private long cObject;
     private Common.ComputeDevice cDevice;
-
     public RowAccessor(long cObject, Common.ComputeDevice device) {
         this.cObject = cObject;
-        this.cDevice =device;
+        this.cDevice = device;
     }
 
     public double[] pullDouble(){
@@ -39,4 +38,5 @@ private native float[] cPullFloat(long cObject, long cRowStartIndex,
                                       long cRowEndIndex, int computeDeviceIndex);
     private native int[] cPullInt(long cObject, long cRowStartIndex,
                                   long cRowEndIndex, int computeDeviceIndex);
+
 }

mllib-dal/src/main/native/OneDAL.cpp

@@ -16,8 +16,10 @@
 
 #include <cstring>
 #include <iostream>
+#include <set>
 
 #include "com_intel_oap_mllib_OneDAL__.h"
+
 #include "service.h"
 
 using namespace daal;
@@ -150,4 +152,4 @@ Java_com_intel_oap_mllib_OneDAL_00024_cNewCSRNumericTableDouble(
     CSRNumericTablePtr *ret = new CSRNumericTablePtr(numericTable);
 
     return (jlong)ret;
-}
+}

mllib-dal/src/main/native/oneapi/dal/ColumnAccessorImpl.cpp

@@ -25,6 +25,7 @@
 #ifdef CPU_GPU_PROFILE
 #include "GPU.h"
 #endif
+
 #ifndef ONEDAL_DATA_PARALLEL
 #define ONEDAL_DATA_PARALLEL
 #endif
@@ -38,6 +39,7 @@ using namespace std;
 using namespace oneapi::dal;
 typedef std::shared_ptr<homogen_table> homogenPtr;
 
+
 /*
  * Class:     com_intel_oneapi_dal_table_ColumnAccessor
  * Method:    cPull
@@ -124,11 +126,11 @@ JNIEXPORT jfloatArray JNICALL Java_com_intel_oneapi_dal_table_ColumnAccessor_cPu
 JNIEXPORT jintArray JNICALL Java_com_intel_oneapi_dal_table_ColumnAccessor_cPullInt
 (JNIEnv *env, jobject, jlong cTableAddr, jlong cColumnIndex, jlong cRowStartIndex,
  jlong cRowEndIndex, jint cComputeDevice) {
-printf("ColumnAccessor PullInt \n");
-homogen_table htable = *((homogen_table *)cTableAddr);
-column_accessor<const int> acc { htable };
-oneapi::dal::array<int> col_values;
-jintArray newIntArray;
+    printf("ColumnAccessor PullInt \n");
+    homogen_table htable = *((homogen_table *)cTableAddr);
+    column_accessor<const int> acc { htable };
+    oneapi::dal::array<int> col_values;
+    jintArray newIntArray;
 switch(getComputeDevice(cComputeDevice)) {
      case compute_device::host:{
             col_values = acc.pull(cColumnIndex, {cRowStartIndex, cRowEndIndex});
@@ -149,7 +151,7 @@ switch(getComputeDevice(cComputeDevice)) {
      default: {
            return newIntArray;
      }
-  }
+}
     newIntArray = env->NewIntArray(col_values.get_count());
     env->SetIntArrayRegion(newIntArray, 0, col_values.get_count(), col_values.get_data());
     return newIntArray;