#!/usr/bin/perl

use File::Basename;
use File::Copy;
use Data::Dumper;
use Time::localtime;
use Switch;
use POSIX qw(strftime);

## the defs

chomp($CURRENT_DIR  = `pwd`);
chomp($BIN_PATH     = `dirname $0`);
chomp($OS           = `uname -o`);
chomp($CORES        = `ls -d /sys/devices/system/cpu/cpu[[:digit:]]* | wc -w`);

require "$BIN_PATH/ccd_defs.pl";

$BIN_PATH = $BIN_PATH."/bin";

my %objectStats    = {
    count           => 0,
    good            => 0,
    bad             => 0,
    minWidth        => 0,
    minHeight       => 0,
    maxWidth        => 0,
    maxHeight       => 0
					
};

my %jobOptions    = {
    resizeTo        => 0,
    srcDir          => $CURRENT_DIR
};

my %args = {};

$jobOptions{srcDir} = "$CURRENT_DIR";

sub run {
    system($_[0]);
    
    if($? != 0){
        die "\n\nquitting cause: \n\t$_[0]\nreturned with code ".$?."\n";
    }
}
sub now {    
    (@date) = localtime;
    $date[5] += 1900;
    $date[4]++;
    
    print sprintf("%0.2d-%0.2d-%0.4d %0.2d:%0.2d:%0.2d",$date[3],$date[4]++, $date[5], $date[2],$date[1],$date[0]);;
}

