#########################################################################
#
#  Script to prepare party affiliation data
#
#########################################################################

$proot = '/alldev/john/LIBDEMS/';
$file = $proot.$ARGV[0];

$n = scalar @ARGV;

$p1 = 0; if ($n > 1) { $p1 = lc( $ARGV[1] ); } 
$p2 = 0; if ($n > 2) { $p2 = lc( $ARGV[2] ); }
$p3 = 0; if ($n > 3) { $p3 = lc( $ARGV[3] ); }

$party = 0;
if ($p1 eq 'libdem') { $party = 1000; }
if ($p1 eq 'labour') { $party = 2000; }
if ($p1 eq 'tory')   { $party = 3000; }
if ($p1 eq 'green')  { $party = 4000; }
if ($p1 eq 'other')  { $party = 5000; }

if ($p2 eq 'weak')    { $party += 100; }
if ($p2 eq 'average') { $party += 200; }
if ($p2 eq 'strong')  { $party += 300; }

if ($p3 eq 'yellow')  { $party += 10; }
if ($p3 eq 'green')   { $party += 20; }
if ($p3 eq 'blue')    { $party += 30; }
if ($p3 eq 'red')     { $party += 40; }

open HFILE, $file or die "failed to open $file for input ($!)\n";
@line = <HFILE>;
close HFILE;

$file = 'partymap.txt';
open HFILE, '>'.$file or die "failed to open $file for input ($!)\n";

$count = 0;
foreach $line (@line) {
  chomp $line;
  if ($count > 0) {
    print HFILE $line.'|'.$party."\n";
  }
  $count++;
}

close HFILE;