# Copyright 2017 Open Networking Foundation (ONF)
#
# Please refer questions to either the onos test mailing list at <onos-test@onosproject.org>,
# the System Testing Plans and Results wiki page at <https://wiki.onosproject.org/x/voMg>,
# or the System Testing Guide page at <https://wiki.onosproject.org/x/WYQg>
#
#     TestON is free software: you can redistribute it and/or modify
#     it under the terms of the GNU General Public License as published by
#     the Free Software Foundation, either version 2 of the License, or
#     (at your option) any later version.
#
#     TestON is distributed in the hope that it will be useful,
#     but WITHOUT ANY WARRANTY; without even the implied warranty of
#     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#     GNU General Public License for more details.
#
#     You should have received a copy of the GNU General Public License
#     along with TestON.  If not, see <http://www.gnu.org/licenses/>.
#
# If you have any questions, or if you don't understand R,
# please contact Jeremy Ronquillo: j_ronquillo@u.pacific.edu

# **********************************************************
# STEP 1: Data management.
# **********************************************************
print( "**********************************************************" )
print( "STEP 1: Data management." )
print( "**********************************************************" )
has_flow_obj = 1
database_host = 2
database_port = 3
database_u_id = 4
database_pw = 5
test_name = 6
branch_name = 7
batch_size = 8
old_flow = 9
save_directory = 10

# Command line arguments are read.
print( "Reading commmand-line args." )
args <- commandArgs( trailingOnly=TRUE )

# ----------------
# Import Libraries
# ----------------

print( "Importing libraries." )
library( ggplot2 )
library( reshape2 )
library( RPostgreSQL )    # For databases

# -------------------
# Check CLI Arguments
# -------------------

print( "Verifying CLI args." )

if ( is.na( args[ save_directory ] ) ){

    print( paste( "Usage: Rscript SCPFIntentInstallWithdrawRerouteLat.R",
                                  "<isFlowObj>" ,
                                  "<database-host>",
                                  "<database-port>",
                                  "<database-user-id>",
                                  "<database-password>",
                                  "<test-name>",
                                  "<branch-name>",
                                  "<batch-size>",
                                  "<using-old-flow>",
                                  "<directory-to-save-graphs>",
                                  sep=" " ) )
    q()  # basically exit(), but in R
}

# -----------------------------------
# Create File Name and Title of Graph
# -----------------------------------

print( "Creating filename and title of graph." )

chartTitle <- "Intent Install, Withdraw, & Reroute Latencies"
flowObjFileModifier <- ""
errBarOutputFile <- paste( args[ save_directory ],
                    "SCPFIntentInstallWithdrawRerouteLat_",
                    args[ branch_name ],
                    sep="" )

if ( args[ has_flow_obj ] == "y" ){
    errBarOutputFile <- paste( errBarOutputFile, "_fobj", sep="" )
    flowObjFileModifier <- "fobj_"
    chartTitle <- paste( chartTitle, "w/ FlowObj" )
}
if ( args[ old_flow ] == "y" ){
    errBarOutputFile <- paste( errBarOutputFile, "_OldFlow", sep="" )
    chartTitle <- paste( chartTitle,
                         "With Old Flow",
                         sep="\n" )
}
errBarOutputFile <- paste( errBarOutputFile,
                           "_",
                           args[ batch_size ],
                           "-batchSize_graph.jpg",
                           sep="" )

chartTitle <- paste( chartTitle,
                     "\nBatch Size =",
                     args[ batch_size ],
                     sep=" " )

# ------------------
# SQL Initialization
# ------------------

print( "Initializing SQL" )

con <- dbConnect( dbDriver( "PostgreSQL" ),
                  dbname = "onostest",
                  host = args[ database_host ],
                  port = strtoi( args[ database_port ] ),
                  user = args[ database_u_id ],
                  password = args[ database_pw ] )

# ---------------------------------------
# Intent Install and Withdraw SQL Command
# ---------------------------------------
print( "Generating Intent Install and Withdraw SQL Command" )

installWithdrawSQLCommand <- paste( "SELECT * FROM intent_latency_",
                                    flowObjFileModifier,
                                    "tests WHERE batch_size=",
                                    args[ batch_size ],
                                    " AND branch = '",
                                    args[ branch_name ],
                                    "' AND date IN ( SELECT MAX( date ) FROM intent_latency_",
                                    flowObjFileModifier,
                                    "tests WHERE branch='",
                                    args[ branch_name ],
                                    "' AND ",
                                    ( if( args[ old_flow ] == 'y' ) "" else "NOT " ) ,
                                    "is_old_flow",
                                    ")",
                                    sep="" )

print( "Sending Intent Install and Withdraw SQL command:" )
print( installWithdrawSQLCommand )
installWithdrawData <- dbGetQuery( con, installWithdrawSQLCommand )

# --------------------------
# Intent Reroute SQL Command
# --------------------------

print( "Generating Intent Reroute SQL Command" )

rerouteSQLCommand <- paste( "SELECT * FROM intent_reroute_latency_",
                            flowObjFileModifier,
                            "tests WHERE batch_size=",
                            args[ batch_size ],
                            " AND branch = '",
                            args[ branch_name ],
                            "' AND date IN ( SELECT MAX( date ) FROM intent_reroute_latency_",
                            flowObjFileModifier,
                            "tests WHERE branch='",
                            args[ branch_name ],
                            "' AND ",
                            ( if( args[ old_flow ] == 'y' ) "" else "NOT " ) ,
                            "is_old_flow",
                            ")",
                            sep="" )

