R/fractus-helper.R

In btmonier/fractus: Fractal Generation Tool

#------------------------------------------------------------------------------
# Title:         fractus.R - Helper Functions                                  
# Author:        Brandon Monier (brandon.monier@sdstate.edu) 
# Created:       2017-05-15 00:00:01 CDT 
# Last Modified: 2018-05-21 19:49:51 CDT                                
#------------------------------------------------------------------------------

## Return data frame
.chaos.df <- function(n, r, shape, c.vert = TRUE) {
	message('Generating index...')
	
	
	s <- c(0.5, 0)
	
	ls1 <- list() 
	for (i in 1:n) {
		if (shape == 'triangle') {
			ls1[[i]] <- .dice.tern()
		}
		if (shape == 'square') {
			ls1[[i]] <- .dice.four()
		}
		if (shape == 'pentagon') {
			ls1[[i]] <- .dice.five()
		}
	}
	
	tmp <- data.table::rbindlist(ls1)
	
	if (isTRUE(c.vert)) {
		tmp <- tmp
	}
	else {
		tmp <- tmp[with(
			tmp, c(TRUE, diff(as.numeric(interaction(x, y))) != 0)
		), ]
	}
	
	pb <- utils::txtProgressBar(
		min = 0, 
		max = nrow(tmp), 
		style = 3, 
		width = 40
	)
	
	ls2 <- list()
	for (i in 2:nrow(tmp)) {
		ls2[[1]] <- data.frame(
			x = r * sum(c(s[[1]], tmp[1][[1, 1]])), 
			y = r * sum(c(s[[2]], tmp[1][[1, 2]]))
		)
		
		ls2[[i]] <- data.frame(
			x = r * sum(c(tmp[i][[1, 1]], ls2[[i - 1]][1, 1])),
			y = r * sum(c(tmp[i][[1, 2]], ls2[[i - 1]][1, 2]))
		)
		utils::setTxtProgressBar(pb, i)
	}
	
	message('Wrapping things up...')
	
	tmp <- data.table::rbindlist(ls2)
	
	message('Done!\n')
	return(tmp)
}



## 'Dice'
.dice.tern <- function() 
{
	alpha <- c(0, 0)
	beta  <- c(0.5, sqrt(3)/2)
	gamma <- c(1, 0)
	
	tmp <- sample(1:3, 1)
	
	if(tmp == 1) {
		return(data.frame(x = alpha[1], y = alpha[2]))
	} 
	
	if(tmp == 2) {
		return(data.frame(x = beta[1], y = beta[2]))
	}
	
	if(tmp == 3) {
		return(data.frame(x = gamma[1], y = gamma[2]))
	}
}


.dice.four <- function() 
{
	a <- c(0, 0)
	b <- c(0, 1)
	c <- c(1, 1)
	d <- c(1, 0)
	
	tmp <- sample(1:4, 1)
	
	if(tmp == 1) {
		return(data.frame(x = a[1], y = a[2]))
	} 
	
	if(tmp == 2) {
		return(data.frame(x = b[1], y = b[2]))
	}
	
	if(tmp == 3) {
		return(data.frame(x = c[1], y = c[2]))
	}
	
	if(tmp == 4) {
		return(data.frame(x = d[1], y = d[2]))
	}
}


.dice.five <- function() 
{
	a <- c(0.5, 0)
	b <- c(1.5, 0)
	c <- c(1.8122, 0.951)
	d <- c(1, 1.539)
	e <- c(0.1878, 0.951)
	
	tmp <- sample(1:5, 1)
	
	if(tmp == 1) {
		return(data.frame(x = a[1], y = a[2]))
	} 
	
	if(tmp == 2) {
		return(data.frame(x = b[1], y = b[2]))
	}
	
	if(tmp == 3) {
		return(data.frame(x = c[1], y = c[2]))
	}
	
	if(tmp == 4) {
		return(data.frame(x = d[1], y = d[2]))
	}
	
	if(tmp == 5) {
		return(data.frame(x = e[1], y = e[2]))
	}
}