More reckless refactoring and code style tweaks -- mostly adding braces around condit...
[zxing.git] / core / src / com / google / zxing / qrcode / QRCodeWriter.java
index 1521197..874cc0e 100644 (file)
 package com.google.zxing.qrcode;
 
 import com.google.zxing.BarcodeFormat;
+import com.google.zxing.EncodeHintType;
 import com.google.zxing.Writer;
 import com.google.zxing.WriterException;
 import com.google.zxing.common.ByteMatrix;
-import com.google.zxing.common.ByteArray;
 import com.google.zxing.qrcode.encoder.Encoder;
 import com.google.zxing.qrcode.encoder.QRCode;
+import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
 
 import java.util.Hashtable;
 
@@ -57,17 +58,23 @@ public final class QRCodeWriter implements Writer {
           height);
     }
 
-    // TODO: Check hints for error correction level instead of hardcoding
-    int errorCorrectionLevel = QRCode.EC_LEVEL_L;
+    ErrorCorrectionLevel errorCorrectionLevel = ErrorCorrectionLevel.L;
+    if (hints != null) {
+      ErrorCorrectionLevel requestedECLevel = (ErrorCorrectionLevel) hints.get(EncodeHintType.ERROR_CORRECTION);
+      if (requestedECLevel != null) {
+        errorCorrectionLevel = requestedECLevel;
+      }
+    }
+
     QRCode code = new QRCode();
-    Encoder.Encode(new ByteArray(contents), errorCorrectionLevel, code);
+    Encoder.encode(contents, errorCorrectionLevel, code);
     return renderResult(code, width, height);
   }
 
   // Note that the input matrix uses 0 == white, 1 == black, while the output matrix uses
   // 0 == black, 255 == white (i.e. an 8 bit greyscale bitmap).
-  private ByteMatrix renderResult(QRCode code, final int width, final int height) {
-    ByteMatrix input = code.matrix();
+  private ByteMatrix renderResult(QRCode code, int width, int height) {
+    ByteMatrix input = code.getMatrix();
     int inputWidth = input.width();
     int inputHeight = input.height();
     int qrWidth = inputWidth + (QUIET_ZONE_SIZE * 2);
@@ -96,7 +103,7 @@ public final class QRCodeWriter implements Writer {
     }
 
     // 2. Expand the QR image to the multiple
-    final byte[][] inputArray = input.getArray();
+    byte[][] inputArray = input.getArray();
     for (int y = 0; y < inputHeight; y++) {
       // a. Write the white pixels at the left of each row
       for (int x = 0; x < leftPadding; x++) {