pregelix/pregelix-example/src/test/java/edu/uci/ics/pregelix/example/util/Record.java

/*
 * Copyright 2009-2013 by The Regents of the University of California
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * you may obtain a copy of the License from
 * 
 *     http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package edu.uci.ics.pregelix.example.util;

@SuppressWarnings("rawtypes")
public class Record implements Comparable {

    private String recordText;

    public Record(String text) {
        recordText = text;
    }

    @Override
    public int hashCode() {
        return recordText.hashCode();
    }

    @Override
    public String toString() {
        return recordText;
    }

    @Override
    public int compareTo(Object o) {
        if (!(o instanceof Record)) {
            throw new IllegalStateException("uncomparable items");
        }
        Record record = (Record) o;
        boolean equal = equalStrings(recordText, record.recordText);
        if (equal) {
            return 0;
        } else {
            return recordText.compareTo(record.recordText);
        }
    }

    @Override
    public boolean equals(Object o) {
        if (!(o instanceof Record)) {
            return false;
        }
        Record record = (Record) o;
        return equalStrings(recordText, record.recordText);
    }

    private boolean equalStrings(String s1, String s2) {
        String[] rowsOne = s1.split("\n");
        String[] rowsTwo = s2.split("\n");

        if (rowsOne.length != rowsTwo.length)
            return false;

        for (int i = 0; i < rowsOne.length; i++) {
            String row1 = rowsOne[i];
            String row2 = rowsTwo[i];

            if (row1.equals(row2))
                continue;

            boolean spaceOrTab = false;
            spaceOrTab = row1.contains(" ");
            String[] fields1 = spaceOrTab ? row1.split(" ") : row1.split("\t");
            String[] fields2 = spaceOrTab ? row2.split(" ") : row2.split("\t");

            for (int j = 0; j < fields1.length; j++) {
                if (j >= fields2.length) {
                    return false;
                }
                if (fields1[j].equals(fields2[j])) {
                    continue;
                } else if (fields1[j].indexOf('.') < 0) {
                    return false;
                } else {
                    Double double1 = Double.parseDouble(fields1[j]);
                    Double double2 = Double.parseDouble(fields2[j]);
                    float float1 = (float) double1.doubleValue();
                    float float2 = (float) double2.doubleValue();

                    if (Math.abs(float1 - float2) < 1.0e-7)
                        continue;
                    else {
                        return false;
                    }
                }
            }
        }
        return true;
    }

}