# Definice funkce ---------------------------------------------------------

vyska <- function(interval = interval) {
  ## našíst data
  dat <- as.data.frame(gsheet::gsheet2tbl("https://docs.google.com/spreadsheets/d/1b0OLM9XrvLvxEsw1IizhEgRBTQ8SY3Xg1ZjX8t15Nw0/edit?usp=sharing"))
  names(dat) <- c("time", "sex", "vyska", "sum")
  
  ## regrese
  lm_all <- lm(vyska ~ sum, dat)
  lm_muz <- lm(vyska ~ sum, dat[dat$sex == "muž", ])
  lm_zena <- lm(vyska ~ sum, dat[dat$sex == "žena", ])
  
  ## graf pro všechny
  plot(jitter(dat$sum), jitter(dat$vyska), col = "black", 
       xlim = c(min(dat$sum)-1, max(dat$sum)+1), 
       ylim = c(min(dat$vyska)-5, max(dat$vyska)+5), 
       xlab = "hrubé skóre", ylab = "výška v centimetrech", main = "Celý vzorek", las=1)
  lines(lm_all, col = Pal()[3])
  text(x = min(dat$sum)-1, y = max(dat$vyska)+5, adj = c(0,1), 
       paste0("Korelace celého vzorku: r = ", round(cor(dat$sum, dat$vyska), 2)))
  
  ## graf pro ženy
  plot(jitter(dat$sum[dat$sex == "muž"]), jitter(dat$vyska[dat$sex == "muž"]), col = "blue", 
       xlim = c(min(dat$sum)-1, max(dat$sum)+1), 
       ylim = c(min(dat$vyska)-5, max(dat$vyska)+5), 
       xlab = "hrubé skóre", ylab = "výška v centimetrech", main = "Podle pohlaví", las=1)
  points(jitter(dat$sum[dat$sex == "žena"]), jitter(dat$vyska[dat$sex == "žena"]), col = "red")
  lines(lm_muz, col = Pal()[1])
  lines(lm_zena, col = Pal()[2])
  text(x = min(dat$sum)-1, y = max(dat$vyska)+5, adj = c(0,1),
       paste0("Korelace pro muže: r = ", round(cor(dat$sum[dat$sex == "muž"], dat$vyska[dat$sex == "muž"]), 2), "\n",
              "Korelace pro ženy: r = ", round(cor(dat$sum[dat$sex == "žena"], dat$vyska[dat$sex == "žena"]), 2)))
  if(!is.na(.GlobalEnv$interval)) {
    later::later(vyska, .GlobalEnv$interval)
  }
}


# Nakresli graf! ----------------------------------------------------------

library(gsheet)
library(DescTools)
library(later)

interval <- NA ## Pozor, nastavení jiné hodnoty než NA povede k vykreslování grafu až do restartu R session!
windows(15, 8); layout(t(1:2)) ## otevře okno
vyska()

## restart session
.rs.restartR()