sub parseArgs {

 ## defaults
    $args{"--resize-to"}             = "1200";
    
    $args{"--start-with"}            = "resize";
    $args{"--end-with"}              = "pmvs";
    
    $args{"--cmvs-maxImages"}        = 100;
    
    $args{"--pmvs-level"}            = 1;
    $args{"--pmvs-csize"}            = 2;
    $args{"--pmvs-threshold"}        = 0.7;
    $args{"--pmvs-wsize"}            = 7;
    $args{"--pmvs-minImageNum"}       = 3;
    
    for($i = 0; $i <= $#ARGV; $i++) {
        if($ARGV[$i] =~ /^--[^a-z\-]*/){
            $args{"$ARGV[$i]"} = true;
            
            if(!($ARGV[$i+1] =~ /^--[^a-z\-]*/)) {
                if($ARGV[$i] eq "--resize-to"){
                    if($ARGV[$i+1] eq "orig" || $ARGV[$i+1] =~ /^[0-9]*$/){
                        $args{$ARGV[$i]} = $ARGV[$i+1];
                    } else {
                        die "\n invalid parameter for \"".$ARGV[$i]."\": ".$ARGV[$i+1];
                    }
                }
                if($ARGV[$i] eq "--start-with"){
                    if($ARGV[$i+1] eq "resize" || $ARGV[$i+1] eq "getKeypoints" || $ARGV[$i+1] eq "match" || $ARGV[$i+1] eq "bundler" || $ARGV[$i+1] eq "cmvs" || $ARGV[$i+1] eq "pmvs"){
                        $args{$ARGV[$i]} = $ARGV[$i+1];
                    } else {    
                        die "\n invalid parameter for \"".$ARGV[$i]."\": ".$ARGV[$i+1]."\n\t valid values are \"resize\", \"getKeypoints\", \"match\", \"bundler\", \"cmvs\", \"pmvs\"";    
                    }
                }
                if($ARGV[$i] eq "--end-with"){
                    if($ARGV[$i+1] eq "resize" || $ARGV[$i+1] eq "getKeypoints" || $ARGV[$i+1] eq "match" || $ARGV[$i+1] eq "bundler" || $ARGV[$i+1] eq "cmvs" || $ARGV[$i+1] eq "pmvs"){
                        $args{$ARGV[$i]} = $ARGV[$i+1];
                    } else {    
                        die "\n invalid parameter for \"".$ARGV[$i]."\": ".$ARGV[$i+1]."\n\t valid values are \"resize\", \"getKeypoints\", \"match\", \"bundler\", \"cmvs\", \"pmvs\"";
                    }
                }
                if($ARGV[$i] eq "--run-only"){
                    if($ARGV[$i+1] eq "resize" || $ARGV[$i+1] eq "getKeypoints" || $ARGV[$i+1] eq "match" || $ARGV[$i+1] eq "bundler" || $ARGV[$i+1] eq "cmvs" || $ARGV[$i+1] eq "pmvs"){
                        $args{"--start-with"} = $ARGV[$i+1];
                        $args{"--end-with"} = $ARGV[$i+1];
                    } else {    
                        die "\n invalid parameter for \"".$ARGV[$i]."\": ".$ARGV[$i+1]."\n\t valid values are \"resize\", \"getKeypoints\", \"match\", \"bundler\", \"cmvs\", \"pmvs\"";
                    }
                }
                if($ARGV[$i] eq "--cmvs-maxImages"){
                    if($ARGV[$i+1] =~ /^[0-9]*$/){
                        $args{$ARGV[$i]} = $ARGV[$i+1];
                    } else {
                        die "\n invalid parameter for \"".$ARGV[$i]."\": ".$ARGV[$i+1];
                    }
                }
                if($ARGV[$i] eq "--pmvs-level"){
                    if($ARGV[$i+1] =~ /^[0-9]*$/){
                        $args{$ARGV[$i]} = $ARGV[$i+1];
                    } else {
                        die "\n invalid parameter for \"".$ARGV[$i]."\": ".$ARGV[$i+1];
                    }
                }
                if($ARGV[$i] eq "--pmvs-csize"){
                    if($ARGV[$i+1] =~ /^[0-9]*$/){
                        $args{$ARGV[$i]} = $ARGV[$i+1];
                    } else {
                        die "\n invalid parameter for \"".$ARGV[$i]."\": ".$ARGV[$i+1];
                    }
                }
                if($ARGV[$i] eq "--pmvs-threshold"){
                    if($ARGV[$i+1] =~ /^-?[0-9]*\.?[0-9]*$/){
                        $args{$ARGV[$i]} = $ARGV[$i+1];
                    } else {
                        die "\n invalid parameter for \"".$ARGV[$i]."\": ".$ARGV[$i+1];
                    }
                }
                if($ARGV[$i] eq "--pmvs-wsize"){
                    if($ARGV[$i+1] =~ /^[0-9]*$/){
                        $args{$ARGV[$i]} = $ARGV[$i+1];
                    } else {
                        die "\n invalid parameter for \"".$ARGV[$i]."\": ".$ARGV[$i+1];
                    }
                }
                if($ARGV[$i] eq "--pmvs-minImageNum"){
                    if($ARGV[$i+1] =~ /^[0-9]*$/){
                        $args{$ARGV[$i]} = $ARGV[$i+1];
                    } else {
                        die "\n invalid parameter for \"".$ARGV[$i]."\": ".$ARGV[$i+1];
                    }
                }
                
                if($ARGV[$i] eq "--force-focal"){
                    if($ARGV[$i+1] =~ /^[0-9]*\.?[0-9]*$/){
                        $args{$ARGV[$i]} = $ARGV[$i+1];
                    } else {
                        die "\n invalid parameter for \"".$ARGV[$i]."\": ".$ARGV[$i+1];
                    }
                }
                if($ARGV[$i] eq "--force-ccd"){
                    if($ARGV[$i+1] =~ /^[0-9]*\.?[0-9]*$/){
                        $args{$ARGV[$i]} = $ARGV[$i+1];
                    } else {
                        die "\n invalid parameter for \"".$ARGV[$i]."\": ".$ARGV[$i+1];
                    }
                }
            }
        }
    }
    
    if($args{"--help"}){        
        print "\nusgae: run.pl [options]";
        print "\nit should be run from the folder contining the images to which should reconstructed";
        print "\n";
        print "\noptions:";
        print "\n            --help: ";
        print "\n                    prints this screen";
        print "\n";
        
        print "\n       --resize-to: <positive integer|\"orig\">";
        print "\n           default: 1200";
        print "\n                    will resize the images so that the maximum width/height of the images are smaller or equal to the specified number";
        print "\n                    if \"--resize-to orig\" is used it will use the images without resizing";
        print "\n";
        
        print "\n      --start-with: <\"resize\"|\"getKeypoints\"|\"match\"|\"bundler\"|\"cmvs\"|\"pmvs\">";
        print "\n           default: resize";
        print "\n                    will start the sript at the specified step";
        print "\n";
        
        print "\n        --end-with: <\"resize\"|\"getKeypoints\"|\"match\"|\"bundler\"|\"cmvs\"|\"pmvs\">";
        print "\n           default: pmvs";
        print "\n                    will stop the sript after the specified step";
        print "\n";
        
        print "\n        --run-only: <\"resize\"|\"getKeypoints\"|\"match\"|\"bundler\"|\"cmvs\"|\"pmvs\">";
        print "\n                    will only execute the specified step";
        print "\n                    equal to --start-with <step> --end-with <step>";
        print "\n";
        
        print "\n     --force-focal: <positive float>";
        print "\n                    override the focal length information for the images";
        print "\n";
        
        print "\n       --force-ccd: <positive float>";
        print "\n                    override the ccd width information for the images";
        print "\n";
        
        print "\n  --cmvs-maxImages: <positive integer>";
        print "\n           default: 100";
        print "\n                    the maximum number of images per cluster";
        print "\n";
        
        print "\n      --pmvs-level: <positive integer>";
        print "\n           default: 1";
        
        print "\n      --pmvs-csize: <positive integer>";
        print "\n           default: 2";
        
        print "\n  --pmvs-threshold: <float: -1 <= x <= 1>";
        print "\n           default: 0.7";
        
        print "\n      --pmvs-wsize: <positive integer>";
        print "\n           default: 7";
        
        print "\n--pmvs-minImageNum: <positive integer>";
        print "\n           default: 3";
        print "\n                    see http://grail.cs.washington.edu/software/pmvs/documentation.html for an explanation of these parameters";
        print "\n";
        
        
        exit;
    }
    
    print "\n  - configuration:";
    
    foreach $args_key (sort keys %args) {
        if($args{$args_key} ne ""){
            print "\n    $args_key: $args{$args_key}";
        }
    }
    
    print "\n";
    print "\n";
}

