world.go 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136
  1. package main
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "fmt"
  6. "os"
  7. "path"
  8. "strconv"
  9. "strings"
  10. )
  11. type World struct {
  12. Name string `json:"name"`
  13. Width uint64 `json:"width"`
  14. Height uint64 `json:"height"`
  15. TileIndex *TileIndex `json:"-"`
  16. Map []uint64 `json:"-"`
  17. }
  18. func LoadWorld(worlddir string) (*World, error) {
  19. var (
  20. w *World = &World{}
  21. err error
  22. )
  23. payload, err := os.ReadFile(path.Join(worlddir, "world.json"))
  24. if err != nil {
  25. return nil, err
  26. }
  27. err = json.Unmarshal(payload, &w)
  28. if err != nil {
  29. return nil, err
  30. }
  31. w.Map = make([]uint64, 0, w.Width*w.Height)
  32. ti, err := ParseTileIndex(path.Join(worlddir, "tiles.txt"))
  33. if err != nil {
  34. return nil, err
  35. }
  36. w.TileIndex = ti
  37. payload, err = os.ReadFile(path.Join(worlddir, "map.dat"))
  38. if err != nil {
  39. return nil, err
  40. }
  41. var (
  42. line string = ""
  43. linenum int = 0
  44. )
  45. for _, b := range payload {
  46. if b == '\n' {
  47. linenum += 1
  48. parts := strings.Split(line, " ")
  49. if len(parts) != int(w.Width) {
  50. return nil, fmt.Errorf("%d]: Failed parsing Map Data, not enough parts (expected=%d, got=%d), '%s'", linenum, w.Width, len(parts), line)
  51. }
  52. for segnum, p := range parts {
  53. t, err := strconv.ParseUint(p, 10, 64)
  54. if err != nil {
  55. return nil, fmt.Errorf("%d]: Failed parsing Map Data, line segment #%d errored %v, '%s'", linenum, segnum, err, line)
  56. }
  57. w.Map = append(w.Map, t)
  58. }
  59. line = ""
  60. } else {
  61. line += string(b)
  62. }
  63. }
  64. return w, nil
  65. }
  66. func (w *World) Save(dirname string) error {
  67. stat, err := os.Stat(dirname)
  68. if err != nil {
  69. if errors.Is(err, os.ErrNotExist) {
  70. err1 := os.MkdirAll(dirname, 0775)
  71. if err1 != nil {
  72. return fmt.Errorf("%v >> %v", err, err1)
  73. }
  74. } else {
  75. return err
  76. }
  77. }
  78. if stat != nil {
  79. if !stat.IsDir() {
  80. err = os.Remove(dirname)
  81. if err != nil {
  82. return err
  83. }
  84. err := os.MkdirAll(dirname, 0775)
  85. if err != nil {
  86. return err
  87. }
  88. }
  89. }
  90. payload, err := json.Marshal(w)
  91. if err != nil {
  92. return err
  93. }
  94. err = os.WriteFile(path.Join(dirname, "world.json"), payload, 0666)
  95. if err != nil {
  96. return err
  97. }
  98. err = w.TileIndex.Save(path.Join(dirname, "tiles.txt"))
  99. if err != nil {
  100. return err
  101. }
  102. f, err := os.Create(path.Join(dirname, "map.dat"))
  103. if err != nil {
  104. return err
  105. }
  106. defer f.Close()
  107. var idx int = 0
  108. for _, t := range w.Map {
  109. if idx >= int(w.Width) {
  110. _, err = f.WriteString("\n")
  111. if err != nil {
  112. return err
  113. }
  114. idx -= int(w.Width)
  115. }
  116. _, err = f.WriteString(fmt.Sprintf("%d ", t))
  117. if err != nil {
  118. return err
  119. }
  120. idx += 1
  121. }
  122. return nil
  123. }
  124. func (w *World) Get(x, y uint64) uint64 {
  125. idx := (y * w.Width) + x
  126. max := w.Width * w.Height
  127. if idx >= max {
  128. return 0
  129. }
  130. return w.Map[idx]
  131. }