1 files changed, 13 insertions, 2 deletions

diff --git a/draklive b/draklive
index b03ca66..876555e 100755
--- a/draklive
+++ b/draklive
@@ -1112,7 +1112,12 @@ sub need_compressed_image {
 
 sub get_disk_image_path {
     my ($live) = @_;
-    get_disk_master_path($live) . if_(need_compressed_image($live), '.bz2');
+    if ($live->{settings}{compression_method} == 'gzip') {
+	get_disk_master_path($live) . if_(need_compressed_image($live), '.gz');
+    }
+    else {
+	get_disk_master_path($live) . if_(need_compressed_image($live), '.bz2');
+    }
 }
 
 sub create_disk_image {
@@ -1124,7 +1129,13 @@ sub create_disk_image {
     my $master = get_disk_master_path($live);
     my $dest = get_disk_image_path($live);
     mkdir_p(dirname($dest));
-    run_('bzip2', '>', $dest, '-f', '-k', '-c', $master);
+
+    if ($live->{settings}{compression_method} == 'gzip') {
+        run_('gzip', '>', $dest, '-f', '-c', '--fast', $master);
+    }
+    else {
+        run_('bzip2', '>', $dest, '-f', '-k', '-c', $master);
+    }
 }
 
 sub create_image {