sub prepareObjects {
 ## get the source list    
    @source_files = `ls -1 | egrep "\.[jJ]{1}[pP]{1}[eE]{0,1}[gG]{1}"`;
    
    print "\n  - source files - "; now(); print "\n";

    foreach $file (@source_files) {
        chomp($file);
        
        chomp($file_make        = `jhead $file | grep "Camera make"`);
        chomp($file_model       = `jhead $file | grep "Camera model"`);
        chomp($file_focal       = `jhead $file | grep "Focal length"`);
        chomp($file_ccd         = `jhead $file | grep "CCD width"`);
        chomp($file_resolution  = `jhead $file | grep "Resolution"`);
    
        my %fileObject = {};			
    
        chomp(($fileObject{src})        = $file);
        chomp(($fileObject{base})       = $file);
        $fileObject{base}               =~ s/\.[^\.]*$//;
    
        chomp(($fileObject{make})       = $file_make =~ /: ([^\n\r]*)/);
        chomp(($fileObject{model})      = $file_model =~ /: ([^\n\r]*)/);
    
        $fileObject{id}                 = $fileObject{make}." ".$fileObject{model};
    
        ($fileObject{width}, $fileObject{height})    = $file_resolution =~ /: ([0-9]*) x ([0-9]*)/;
            
        if(!$args{"--force-focal"}){
            ($fileObject{focal})      = $file_focal =~ /:[\ ]*([0-9\.]*)mm/;
        } else {
            $fileObject{focal}        = $args{"--force-focal"};
        }
        
        if(!$args{"--force-ccd"}){
            ($fileObject{ccd})        = $file_ccd =~ /:[\ ]*([0-9\.]*)mm/;
            
            if(!$fileObject{ccd}){;
                $fileObject{ccd}      = $ccdWidths{$fileObject{id}};
            }
        } else {
            $fileObject{ccd}          = $args{"--force-ccd"};
        }
    
        if($fileObject{ccd} && $fileObject{focal} && $fileObject{width} && $fileObject{height}){
            if($fileObject{width} > $fileObject{height}){
                $fileObject{focalpx} = $fileObject{width} * ($fileObject{focal} / $fileObject{ccd});
            } else {
                $fileObject{focalpx} = $fileObject{height} * ($fileObject{focal} / $fileObject{ccd});
            }
            
            $fileObject{isOk} = true;
            $objectStats{good}++;
            
            print "\n     using $fileObject{src}     dimensions: $fileObject{width}x$fileObject{height} / focal: $fileObject{focal}mm / ccd: $fileObject{ccd}mm";
        } else {
            $fileObject{isOk} = false;
            $objectStats{bad}++;
            
            print "\n    no CCD width or focal length found for $fileObject{src}";
        }
    
        $objectStats{count}++;
    
        if($objectStats{minWidth} == 0)  { $objectStats{minWidth}  = $fileObject{width}; }
        if($objectStats{minHeight} == 0) { $objectStats{minHeight} = $fileObject{height}; }
    
        $objectStats{minWidth}  = $objectStats{minWidth}  < $fileObject{width}  ? $objectStats{minWidth}  : $fileObject{width};
        $objectStats{minHeight} = $objectStats{minHeight} < $fileObject{height} ? $objectStats{minHeight} : $fileObject{height};
        $objectStats{maxWidth}  = $objectStats{maxWidth}  > $fileObject{width}  ? $objectStats{maxWidth}  : $fileObject{width};
        $objectStats{maxHeight} = $objectStats{maxHeight} > $fileObject{height} ? $objectStats{maxHeight} : $fileObject{height};
            
        push(@objects, \%fileObject);
    }
    
    
    
    if(!$objectStats{good}){
        print "\n\n    found no usable images - quitting\n";
        die;
    } else {
        print "\n\n    found $objectStats{good} usable images";
    }
    
    print "\n";
    
    $jobOptions{resizeTo} = $args{"--resize-to"};

    print "\n  using max image size of $jobOptions{resizeTo} x $jobOptions{resizeTo}";    
    
    $jobOptions{jobDir} = "$jobOptions{srcDir}/reconstruction-with-image-size-$jobOptions{resizeTo}";
    
    $jobOptions{step_1_convert}         = "$jobOptions{jobDir}/_convert.templist.txt";
    $jobOptions{step_1_vlsift}          = "$jobOptions{jobDir}/_vlsift.templist.txt";
    $jobOptions{step_1_gzip}            = "$jobOptions{jobDir}/_gzip.templist.txt";
    
    $jobOptions{step_2_filelist}        = "$jobOptions{jobDir}/_filelist.templist.txt";
    $jobOptions{step_2_matches}         = "$jobOptions{jobDir}/matches.init.txt";
    
    $jobOptions{step_3_filelist}        = "$jobOptions{jobDir}/list.txt";
    $jobOptions{step_3_bundlerOptions}  = "$jobOptions{jobDir}/options.txt";
    
    mkdir($jobOptions{jobDir});
        
    foreach $fileObject (@objects) {
        if($fileObject->{isOk}){
            $fileObject->{step_0_resizedImage}  = "$jobOptions{jobDir}/$fileObject->{base}.jpg";
            
            $fileObject->{step_1_pgmFile}       = "$jobOptions{jobDir}/$fileObject->{base}.pgm";
            $fileObject->{step_1_keyFile}       = "$jobOptions{jobDir}/$fileObject->{base}.key";
            $fileObject->{step_1_gzFile}        = "$jobOptions{jobDir}/$fileObject->{base}.key.gz";
        }
    }
    
#    exit
}

