util.py

import os
from os import path

# Gloable variables
feature_path = ".features.txt"
mode_path = ".mode.txt"
cargo_features = {"aby", "c", "lp", "r1cs", "kahip", "kahypar",
                  "smt", "zok", "datalog", "bellman", "spartan", "poly"}

# Environment variables
ABY_SOURCE = "./../ABY"
KAHIP_SOURCE = "./../KaHIP"
KAHYPAR_SOURCE = "./../kahypar"


def set_env(features):
    for f in features:
        if f == 'aby':
            if not os.getenv("ABY_SOURCE"):
                os.environ["ABY_SOURCE"] = ABY_SOURCE
        if f == 'kahip':
            if not os.getenv("KAHIP_SOURCE"):
                os.environ["KAHIP_SOURCE"] = KAHIP_SOURCE
        if f == 'kahypar':
            if not os.getenv("KAHYPAR_SOURCE"):
                os.environ["KAHYPAR_SOURCE"] = KAHYPAR_SOURCE


def save_mode(mode):
    """ Save mode to file """
    with open(mode_path, 'w') as f:
        f.write(mode)


def load_mode():
    """ Load mode from file """
    if path.exists(mode_path):
        with open(mode_path, 'r') as f:
            return f.read().strip()
    else:
        return ""


def save_features(features):
    """ Save features to file """
    with open(feature_path, 'w') as f:
        feature_str = "\n".join(features)
        f.write(feature_str)


def load_features():
    """ Load features from file """
    if path.exists(feature_path):
        with open(feature_path, 'r') as f:
            features = f.read().splitlines()
            return features
    else:
        return []