source("/home/user/documents/R_scripts/analysis_functions.R")
library("data.table")
library("zoo")
library("remotes")
library("ggplot2")
library("gridExtra")

#'Example composition of the video file path.
Folder="/home/user/documents/DLC_results/"
TIteration="175000"
Shuffle="shuffle7_"
Reso="/100_reso"
Prefix="/pose_predictions/PB_T3_"
Middle="_P3_RotatingBeam_"
Network="croppedDLC_resnet50_RotatingBeam3.1Jul12"
#Network="croppeddownsampledDLC_resnet50_RotatingBeam3.1Jul12"
End="_filtered.csv"
PicFolder=paste0(Folder, TIteration, Reso, "/results_statistical_analysis/")

#'At the moment only 20 videos can analyzed in one batch.
Number_of_videos=20

#'Pasting of the video file paths. 
#'All video names should follow the same scheme.
#'Do NOT change the variable names.
T3_3_1_1=paste0(Folder, TIteration, Reso, Prefix, "3_1", Middle, "1", Network, Shuffle, TIteration, End)
T3_3_2_2=paste0(Folder, TIteration, Reso, Prefix, "3_2", Middle, "2", Network, Shuffle, TIteration, End)
T3_4_2_4=paste0(Folder, TIteration, Reso, Prefix, "4_2", Middle, "4", Network, Shuffle, TIteration, End)
T3_4_3_2=paste0(Folder, TIteration, Reso, Prefix, "4_3", Middle, "2", Network, Shuffle, TIteration, End)
T3_4_4_1=paste0(Folder, TIteration, Reso, Prefix, "4_4", Middle, "1", Network, Shuffle, TIteration, End)
T3_4_5_1=paste0(Folder, TIteration, Reso, Prefix, "4_5", Middle, "1", Network, Shuffle, TIteration, End)
T3_5_2_1=paste0(Folder, TIteration, Reso, Prefix, "5_2", Middle, "1", Network, Shuffle, TIteration, End)
T3_6_1_1=paste0(Folder, TIteration, Reso, Prefix, "6_1", Middle, "1", Network, Shuffle, TIteration, End)
T3_7_2_4=paste0(Folder, TIteration, Reso, Prefix, "7_2", Middle, "4", Network, Shuffle, TIteration, End)
T3_8_4_4=paste0(Folder, TIteration, Reso, Prefix, "8_4", Middle, "4", Network, Shuffle, TIteration, End)
T3_9_3_3=paste0(Folder, TIteration, Reso, Prefix, "9_3", Middle, "3", Network, Shuffle, TIteration, End)
T3_9_4_2=paste0(Folder, TIteration, Reso, Prefix, "9_4", Middle, "2", Network, Shuffle, TIteration, End)
T3_10_2_2=paste0(Folder, TIteration, Reso, Prefix, "10_2", Middle, "2", Network, Shuffle, TIteration, End)
T3_10_3_3=paste0(Folder, TIteration, Reso, Prefix, "10_3", Middle, "3", Network, Shuffle, TIteration, End)
T3_12_1_2=paste0(Folder, TIteration, Reso, Prefix, "12_1", Middle, "2", Network, Shuffle, TIteration, End)
T3_13_4_3=paste0(Folder, TIteration, Reso, Prefix, "13_4", Middle, "3", Network, Shuffle, TIteration, End)
T3_14_1_1=paste0(Folder, TIteration, Reso, Prefix, "14_1", Middle, "1", Network, Shuffle, TIteration, End)
T3_14_3_2=paste0(Folder, TIteration, Reso, Prefix, "14_3", Middle, "2", Network, Shuffle, TIteration, End)
T3_16_3_2=paste0(Folder, TIteration, Reso, Prefix, "16_3", Middle, "2", Network, Shuffle, TIteration, End)
T3_17_1_4=paste0(Folder, TIteration, Reso, Prefix, "17_1", Middle, "4", Network, Shuffle, TIteration, End)

