X-Git-Url: http://git.equinoxoli.org/?p=migration-tools.git;a=blobdiff_plain;f=match_fingerprints;h=8f0d485831f815dd4d312230cafdaa4368283aa4;hp=4300429df383a55416d9cbd59479b92aab83c387;hb=HEAD;hpb=3ec4fd758b018abda104bbeffa5ebf02b832030b

diff --git a/match_fingerprints b/match_fingerprints
index 4300429..8f0d485 100755
--- a/match_fingerprints
+++ b/match_fingerprints
@@ -1,4 +1,22 @@
 #!/usr/bin/perl
+
+# Copyright 2009-2012, Equinox Software, Inc.
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License
+# as published by the Free Software Foundation; either version 2
+# of the License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+
+
 use strict;
 use warnings;
 use open ':utf8';
@@ -14,6 +32,8 @@ my @recs = (); # fingerprints belonging to each record
 my %seen = (); # records we've already seen
 my $lastscore = 0; # previous fingerprint's score
 
+my %subs  = (); # error-checking hashe
+
 open FP, '<', $ARGV[0] or die "Can't open input file: $!\n";
 
 print "Loading and ranking fingerprints\n";
@@ -42,7 +62,7 @@ sub populate_fingerprint {
     $fp{sha1}    = sha1_base64($stripped);
 
     # make sure file is sorted properly
-    if ($lastscore and ($fp{compact} > $lastscore)) {
+    if ($lastscore and ($fp{compact} < $lastscore)) {
         print "Input file is sorted improperly or unsorted.\n";
         die "Sort descending (sort -r) and rerun this script.\n";
     }
@@ -59,7 +79,7 @@ sub rank_fingerprint {
     my $id   = $fp->{id};
 
     # only process records which haven't already been seen
-    unless (defined $seen{$id}) {
+    #unless (defined $seen{$id}) {
         unless (defined $fps{$sha1}) {
             # haven't seen this fp before. create a new listref to hold subs
             # and stow the hash of the fingerprint that we're lead of
@@ -70,7 +90,7 @@ sub rank_fingerprint {
             push @{ $fps{$sha1} }, $id;
         }
         $seen{$id} = 1;
-    }
+    #}
 }
 
 
@@ -85,8 +105,41 @@ sub dump_records {
     open OUT, '>', $conf->{output}
       or die "Can't open ", $conf->{output}, "$!\n";
     for my $rec (@recs) {
-        print OUT $rec->{id}, "\t$_\n"
-          for ( @{ $fps{ $rec->{sha1} } } );
+        for ( @{ $fps{ $rec->{sha1} } } ) {
+            # check for dupes and die if they exist
+            #die "Collision: dupe sub record $_\n" if $subs{$_};
+            if( not exists ($subs{$_}) ) {
+                #die "Collision: lead in sub list ", $rec->{id}, "\n"
+                #  if $subs{ $rec->{id} };
+                my $lead = $rec->{id};
+
+                # we don't need to match onto itself
+                next if ($lead eq $_);
+
+                if ($subs{ $rec->{id} })
+                {
+                    print "moving ".$lead." to ".$subs{$rec->{id}}."\n";
+                    $lead = $subs{$rec->{id}};
+                }
+
+                if($_ eq '1459051')
+                {
+                    print "looking at ".$rec->{sha1};
+                }
+
+                # we don't want subs below threshold
+                next if ($_ < $conf->{threshold});
+
+                # we don't need to match onto itself
+                next if ($lead eq $_);
+
+                # record this sub having this leader
+                $subs{$_} = $lead;
+
+                # still here? output.
+                print OUT $lead, "\t$_\n"
+            }
+        }
     }
 }
 
@@ -99,6 +152,7 @@ sub initialize {
 
     my $rc = GetOptions( $c,
                          'output|o=s',
+                         'threshold|t=i',
                          'help|h',
                        );
     show_help() unless $rc;
@@ -112,11 +166,13 @@ sub initialize {
         print "Required option: ", join(', ', @missing), " missing!\n";
         show_help();
     }
+
+    $c->{threshold} = 0 unless $c->{threshold};
 }
 
 sub show_help {
     print <<HELP;
-Usage is: compress_fingerprints -o OUTPUTFILE INPUTFILE
+Usage is: compress_fingerprints [-t THRESHOLD] -o OUTPUTFILE INPUTFILE
 HELP
 exit;
 }