import argparse
import csv
import sys


def read_matrices_from_csv(file_path):
    with open(file_path, newline='') as csvfile:
        reader = csv.reader(csvfile, delimiter=' ', skipinitialspace=True)
        matrices, matrix, line_numbers = [], [], []
        line_number = 0

        for row in reader:
            line_number += 1
            if row:
                matrix.append([float(item) for item in row])
                line_numbers.append(line_number)
            else:
                if matrix:
                    matrices.append((matrix, line_numbers))
                    matrix, line_numbers = [], []

        if matrix:  # Add the last matrix if there's no blank line at the end
            matrices.append((matrix, line_numbers))

        return matrices

def compare_matrices(matrix_a_tuple, matrix_b_tuple, tolerance=0.01):
    data_a, line_numbers_a = matrix_a_tuple
    data_b, line_numbers_b = matrix_b_tuple
    max_diff = 0

    if len(data_a) != len(data_b):
        return False, f"Matrix size mismatch: Matrix A has {len(data_a)} rows, Matrix B has {len(data_b)} rows.", None, None, max_diff

    for row_index, (row_a, row_b, line_num_a, line_num_b) in enumerate(zip(data_a, data_b, line_numbers_a, line_numbers_b)):
        if len(row_a) != len(row_b):
            return False, f"Row size mismatch at line {line_num_a} (File A) and line {line_num_b} (File B).", line_num_a, line_num_b, max_diff

        for col_index, (elem_a, elem_b) in enumerate(zip(row_a, row_b)):
            if abs(elem_a - elem_b) > max_diff:
                max_diff = abs(elem_a - elem_b)
            if abs(elem_a - elem_b) > tolerance:
                return False, f"Mismatch at line {line_num_a}, column {col_index+1} (File A) vs line {line_num_b}, column {col_index+1} (File B):\nA({elem_a}) vs B({elem_b}).", line_num_a, line_num_b, max_diff

    return True, "Matrices are equal.", None, None, max_diff

def compare_csv_files(file_path1, file_path2, tolerance=0.01):
    matrices1 = read_matrices_from_csv(file_path1)
    matrices2 = read_matrices_from_csv(file_path2)
    max_difference = 0

    if len(matrices1) != len(matrices2):
        return False, "Number of matrices in files do not match.", None, None, max_difference

    for index, (matrix_a_tuple, matrix_b_tuple) in enumerate(zip(matrices1, matrices2)):
        equal, message, line_num_a, line_num_b, max_diff = compare_matrices(matrix_a_tuple, matrix_b_tuple, tolerance)
        if max_diff > max_difference:
            max_difference = max_diff
        if not equal:
            return False, f"In Matrix pair {index}: {message}", line_num_a, line_num_b, max_difference

    return True, "All matrices are equal.", None, None, max_difference


def main():
    parser = argparse.ArgumentParser(description='Compare two CSV files containing matrices.')
    parser.add_argument('file_a', help='Path to File A')
    parser.add_argument('file_b', help='Path to File B')
    parser.add_argument('--tolerance', type=float, default=0.01, help='Tolerance for comparison (default: 0.01)')
    parser.add_argument('--silent', action='store_true', help='Run in silent mode without printing details')
    args = parser.parse_args()

    are_equal, message, line_num_a, line_num_b, max_difference = compare_csv_files(args.file_a, args.file_b, args.tolerance)

    if not args.silent:
        print(message)

    if not are_equal:
        sys.exit(1)

if __name__ == "__main__":
    main()