solver_trace.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <ostream>
#include <utility>
#include <vector>
 
#include <boost/rational.hpp>
 
#include "graphs.hpp"
 
namespace discrete_modulus {
 
using namespace boost;
 
struct TraceRound {
    rational<long> theta;
    std::vector<Vertex> vertices;                  
    std::vector<std::pair<Vertex, Vertex>> crit_set;  
};
 
struct SolverTrace {
    std::vector<TraceRound> rounds;
};
 
inline void write_trace_json(std::ostream& os, const SolverTrace& trace) {
    os << "{\n  \"version\": 1,\n  \"rounds\": [\n";
    for (std::size_t r = 0; r < trace.rounds.size(); ++r) {
        const TraceRound& round = trace.rounds[r];
 
        os << "    {\n      \"vertices\": [";
        for (std::size_t i = 0; i < round.vertices.size(); ++i) {
            if (i > 0) {
                os << ", ";
            }
            os << round.vertices[i];
        }
 
        os << "],\n      \"crit_set\": [";
        for (std::size_t i = 0; i < round.crit_set.size(); ++i) {
            if (i > 0) {
                os << ", ";
            }
            os << "[" << round.crit_set[i].first << ", " << round.crit_set[i].second << "]";
        }
 
        os << "],\n      \"theta\": [" << round.theta.numerator() << ", " << round.theta.denominator() << "]\n";
        os << "    }" << (r + 1 < trace.rounds.size() ? ",\n" : "\n");
    }
    os << "  ]\n}\n";
}
 
}  // namespace discrete_modulus