fixing sequence script

1 files changed, 18 insertions, 8 deletions

diff --git a/splice.pl b/splice.pl
index 3a63180..50fc170 100644
--- a/splice.pl
+++ b/splice.pl
@@ -49,14 +49,24 @@ mkdir("./datasets/$dataset/test_A");
 mkdir("./datasets/$dataset/test_B");
 mkdir("./sequences/$dataset");
 
-my $dataset_lines = `ls -1v sequences/$dataset/*.png`;
-my @dataset_files = split("\n", $dataset_lines);
+sub read_dir {
+  my $path = shift;
+  my $files = [];
+  opendir my $dir, $path or die "Cannot open directory: $!";
+  while (my $file = readdir($dir)) {
+    if ($file =~ /\.png$/) {
+      push @$files, $file;
+    }
+  }
+  closedir $dir;
+  return $files;
+}
 
-my $sequence_lines = `ls -1v sequences/$sequence/*.png`;
-my @sequence_files = split('\n', $sequence_lines);
+my $sequence_files = read_dir("sequences/$sequence");
+my $dataset_files = read_dir("sequences/$dataset");
 
-print "Sequence count: " . (scalar @sequence_files) . "\n";
-print "Dataset count: " . (scalar @dataset_files) . "\n";
+print "Sequence count: " . (scalar @$sequence_files) . "\n";
+print "Dataset count: " . (scalar @$dataset_files) . "\n";
 
 my $pwd = `pwd`;
 chomp $pwd;
@@ -64,9 +74,9 @@ $pwd .= '/';
 print $pwd . "\n";
 
 my $sequence_i = 0;
-my $dataset_i = scalar @dataset_files;
+my $dataset_i = scalar @$dataset_files;
 my $last_fn = undef;
-for my $filename (@sequence_files) {
+for my $filename (sort @$sequence_files) {
   if ($start_frame <= $sequence_i and $sequence_i <= $end_frame) {
     chomp $filename;
     # print($pwd . $filename. "\n");