v/vlib/stbi/stbi.v

78 lines
1.6 KiB
V
Raw Normal View History

2020-01-23 21:04:46 +01:00
// Copyright (c) 2019-2020 Alexander Medvednikov. All rights reserved.
2019-06-23 04:21:30 +02:00
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
2019-06-22 20:20:28 +02:00
module stbi
2019-09-23 12:42:20 +02:00
// note we might need special case for this
// import gl
2019-06-22 20:20:28 +02:00
#flag -I @VROOT/thirdparty/stb_image
2019-06-23 14:08:17 +02:00
2019-06-22 20:20:28 +02:00
#define STB_IMAGE_IMPLEMENTATION
2019-06-23 14:08:17 +02:00
#include "stb_image.h"
2019-10-27 12:05:50 +01:00
pub struct Image {
2019-06-22 20:20:28 +02:00
mut:
width int
height int
nr_channels int
ok bool
data voidptr
ext string
}
2019-11-24 04:27:02 +01:00
fn C.stbi_load() voidptr
fn C.stbi_load_from_memory() voidptr
2019-11-24 04:27:02 +01:00
fn C.stbi_image_free()
fn C.stbi_set_flip_vertically_on_load()
2019-06-26 17:49:50 +02:00
pub fn load(path string) Image {
2019-06-22 20:20:28 +02:00
ext := path.all_after('.')
mut res := Image {
ok: true
ext: ext
data: 0
}
2019-08-22 23:00:31 +02:00
flag := if ext == 'png' { C.STBI_rgb_alpha } else { 0 }
res.data = C.stbi_load(path.str, &res.width, &res.height, &res.nr_channels, flag)
2019-06-22 20:20:28 +02:00
if isnil(res.data) {
2019-08-22 23:00:31 +02:00
println('stbi image failed to load')
2019-06-23 10:41:42 +02:00
exit(1)
2019-06-22 20:20:28 +02:00
}
return res
}
//pub fn load_from_memory(buf []byte) Image {
pub fn load_from_memory(buf byteptr) Image {
mut res := Image {
ok: true
data: 0
}
flag := C.STBI_rgb_alpha
res.data = C.stbi_load_from_memory(buf, 3812, &res.width, &res.height, &res.nr_channels, flag)
if isnil(res.data) {
println('stbi image failed to load from memory')
exit(1)
}
return res
}
2019-06-26 17:49:50 +02:00
pub fn (img Image) free() {
2019-06-22 20:20:28 +02:00
C.stbi_image_free(img.data)
}
2019-06-26 17:49:50 +02:00
pub fn (img Image) tex_image_2d() {
2019-08-22 23:00:31 +02:00
mut rgb_flag := C.GL_RGB
2019-06-22 20:20:28 +02:00
if img.ext == 'png' {
2019-08-22 23:00:31 +02:00
rgb_flag = C.GL_RGBA
2019-06-22 20:20:28 +02:00
}
2019-08-22 23:00:31 +02:00
C.glTexImage2D(C.GL_TEXTURE_2D, 0, rgb_flag, img.width, img.height, 0,
rgb_flag, C.GL_UNSIGNED_BYTE, img.data)
2019-06-22 20:20:28 +02:00
}
2019-06-26 17:49:50 +02:00
pub fn set_flip_vertically_on_load(val bool) {
2019-06-22 20:20:28 +02:00
C.stbi_set_flip_vertically_on_load(val)
}