#include <iostream>
#include <fstream>
#include <ShapeUpgrade_UnifySameDomain.hxx>
#include <STEPControl_Reader.hxx>
#include <STEPControl_Writer.hxx>
#include <TopoDS.hxx>

int main()
{
    const char* inputPath = "ball-valve.step";
    const char* outputPath = "output.step";

    // Initialize the STEP reader
    STEPControl_Reader reader;
    if (reader.ReadFile(inputPath) != IFSelect_RetDone)
    {
        std::cout << "Error: cannot read file." << std::endl;
        return 1;
    }

    // Check the load status
    reader.PrintCheckLoad(true, IFSelect_ItemsByEntity);
    if (reader.TransferRoots() != reader.NbRootsForTransfer())
    {
        std::cout << "Error: roots transferred unsuccessfully." << std::endl;
        return 1;
    }

    // Get the shape from the reader
    TopoDS_Shape shape = reader.Shape();

    // Perform "unify domains" algorithm
    ShapeUpgrade_UnifySameDomain unifyDomains(shape);
    unifyDomains.Build();
    TopoDS_Shape unifiedShape = unifyDomains.Shape();

    // Write the unified shape to STEP file
    STEPControl_Writer writer;
    writer.Transfer(unifiedShape, STEPControl_AsIs);
    if (!writer.Write(outputPath))
    {
        std::cout << "Error: failed to write output file." << std::endl;
        return 1;
    }

    return 0;
}