#!/usr/bin/perl

use strict;

our $images_dir = "/home/ubuntu/Desktop/wood/";
our $thumbs_dir = "/home/ubuntu/Desktop/thumbs/wood/";

our $dt = 1;

mkdir($thumbs_dir);
mkdir($thumbs_dir . "A/");
mkdir($thumbs_dir . "B/");
mkdir($thumbs_dir . "A/train/");
mkdir($thumbs_dir . "B/train/");
mkdir($thumbs_dir . "A/val/");
mkdir($thumbs_dir . "B/val/");
mkdir($thumbs_dir . "A/test/");
mkdir($thumbs_dir . "B/test/");

our @files = ();

opendir DIR, $images_dir;
while (readdir DIR) {
  next if /^\./;
  push(@files, $_);
}
closedir DIR;

our @images = sort @files;

my $count = scalar(@images) - $dt;

my $i; my $x; my $y; my $dir;
for ($i = 0; $i < $count; $i++) {
  my $id = $i;
  if ( ($id % 7) == 3) {
    $dir = "test/";
  } elsif ( ($id % 7) == 6) {
    $dir = "val/";
  } else {
    $dir = "train/";
  }
  if ( $i && ($i % 1000) == 0) {
    print($id . "...\n")
  }

  my $a_frame = $images_dir . $images[$id];
  my $b_frame = $images_dir . $images[$id+$dt];
  my $fn = sprintf("frame_%05d.png", $i);
  #print "$fn $x $y $a_frame\n";
  system("convert", $a_frame, "-resize", '256x256!', '-canny', '0x1+10%+30%', $thumbs_dir . "A/" . $dir . $fn);
  system("convert", $b_frame, "-resize", '256x256!', $thumbs_dir . "B/" . $dir . $fn);
}
print $count . "\n";