GIF Frames for Contact and Misses

#################################################################################
# Generate images for a GIF for the movement of the pitch based on contact/miss #
#################################################################################


#############
# LIBRARIES #
#############

library(RODBC);
library(fields);
library(diagram);
library(calibrate);

####################
# ADDITIONAL FILES #
####################

source("Movement Scripts/Mvt_Fcns.r");

########
# MAIN #
########

# Set the pitcher name
first <- 'Julio';
last <- 'Urias';

# Set the pitch type based on the two-letter designation
pitch_type <- 'CU';

# Set the handedness of the batter
stand <- 'R';

# Set the year
year <- '2016';

# Set up the database name
database <- paste("MLB",year,sep="");

#Set the number of frames
Ny <- 30;

# Set the increment in the y-direction
y_seq <- seq(55,17/12,length=Ny);

# Set the movement type
# 1 <- PITCHf/x movement
# 2 <- PITCHf/x movement without drag
# 3 <- planar movement
# 4 <- planar movement without drag
mvt_type <- 1;

# Set the descriptor based on the type of movement
mvt_desc <- switch(mvt_type,'PFX','PFX D','W','W D');

# Set the quadrilateral core of the strike zone
if (stand == 'L'){
    UL <- c(-0.47,2.85);
    UR <- c(0.22,2.87);
    LL <- c(-0.5,2.04);
    LR <- c(0.24,2.16);
} else {
    UL <- c(-0.38,2.88);
    UR <- c(0.35,2.81);
    LL <- c(-0.46,2.1);
    LR <- c(0.35,2.09);
}

# Associate the pitch type with its full name
pitch_name <- switch(pitch_type,
       FA = 'Fastballs',
       FF = 'Four-Seam FBs',
       FT = 'Two-Seam FBs',
       FC = 'Cut Fastballs',
       FS = 'Split-Finger FBs',
       FO = 'Forkballs',
       SI = 'Sinkers',
       SL = 'Sliders',
       CU = 'Curveballs',
       KC = 'Knuckle Curves',
       EP = 'Ephuuses',
       CH = 'Changeups',
       SC = 'Screwballs',
       KN = 'Knuckleballs',
       UN = 'Unknown Pitches');

# Set the cases where a batter has made contact with a pitch
contact_cases <- "(des = 'In play, no out' OR
                   des = 'In play, out(s)' OR
                   des = 'In play, run(s)' OR
                   des = 'Foul' OR
                   des = 'Foul (Runner Going)' OR
                   des = 'Foul Bunt' OR
                   des = 'Foul Tip')";

# Set the cases where a batter has swung and missed at a pitch
miss_cases <- "(des = 'Swinging Strike' OR
                des = 'Swinging Pitchout' OR
                des = 'Missed Bunt' OR
                des = 'Swinging Strike (Blocked)')";

# Open the channel
channel <- odbcConnect(database);

