R/addTable.R

In davidgohel/R2DOCX: Create MS docx documents from R

# Method Docx.addTable 
# 
# Author: David GOHEL <david.gohel@lysis-consultants.fr>
# Date: 1 avr. 2013
# Version: 0.1
###############################################################################

setMethod("addTable", "Docx", function(x 
				, data
				, formats
				, header.labels
				, grouped.cols=list()
				, span.columns = character(0)
				, col.types
				, col.colors
				, col.fontcolors
				, bookmark
				, ...) {
			
			#------ controls
			
			check.args = list()
			
			if( !missing( data ) )
				check.args$data = data
			
			if( missing( formats ) ) 
				formats = get.default.formats()
			check.args$formats = formats
			
			if( !missing( header.labels ) )
				check.args$header.labels = header.labels
			if( !missing( grouped.cols ) )
				check.args$grouped.cols = grouped.cols
			if( !missing( span.columns ) )
				check.args$span.columns = span.columns
			if( !missing( col.types ) )
				check.args$col.types = col.types
			if( !missing( col.colors ) )
				check.args$col.colors = col.colors
			if( !missing( col.fontcolors ) )
				check.args$col.fontcolors = col.fontcolors
			
			args = do.call( addTable.check.arg, check.args )
			
			.jformats.object = .jinitialize.TableFormat( args$formats )
			
			obj = .jnew("com/lysis/docx4r/elements/DataTable", .jformats.object  )

			for( j in names( args$data ) ){
				if( is.factor(args$data[, j] ) ) tempdata = as.character( args$data[, j] )
				else tempdata = args$data[, j]
				
				if( args$col.types[[j]] == "percent" )
					.jcall( obj , "V", "setPercentData", j, args$header.labels[[j]], .jarray(tempdata*100) )
				else if( args$col.types[[j]] == "double" )
					.jcall( obj , "V", "setData", j, args$header.labels[[j]], .jarray( as.double(tempdata)) )
				else if( args$col.types[[j]] == "integer" )
					.jcall( obj , "V", "setData", j, args$header.labels[[j]], .jarray( as.integer(tempdata)) )
				else if( args$col.types[[j]] == "character" )
					.jcall( obj , "V", "setData", j, args$header.labels[[j]], .jarray( as.character(tempdata)) )
				else if( args$col.types[[j]] == "date" )
					.jcall( obj , "V", "setData", j, args$header.labels[[j]], .jarray( format( tempdata, "%Y-%m-%d" ) ) )
				else if( args$col.types[[j]] == "datetime" )
					.jcall( obj , "V", "setData", j, args$header.labels[[j]], .jarray( format( tempdata, "%Y-%m-%d " ) ) )
			}#TODO: manque les dates
			
			if( length( args$grouped.cols ) > 0 ){
				for( j in names( args$grouped.cols ) ){ 
					.jcall( obj , "V", "setGroupedCols", j, .jarray( args$grouped.cols[[j]] ) )
				}
			}
			
			if( length( args$col.colors ) > 0 ){
				for( j in names( args$col.colors ) ){ 
					.jcall( obj , "V", "setFillColors", j, .jarray( args$col.colors[[j]] ) )
				}
			}
			
			if( length( args$col.fontcolors ) > 0 ){
				for( j in names( args$col.fontcolors ) ){ 
					.jcall( obj , "V", "setFontColors", j, .jarray( args$col.fontcolors[[j]] ) )
				}
			}
			
			for(j in args$span.columns ){
				 instructions = list()
				 current.col = args$data[, j]
				 groups = cumsum( c(TRUE, current.col[-length(current.col)] != current.col[-1] ) )
				 groups.counts = tapply( groups, groups, length )
				 for(i in 1:length( groups.counts )){
				   if( groups.counts[i] == 1 ) instructions[[i]] = 0
				   else {
				     instructions[[i]] = c(1 , rep(2, groups.counts[i]-1 ) )
				   }
				 }
				.jcall( obj , "V", "setMergeInstructions", j, .jarray( as.integer( unlist(instructions) ) ) )
			}
			if( missing( bookmark ) )
				.jcall( x@obj, "V", "add", obj )
			else .jcall( x@obj, "V", "insert", bookmark, obj )
			x
		})