TreatmentData: Treatment Plots

# Create some data for testing purposes
patientData = data.frame(
  patid = paste0("PT", 1:20), 
  treatment = c(rep("Placebo",10), rep("Treatment", 10)), 
  zeroDate = as.Date(rep(15000,20), origin = "1970-01-01"), stringsAsFactors = FALSE)

# End of treatment event
eventData = data.frame(
  patid = paste0("PT", 1:20), 
  EOT = as.Date(floor(runif(20,15040,15050)), origin = "1970-01-01", stringsAsFactors = FALSE)
)

# treatment X
treatmentData = data.frame(
  patid = paste0("PT", 1:20), 
  SOT1 = as.Date(floor(runif(20,15005,15010)), origin = "1970-01-01"),
  EOT1 = as.Date(floor(runif(20,15011,15015)), origin = "1970-01-01"),
  SOT2 = as.Date(floor(runif(20,15020,15025)), origin = "1970-01-01"),
  EOT2 = eventData$EOT,
  stringsAsFactors = FALSE)

# pauses to treatment x
treatmentPauses = data.frame(
  patid = sample(paste0("PT", 1:20), size = 10),
  SOT = as.Date(floor(runif(20,15030,15033)), origin = "1970-01-01"),
  EOT = as.Date(floor(runif(20,15034,15038)), origin = "1970-01-01"),
  stringsAsFactors = FALSE)


# treatment y
treatmentData2 = data.frame(
  patid = paste0("PT", 1:20), 
  SOT1 = as.Date(floor(runif(20,15005,15010)), origin = "1970-01-01"),
  EOT1 = as.Date(floor(runif(20,15011,15015)), origin = "1970-01-01"),
  SOT2 = as.Date(floor(runif(20,15020,15025)), origin = "1970-01-01"),
  EOT2 = eventData$EOT
  , stringsAsFactors = FALSE)


# treatment z, these are single day events with two possible doses
n = 300
treatmentDataSingle = data.frame(
  patid = paste0("PT", sample(1:20,size = n,replace = TRUE)), 
  treatmentTime = as.Date(floor(runif(n,15005,15040)), origin = "1970-01-01"),
  dose = sample(c(50,100), size = n, replace = TRUE),
  stringsAsFactors = FALSE
)

##########################################################################
# Create some EventDate objects for single date events
# Create some EventDuration objects for events over a number of days

# treatment x
tDuration = EventDuration(
  data = treatmentData,
  startDateColumns = c("SOT1","SOT2"),
  endDateColumns = c("EOT1", "EOT2"),
  label = "Treatment X",
  ypos = c(1/3,2/3),
  col = "blue"
)

# pause in treatment x, use white to clear this area of plot
tDurationPause = EventDuration(
  data = treatmentPauses,
  startDateColumns = c("SOT"),
  endDateColumns = c("EOT"),
  label = "Treatment X",
  ypos = c(1/3,2/3),
  col = "white"
)

# treatment y
tDuration2 = EventDuration(
  data = treatmentData2,
  startDateColumns = c("SOT1","SOT2"),
  endDateColumns = c("EOT1", "EOT2"),
  label = "Treatment Y",
  ypos = c(0,1/3),
  col = "orange"
)



library(sqldf)
# treatment z, dose 50mg
singleData = EventDate(
  sqldf("select * from treatmentDataSingle where dose == 50"),
  columns = "treatmentTime",
  label = "treatment Z (50mg)",
  col = "green",
  lwd = 3
)

# treatment z dose 100mg
singleData2 = EventDate(
  sqldf("select * from treatmentDataSingle where dose == 100"),
  columns = "treatmentTime",
  label = "treatment Z (100mg)",
  col = "green4",
  lwd = 3
)

# end of treatment event
EOTevent = EventDate(
  data = eventData,
  columns = "EOT",
  label = "End of treatment",
  col = "red"
)

##########################################################################
# The base plot just needs the patientKey, treatment and zeroDate

# Creat the the reference object
treatmentPlot = TreatmentPlot(patientData, "patid", "treatment","zeroDate")

# Prepare the plot area.
treatmentPlot$CreatePlot(xlim = c(0,50), numberRowsLegend = 2)
# Note: this function uses screens. It may be added to a current screen in a
# multiscreen plot. Unitentionally leaving screens open may cause problems.
# see ?split.screen for more details on screens

# add the event duration elements for treatments x and y
treatmentPlot$AddEventDuration(tDuration)
treatmentPlot$AddEventDuration(tDurationPause)
treatmentPlot$AddEventDuration(tDuration2)

# add the events for treatment z at both doses
treatmentPlot$AddEvent(singleData, ypos = c(2/3,1))
treatmentPlot$AddEvent(singleData2, ypos = c(2/3,1))

# add the end of treatment event
treatmentPlot$AddEvent(EOTevent)

EOTevent@data$label = "EOT"
treatmentPlot$AddEventLabel(EOTevent,"label")
# The following functions tidy up the plot
treatmentPlot$AddGrid(xDistance = 21, xSubDistance = 7)
treatmentPlot$AddTreatmentYAxis()
treatmentPlot$AddPatientKey()
treatmentPlot$AddXAxis()
treatmentPlot$AddXLabel("Time from Registration (days)")

# Add the legend
treatmentPlot$AddLegend(itemsOnRow = 3, xoffset = 0.05)
# legend items are stored everytime an event is added so we
# don't need to add extras here

# close the screens used by treatmentPlot
treatmentPlot$CloseScreens()
# Close all screens (not necesary for the example)
close.screen(all.screens = TRUE)