58 lines
1.0 KiB
Go
58 lines
1.0 KiB
Go
package main
|
|
|
|
import (
|
|
"image"
|
|
"log"
|
|
"math/rand"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
// load an image from a file into an Image value
|
|
func loadImage(filename string) image.Image {
|
|
img, err := os.Open(filename)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer img.Close()
|
|
|
|
imgData, _, err := image.Decode(img)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
log.Println("Loading :", filename)
|
|
|
|
return imgData
|
|
}
|
|
|
|
// Walk through a folder recursively and returns a list of image paths
|
|
func getImagesList(path string) []string {
|
|
var imgs []string
|
|
err := filepath.Walk(path,
|
|
func(path string, info os.FileInfo, err error) error {
|
|
ext := strings.ToLower(filepath.Ext(path))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if ext == ".jpg" || ext == ".png" {
|
|
imgs = append(imgs, path)
|
|
}
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
|
|
return imgs
|
|
}
|
|
|
|
// Randomly choose x number of image from a given folder
|
|
func getRandomImage() string {
|
|
dir := getImagesList(ConfigRegister.InputDir)
|
|
index := rand.Intn(len(dir))
|
|
return dir[index]
|
|
}
|