import java.text.DecimalFormat;
import java.util.Hashtable;

/**
 * 文件大小單位轉(zhuǎn)換
 * @author Administrator
 * 
 */
public class UnitsConversion extends DecimalFormat {

    private static final long serialVersionUID = 3168068393840262910L;
    /**
     * 存放有效單位的數(shù)組
     */
    private static Hashtable<String, String> validUnits = new Hashtable<String, String>();
    /**
     * 限制文件大小上限為1G
     */
    private static int GB_MAX_SIZE = 1;
    /**
     * 最大的MB值
     */
    private static int MB_MAX_SIZE = GB_MAX_SIZE * 1024;
    /**
     * 最大的KB值
     */
    private static int KB_MAX_SIZE = MB_MAX_SIZE * 1024;
    /**
     * 最大的Bytes值
     */
    private static int BYTES_MAX_SIZE = KB_MAX_SIZE * 1024;
    /**
     * 數(shù)字部分的值
     */
    private Double numPart;
    /**
     * 原始的單位字符串
     */
    private String originalUnit;
    /**
     * 標(biāo)準(zhǔn)的單位字符串
     */
    private String unit;
    /**
     * 轉(zhuǎn)換后的結(jié)果
     */
    private String result;

    // 添加所有有效單位
    static {
        validUnits.put("字節(jié)", "Bytes");
        validUnits.put("bytes", "Bytes");
        validUnits.put("byte", "Bytes");
        validUnits.put("kb", "KB");
        validUnits.put("k", "KB");
        validUnits.put("兆", "MB");
        validUnits.put("mb", "MB");
        validUnits.put("m", "MB");
        validUnits.put("gb", "GB");
        validUnits.put("g", "GB");
    }

    /**
     * 構(gòu)造方法：指定了數(shù)字格式，初始所有屬性為NULL
     */
    public UnitsConversion() {
        super("########.##");
        numPart = null;
        result = null;
        unit = null;
        originalUnit = null;
    }

    /**
     * 根據(jù)單位、數(shù)字大小按照常用的轉(zhuǎn)換原則進(jìn)行轉(zhuǎn)換
     * 
     * @param input
     * @return 成功轉(zhuǎn)換后的結(jié)果是非空字符串；若失敗了，結(jié)果為空字符串
     */
    public String defaultConversion(String input) {
        analyzeString(input);
        if (result != null) {
            return result;
        }
        // 單位 Bytes
        if (unit.equals("Bytes")) {
            int numPart2Int = numPart.intValue();
            // 輸入大小與1G相差0.5M之內(nèi)，返回1GB
            if ((BYTES_MAX_SIZE - numPart2Int) < (1024 * 1024) / 2) {
                return "1 GB";
            }
            // (0,1KB)
            if (numPart2Int < 1024) {
                return numPart2Int + " Bytes";
            }
            // [1KB,1023KB]
            if (numPart2Int >= 1024 && numPart2Int <= (1024 - 1) * 1024) {
                return format(numPart / 1024) + " KB";
            }
            // (1023KB,1GB)
            if (numPart2Int > (1024 - 1) * 1024 && numPart2Int < BYTES_MAX_SIZE) {
                return format(numPart / (1024 * 1024)) + " MB";
            } else
                result = "";
            return result;
        }

        if (unit.equals("KB")) {
            return "還沒(méi)實(shí)現(xiàn)....";
        }

        if (unit.equals("MB")) {
            return "還沒(méi)實(shí)現(xiàn)....";
        }

        if (unit.equals("GB")) {
            return "還沒(méi)實(shí)現(xiàn)....";
        }
        result = "";
        return result;
    }

    /**
     * 把字符串的數(shù)字部分與單位分離，并對(duì)數(shù)字、單位的有效性進(jìn)行檢驗(yàn)， 若有非法狀況，把結(jié)果賦值為 "" ，將其返回給用戶
     * 
     * @param input
     */
    public void analyzeString(String input) {
        // 初步檢驗(yàn)輸入的字符串
        if (input == null || input.trim().length() < 2) {
            System.out.println("輸入的字符串有誤");
            result = "";
            return;
        }
        input = input.replaceAll(" ", "");
        int firstIndexOfUnit;// 單位的起始位置
        String strOfNum;// 數(shù)字部分的字符串
        // 從尾部開(kāi)始遍歷字符串
        for (int i = input.length() - 1; i >= 0; i--) {
            if (Character.isDigit(input.charAt(i))) {
                firstIndexOfUnit = i + 1;
                originalUnit = input.substring(firstIndexOfUnit,
                        input.length()).toLowerCase();
                if (!isValidUnit(originalUnit)) {
                    System.out.println("無(wú)效單位。");
                    result = "";
                    return;
                }
                unit = validUnits.get(originalUnit);
                strOfNum = input.substring(0, firstIndexOfUnit);
                numPart = Double.parseDouble(strOfNum);
                if (!isValidNum(numPart, unit)) {
                    System.out.println("文件大小非法");
                    result = "";
                    return;
                }
                if (numPart == 0) {
                    result = "0 Bytes";
                    return;
                }
                break;
            }
        }
        if (unit == null || numPart == null) {
            System.out.println("輸入的字符串有誤");
            result = "";
            return;
        }

    }

    /**
     * 文件大小越界檢查
     * 
     * @param num
     * @param unit
     * @return 在1G范圍內(nèi)（包括1G），返回true；否則返回false
     */
    public boolean isValidNum(Double num, String unit) {
        if (num == null || num < 0 || num > BYTES_MAX_SIZE) {
            return false;
        }
        if (unit.equals("KB") && num > KB_MAX_SIZE) {
            return false;
        }
        if (unit.equals("MB") && num > MB_MAX_SIZE) {
            return false;
        }
        if (unit.equals("GB") && num > GB_MAX_SIZE) {
            return false;
        }
        return true;
    }

    /**
     * 檢查原始單位originalUnit是否有效
     * 
     * @param originalUnit
     * @return 若originalUnit為空，那么會(huì)給他賦默認(rèn)值 bytes ，并返回true；<br>
     *         若originalUnit是有效單位集合中之一，返回true。
     */
    public boolean isValidUnit(String originalUnit) {
        if (originalUnit == null || originalUnit.trim().length() < 1) {
            originalUnit = "bytes";
            return true;
        }
        for (String validUnit : validUnits.keySet()) {
            if (validUnit.equalsIgnoreCase(originalUnit)) {
                return true;
            }
        }
        return false;
    }
    //測(cè)試
    public static void main(String[] args) {
        System.out.println("-------------");
        for (int i = 1020 * 1024; i <= 1024 * 1111; i += 9) {
            String input = i + " ";
            System.out.println(input + " ---> "
                    + new UnitsConversion().defaultConversion(input));
        }
    }

}