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;

}