# Set the query for contact
Contact_Query = paste("SELECT x0, y0, z0, vx0, vy0, vz0, ax, ay, az
                      FROM pitches WHERE pitch_type = '",pitch_type,"' AND
                      ",contact_cases," AND ab_id IN
                      (SELECT ab_id FROM atbats WHERE
                      pitcher = (SELECT eliasid FROM players WHERE first = '",
                      first,"' AND last = '",last,"'))",sep="");

# Download the contact
Contact <- sqlQuery(channel,Contact_Query);

# Close the channel
close(channel);

# Project the contact data
if (mvt_type == 1){
    Contact_xz <- Pitch_Project_PFX(Contact,Ny,stand);
} else if (mvt_type == 2){
    Contact_xz <- Pitch_Project_PFX_Drag(Contact,Ny,stand);
} else if (mvt_type == 3){
    Contact_xz <- Pitch_Project_W(Contact,Ny,stand);
} else {
    Contact_xz <- Pitch_Project_W_Drag(Contact,Ny,stand);
}

# Set the x and z coordinates of the contact projections
x_C <- Contact_xz$x_PL;
z_C <- Contact_xz$z_PL;

# Open the channel
channel <- odbcConnect(database);

# Set the query for misses
Miss_Query = paste("SELECT x0, y0, z0, vx0, vy0, vz0, ax, ay, az
                   FROM pitches WHERE pitch_type = '",pitch_type,"' AND
                   ",miss_cases," AND ab_id IN
                   (SELECT ab_id FROM atbats WHERE
                   pitcher = (SELECT eliasid FROM players WHERE first = '",
                   first,"' AND last = '",last,"'))",sep="");
qu
# Download the data for misses
Miss <- sqlQuery(channel,Miss_Query);

# Close the channel
close(channel);

# Project the contact data
if (mvt_type == 1){
    Miss_xz <- Pitch_Project_PFX(Miss,Ny,stand);
} else if (mvt_type == 2){
    Miss_xz <- Pitch_Project_PFX_Drag(Miss,Ny,stand);
} else if (mvt_type == 3){
    Miss_xz <- Pitch_Project_W(Miss,Ny,stand);
} else {
    Miss_xz <- Pitch_Project_W_Drag(Miss,Ny,stand);
}

# Set the x and z coordinates of the contact projections
x_M <- Miss_xz$x_PL;
z_M <- Miss_xz$z_PL;

# Find the ball/strike indices
Contact_Ball_Str <- ball_strike_prob(Contact,stand);
Miss_Ball_Str <- ball_strike_prob(Miss,stand);

# Parse the indices
C_Str_ind <- Contact_Ball_Str$S;
C_Ball_ind <- Contact_Ball_Str$B;
M_Str_ind <- Miss_Ball_Str$S;
M_Ball_ind <- Miss_Ball_Str$B;

# Set up an array to draw a strike zone contour
dx <- 0.005;
X <- seq(-3+dx/2,3-dx/2,length=1200);
Y <- seq(dx/2,6-dx/2,length=1200);
k_sz <- 4;
Str_Prb_Array <- array(0,dim=c(1200,1200));
for (i in 1:1200) {
    for (j in 1:1200){
        P <- c(X[i],Y[j]);
        d_sz <- Min_Dist_Quad(P,UL,UR,LL,LR);
        Str_Prb_Array[i,j] <- exp(-k_sz*d_sz^4);
    }
}

# Lower the opacity
red_white <- intpalette(c('white','red'),n=11);
orange_white <- intpalette(c('white','darkorange'),n=11);
green_white <- intpalette(c('white','green'),n=11);
blue_white <- intpalette(c('white','blue'),n=11);
fade_ind <- 3;
no_s <- 0.1;

# Generate the images
for (j in 1:Ny){
    if (j < 10){
        filename <- paste(last,"_",pitch_type,"_CM_",stand,"_0",j,".jpg",sep="");
    }
    else {
        filename <- paste(last,"_",pitch_type,"_CM_",stand,"_",j,".jpg",sep="");
    }
    jpeg(filename);
    plot(x_C[C_Str_ind,j],z_C[C_Str_ind,j],pch=20,col="green",xlim=c(-3,3),ylim=c(0,6),
         xlab='Horizontal Location (Feet)',ylab='Vertical Location (Feet)',
         main=paste(first," ",last,": ",pitch_name," v. ",stand,"HB at ",round(y_seq[j],digits=2)," Feet (",mvt_desc,")",sep=""),
         xaxs='i',yaxs='i');
    par(new=TRUE);
    plot(x_C[C_Ball_ind,j],z_C[C_Ball_ind,j],pch=20,col="blue",xlim=c(-3,3),ylim=c(0,6),
         xlab='',ylab='',xaxs='i',yaxs='i');
    par(new=TRUE);
    plot(x_M[M_Str_ind,j],z_M[M_Str_ind,j],pch=20,col="red",xlim=c(-3,3),ylim=c(0,6),
         xlab='',ylab='',xaxs='i',yaxs='i');
    par(new=TRUE);
    plot(x_M[M_Ball_ind,j],z_M[M_Ball_ind,j],pch=20,col="darkorange",xlim=c(-3,3),ylim=c(0,6),
         xlab='',ylab='',xaxs='i',yaxs='i');
    par(new=TRUE);
    contour(X,Y,Str_Prb_Array,levels=c(0.5),drawlabels=FALSE,col='black',xlim=c(-3,3),ylim=c(0,6),xaxs='i',yaxs='i',lwd=2);
    par(new=FALSE);
    dev.off();
}