tbcs.hpp

Go to the documentation of this file.

 
#ifndef TBCS_HPP_
#define TBCS_HPP_
 
#include <libff/common/profiling.hpp>
#include <libsnark/relations/variable.hpp>
 
namespace libsnark
{
 
/*********************** BACS variable assignment ****************************/
 
typedef std::vector<bool> tbcs_variable_assignment;
 
/**************************** TBCS gate **************************************/
 
typedef size_t tbcs_wire_t;
 
enum tbcs_gate_type {
    TBCS_GATE_CONSTANT_0 = 0,
    TBCS_GATE_AND = 1,
    TBCS_GATE_X_AND_NOT_Y = 2,
    TBCS_GATE_X = 3,
    TBCS_GATE_NOT_X_AND_Y = 4,
    TBCS_GATE_Y = 5,
    TBCS_GATE_XOR = 6,
    TBCS_GATE_OR = 7,
    TBCS_GATE_NOR = 8,
    TBCS_GATE_EQUIVALENCE = 9,
    TBCS_GATE_NOT_Y = 10,
    TBCS_GATE_IF_Y_THEN_X = 11,
    TBCS_GATE_NOT_X = 12,
    TBCS_GATE_IF_X_THEN_Y = 13,
    TBCS_GATE_NAND = 14,
    TBCS_GATE_CONSTANT_1 = 15
};
 
static const int num_tbcs_gate_types = 16;
 
class tbcs_gate
{
public:
    tbcs_wire_t left_wire;
    tbcs_wire_t right_wire;
 
    tbcs_gate_type type;
 
    tbcs_wire_t output;
 
    bool is_circuit_output;
 
    bool evaluate(const tbcs_variable_assignment &input) const;
    void print(
        const std::map<size_t, std::string> &variable_annotations =
            std::map<size_t, std::string>()) const;
    bool operator==(const tbcs_gate &other) const;
 
    friend std::ostream &operator<<(std::ostream &out, const tbcs_gate &g);
    friend std::istream &operator>>(std::istream &in, tbcs_gate &g);
};
 
/****************************** TBCS inputs **********************************/
 
typedef tbcs_variable_assignment tbcs_primary_input;
 
typedef tbcs_variable_assignment tbcs_auxiliary_input;
 
/************************** TBCS circuit *************************************/
 
class tbcs_circuit
{
public:
    size_t primary_input_size;
    size_t auxiliary_input_size;
    std::vector<tbcs_gate> gates;
 
    tbcs_circuit() : primary_input_size(0), auxiliary_input_size(0) {}
 
    size_t num_inputs() const;
    size_t num_gates() const;
    size_t num_wires() const;
 
    std::vector<size_t> wire_depths() const;
    size_t depth() const;
 
#ifdef DEBUG
    std::map<size_t, std::string> gate_annotations;
    std::map<size_t, std::string> variable_annotations;
#endif
 
    bool is_valid() const;
    bool is_satisfied(
        const tbcs_primary_input &primary_input,
        const tbcs_auxiliary_input &auxiliary_input) const;
 
    tbcs_variable_assignment get_all_wires(
        const tbcs_primary_input &primary_input,
        const tbcs_auxiliary_input &auxiliary_input) const;
    tbcs_variable_assignment get_all_outputs(
        const tbcs_primary_input &primary_input,
        const tbcs_auxiliary_input &auxiliary_input) const;
 
    void add_gate(const tbcs_gate &g);
    void add_gate(const tbcs_gate &g, const std::string &annotation);
 
    bool operator==(const tbcs_circuit &other) const;
 
    void print() const;
    void print_info() const;
 
    friend std::ostream &operator<<(
        std::ostream &out, const tbcs_circuit &circuit);
    friend std::istream &operator>>(std::istream &in, tbcs_circuit &circuit);
};
 
} // namespace libsnark
 
#endif // TBCS_HPP_