dataset only process audio

1 files changed, 4 insertions, 0 deletions

diff --git a/datasets/dataset.pl b/datasets/dataset.pl
index 4e3b9a9..8b77af9 100755
--- a/datasets/dataset.pl
+++ b/datasets/dataset.pl
@@ -20,6 +20,10 @@ FMT
 sub process($) {
   my $filename = shift or die "Usage: $0 [...filenames]\n";
   my ($name, $ext) = split(/\./, $filename, 2);
+  if ($ext !~ /(wav|mp3|aiff?|flac)/) {
+    print "not a valid file extension: $ext\n";
+    return;
+  }
 
   my $soxi = `soxi $filename`;
   my @lines = split("\n", $soxi);