print( "Sending Intent Reroute SQL command:" )
print( rerouteSQLCommand )
rerouteData <- dbGetQuery( con, rerouteSQLCommand )

# **********************************************************
# STEP 2: Organize Data.
# **********************************************************

print( "**********************************************************" )
print( "STEP 2: Organize Data." )
print( "**********************************************************" )

# -------------------------------------------------------
# Combining Install, Withdraw, and Reroute Latencies Data
# -------------------------------------------------------

print( "Combining Install, Withdraw, and Reroute Latencies Data" )

if ( ncol( rerouteData ) == 0 ){  # Checks if rerouteData exists, so we can exclude it if necessary
    avgs <- c( installWithdrawData[ 'install_avg' ],
               installWithdrawData[ 'withdraw_avg' ] )
} else{
    colnames( rerouteData ) <- c( "date",
                                  "name",
                                  "date",
                                  "branch",
                                  "is_old_flow",
                                  "commit",
                                  "scale",
                                  "batch_size",
                                  "reroute_avg",
                                  "reroute_std" )

    avgs <- c( installWithdrawData[ 'install_avg' ],
               installWithdrawData[ 'withdraw_avg' ],
               rerouteData[ 'reroute_avg' ] )
}

# Combine lists into data frames.
dataFrame <- melt( avgs )

# --------------------
# Construct Data Frame
# --------------------

print( "Constructing data frame." )

if ( ncol( rerouteData ) == 0 ){  # Checks if rerouteData exists (due to batch size) for the dataFrame this time
    dataFrame$scale <- c( installWithdrawData$scale,
                          installWithdrawData$scale )

    dataFrame$stds <- c( installWithdrawData$install_std,
                         installWithdrawData$withdraw_std )
} else{
    dataFrame$scale <- c( installWithdrawData$scale,
                          installWithdrawData$scale,
                          rerouteData$scale )

    dataFrame$stds <- c( installWithdrawData$install_std,
                         installWithdrawData$withdraw_std,
                         rerouteData$reroute_std )
}

colnames( dataFrame ) <- c( "ms",
                            "type",
                            "scale",
                            "stds" )

# Format data frame so that the data is in the same order as it appeared in the file.
dataFrame$type <- as.character( dataFrame$type )
dataFrame$type <- factor( dataFrame$type, levels=unique( dataFrame$type ) )

dataFrame <- na.omit( dataFrame ) # Omit any data that doesn't exist

print( "Data Frame Results:" )
print( dataFrame )

# **********************************************************
# STEP 3: Generate graph.
# **********************************************************

print( "**********************************************************" )
print( "STEP 3: Generate Graph." )
print( "**********************************************************" )

# -------------------
# Main Plot Generated
# -------------------

print( "Creating the main plot." )

mainPlot <- ggplot( data = dataFrame, aes( x = scale,
                                           y = ms,
                                           ymin = ms,
                                           ymax = ms + stds,
                                           fill = type ) )

# ------------------------------
# Fundamental Variables Assigned
# ------------------------------

print( "Generating fundamental graph data." )

theme_set( theme_grey( base_size = 22 ) )
barWidth <- 1.3
xScaleConfig <- scale_x_continuous( breaks = c( 1, 3, 5, 7, 9) )
xLabel <- xlab( "Scale" )
yLabel <- ylab( "Latency (ms)" )
fillLabel <- labs( fill="Type" )
title <- ggtitle( chartTitle )
imageWidth <- 15
imageHeight <- 10
imageDPI <- 200
errorBarColor <- rgb( 140, 140, 140, maxColorValue=255 )

theme <- theme( plot.title=element_text( hjust = 0.5, size = 32, face='bold' ),
                legend.position="bottom",
                legend.text=element_text( size=22 ),
                legend.title = element_blank(),
                legend.key.size = unit( 1.5, 'lines' ) )

colors <- scale_fill_manual( values=c( "#F77670",
                                       "#619DFA",
                                       "#18BA48" ) )

# Store plot configurations as 1 variable
fundamentalGraphData <- mainPlot +
                        xScaleConfig +
                        xLabel +
                        yLabel +
                        fillLabel +
                        theme +
                        title +
                        colors

# ---------------------------
# Generating Bar Graph Format
# ---------------------------

print( "Generating bar graph with error bars." )

barGraphFormat <- geom_bar( stat = "identity",
                            width = barWidth,
                            position = "dodge" )

errorBarFormat <- geom_errorbar( width = barWidth,
                                 position = position_dodge( barWidth ),
                                 color = errorBarColor )

values <- geom_text( aes( x = dataFrame$scale,
                          y = dataFrame$ms + 0.035 * max( dataFrame$ms ),
                          label = format( dataFrame$ms,
                                          digits = 3,
                                          big.mark = ",",
                                          scientific = FALSE ) ),
                          position = position_dodge( width = barWidth ),
                          size = 5.5,
                          fontface = "bold" )

wrapLegend <- guides( fill = guide_legend( nrow = 1, byrow = TRUE ) )

result <- fundamentalGraphData +
          barGraphFormat +
          errorBarFormat +
          values +
          wrapLegend

# -----------------------
# Exporting Graph to File
# -----------------------

print( paste( "Saving bar chart with error bars to", errBarOutputFile ) )

ggsave( errBarOutputFile,
        width = imageWidth,
        height = imageHeight,
        dpi = imageDPI )

print( paste( "[SUCCESS] Successfully wrote bar chart with error bars out to", errBarOutputFile ) )