support multiple images generation with time lapse for domainwrap

arts/domainwrap.go

@@ -1,9 +1,11 @@
 package arts
 
 import (
+ "fmt"
  "github.com/jdxyw/generativeart"
  "github.com/jdxyw/generativeart/common"
  "image/color"
+ "log"
 )
 
 // ColorMapping maps some parameters to color space.
@@ -15,6 +17,12 @@ type domainWrap struct {
  scale2 float64
  xOffset, yOffset float64
  fn ColorMapping
+ // How many images would be created in this generation.
+ numImages int
+ // Use these parameters to create images with time lapse.
+ xOffsetStep, yOffsetStep float64
+ // The imagPath for generative images.
+ imgPath string
 }
 
 // NewDomainWrap returns a domainWrap object.
@@ -29,9 +37,34 @@ func NewDomainWrap(scale, scale2, xOffset, yOffset float64, cmap ColorMapping) *
  }
 }
 
+func (d *domainWrap) SetDynamicParameter(xstep, ystep float64, n int, path string) {
+ d.xOffsetStep = xstep
+ d.yOffsetStep = ystep
+ d.numImages = n
+ d.imgPath = path
+}
+
 // Generative draws a domain warp image.
 // Reference: https://www.iquilezles.org/www/articles/warp/warp.htm
 func (d *domainWrap) Generative(c *generativeart.Canva) {
+ if d.numImages == 0 && len(d.imgPath) == 0 {
+ d.generative(c)
+ return
+ }
+
+ if d.numImages > 0 && len(d.imgPath) == 0 {
+ log.Fatal("Missing the parameters numImages or imgPath")
+ }
+ for i := 0; i < d.numImages; i++ {
+ imgfile := fmt.Sprintf("%v/domainwrap%03d.PNG", d.imgPath, i)
+ d.xOffset += d.xOffsetStep * float64(i)
+ d.yOffset += d.yOffsetStep * float64(i)
+ d.generative(c)
+ c.ToPNG(imgfile)
+ }
+}
+
+func (d *domainWrap) generative(c *generativeart.Canva) {
  for h := 0.0; h < float64(c.Height()); h += 1.0 {
  for w := 0.0; w < float64(c.Width()); w += 1.0 {
  r, m1, m2 := d.pattern(w*d.scale, h*d.scale, d.xOffset, d.yOffset)

example/example_domainwrap2.go

@@ -0,0 +1,30 @@
+package main
+
+import (
+ "github.com/jdxyw/generativeart"
+ "github.com/jdxyw/generativeart/arts"
+ "github.com/jdxyw/generativeart/common"
+ "image/color"
+ "math/rand"
+ "time"
+)
+
+func cmap(r, m1, m2 float64) color.RGBA {
+ rgb := color.RGBA{
+ R: uint8(common.Constrain(m1*200*r, 0, 255)),
+ G: uint8(common.Constrain(r*200, 0, 255)),
+ B: uint8(common.Constrain(m2*255*r, 70, 255)),
+ A: 255,
+ }
+ return rgb
+}
+
+func main() {
+ rand.Seed(time.Now().Unix())
+ c := generativeart.NewCanva(500, 500)
+ c.SetBackground(common.Black)
+ c.FillBackground()
+ d := arts.NewDomainWrap(0.01, 4, 4, 20, cmap)
+ d.SetDynamicParameter(0.005, 0, 100, "./temp")
+ c.Draw(d)
+}

generativeart.go

@@ -131,6 +131,10 @@ func (c *Canva) Draw(e Engine) {
  e.Generative(c)
 }
 
+func (c *Canva) DrawTimelapse(e Engine) {
+
+}
+
 // FillBackground fills the background of the Canva.
 func (c *Canva) FillBackground() {
  draw.Draw(c.Img(), c.Img().Bounds(), &image.Uniform{c.Opts().Background()}, image.ZP, draw.Src)
@@ -142,12 +146,12 @@ func (c *Canva) ToPNG(fpath string) error {
  if err != nil {
  return err
  }
- if err := png.Encode(f, c.Img()); err != nil {
+ if err = png.Encode(f, c.Img()); err != nil {
  f.Close()
  return err
  }
 
- if err := f.Close(); err != nil {
+ if err = f.Close(); err != nil {
  return err
  }