sub resize {
    print "\n";
    print "\n  - preparing images - "; now(); print "\n";
    print "\n";

    chdir($jobOptions{jobDir});
    
    foreach $fileObject (@objects) {
        if($fileObject->{isOk}){
            if($jobOptions{resizeTo} != "orig" && (($fileObject->{width} > $jobOptions{resizeTo}) || ($fileObject->{height} > $jobOptions{resizeTo}))){
                print "\n    resising $fileObject->{src} \tto $fileObject->{step_0_resizedImage}";
                
                run("convert -resize $jobOptions{resizeTo}x$jobOptions{resizeTo} -quality 100 \"$jobOptions{srcDir}/$fileObject->{src}\" \"$fileObject->{step_0_resizedImage}\"");

            } else {
                print "\n     copying $fileObject->{src} \tto $fileObject->{step_0_resizedImage}";
                
                copy("$CURRENT_DIR/$fileObject->{src}", "$fileObject->{step_0_resizedImage}");
            }
            
            chomp($file_resolution    = `jhead $fileObject->{step_0_resizedImage} | grep "Resolution"`);
            ($fileObject->{width}, $fileObject->{height})    = $file_resolution =~ /: ([0-9]*) x ([0-9]*)/;
            print "\t ($fileObject->{width} x $fileObject->{height})";
        }
    }
    
    if($args{"--end-with"} ne "resize"){
        getKeypoints();
    }
}

