|
28 | 28 |
|
29 | 29 | static inline uint32_t crc32c_bmp(const char* filename) {
|
30 | 30 | std::ifstream in{ filename, std::ios::binary | std::ios::ate };
|
| 31 | + if (!in) return 0; |
31 | 32 | auto size = in.tellg();
|
32 |
| - in.seekg(0); |
33 |
| - std::vector<uint8_t> buf(size); |
| 33 | + in.seekg(0, std::ios::beg); |
| 34 | + |
| 35 | + std::vector<uint8_t> buf(static_cast<size_t>(size)); |
34 | 36 | in.read(reinterpret_cast<char*>(buf.data()), size);
|
35 | 37 |
|
36 |
| - // BMP offset at bytes 10–13 |
| 38 | + // offset in BMP header (bytes 10–13) |
37 | 39 | uint32_t offset = *reinterpret_cast<uint32_t*>(buf.data() + 10);
|
38 | 40 | uint8_t* bmp = buf.data() + offset;
|
39 |
| - size_t len = buf.size() - offset; |
| 41 | + size_t len = buf.size() - offset; |
40 | 42 |
|
41 |
| - uint64_t crcReg = 0xFFFFFFFFull; |
42 |
| - size_t qwords = len / 8; |
43 |
| - auto ptr = reinterpret_cast<uint64_t*>(bmp); |
44 |
| - for (size_t i = 0; i < qwords; ++i) |
45 |
| - crcReg = _mm_crc32_u64(crcReg, ptr[i]); |
| 43 | +#if defined(_M_X64) || defined(__x86_64__) |
| 44 | + uint64_t crc64 = 0xFFFFFFFFull; |
| 45 | + auto q64 = len / 8; |
| 46 | + auto const* p64 = reinterpret_cast<uint64_t const*>(bmp); |
| 47 | + for (size_t i = 0; i < q64; ++i) |
| 48 | + crc64 = _mm_crc32_u64(crc64, p64[i]); |
46 | 49 |
|
47 |
| - uint32_t crc = static_cast<uint32_t>(crcReg); |
48 |
| - auto tail = reinterpret_cast<uint8_t*>(ptr + qwords); |
| 50 | + uint32_t crc = static_cast<uint32_t>(crc64); |
| 51 | + auto const* tail = reinterpret_cast<uint8_t const*>(p64 + q64); |
49 | 52 | for (size_t i = 0, r = len & 7; i < r; ++i)
|
50 | 53 | crc = _mm_crc32_u8(crc, tail[i]);
|
| 54 | +#else |
| 55 | + uint32_t crc = 0xFFFFFFFFu; |
| 56 | + auto q32 = len / 4; |
| 57 | + auto const* p32 = reinterpret_cast<uint32_t const*>(bmp); |
| 58 | + for (size_t i = 0; i < q32; ++i) |
| 59 | + crc = _mm_crc32_u32(crc, p32[i]); |
| 60 | + |
| 61 | + auto const* tail = reinterpret_cast<uint8_t const*>(p32 + q32); |
| 62 | + for (size_t i = 0, r = len & 3; i < r; ++i) |
| 63 | + crc = _mm_crc32_u8(crc, tail[i]); |
| 64 | +#endif |
51 | 65 |
|
52 | 66 | return crc ^ 0xFFFFFFFFu;
|
53 | 67 | }
|
54 | 68 |
|
55 | 69 | int main(int argc, char** argv) {
|
56 | 70 | for (int i = 1; i < argc; ++i) {
|
57 | 71 | uint32_t h = crc32c_bmp(argv[i]);
|
58 |
| - std::cout << argv[i] << ": 0x" << std::hex << std::uppercase << h << "\n"; |
| 72 | + std::cout << argv[i] |
| 73 | + << ": 0x" << std::hex << std::uppercase << h |
| 74 | + << std::dec << "\n"; |
59 | 75 | }
|
| 76 | + return 0; |
60 | 77 | }
|
0 commit comments