2 # Ikiwiki Sweave-alike plugin
3 # under the terms of the GPL version 2, or any later version at your
5 # Copyright 2012 by Don Armstrong <don@donarmstrong.com>.
8 package IkiWiki::Plugin::sweavealike;
12 sweavealike -- A Sweave-alike plugin which allows for embedding R code in IkiWiki
16 sweavealike allows you to embed R code in IkiWiki.
19 [[!sweavealike code='''
28 =head2 Available options
34 R code to execute. Required to be present. [If you didn't want to
35 execute R code, why would you use this directive?]
39 Echo the R code. [Basically, this escapes the code with >, and then
40 adds spaces so that the output is interpreted as a code fragment.]
44 Suppress all non-figure output
48 Defaults to show the results (what R wrote to stdout) of a particular
53 If present, code is assumed to produce a figure which is automatically
58 Integer width of included figure; defaults to 400
62 Integer height of included figure; defaults to 400
69 R is a complete language, and no attempt is made to control what you
70 can do. Reading and writing arbitrary files, as well as exhausting
71 available memory and CPU are all trivially possible. Thus, you should
72 NEVER use this plugin on a publicly editable IkiWiki instance.
74 You should be able to refer to previously created figures without
86 use Encode qw(decode);
87 use Digest::MD5 qw(md5_hex);
89 my $id = "sweavealike";
91 hook(type => "getsetup", id => $id, call => \&getsetup);
92 hook(type => "preprocess", id => $id, call => \&preprocess);
93 hook(type => "htmlize", id => $id, call => \&htmlize);
94 hook(type => "savestate", id => $id, call => \&savestate);
98 return(plugin => {safe => 0,
101 link => "http://git.donarmstrong.com/?p=ikiwiki_plugins.git;a=blob;f=sweavealike.pm;hb=HEAD",
102 description => "sweavealike plugin",
108 return(md5_hex(map {decode('utf8',$_)} @_));
114 if (not defined $pagestate{$param{page}}{$id}{R}) {
115 $pagestate{$param{page}}{$id}{R} = Statistics::R->new(shared => 1)
116 or error("Unable to create an R process");
118 # we currently don't bother to support anything but outputing the
119 # entire segment of code and its R output
121 if (not exists $param{code}
122 or not defined $param{code}
123 or not length $param{code}) {
124 error("There wasn't any R code supplied");
128 if (exists $param{fig}) {
129 $pagestate{$param{page}}{$id}{fignum}++;
130 $param{width} = '400' unless exists $param{width} and defined $param{width};
131 $param{height} = '400' unless exists $param{height} and defined $param{height};
132 for (qw(width height)) {
133 if ($param{$_} !~ /^\d+$/) {
134 error("invalid $_; must be an integer: $param{$_}");
137 # because even if the code is duplicated, the figure could still be the same.
138 my $md5 = code_md5($param{code},$param{width},$param{height},$pagestate{$param{page}}{$id}{fignum});
139 $image_loc = "$param{page}/${md5}.png";
140 my $image_loc_esc = $image_loc;
141 $image_loc_esc =~ s/"/\\"/g;
142 will_render($param{page},$image_loc);
144 $pagestate{$param{page}}{$id}{R}->run(qq{png(filename="$image_loc_esc",width=$param{width},height=$param{height});});
147 error("code 'png(filename="$image_loc_esc",width=$param{width},height=$param{height});' (from internal figure handling) produced error '$@'");
152 $code_result = $pagestate{$param{page}}{$id}{R}->run($param{code});
155 error("code '$param{code}' produced error '$@'");
159 if (exists $param{fig}) {
161 $pagestate{$param{page}}{$id}{R}->run(qq{dev.off();});
164 error("code 'dev.off()' (from internal figure handling) produced error '$@'");
166 $fig_output = qq(\n<img class="sweavealike" src=").urlto($image_loc,$param{destpage}).qq(" />);
168 if (exists $param{nooutput}) {
169 return($output.$fig_output);
171 if (exists $param{echo}) {
172 $output .= $param{code};
173 $output =~ s/^/> /mg;
176 if (not exists $param{results} or
177 (defined $param{results} and
178 $param{results} !~ /^(hide|false)$/i;)) {
179 $output .= $code_result;
181 if (exists $param{echo} or
182 exists $param{results}) {
188 # stop any started R processes here
191 if (exists $pagestate{$param{page}} and
192 exists $pagestate{$param{page}}{$id} and
193 exists $pagestate{$param{page}}{$id}{R}) {
194 if (defined $pagestate{$param{page}}{$id}{R}
195 and $pagestate{$param{page}}{$id}{R}->is_started()) {
196 $pagestate{$param{page}}{$id}{R}->stop();
198 delete $pagestate{$param{page}}{$id}{R};
203 # make sure we never try to save an R process
204 for my $page (keys %pagestate) {
205 next unless exists $pagestate{$page}{$id};
206 next unless exists $pagestate{$page}{$id}{R};
207 if (defined $pagestate{$page}{$id}{R}
208 and $pagestate{$page}{$id}{R}->is_started()) {
209 $pagestate{$page}{$id}{R}->stop;
211 delete $pagestate{$page}{$id}{R};