Skip to content

Commit b2335e2

Browse files
committed
shorten shader floats, adjust cam speed, fill window, -shtest
1 parent 035f55d commit b2335e2

File tree

13 files changed

+379
-474
lines changed

13 files changed

+379
-474
lines changed

Main.ipynb

Lines changed: 14 additions & 14 deletions
Large diffs are not rendered by default.

gen_shaders.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ def gen(model):
2222
int channelGroup = int(screen_coords.x / screenSize.x);
2323
vec2 uv = mod(screen_coords, screenSize) / screenSize;
2424
uv -= 0.5;
25-
uv.x = -uv.x;
25+
uv.x = -uv.x * screenSize.x / screenSize.y;
2626
mat3 ca = setCamera(camPos, camTarget, 0.0);
2727
vec3 ro = camPos;
2828
vec3 rd = ca * normalize(vec3(uv, 1.0));
@@ -88,9 +88,9 @@ def gen(model):
8888
weights = w[i * 4 + j]
8989
s += f'\t\tfloat c{j} = ' \
9090
+ '+'.join([
91-
f'h{k // 4}[{k % 4}]*({v})'
91+
f'h{k // 4}[{k % 4}]*({v:.8f})'
9292
for k, v in enumerate(weights)]) \
93-
+ f'+({b[i * 4 + j]});\n'
93+
+ f'+({b[i * 4 + j]:.8f});\n'
9494
if layer == model.layers[-1]:
9595
s += """\t\treturn clamp(vec4(c0, c1, c2, c3), 0.0, 1.0);
9696
} else """

viewer/camera.lua

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,16 +5,16 @@ function camera.load()
55
camera.pos = { x=-3.7, y=-1.6, z=-0.3 }
66
camera.rot = { x=0, y=0.2, z=0 }
77
camera.target = nil -- set in update
8-
camera.moveSpeed = 2
8+
camera.moveSpeed = 20
99
camera.turnSpeed = 1 / 200
1010
end
1111

1212
function camera.update(dt)
1313
local speed = camera.moveSpeed
1414
if love.keyboard.isScancodeDown('lctrl') then
15-
speed = speed / 6
15+
speed = speed / 4
1616
elseif love.keyboard.isScancodeDown('lshift') then
17-
speed = speed * 6
17+
speed = speed * 4
1818
end
1919

2020
if love.keyboard.isScancodeDown('space') then

viewer/main.lua

Lines changed: 2 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,6 @@
11

22
camera = require 'camera'
3-
_renderer = require 'renderer'
4-
shtest = require 'shtest'
5-
renderer = _renderer
3+
renderer = require 'renderer'
64

75
ssx, ssy = love.graphics.getDimensions()
86

@@ -15,16 +13,10 @@ love.graphics.setDefaultFilter('nearest', 'nearest')
1513
canvases = {}
1614
shaders = {}
1715

18-
fonts = {
19-
f24 = love.graphics.newFont(24),
20-
f32 = love.graphics.newFont(32)
21-
}
22-
2316
function love.load()
2417
camera.load()
2518
renderer.load()
26-
shtest.load()
27-
fpsLimit = -60
19+
fpsLimit = 0
2820
fpsTimer = 0
2921
end
3022

@@ -63,9 +55,6 @@ function love.keypressed(k, scancode, isRepeat)
6355
love.event.quit()
6456
elseif k == 'r' then
6557
camera.load()
66-
renderer.load()
67-
elseif k == 'tab' then
68-
renderer = renderer == _renderer and shtest or _renderer
6958
end
7059
end
7160

viewer/renderer.lua

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,19 +6,22 @@ local function nc32(w, h)
66
end
77

88
function renderer.load()
9-
renderer.canvases = {nc32(256 * 13, 256)}
9+
rx = ssx / 4
10+
ry = ssy / 4
11+
renderer.canvases = {nc32(rx * 13, ry)}
1012
renderer.shaders = {love.graphics.newShader('shaders/x.glsl')}
11-
renderer.shaders[1]:send('screenSize', { 256, 256 })
13+
renderer.shaders[1]:send('screenSize', { rx, ry })
1214
nFeats = 64
1315
for i=1, 6 do
1416
if i == 6 then
15-
table.insert(renderer.canvases, love.graphics.newCanvas(256, 256))
17+
table.insert(renderer.canvases, love.graphics.newCanvas(rx, ry))
1618
else
17-
table.insert(renderer.canvases, nc32(256 * 64 / 4, 256))
19+
table.insert(renderer.canvases, nc32(rx * nFeats / 4, ry))
1820
end
1921
table.insert(renderer.shaders, love.graphics.newShader('shaders/l' .. i-1 .. '.glsl'))
20-
renderer.shaders[#renderer.shaders]:send('screenSize', { 256, 256 })
22+
renderer.shaders[#renderer.shaders]:send('screenSize', { rx, ry })
2123
end
24+
renderer.canvases[#renderer.canvases]:setFilter('linear', 'linear')
2225
end
2326

2427
function renderer.updateImage()
@@ -47,7 +50,7 @@ end
4750
function renderer.draw()
4851
love.graphics.setColor(1, 1, 1)
4952
love.graphics.draw(renderer.canvases[#renderer.canvases],
50-
ssx / 2, ssy / 2, 0, 2, 2, 128, 128)
53+
ssx / 2, ssy / 2, 0, 4, 4, ssx / 8, ssy / 8)
5154
end
5255

5356
return renderer

viewer/shaders/l0.glsl

Lines changed: 64 additions & 64 deletions
Large diffs are not rendered by default.

viewer/shaders/l1.glsl

Lines changed: 64 additions & 64 deletions
Large diffs are not rendered by default.

viewer/shaders/l2.glsl

Lines changed: 64 additions & 64 deletions
Large diffs are not rendered by default.

viewer/shaders/l3.glsl

Lines changed: 64 additions & 64 deletions
Large diffs are not rendered by default.

viewer/shaders/l4.glsl

Lines changed: 64 additions & 64 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)