import "github.com/wangningkai/go-shp"

reader, err := shp.Open("file.shp")
if err != nil {
    log.Fatal(err)
}
defer reader.Close()

for reader.Next() {
    n, shape := reader.Shape()
    // 处理几何对象

    // 读取属性
    attrs := reader.ReadAttribute(n)
}

writer, err := shp.Create("output.shp", shp.POINT)
if err != nil {
    log.Fatal(err)
}
defer writer.Close()

// 设置字段
fields := []shp.Field{
    shp.StringField("NAME", 50),
    shp.NumberField("ID", 10),
}
writer.SetFields(fields)

// 写入数据
row := writer.Write(&shp.Point{X: 1.0, Y: 2.0})
writer.WriteAttribute(int(row), 0, "Point A")
writer.WriteAttribute(int(row), 1, 123)

// Shapefile 转 GeoJSON
err := shp.ConvertShapefileToGeoJSON("input.shp", "output.geojson")

// GeoJSON 转 Shapefile
err = shp.ConvertGeoJSONToShapefile("input.geojson", "output.shp")

// 基础使用 - 使用默认配置
reader, err := shp.Open("file.shp")

// 容错模式 - 跳过损坏的数据
reader, err := shp.Open("file.shp", 
    shp.WithIgnoreCorruptedShapes(true))

// 内存限制 - 限制最大内存使用量（字节）
// 当超过限制时，Next() 返回错误
reader, err := shp.Open("file.shp",
    shp.WithMaxMemoryUsage(50*1024*1024)) // 50MB

// 缓冲配置 - 调整I/O缓冲大小
reader, err := shp.Open("file.shp",
    shp.WithBuffering(true, 128*1024)) // 128KB缓冲

// 调试模式 - 输出详细日志
reader, err := shp.Open("file.shp",
    shp.WithDebug(true))

// 组合使用多个选项
reader, err := shp.Open("file.shp",
    shp.WithIgnoreCorruptedShapes(true),
    shp.WithMaxMemoryUsage(100*1024*1024),
    shp.WithDebug(true))

// 基础使用
writer, err := shp.Create("output.shp", shp.POINT)

// 启用数据验证
writer, err := shp.CreateWithConfig("output.shp", shp.POINT,
    shp.DefaultWriterConfig(),
    shp.WithValidation(true))

// 同步写入 - 每次写入后立即同步到磁盘（较慢但安全）
writer, err := shp.CreateWithConfig("output.shp", shp.POINT,
    shp.DefaultWriterConfig(),
    shp.WithSync(true))

import (
    "errors"
    "github.com/wangningkai/go-shp"
)

reader, err := shp.Open("file.shp")
if err != nil {
    var shapeErr *shp.ShapeError
    if errors.As(err, &shapeErr) {
        // 根据错误类型处理
        switch shapeErr.Type {
        case shp.ErrInvalidFormat:
            log.Fatal("文件格式不正确")
        case shp.ErrCorruptedFile:
            log.Fatal("文件已损坏")
        case shp.ErrExceedsMemoryLimit:
            log.Fatal("超过内存限制，请增加MaxMemoryUsage或减小文件")
        case shp.ErrIO:
            log.Fatal("I/O错误:", shapeErr.Cause)
        }
    }
    log.Fatal(err)
}

// 在读取过程中检查错误
for reader.Next() {
    // 处理数据
}
if reader.Err() != nil {
    log.Fatal("读取失败:", reader.Err())
}

// 流式读取大文件，限制内存使用
reader, err := shp.Open("large.shp",
    shp.WithMaxMemoryUsage(100*1024*1024),  // 100MB 限制
    shp.WithBuffering(true, 256*1024))      // 256KB 缓冲，减少I/O次数
if err != nil {
    log.Fatal(err)
}
defer reader.Close()

// 逐条处理，避免一次性加载到内存
processed := 0
for reader.Next() {
    n, shape := reader.Shape()
    // 处理单条记录
    processShape(shape)
    
    // 定期检查内存是否达到限制
    if processed%10000 == 0 {
        log.Printf("Processed %d shapes\n", processed)
    }
    processed++
}
if reader.Err() != nil {
    log.Fatal(reader.Err())
}

// 从 .shp 流式写出到 .geojson（边读边写，内存占用恒定）
f, _ := os.Create("output.geojson")
defer f.Close()
conv := shp.GeoJSONConverter{}
err := conv.ShapefileToGeoJSONStream("large.shp", f, 
    shp.WithIgnoreCorruptedShapes(true))
if err != nil {
    log.Fatal(err)
}

reader, err := shp.Open("corrupted.shp",
    shp.WithIgnoreCorruptedShapes(true),  // 跳过损坏的shape
    shp.WithDebug(true))                  // 输出详细日志，帮助定位问题
if err != nil {
    log.Fatal(err)
}
defer reader.Close()

skipped := 0
for reader.Next() {
    // 正常处理
}
// 注意：损坏的shape被跳过，不会导致整个读取过程中断
log.Printf("Read completed (some records may have been skipped)")

reader, err := shp.OpenWithConfig("input.shp", shp.DefaultReaderConfig(),
    shp.WithIgnoreCorruptedShapes(true))

validShapes := 0
for reader.Next() {
    validShapes++
}
log.Printf("Successfully read %d valid shapes\n", validShapes)