sub getKeypoints {
    print "\n";
    print "\n  - finding keypoints - "; now(); print "\n";
    print "\n\n";
    
    chdir($jobOptions{jobDir});
    
    $vlsiftJobs    = "";
    
    foreach $fileObject (@objects) {
        if($fileObject->{isOk}){
            $vlsiftJobs    .= "convert -format pgm \"$fileObject->{step_0_resizedImage}\" \"$fileObject->{step_1_pgmFile}\"";
            $vlsiftJobs    .= " && $BIN_PATH/vlsift \"$fileObject->{step_1_pgmFile}\" -o \"$fileObject->{step_1_keyFile}.sift\" > /dev/null && perl $BIN_PATH/../convert_vlsift_to_lowesift.pl \"$fileObject->{base}\"";
            $vlsiftJobs    .= " && gzip -f \"$fileObject->{step_1_keyFile}\"";
            $vlsiftJobs    .= " && rm -f \"$fileObject->{step_1_pgmFile}\"";
            $vlsiftJobs    .= " && rm -f \"$fileObject->{step_1_keyFile}.sift\"\n";
        }
    }
    
    system("echo \"$vlsiftJobs\"    > $jobOptions{step_1_vlsift}");
    
    run("\"$BIN_PATH/parallel\" --halt-on-error 1 -j+0 < \"$jobOptions{step_1_vlsift}\"");
    
    if($args{"--end-with"} ne "getKeypoints"){
        match();
    }
}

sub match {
    print "\n";
    print "\n  - matching keypoints - "; now(); print "\n";
    print "\n";

    chdir($jobOptions{jobDir});
    
    $filesList = "";
    
    foreach $fileObject (@objects) {
        if($fileObject->{isOk}){
            if($fileObject->{isOk}){
                $filesList        .= "\"$fileObject->{step_1_keyFile}\"\n";
            }
        }
    }
    
    system("echo \"$filesList\"     > $jobOptions{step_2_filelist}     ");
    run("\"$BIN_PATH/KeyMatchFull\" \"$jobOptions{step_2_filelist}\" \"$jobOptions{step_2_matches}\"    ");
    
    if($args{"--end-with"} ne "match"){
        bundler();
    }
}

