#!/usr/bin/perl
# cobol_count - count physical lines of code.
# Usage: cobol_count [-f file] [list_of_files]
#  file: file with a list of files to count (if "-", read list from stdin)
#  list_of_files: list of files to count
#  -f file or list_of_files can be used, or both
# This is a trivial/naive program for scripts, etc.

$total_sloc = 0;

# Do we have "-f" (read list of files from second argument)?
if (($#ARGV >= 1) && ($ARGV[0] eq "-f")) {
  # Yes, we have -f
  if ($ARGV[1] eq "-") {
    # The list of files is in STDIN
    while (<STDIN>) {
      chomp ($_);
      &count_file ($_);
    }
  } else {
    # The list of files is in the file $ARGV[1]
    open (FILEWITHLIST, $ARGV[1]) || die "Error: Could not open $filewithlist\n";
    while (<FILEWITHLIST>) {
      chomp ($_);
      &count_file ($_);
    }
    close FILEWITHLIST;
  }
  shift @ARGV; shift @ARGV;
}
# Process all (remaining) arguments as file names
while ($file = shift @ARGV) {
  &count_file ($file);
}

print "Total:\n";
print "$total_sloc\n";

sub count_file {
  my ($file) = @_;
  my $sloc = 0;
  my $free_format = 0;  # Support "free format" source code.

  open (FILE, $file);
  while (<FILE>) {
    if (m/^......\$.*SET.*SOURCEFORMAT.*FREE/i) {$free_format = 1;}
    if ($free_format) {
      if (m/^\s*[\*\/]/) { }  # Saw a comment.
      elsif (m/\S/) {$sloc++;} # Saw a non-whitespace, it's SLOC.
    } else {
      if (m/^......[\*\/]/) {}  # Saw a comment - marked in indicator area.
      elsif (m/^[\*\/]/) {}     # Saw a comment.
      elsif (m/^........*\S/) {$sloc++;} # Saw a non-whitespace, it's SLOC.
    }
  }
  print "$sloc $file\n";
  $total_sloc += $sloc;
  close (FILE);
}