annotate cluster.tools/cluster.2.centroid.R @ 2:b442996b66ae draft

Uploaded
author peter-waltman
date Wed, 27 Feb 2013 20:17:04 -0500
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
2
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
1 #!/usr/bin/env Rscript
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
2 argspec <- c("tab.2.cdt.R converts a data matrix to cdt format
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
3
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
4 Usage:
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
5
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
6 Optional:
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
7
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
8 \n\n")
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
9 args <- commandArgs(TRUE)
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
10 if ( length( args ) == 1 && args =="--help") {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
11 write(argspec, stderr())
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
12 q();
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
13 }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
14
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
15 lib.load.quiet <- function( package ) {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
16 package <- as.character(substitute(package))
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
17 suppressPackageStartupMessages( do.call( "library", list( package=package ) ) )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
18 }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
19
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
20 lib.load.quiet( getopt )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
21 lib.load.quiet( amap )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
22 if ( any( c( 'flashClust', 'fastcluster' ) %in% installed.packages() ) ) {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
23 if ( 'flashClust' %in% installed.packages() ) {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
24 lib.load.quiet( flashClust )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
25 } else {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
26 if ( 'fastcluster' %in% installed.packages() ) {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
27 lib.load.quiet( fastcluster )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
28 }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
29 }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
30 }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
31
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
32 spec <- matrix( c( "dataset", "d", 1, "character",
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
33 "gen.new.dgram", "g", 2, "character",
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
34 "output.fname", "o", 2, "character"
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
35 ),
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
36 nc=4,
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
37 byrow=TRUE
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
38 )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
39
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
40
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
41 opt <- getopt( spec=spec )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
42 if ( is.null( opt$output.report.dir ) ) { opt$output.report.dir <- "report" }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
43 if ( is.null( opt$output.fname ) ) { opt$output.fname <- file.path( opt$output.report.dir, paste( "data", opt$output.format, sep="." ) ) }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
44 if ( is.null( opt$gen.new.dgram ) ) {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
45 opt$gen.new.dgram <- FALSE
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
46 } else {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
47 if ( ! opt$gen.new.dgram %in% c( "no", "yes" ) ) {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
48 stop( "invalid input to gen.new.dgram param", opt$gen.new.dgram, "\n" )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
49 }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
50 ## set to TRUE/FALSE
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
51 opt$gen.new.dgram <- ( opt$gen.new.dgram == "yes" )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
52 }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
53
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
54
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
55 load( opt$dataset ) ## should load the cl, treecl.res (or partcl.res) and data
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
56
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
57 if ( ! exists( 'data' ) ) stop( "No data object in the rdata file provided for", opt$output.format, "format!!\n" )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
58 if ( inherits( data, "dist" ) ) stop( "data provided is a distance matrix - not a data matrix. Can't generate TreeView or Tab-delimited files w/distance matrices!\n" )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
59
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
60 ## the rest of this is for the remaining output formats
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
61 ## pre-set the cluster results for rows & cols to NULL
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
62 direction <- NULL
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
63 if ( exists( 'treecl.res' ) ) {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
64 cl.res <- treecl.res
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
65 if ( is.null( treecl.res$dist.method ) ) treecl.res$dist.method <- 'euclidean' # just set it to some stub so that the ctc fn's don't complain
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
66 } else {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
67 if ( exists( 'partcl.res' ) ) {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
68 cl.res <- partcl.res
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
69 }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
70 else {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
71 stop( 'could not find a valid cluster result to use for primary direction\n' )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
72 }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
73 }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
74
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
75 if ( all( names( cl ) %in% rownames( data ) ) ) {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
76 direction <- "rows"
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
77 } else if ( all( names( cl ) %in% colnames( data ) ) ) {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
78 direction <- "cols"
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
79 data <- t( data )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
80 } else {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
81 stop( "Specified cluster result does not come from this data set\n" )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
82 }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
83
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
84
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
85 centroids <- NULL
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
86 cl <- sort( cl )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
87 if ( inherits( cl.res, "kmeans" ) ) {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
88 ## already comes pre-calculated for us!!
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
89 centroids <- cl.res$centers
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
90 } else {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
91 data <- data[ names( cl ), ]
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
92 cl.list <- unique( cl )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
93 cl.list <- lapply( cl.list, function(i) cl[ cl %in% i ] )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
94 centroids <- sapply( cl.list,
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
95 function(x) {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
96 return( apply( data[ names(x), ], 2, mean, na.rm=T ) )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
97 }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
98 )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
99 centroids <- t( centroids ) ## get them back to the same number of columns that data has now
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
100 }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
101
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
102 data <- centroids
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
103 colnames( data ) <- sapply( 1:max( cl ), function(i) sprintf( "cluster-%02d", i ) )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
104
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
105 if ( opt$gen.new.dgram ) {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
106 distance <- 'euclidean'
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
107 if ( inherits( cl.res, 'hclust' ) ) {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
108 distance <- cl.res$dist.method
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
109 }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
110 dmat <- Dist( data, distance )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
111 treecl.res <- hclust( dmat )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
112 cl <- cutree( treecl.res )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
113 }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
114
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
115 if ( direction == "cols" ) {
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
116 data <- t( data )
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
117 }
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
118
b442996b66ae Uploaded
peter-waltman
parents:
diff changeset
119 save( file=opt$output.name, treecl.res, cl, data )