engine/generator_thread.lua at master · nagolove/engine · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
local _tl_compat; if (tonumber((_VERSION or ''):match('[%d.]*$')) or 0) < 5.3 then local p, m = pcall(require, 'compat53.module'); if p then _tl_compat = m end end; local string = _tl_compat and _tl_compat.string or string


print('generator_thread: started')

local args = { ... }

local inspect = require('inspect')
print('args', inspect(args))

require('love.timer')
require("love.thread")
require("love.math")
require('common')

local gen_status_channel = love.thread.getChannel("gen_status_channel")

local das = require("diamond_and_square")
local serpent = require('serpent')

local mapn = args[1]

if type(mapn) ~= 'number' then
   error("'mapn' type should be a number, not " .. type(mapn))
end
local low_range, high_range = 1, 12
if mapn < low_range or mapn > high_range then
   error(string.format(
   'mapn out on range(%d..%d) - %d',
   low_range, high_range,
   mapn))

end

local rng_state = args[2]

if type(rng_state) ~= 'string' then
   error("'rng_state' should be a string nor " .. type(rng_state))
end

local rng = love.math.newRandomGenerator()
rng:setState(rng_state)

print('mapn, state', mapn, rng_state)

local gen = das.new(
mapn,
function()
   return rng:random()
end)


function write2file()

   local map = {}
   for i = 1, gen:get_mapsize() do
      map[#map + 1] = {}
      for j = 1, gen:get_mapsize() do
         map[#map][j] = gen:get(i - 1, j - 1)
      end
   end


   local dir_name = zerofyNum(mapn) .. "_" .. rng_state
   love.filesystem.createDirectory(dir_name)

   local fname = dir_name .. "/map.data.bin"
   print('write2file: fname', fname)

   local compress = love.data.compress
   local struct = require('struct')
   local packed_mapsize = struct.pack("L", gen:get_mapsize())

   love.filesystem.write(fname, "", 0)
   love.filesystem.append(fname, packed_mapsize, #packed_mapsize)

   print('map', #map)

   for i = 1, #map do
      local row = map[i]
      local uncompressed = serpent.dump(row)
      local compressed = compress('string', 'gzip', uncompressed, 9)


      local packed_rowlen = struct.pack("L", #compressed)
      love.filesystem.append(fname, packed_rowlen, #packed_rowlen)
      love.filesystem.append(fname, compressed, #compressed)
   end

end

local tstart = love.timer.getTime()
gen:eval()
local tfinish = love.timer.getTime()
print('map generated for', (tfinish - tstart) * 1000., "sec.")


write2file()


gen_status_channel:push(true)

print('generator_thread: finished')