c-kzg-4844/scripts/convert_trusted_setup.py

#!/usr/bin/env python3

import argparse
import json
from typing import TextIO


def convert(ts_json: TextIO, ts_text: TextIO) -> None:
    """
    Convert trusted setup to text format.
    """
    trusted_setup = json.load(ts_json)
    g1_monomial = trusted_setup["g1_monomial"]
    g1_lagrange = trusted_setup["g1_lagrange"]
    g2_monomial = trusted_setup["g2_monomial"]

    if len(g1_monomial) != len(g1_lagrange):
        raise Exception("len(g1_monomial) != len(g1_lagrange)")

    print(len(g1_monomial), file=ts_text)
    print(len(g2_monomial), file=ts_text)
    for g1 in g1_lagrange:
        print(g1.replace("0x", ""), file=ts_text)
    for g2 in g2_monomial:
        print(g2.replace("0x", ""), file=ts_text)
    for g1 in g1_monomial:
        print(g1.replace("0x", ""), file=ts_text)


if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description="Convert trusted setup from JSON to text format.",
    )
    parser.add_argument(
        "--input",
        required=True,
        type=argparse.FileType("r"),
        help="the trusted setup in JSON format",
    )
    parser.add_argument(
        "--output",
        required=True,
        type=argparse.FileType("w"),
        help="the trusted setup in text format",
    )
    args = parser.parse_args()

    try:
        convert(args.input, args.output)
    finally:
        args.input.close()
        args.output.close()