odboy
f5e3f6b4ef
- 移动文件改用 uuid+原文件名，避免同名覆盖丢失数据
- BMP 负高度取绝对值，自顶向下格式解析正确
- 新增 {目录名}-picker 白名单输出目录，整理后跳过重复扫描
- 分类目录名去掉 uuid 前缀，重复运行并入已有目录
88 lines
2.4 KiB
Go
88 lines
2.4 KiB
Go
package core
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"fmt"
|
|
"os"
|
|
)
|
|
|
|
// BMPFileHeader represents the bitmap file header structure.
|
|
type BMPFileHeader struct {
|
|
FileType [2]byte
|
|
FileSize uint32
|
|
Reserved1 uint16
|
|
Reserved2 uint16
|
|
DataOffset uint32
|
|
}
|
|
|
|
// BMPInfoHeader represents the bitmap info header structure.
|
|
type BMPInfoHeader struct {
|
|
Size uint32
|
|
Width int32
|
|
Height int32
|
|
Planes uint16
|
|
BitsPerPixel uint16
|
|
Compression uint32
|
|
SizeImage uint32
|
|
XPelsPerMeter int32
|
|
YPelsPerMeter int32
|
|
ColorsUsed uint32
|
|
ColorsImportant uint32
|
|
}
|
|
|
|
func ReadBmpImage(filePath string) (error, int32, int32) {
|
|
file, err := os.Open(filePath)
|
|
if err != nil {
|
|
fmt.Printf("=== Failed to open file: %v\n", err)
|
|
return err, 0, 0
|
|
}
|
|
defer file.Close()
|
|
|
|
// Read the file header.
|
|
var fileHeader BMPFileHeader
|
|
if err := binary.Read(file, binary.LittleEndian, &fileHeader); err != nil {
|
|
fmt.Printf("=== Failed to Read file: %v\n", err)
|
|
return err, 0, 0
|
|
}
|
|
|
|
// Check if it's a valid BMP file by verifying the file type.
|
|
if string(fileHeader.FileType[:]) != "BM" {
|
|
return fmt.Errorf("not a valid BMP file: file type is %q, expected \"BM\"", string(fileHeader.FileType[:])), 0, 0
|
|
}
|
|
|
|
// Read the info header.
|
|
var infoHeader BMPInfoHeader
|
|
if err := binary.Read(file, binary.LittleEndian, &infoHeader); err != nil {
|
|
fmt.Printf("=== Failed to Read file: %v\n", err)
|
|
return err, 0, 0
|
|
}
|
|
// 自顶向下 BMP 的高度为负值，取绝对值避免影响后续解析
|
|
return err, absInt32(infoHeader.Width), absInt32(infoHeader.Height)
|
|
//fmt.Printf("=== File size: %d bytes\n", fileHeader.FileSize)
|
|
//fmt.Printf("=== Image dimensions: %dx%d\n", )
|
|
//fmt.Printf("=== Bits per pixel: %d\n", infoHeader.BitsPerPixel)
|
|
|
|
// At this point, you would typically read the pixel data into a slice,
|
|
// taking into account any padding required for alignment and the specific
|
|
// pixel format (e.g., RGB, RGBA, indexed color). This part is omitted here.
|
|
|
|
// You may also need to skip over a possible color palette if present.
|
|
// The size of the palette can be inferred from the number of colors used and
|
|
// the bits per pixel value.
|
|
|
|
// For simplicity, we'll just read the rest of the file as raw bytes.
|
|
//data, err := ioutil.ReadAll(file)
|
|
//if err != nil {
|
|
// panic(err)
|
|
//}
|
|
//fmt.Printf("=== Read %d bytes of pixel data.\n", len(data))
|
|
}
|
|
|
|
// absInt32 返回 int32 的绝对值
|
|
func absInt32(v int32) int32 {
|
|
if v < 0 {
|
|
return -v
|
|
}
|
|
return v
|
|
}
|