Update data types when using local recoding

arx-deidentifier · Oct 8, 2015 · 0573c69 · 0573c69
1 parent 5197114
commit 0573c69
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 2 deletions.
diff --git a/data/example2.deid b/data/example2.deid
diff --git a/src/main/org/deidentifier/arx/ARXResult.java b/src/main/org/deidentifier/arx/ARXResult.java
@@ -584,6 +584,9 @@ public int optimize(DataHandle handle, double gsFactor, ARXListener listener) th
                 }
             }
 
+            // Update data types
+            output.updateDataTypes(result.optimum.getGeneralization());
+
             // Mark as optimized
             if (optimized != 0) {
                 output.setOptimized(true);

diff --git a/src/main/org/deidentifier/arx/DataHandleOutput.java b/src/main/org/deidentifier/arx/DataHandleOutput.java
@@ -443,15 +443,15 @@ protected int[][] getInputBuffer() {
         checkRegistry();
         return registry.getInputHandle().getInputBuffer();
     }
-        
+
     /**
      * Returns the output buffer
      * @return
      */
     protected Data getOutputBufferGeneralized() {
         return outputGeneralized;
     }
-
+        
     /**
      * Returns the output buffer
      * @return
@@ -621,4 +621,22 @@ protected void internalSwap(final int row1, final int row2) {
     protected void setOptimized(boolean optimized) {
         this.optimized = true;
     }
+
+    /**
+     * Used to update data types after local recoding
+     * @param transformation
+     */
+    protected void updateDataTypes(int[] transformation) {
+
+        for (int i = 0; i < dataTypes.length; i++) {
+            DataType<?>[] type = dataTypes[i];
+            if (type != null) {
+                for (int j = 0; j < type.length; j++) {
+                    if ((i == AttributeTypeInternal.QUASI_IDENTIFYING_GENERALIZED && transformation[j] > 0)) {
+                        dataTypes[i][j] = DataType.STRING;
+                    }
+                }
+            }
+        }
+    }
 }