Skip to content

Minor fixes 0 #43

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Pokemonpower92 merged 2 commits into from
Mar 2, 2025
Merged

Minor fixes 0 #43

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
ebfa4fb
rename seed set
Pokemonpower92 Mar 2, 2025
6d114c8
better generation of seed data
Pokemonpower92 Mar 2, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
207 changes: 59 additions & 148 deletions internal/database/seed.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ import (
"image/color"
"image/draw"
"image/png"
"math/rand"

"github.com/google/uuid"
"github.com/pokemonpower92/collagegenerator/config"
Expand All @@ -17,174 +18,78 @@ import (
sqlc "github.com/pokemonpower92/collagegenerator/internal/sqlc/generated"
)

type RandomImage struct {
id uuid.UUID
image *image.RGBA
}

// Generates a 50x50 pixel image of a random, uniform color
// and stores it in the provided store.
func generateRandomImage(store datastore.Store) *RandomImage {
id := uuid.New()
color := color.RGBA{
R: uint8(rand.Intn(255)),
G: uint8(rand.Intn(255)),
B: uint8(rand.Intn(255)),
A: 255,
}
img := image.NewRGBA(image.Rect(0, 0, 50, 50))
draw.Draw(
img,
img.Bounds(),
&image.Uniform{color},
image.Point{},
draw.Src,
)
var buf bytes.Buffer
if err := png.Encode(&buf, img); err != nil {
panic(err)
}
err := store.PutFile(id, &buf)
if err != nil {
panic(err)
}
return &RandomImage{
id: id,
image: img,
}
}

func Seed() {
config.LoadEnvironmentVariables()
c := config.NewPostgresConfig()
ctx := context.Background()

isRepo, err := repository.NewImageSetRepository(c, ctx)
if err != nil {
panic(err)
}
defer isRepo.Close()

imSet, err := isRepo.Create(sqlc.CreateImageSetParams{
Name: uuid.NewString(),
Name: "SeedSet",
Description: "A seeded imageset",
})
if err != nil {
panic(err)
}
store := datastore.NewStore()
images := []struct {
id uuid.UUID
color color.RGBA
}{
// Three shades of red
{
id: uuid.New(),
color: color.RGBA{R: 255, G: 0, B: 0, A: 255}, // Bright red
},
{
id: uuid.New(),
color: color.RGBA{R: 192, G: 0, B: 0, A: 255}, // Medium red
},
{
id: uuid.New(),
color: color.RGBA{R: 128, G: 0, B: 0, A: 255}, // Dark red
},
// Three shades of green
{
id: uuid.New(),
color: color.RGBA{R: 0, G: 255, B: 0, A: 255}, // Bright green
},
{
id: uuid.New(),
color: color.RGBA{R: 0, G: 192, B: 0, A: 255}, // Medium green
},
{
id: uuid.New(),
color: color.RGBA{R: 0, G: 128, B: 0, A: 255}, // Dark green
},
// Three shades of blue
{
id: uuid.New(),
color: color.RGBA{R: 0, G: 0, B: 255, A: 255}, // Bright blue
},
{
id: uuid.New(),
color: color.RGBA{R: 0, G: 0, B: 192, A: 255}, // Medium blue
},
{
id: uuid.New(),
color: color.RGBA{R: 0, G: 0, B: 128, A: 255}, // Dark blue
},
// Three grayscale values
{
id: uuid.New(),
color: color.RGBA{R: 255, G: 255, B: 255, A: 255}, // White
},
{
id: uuid.New(),
color: color.RGBA{R: 128, G: 128, B: 128, A: 255}, // Medium gray
},
{
id: uuid.New(),
color: color.RGBA{R: 0, G: 0, B: 0, A: 255}, // Black
},
// Flesh tones
{
id: uuid.New(),
color: color.RGBA{R: 255, G: 224, B: 196, A: 255}, // Light flesh tone
},
{
id: uuid.New(),
color: color.RGBA{R: 238, G: 207, B: 180, A: 255}, // Medium light flesh tone
},
{
id: uuid.New(),
color: color.RGBA{R: 224, G: 172, B: 105, A: 255}, // Medium flesh tone
},
{
id: uuid.New(),
color: color.RGBA{R: 141, G: 85, B: 36, A: 255}, // Dark flesh tone
},
{
id: uuid.New(),
color: color.RGBA{R: 198, G: 134, B: 66, A: 255}, // Medium dark flesh tone
},
{
id: uuid.New(),
color: color.RGBA{R: 89, G: 47, B: 42, A: 255}, // Deep flesh tone
},
// Purples
{
id: uuid.New(),
color: color.RGBA{R: 147, G: 112, B: 219, A: 255}, // Medium purple
},
{
id: uuid.New(),
color: color.RGBA{R: 75, G: 0, B: 130, A: 255}, // Indigo
},
{
id: uuid.New(),
color: color.RGBA{R: 186, G: 85, B: 211, A: 255}, // Medium orchid
},
// Yellows
{
id: uuid.New(),
color: color.RGBA{R: 255, G: 255, B: 0, A: 255}, // Bright yellow
},
{
id: uuid.New(),
color: color.RGBA{R: 255, G: 215, B: 0, A: 255}, // Gold
},
{
id: uuid.New(),
color: color.RGBA{R: 218, G: 165, B: 32, A: 255}, // Goldenrod
},
// Pinks
{
id: uuid.New(),
color: color.RGBA{R: 255, G: 192, B: 203, A: 255}, // Pink
},
{
id: uuid.New(),
color: color.RGBA{R: 255, G: 20, B: 147, A: 255}, // Deep pink
},
{
id: uuid.New(),
color: color.RGBA{R: 219, G: 112, B: 147, A: 255}, // Pale violet red
},
}
img := image.NewRGBA(image.Rect(0, 0, 50, 50))
for _, im := range images {
draw.Draw(
img,
img.Bounds(),
&image.Uniform{im.color},
image.Point{},
draw.Src,
)
var buf bytes.Buffer
if err := png.Encode(&buf, img); err != nil {
panic(err)
}
store.PutFile(im.id, &buf)
}

acRepo, err := repository.NewAverageColorRepository(c, ctx)
if err != nil {
panic(err)
}
defer acRepo.Close()
for _, im := range images {
imageSetImage, err := store.GetRGBA(im.id)
if err != nil {
panic(err)
}
average := imageprocessing.CalculateAverageColor(imageSetImage)

// Generates 100 random images for the seed image set
// and creates the average color records for them.
store := datastore.NewStore()
for range 100 {
randomImage := generateRandomImage(store)
average := imageprocessing.CalculateAverageColor(randomImage.image)
_, err = acRepo.Create(sqlc.CreateAverageColorParams{
ID: im.id,
ID: randomImage.id,
ImagesetID: imSet.ID,
FileName: im.id.String(),
FileName: randomImage.id.String(),
R: int32(average.R),
G: int32(average.G),
B: int32(average.B),
Expand All @@ -194,30 +99,36 @@ func Seed() {
panic(err)
}
}

tiRepo, err := repository.NewTargetImageRepository(c, ctx)
if err != nil {
panic(err)
}
defer tiRepo.Close()

randomTargetImage := generateRandomImage(store)
targetImage, err := tiRepo.Create(sqlc.CreateTargetImageParams{
ID: images[len(images)-1].id,
Name: "Grey",
ID: randomTargetImage.id,
Name: "SeedTargetImage",
Description: "A seeded target image",
})
if err != nil {
panic(err)
}

cRepo, err := repository.NewCollageRepository(c, ctx)
if err != nil {
panic(err)
}
defer cRepo.Close()

col, err := cRepo.Create(sqlc.CreateCollageParams{
Name: "Seed Collage",
Description: "A seeded collage",
ImageSetID: imSet.ID,
TargetImageID: targetImage.ID,
})

service.CreateCollageMetaData(col)
if err != nil {
panic(err)
Expand Down