sub bundler {
    print "\n";
    print "\n  - running bundler - "; now(); print "\n";
    print "\n";

    chdir($jobOptions{jobDir});
    
    mkdir($jobOptions{jobDir}."/bundle");
    mkdir($jobOptions{jobDir}."/pmvs");
    mkdir($jobOptions{jobDir}."/pmvs/txt");
    mkdir($jobOptions{jobDir}."/pmvs/visualize");
    mkdir($jobOptions{jobDir}."/pmvs/models");
    
    $filesList = "";
    
    foreach $fileObject (@objects) {
        if($fileObject->{isOk}){
            if($fileObject->{isOk}){
                $filesList        .= sprintf("\./%s.jpg 0 %0.5f\n", $fileObject->{base}, $fileObject->{focalpx});
            }
        }
    }
    
    chomp($filesList);
    
    $bundlerOptions  = "--match_table matches.init.txt\n";
    $bundlerOptions .= "--output bundle.out\n";
    $bundlerOptions .= "--output_all bundle_\n";
    $bundlerOptions .= "--output_dir bundle\n";
    $bundlerOptions .= "--variable_focal_length\n";
    $bundlerOptions .= "--use_focal_estimate\n";
    $bundlerOptions .= "--constrain_focal\n";
    $bundlerOptions .= "--constrain_focal_weight 0.0001\n";
    $bundlerOptions .= "--estimate_distortion\n";
    $bundlerOptions .= "--run_bundle";
        
    system("echo \"$bundlerOptions\" > \"$jobOptions{step_3_bundlerOptions}\"");
    system("echo \"$filesList\" > \"$jobOptions{step_3_filelist}\"");
    
    run("\"$BIN_PATH/bundler\" \"$jobOptions{step_3_filelist}\" --options_file \"$jobOptions{step_3_bundlerOptions}\" > bundle/out");
    run("\"$BIN_PATH/Bundle2PMVS\" \"$jobOptions{step_3_filelist}\" bundle/bundle.out");
    run("\"$BIN_PATH/RadialUndistort\" \"$jobOptions{step_3_filelist}\" bundle/bundle.out pmvs");
    
    $i = 0;
    
    foreach $fileObject (@objects) {
        if($fileObject->{isOk}){
            if($fileObject->{isOk}){
                if (-e "pmvs/$fileObject->{base}.rd.jpg"){
                    $nr = sprintf("%08d", $i++);
                
                    system("mv pmvs/$fileObject->{base}.rd.jpg pmvs/visualize/$nr.jpg");
                    system("mv pmvs/$nr.txt pmvs/txt/$nr.txt");
                }
            }
        }
    }
    
    system("\"$BIN_PATH/Bundle2Vis\" pmvs/bundle.rd.out pmvs/vis.dat");
    
    if($args{"--end-with"} ne "bundler"){
        cmvs();
    }
}

sub cmvs {
    print "\n";
    print "\n  - running cmvs - "; now(); print "\n";
    print "\n";

    chdir($jobOptions{jobDir});
    
    run("\"$BIN_PATH/cmvs\" pmvs/ $args{'--cmvs-maxImages'} $CORES");
    run("\"$BIN_PATH/genOption\" pmvs/ $args{'--pmvs-level'} $args{'--pmvs-csize'} $args{'--pmvs-threshold'} $args{'--pmvs-wsize'} $args{'--pmvs-minImageNum'} $CORES");
    
    if($args{"--end-with"} ne "cmvs"){
        pmvs();
    }
}

sub pmvs {
    print "\n";
    print "\n  - running pmvs - ";
    print "\n";

    chdir($jobOptions{jobDir});
    
    run("\"$BIN_PATH/pmvs2\" pmvs/ option-0000");
    
    system("cp -Rf \"$jobOptions{jobDir}/pmvs/models\" \"$jobOptions{jobDir}-results\"");
}

parseArgs();
prepareObjects();
    
chdir($jobOptions{jobDir});
    
switch ($args{"--start-with"}) {
    case "resize"          { resize();          }
    case "getKeypoints"    { getKeypoints();    }
    case "match"           { match();           }
    case "bundler"         { bundler();         }
    case "cmvs"            { cmvs();            }
    case "pmvs"            { pmvs();            }
}

print "\n";
print "\n  - done - "; now(); print "\n";
print "\n";