/*
 * Copyright 2012 ZXing authors
 *
 * 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.
 */

#import "ZXDecoderResult.h"
#import "ZXErrors.h"
#import "ZXMaxiCodeDecodedBitStreamParser.h"

const unichar SHIFTA = 0xFFF0;
const unichar SHIFTB = 0xFFF1;
const unichar SHIFTC = 0xFFF2;
const unichar SHIFTD = 0xFFF3;
const unichar SHIFTE = 0xFFF4;
const unichar TWOSHIFTA = 0xFFF5;
const unichar THREESHIFTA = 0xFFF6;
const unichar LATCHA = 0xFFF7;
const unichar LATCHB = 0xFFF8;
const unichar LOCK = 0xFFF9;
const unichar ECI = 0xFFFA;
const unichar NS = 0xFFFB;
const unichar PAD = 0xFFFC;
const unichar FS = 0x001C;
const unichar GS = 0x001D;
const unichar RS = 0x001E;

const unichar SETS[1][383] = {
  '\n', 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W',
  'X', 'Y', 'Z', ECI, FS, GS, RS, NS, ' ', PAD, '"', '#', '$', '%', '&', '\'', '(', ')', '*', '+', ',', '-', '.', '/', '0',
  '1', '2', '3', '4', '5', '6', '7', '8', '9', ':', SHIFTB, SHIFTC, SHIFTD, SHIFTE, LATCHB,
  '`', 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p' , 'q', 'r', 's', 't', 'u', 'v', 'w',
  'x', 'y', 'z', ECI, FS, GS, RS, NS, '{', PAD, '}', '~', 0x007F, ';', '<', '=', '>', '?', '[', '\\', ']', '^', '_', ' ',
  ',', '.', '/', ':', '@', '!', '|', PAD, TWOSHIFTA, THREESHIFTA, PAD, SHIFTA, SHIFTC, SHIFTD, SHIFTE, LATCHA,
  0x00C0, 0x00C1, 0x00C2, 0x00C3, 0x00C4, 0x00C5, 0x00C6, 0x00C7, 0x00C8, 0x00C9, 0x00CA, 0x00CB, 0x00CC, 0x00CD, 0x00CE,
  0x00CF, 0x00D0, 0x00D1, 0x00D2, 0x00D3, 0x00D4, 0x00D5, 0x00D6, 0x00D7, 0x00D8, 0x00D9, 0x00DA, ECI, FS, GS, RS, 0x00DB,
  0x00DC, 0x00DD, 0x00DE, 0x00DF, 0x00AA, 0x00AC, 0x00B1, 0x00B2, 0x00B3, 0x00B5, 0x00B9, 0x00BA, 0x00BC, 0x00BD, 0x00BE,
  0x0080, 0x0081, 0x0082, 0x0083, 0x0084, 0x0085, 0x0086, 0x0087, 0x0088, 0x0089, LATCHA, ' ', LOCK, SHIFTD, SHIFTE, LATCHB,
  0x00E0, 0x00E1, 0x00E2, 0x00E3, 0x00E4, 0x00E5, 0x00E6, 0x00E7, 0x00E8, 0x00E9, 0x00EA, 0x00EB, 0x00EC, 0x00ED, 0x00EE,
  0x00EF, 0x00F0, 0x00F1, 0x00F2, 0x00F3, 0x00F4, 0x00F5, 0x00F6, 0x00F7, 0x00F8, 0x00F9, 0x00FA, ECI, FS, GS, RS, NS,
  0x00FB, 0x00FC, 0x00FD, 0x00FE, 0x00FF, 0x00A1, 0x00A8, 0x00AB, 0x00AF, 0x00B0, 0x00B4, 0x00B7, 0x00B8, 0x00BB, 0x00BF,
  0x008A, 0x008B, 0x008C, 0x008D, 0x008E, 0x008F, 0x0090, 0x0091, 0x0092, 0x0093, 0x0094, LATCHA, ' ', SHIFTC, LOCK, SHIFTE,
  LATCHB, 0x0000, 0x0001, 0x0002, 0x0003, 0x0004, 0x0005, 0x0006, 0x0007, 0x0008, 0x0009, '\n', 0x000B, 0x000C, '\r',
  0x000E, 0x000F, 0x0010, 0x0011, 0x0012, 0x0013, 0x0014, 0x0015, 0x0016, 0x0017, 0x0018, 0x0019, 0x001A, ECI, PAD, PAD,
  0x001B, NS, FS, GS, RS, 0x001F, 0x009F, 0x00A0, 0x00A2, 0x00A3, 0x00A4, 0x00A5, 0x00A6, 0x00A7, 0x00A9, 0x00AD, 0x00AE,
  0x00B6, 0x0095, 0x0096, 0x0097, 0x0098, 0x0099, 0x009A, 0x009B, 0x009C, 0x009D, 0x009E, LATCHA, ' ', SHIFTC, SHIFTD, LOCK,
  LATCHB, 0x0000, 0x0001, 0x0002, 0x0003, 0x0004, 0x0005, 0x0006, 0x0007, 0x0008, 0x0009, '\n', 0x000B, 0x000C, '\r',
  0x000E, 0x000F, 0x0010, 0x0011, 0x0012, 0x0013, 0x0014, 0x0015, 0x0016, 0x0017, 0x0018, 0x0019, 0x001A, 0x001B, 0x001C,
  0x001D, 0x001E, 0x001F, 0x0020, 0x0021, '"', 0x0023, 0x0024, 0x0025, 0x0026, 0x0027, 0x0028, 0x0029, 0x002A, 0x002B,
  0x002C, 0x002D, 0x002E, 0x002F, 0x0030, 0x0031, 0x0032, 0x0033, 0x0034, 0x0035, 0x0036, 0x0037, 0x0038, 0x0039, 0x003A,
  0x003B, 0x003C, 0x003D, 0x003E, 0x003F
};

