Subversion Repositories HomeAutomation

#!/usr/bin/perl

# Written by Mattias Runge 2008-2010

use Cwd 'abs_path';
use File::Basename;
use Time::HiRes qw( usleep );

$| = 1;

$numArgs = $#ARGV + 1;

my $hostname = "localhost";
my $port = 1201;
my $filename = "";
my $hwid = "";
my $bios = 0;
my $reset = "false";

foreach $argnum (0 .. $#ARGV)
{
    if ($ARGV[$argnum] eq "-h")
    {
        $hostname = $ARGV[$argnum+1];
        next;

    }
    elsif ($ARGV[$argnum] eq "-p")
    {
        $port = $ARGV[$argnum+1];
        next;

    }
    elsif ($ARGV[$argnum] eq "-i")
    {
        $hwid = $ARGV[$argnum+1];
        next;

    }
    elsif ($ARGV[$argnum] eq "-f")
    {
        $filename = abs_path($ARGV[$argnum+1]);
        next;

    }
    elsif ($ARGV[$argnum] eq "-b")
    {
        $bios = 1;
    }
    elsif ($ARGV[$argnum] eq "-r")
    {
        $reset = "true";
    }
}

if ($hwid eq "")
{
    die "You must specify a hardware id!\n";
    exit 1;
}

my $path = dirname(abs_path($0));

# Read atom functions from separate file
if (-e $path."/setConfig/AtomLib.pl") {
    require $path."/setConfig/AtomLib.pl";
} else {
    print "Error: No AtomLib.pl was found in ".$path.". \n\n";
    exit 1;
}

#0x14f90d1f to 0x1f0df914
@hwidChars = split(//, $hwid);
$hwid = $hwidChars[0]. $hwidChars[1]. uc($hwidChars[8]. $hwidChars[9]. $hwidChars[6]. $hwidChars[7]. $hwidChars[4]. $hwidChars[5]. $hwidChars[2]. $hwidChars[3]);

print "atomDude settings:\n";
print "Hardware id: " . $hwid . "\n";
print "Atom address: " . $hostname . ":" . $port . "\n";
print "Hexfile: " . $filename . "\n";
#print "Parameters: " . "\n";

############################## RESET ##############################
if ($reset eq "true" && $filename eq "")
{
    $socket = atomd_initialize($hostname, $port);
    if ($port == 1203)
    {
        atomjs_write($socket, "Node_ResetNative('".$hwid."');\n");
        $response = atomjs_read_line($socket);
        if ($response =~ m/Error/)
        {
            die $response;
        }
        print $response;

        $success = 0;
        for ($count = 0; $count < 15; $count++)
        {
            atomjs_write($socket, "Node_ResetPollNative();\n");
            if (atomjs_read_line($socket) =~ m/true/)
            {
                print "\033[32m" . $hwid . " started okay.\033[0m\n";
                $success = 1;
                last;
            }
            usleep(200000);
        }
        if ($success == 0)
        {
            print "\033[31mTimed out waiting for reset.\033[0m\n";
        }
    }
    else
    {
        usleep(100000);
        atomd_kill_promt($socket);
        atomd_send_command($socket, "Node_Reset $hwid");
        print atomd_read_command_response($socket);
        #my $result = system("atomic -s $hostname -p $port -c \"Node_Reset $hwid\"");
        #exit($result);
    }
}
############################## REPROGRAM ##############################
else
{
    open (FP, "+<".$filename) or die "Error: Cannot open file $filename\n";
    @filecontents = <FP>;
    close FP;
    $socket = atomd_initialize($hostname, $port);

    if ($port == 1203)
    {
        atomjs_write($socket, "Node_ClearHex();\n");
        atomjs_read_line($socket);
        my $linenums = 0;
        foreach $filerow (@filecontents) 
        {
            $filerow =~ s/^\s+//;
            $filerow =~ s/\s+$//;
            $linenums += 1;
            $command = "Node_AppendHexNative('".$filerow."');\n";
            atomjs_write($socket, $command);
            $response = atomjs_read_line($socket);
            if ($response =~ m/Error/)
            {
                die $response;
            }
        }
        $command = "Node_ProgramNative('".$hwid."', '".$bios."', '".$linenums."');\n";
        atomjs_write($socket, $command);
        #print $command;
        $response = atomjs_read_line($socket);
        if ($response =~ m/Error/)
        {
            die $response;
        }
        print $response;
#atomd_disconnect($socket);
        $success = 0;
        $alreadyprintfull = 0;
        for ($count = 0; $count < 150; $count++)
        {
            atomjs_write($socket, "Node_ProgramPollNative();\n");
            if (atomjs_read_line($socket) =~ m/true/)
            {
                print progress_bar( 100, 100, 40, '=' );
                print "\n\033[32m" . $hwid . " started okay.\033[0m\n";
                $success = 1;
                last;
            }
            atomjs_write($socket, "Node_GetProgramProgress('".$hwid."');\n");
            $response = atomjs_read_line($socket);
            if ($response =~ m/Error/)
            {
                die $response;
            }
            $response =~ s/^\s+//;
            $response =~ s/\s+$//;
            if ($response > 99)
            {
                $response = 99;
            }
            #print $response."\n";
            print progress_bar( $response, 100, 40, '=' );
            
            usleep(200000);
        }
        if ($success == 0)
        {
            print "\n\033[31mTimed out waiting for program to finish.\033[0m\n";
        }
    }
    else
    {
        usleep(100000);
        atomd_kill_promt($socket);
    
        #print "Sending Node_ClearHex\n";
        atomd_send_command($socket, "Node_ClearHex");
        atomd_kill_promt($socket);
    
        my $linenums = 0;
        foreach $filerow (@filecontents) 
        {
            #print "Sending Node_AppendHex $filerow \n";
            #$filerow =~ s/^\s+//;
            #$filerow =~ s/\s+$//;
            atomd_send_command($socket, "Node_AppendHex $filerow");
            atomd_kill_promt($socket);
            $linenums += 1;
        }
        #print "Sending Node_ProgramHex $hwid $bios $linenums \n";
        atomd_send_command($socket, "Node_ProgramHex $hwid $bios $linenums");

        print atomd_read_command_response($socket);

#       my $result = system("atomic -s $hostname -p $port -c \"Node_Program $hwid $bios $filename\"");
#       exit($result);
    }
}

exit(0);

# wget-style. routine by tachyon
# at http://tachyon.perlmonk.org/
sub progress_bar {
    my ( $got, $total, $width, $char ) = @_;
    $width ||= 25; $char ||= '=';
    my $num_width = length $total;
    sprintf "|%-${width}s| (%.2f%%)\r", 
        $char x (($width-1)*$got/$total). '>', 
        $got, $total, 100*$got/+$total;
}