SPI FLASH 容量计算

//0x9f 的ID2  计算FLASH容量
u32  CalcFlashCapacity(u8 CapcaityID, u8* u8Unit)
{
    u8 i = 0;
    u32 u32Result = 2 << CapcaityID;


    if (u32Result < 1024)
    {
        *u8Unit = 'B';
    }
    else
    {
        u32Result >>= 10;
        if (u32Result <1024)
        {
            *u8Unit = 'K';
        }
        else
        {
            u32Result >>= 10;
            if (u32Result < 1024)
            {
                *u8Unit = 'M';
            }
            else
            {
                *u8Unit = 'G';
            }
        }
    }
    return u32Result;

}

//此计算方法不一定完全正确。若大神们有好方法，请指点，谢谢！