@interface ZXMaxiCodeDecodedBitStreamParser ()

+ (int)bit:(int)bit bytes:(unsigned char *)bytes length:(unsigned int)length;
+ (int)integer:(unsigned char *)bytes length:(unsigned int)length x:(unsigned char *)x xLength:(unsigned int)xLength;
+ (int)country:(unsigned char *)bytes length:(unsigned int)length;
+ (int)serviceClass:(unsigned char *)bytes length:(unsigned int)length;
+ (int)postCode2Length:(unsigned char *)bytes length:(unsigned int)length;
+ (int)postCode2:(unsigned char *)bytes length:(unsigned int)length;
+ (NSString *)postCode3:(unsigned char *)bytes length:(unsigned int)length;
+ (NSString *)message:(unsigned char *)bytes length:(unsigned int)length start:(int)start len:(int)len;

@end

@implementation ZXMaxiCodeDecodedBitStreamParser

+ (ZXDecoderResult *)decode:(unsigned char *)bytes length:(unsigned int)length mode:(int)mode {
  NSMutableString *result = [NSMutableString stringWithCapacity:144];
  switch (mode) {
    case 2:
    case 3: {
      NSString *postcode;
      if (mode == 2) {
        int pc = [self postCode2:bytes length:length];
        postcode = [NSString stringWithFormat:@"%9d", pc];
      } else {
        postcode = [self postCode3:bytes length:length];
      }
      NSString *country = [NSString stringWithFormat:@"%3d", [self country:bytes length:length]];
      NSString *service = [NSString stringWithFormat:@"%3d", [self serviceClass:bytes length:length]];
      [result appendString:[self message:bytes length:length start:10 len:84]];
      if ([result hasPrefix:[NSString stringWithFormat:@"[)>%C01%C", RS, GS]]) {
        [result insertString:[NSString stringWithFormat:@"%@%C%@%C%@%C", postcode, GS, country, GS, service, GS] atIndex:9];
      } else {
        [result insertString:[NSString stringWithFormat:@"%@%C%@%C%@%C", postcode, GS, country, GS, service, GS] atIndex:0];
      }
      break;
    }
    case 4:
      [result appendString:[self message:bytes length:length start:1 len:93]];
      break;
    case 5:
      [result appendString:[self message:bytes length:length start:1 len:77]];
      break;
  }
  return [[[ZXDecoderResult alloc] initWithRawBytes:bytes
                                             length:length
                                               text:result
                                       byteSegments:nil
                                            ecLevel:[NSString stringWithFormat:@"%d", mode]] autorelease];
}

+ (int)bit:(int)bit bytes:(unsigned char *)bytes length:(unsigned int)length {
  bit--;
  return (bytes[bit / 6] & (1 << (5 - (bit % 6)))) == 0 ? 0 : 1;
}

+ (int)integer:(unsigned char *)bytes length:(unsigned int)length x:(unsigned char *)x xLength:(unsigned int)xLength {
  int val = 0;
  for (int i = 0; i < xLength; i++) {
    val += [self bit:x[i] bytes:bytes length:length] << (xLength - i - 1);
  }
  return val;
}