#'Video names that appear in the result data frame.
DLCnames3<-c("PB_T3_3_1_P3_RotatingBeam_1", "PB_T3_3_2_P3_RotatingBeam_2", "PB_T3_4_2_P3_RotatingBeam_4", "PB_T3_4_3_P3_RotatingBeam_2", "PB_T3_4_4_P3_RotatingBeam_1", "PB_T3_4_5_P3_RotatingBeam_1", "PB_T3_5_2_P3_RotatingBeam_1", "PB_T3_6_1_P3_RotatingBeam_1", "PB_T3_7_2_P3_RotatingBeam_4", "PB_T3_8_4_P3_RotatingBeam_4", "PB_T3_9_3_P3_RotatingBeam_3", "PB_T3_9_4_P3_RotatingBeam_2", "PB_T3_10_2_P3_RotatingBeam_2", "PB_T3_10_3_P3_RotatingBeam_3", "PB_T3_12_1_P3_RotatingBeam_2", "PB_T3_13_4_P3_RotatingBeam_3", "PB_T3_14_1_P3_RotatingBeam_1", "PB_T3_14_3_P3_RotatingBeam_2", "PB_T3_16_3_P3_RotatingBeam_2", "PB_T3_17_1_P3_RotatingBeam_4")

ResolutionHeight=1080
ResolutionWidth=1920

#'FPS of each video. Same order as above!
DLCFPS<-c(31, 32, 32, 31, 32, 31, 32, 43, 46, 45, 32, 31, 32, 32, 30, 30, 30, 30, 30, 30)


###DO NOT CHANGE ANYTHING FROM THIS POINT ON###

DLCnames2<-c("311", "322", "424", "432", "441", "451", "521", "611", "724", "844", "933", "942", "1022", "1033", "1212", "1343", "1411", "1432", "1632", "1714")
DLCnames4<-c(T3_3_1_1, T3_3_2_2, T3_4_2_4, T3_4_3_2, T3_4_4_1, T3_4_5_1, T3_5_2_1, T3_6_1_1, T3_7_2_4, T3_8_4_4, T3_9_3_3, T3_9_4_2, T3_10_2_2, T3_10_3_3, T3_12_1_2, T3_13_4_3, T3_14_1_1, T3_14_3_2, T3_16_3_2, T3_17_1_4)
for(i in 1:Number_of_videos){
x<-prep.dlc(DLCnames4[i], Height=ResolutionHeight, Width=ResolutionWidth, FPS=DLCFPS[i])
assign(paste("T", as.character(DLCnames2[i]), sep=""), x)
}

DLCnames<-list(T311, T322, T424, T432, T441, T451, T521, T611, T724, T844, 
T933, T942, T1022, T1033, T1212, T1343, T1411, T1432, T1632, T1714)
DLCcolnames<-c("Video", "FPS", "Start", "End", "Reason for End", "Distance", "Speed", "Hindlimb Drops")
Units<-c("Unit", "1/s", "Frame #", "Frame #", " ", "cm", "cm/s", "Frame #", "L/R")
TDLC1<-data.frame()
for(i in DLCcolnames){TDLC1[[i]]<-as.character()}
for(i in 1:ncol(TDLC1)){TDLC1[1, i]=Units[i]}
for(i in 1:length(DLCnames3)){
  TDLC1[i+1,1]=DLCnames3[i]
  TDLC1[i+1,2]=DLCFPS[i]
  TDLC1[i+1,3]=start(DLCnames[[i]])
  TDLC1[i+1, 4]=end(DLCnames[[i]])
  TDLC1[i+1, 5]=rfend(DLCnames[[i]])
  TDLC1[i+1, 6]=distance(DLCnames[[i]],DLCnames[[i]]$forepaw_right_x[end(DLCnames[[i]])+1])
  TDLC1[i+1, 7]=speed(DLCnames[[i]], end(DLCnames[[i]]),time(DLCnames[[i]], start(DLCnames[[i]]), end(DLCnames[[i]])))
  hld<-c()
  hld<-hdrop(DLCnames[[i]], hld)
  if(length(hld)>0){
    for(j in 1:length(hld)){
      if(as.numeric(hld[j])>as.numeric(TDLC1[i+1, 4])){
        next()
        }
        else{
          TDLC1[i+1, 7+j]=as.numeric(hld[j])
        }
      }
   }
}
View(TDLC1)
source("/home/user/Documents/statistical_analysis.R")