package com.ekreutz.barcodescanner.util;

import com.google.android.gms.vision.barcode.Barcode;

import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

/**
 * Maps barcode FORMATS from Google's Barcode class, to a set of human readable strings.
 */
public class BarcodeFormat {

    public static final Map<Integer, String> FORMATS;
    public static final Map<String, Integer> REVERSE_FORMATS;

    private static final String UNKNOWN_FORMAT_STRING = "UNKNOWN_FORMAT";
    private static final int UNKNOWN_FORMAT_INT = -1;

    static {
        // Initialize integer to string map
        Map<Integer, String> map = new HashMap<>();
        map.put(Barcode.CODE_128, "CODE_128");
        map.put(Barcode.CODE_39, "CODE_39");
        map.put(Barcode.CODE_93, "CODE_93");
        map.put(Barcode.CODABAR, "CODABAR");
        map.put(Barcode.DATA_MATRIX, "DATA_MATRIX");
        map.put(Barcode.EAN_13, "EAN_13");
        map.put(Barcode.EAN_8, "EAN_8");
        map.put(Barcode.ITF, "ITF");
        map.put(Barcode.QR_CODE, "QR_CODE");
        map.put(Barcode.UPC_A, "UPC_A");
        map.put(Barcode.UPC_E, "UPC_E");
        map.put(Barcode.PDF417, "PDF417");
        map.put(Barcode.AZTEC, "AZTEC");
        FORMATS = Collections.unmodifiableMap(map);


        // Initialize string to integer map
        Map<String, Integer> rmap = new HashMap<>();
        for (Map.Entry<Integer, String> entry : FORMATS.entrySet()) {
            rmap.put(entry.getValue(), entry.getKey());
        }

        rmap.put("ALL", 0);
        REVERSE_FORMATS = Collections.unmodifiableMap(rmap);
    }

    public static String get(int format) {
        if (FORMATS.containsKey(format)) {
            return FORMATS.get(format);
        }

        return UNKNOWN_FORMAT_STRING;
    }

    public static int get(String format) {
        if (REVERSE_FORMATS.containsKey(format)) {
            return REVERSE_FORMATS.get(format);
        }

        return UNKNOWN_FORMAT_INT;
    }
}