#define COUNTRY_ARRAY_LEN 10
+ (int)country:(unsigned char *)bytes length:(unsigned int)length {
  unsigned char array[COUNTRY_ARRAY_LEN] = {53, 54, 43, 44, 45, 46, 47, 48, 37, 38};

  return [self integer:bytes length:length x:array xLength:COUNTRY_ARRAY_LEN];
}

#define SERVICE_ARRAY_LEN 10
+ (int)serviceClass:(unsigned char *)bytes length:(unsigned int)length {
  unsigned char array[SERVICE_ARRAY_LEN] = {55, 56, 57, 58, 59, 60, 49, 50, 51, 52};

  return [self integer:bytes length:length x:array xLength:SERVICE_ARRAY_LEN];
}

#define POST_CODE2_LENGTH_LEN 10
+ (int)postCode2Length:(unsigned char *)bytes length:(unsigned int)length {
  unsigned char array[POST_CODE2_LENGTH_LEN] = {39, 40, 41, 42, 31, 32};

  return [self integer:bytes length:length x:array xLength:POST_CODE2_LENGTH_LEN];
}

#define POST_CODE2_LEN 30
+ (int)postCode2:(unsigned char *)bytes length:(unsigned int)length {
  unsigned char array[POST_CODE2_LEN] = {33, 34, 35, 36, 25, 26, 27, 28, 29, 30, 19,
    20, 21, 22, 23, 24, 13, 14, 15, 16, 17, 18, 7, 8, 9, 10, 11, 12, 1, 2};

  return [self integer:bytes length:length x:array xLength:POST_CODE2_LEN];
}

#define POST_CODE3_LEN 6
+ (NSString *)postCode3:(unsigned char *)bytes length:(unsigned int)length {
  unsigned char array[POST_CODE3_LEN][POST_CODE3_LEN] = {
    {39, 40, 41, 42, 31, 32},
    {33, 34, 35, 36, 25, 26},
    {27, 28, 29, 30, 19, 20},
    {21, 22, 23, 24, 13, 14},
    {15, 16, 17, 18,  7,  8},
    { 9, 10, 11, 12,  1,  2}
  };

  return [NSString stringWithFormat:@"%C%C%C%C%C%C",
          SETS[0][[self integer:bytes length:length x:array[0] xLength:POST_CODE3_LEN]],
          SETS[0][[self integer:bytes length:length x:array[1] xLength:POST_CODE3_LEN]],
          SETS[0][[self integer:bytes length:length x:array[2] xLength:POST_CODE3_LEN]],
          SETS[0][[self integer:bytes length:length x:array[3] xLength:POST_CODE3_LEN]],
          SETS[0][[self integer:bytes length:length x:array[4] xLength:POST_CODE3_LEN]],
          SETS[0][[self integer:bytes length:length x:array[5] xLength:POST_CODE3_LEN]]];
}

+ (NSString *)message:(unsigned char *)bytes length:(unsigned int)length start:(int)start len:(int)len {
  NSMutableString *sb = [NSMutableString string];
  int shift = -1;
  int set = 0;
  int lastset = 0;
  for (int i = start; i < start + len; i++) {
    unichar c = SETS[set][bytes[i]];
    switch (c) {
      case LATCHA:
        set = 0;
        shift = -1;
        break;
      case LATCHB:
        set = 1;
        shift = -1;
        break;
      case SHIFTA:
      case SHIFTB:
      case SHIFTC:
      case SHIFTD:
      case SHIFTE:
        lastset = set;
        set = c - SHIFTA;
        shift = 1;
        break;
      case TWOSHIFTA:
        lastset = set;
        set = 0;
        shift = 2;
        break;
      case THREESHIFTA:
        lastset = set;
        set = 0;
        shift = 3;
        break;
      case NS: {
        int nsval = (bytes[++i] << 24) + (bytes[++i] << 18) + (bytes[++i] << 12) + (bytes[++i] << 6) + bytes[++i];
        [sb appendFormat:@"%9d", nsval];
        break;
      }
      case LOCK:
        shift = -1;
        break;
      default:
        [sb appendFormat:@"%C", c];
    }
    if (shift-- == 0) {
      set = lastset;
    }
  }
  while (sb.length > 0 && [sb characterAtIndex:sb.length - 1] == PAD) {
    [sb deleteCharactersInRange:NSMakeRange(sb.length - 1, 1)];
  }
  return sb;
}

@end
