splice scirpt working

1 files changed, 29 insertions, 9 deletions

diff --git a/splice.pl b/splice.pl
index 8189cab..dae44da 100644
--- a/splice.pl
+++ b/splice.pl
@@ -29,11 +29,19 @@ if ($dataset eq 'PLACEHOLDER' or $sequence eq 'PLACEHOLDER' || ! -e "sequences/$
   exit(1);
 }
 
-if ($start_frame == 0 or !$end_frame == 0) {
+if ($start_frame == 0 or $end_frame == 0) {
   print "No frames specified\n";
   exit(1);
 }
 
+print "module: $module\n";
+print "dataset: $dataset\n";
+print "sequence: $sequence\n";
+print "folder_id: $folder_id\n";
+print "start_frame: $start_frame\n";
+print "end_frame: $end_frame\n";
+print "endpoint: $endpoint\n";
+
 mkdir("./datasets/$dataset");
 mkdir("./datasets/$dataset/train_A");
 mkdir("./datasets/$dataset/train_B");
@@ -41,31 +49,43 @@ mkdir("./datasets/$dataset/test_A");
 mkdir("./datasets/$dataset/test_B");
 mkdir("./sequences/$dataset");
 
-my $dataset_lines = `ls -1v sequences/$sequence/*.png`;
+my $dataset_lines = `ls -1v sequences/$dataset/*.png`;
 my @dataset_files = split("\n", $dataset_lines);
 
 my $sequence_lines = `ls -1v sequences/$sequence/*.png`;
 my @sequence_files = split('\n', $sequence_lines);
 
+print "Sequence count: " . (scalar @sequence_files) . "\n";
+print "Dataset count: " . (scalar @dataset_files) . "\n";
+
 my $pwd = `pwd`;
 chomp $pwd;
 $pwd .= '/';
 print $pwd . "\n";
 
-my $i = scalar @dataset_files;
+my $sequence_i = 0;
+my $dataset_i = scalar @dataset_files;
 my $last_fn = undef;
 for my $filename (@sequence_files) {
-  if ($start_frame <= $i and $i <= $end_frame) {
+  if ($start_frame <= $sequence_i and $sequence_i <= $end_frame) {
     chomp $filename;
-    print($pwd . $filename);
-    system('ln', '-s', $pwd . $filename, sprintf('./datasets/$dataset/frame_%05d.png', $i));
+    print($pwd . $filename. "\n");
+    system('ln', '-s', $pwd . $filename, sprintf('./sequences/%s/frame_%05d.png', $dataset, $dataset_i));
     if (defined $last_fn) {
-      system('ln', '-s', $pwd . $last_fn, sprintf('./datasets/$dataset/frame_%05d.png', $i));
-      system('ln', '-s', $pwd . $filename, sprintf('./datasets/$dataset/frame_%05d.png', $i));
+      my $a_fn = sprintf('./datasets/%s/train_A/%s_%05d.png', $dataset, $sequence, $sequence_i);
+      my $b_fn = sprintf('./datasets/%s/train_B/%s_%05d.png', $dataset, $sequence, $sequence_i);
+      if (! -e $a_fn && ! -e $b_fn) {
+        system('ln', '-s', $pwd . $last_fn, $a_fn);
+        system('ln', '-s', $pwd . $filename, $b_fn);
+      }
     }
     $last_fn = $filename;
-    $i += 1;
+    $dataset_i += 1;
+  }
+  if ($end_frame < $sequence_i) {
+    break;
   }
+  $sequence_i += 1;
 }
 
 if (defined $endpoint) {