package xdelta


import (
	"io"
	"math/rand"
)

const (
	blocksize = 16380
)

func WriteRstreams(t *TestGroup, seed, offset, len int64,
	src, tgt io.WriteCloser) {
	t.Go("src", func (g Goroutine) {
		go writeOne(g, seed, 0, len, src)
	})
	t.Go("tgt", func (g Goroutine) {
		go writeOne(g, seed, offset, len, tgt)
	})
}

func writeOne(g Goroutine, seed, offset, len int64, stream io.WriteCloser) {
	if offset != 0 {
		// Fill with other random data until the offset
		if err := writeRand(rand.New(rand.NewSource(^seed)), offset, stream); err != nil {
			g.Panic(err)
		}
	}
	if err := writeRand(rand.New(rand.NewSource(seed)),
		len - offset, stream); err != nil {
		g.Panic(err)
	}
	if err := stream.Close(); err != nil {
		g.Panic(err)
	}
	g.OK()
}

func writeRand(r *rand.Rand, len int64, s io.Writer) error {
	blk := make([]byte, blocksize)
	for len > 0 {
		fillRand(r, blk)
		c := blocksize
		if len < blocksize {
			c = int(len)
		}
		if _, err := s.Write(blk[0:c]); err != nil {
			return err
		}
		len -= int64(c)
	}
	return nil
}

func fillRand(r *rand.Rand, blk []byte) {
	for p := 0; p < blocksize; {
		v := r.Int63()
		for i := 7; i != 0; i-- {
			blk[p] = byte(v)
			p++
			v >>= 8
		}
	}
}