pblib-rs 0.1.0

Rust safe bindings for pblib
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

#include "cpblib.h"

extern "C"
{
    int32_t* encode_formula(std::vector< std::vector<int32_t> > formula_vec, int32_t firstAuxiliaryVariable);

    CPB2CNF* newPB2CNF()
    {
        PB2CNF* pb2cnf = new PB2CNF();
        return reinterpret_cast<CPB2CNF*>(pb2cnf);
    }

    int32_t* encodeLeq(
        CPB2CNF* cpb2cnf,
        int64_t* weights,
        int32_t weights_len,
        int32_t* literals,
        int32_t literals_len,
        int64_t leq,
        int32_t firstAuxiliaryVariable
    ) {
        PB2CNF *pb2cnf = reinterpret_cast<PB2CNF *>(cpb2cnf);
        std::vector<int64_t> weights_vec(weights, weights + weights_len);
        std::vector<int32_t> literals_vec(literals, literals + literals_len);
        std::vector< std::vector<int32_t> > formula_vec;
        firstAuxiliaryVariable = pb2cnf->encodeLeq(weights_vec, literals_vec, leq, formula_vec, firstAuxiliaryVariable) + 1;
        return encode_formula(formula_vec, firstAuxiliaryVariable);
    }

    int32_t* encodeGeq(
        CPB2CNF* cpb2cnf,
        int64_t* weights,
        int32_t weights_len,
        int32_t* literals,
        int32_t literals_len,
        int64_t geq,
        int32_t firstAuxiliaryVariable
    ) {
        PB2CNF *pb2cnf = reinterpret_cast<PB2CNF *>(cpb2cnf);
        std::vector<int64_t> weights_vec(weights, weights + weights_len);
        std::vector<int32_t> literals_vec(literals, literals + literals_len);
        std::vector< std::vector<int32_t> > formula_vec;
        firstAuxiliaryVariable = pb2cnf->encodeGeq(weights_vec, literals_vec, geq, formula_vec, firstAuxiliaryVariable) + 1;
        return encode_formula(formula_vec, firstAuxiliaryVariable);
    }

    int32_t* encodeBoth(
        CPB2CNF* cpb2cnf,
        int64_t* weights,
        int32_t weights_len,
        int32_t* literals,
        int32_t literals_len,
        int64_t leq,
        int64_t geq,
        int32_t firstAuxiliaryVariable
    ) {
        PB2CNF *pb2cnf = reinterpret_cast<PB2CNF *>(cpb2cnf);
        std::vector<int64_t> weights_vec(weights, weights + weights_len);
        std::vector<int32_t> literals_vec(literals, literals + literals_len);
        std::vector< std::vector<int32_t> > formula_vec;
        firstAuxiliaryVariable = pb2cnf->encodeBoth(weights_vec, literals_vec, leq, geq, formula_vec, firstAuxiliaryVariable) + 1;
        return encode_formula(formula_vec, firstAuxiliaryVariable);
    }

    int32_t* encodeAtMostK(
        CPB2CNF* cpb2cnf,
        int32_t* literals,
        int32_t literals_len,
        int64_t k,
        int32_t firstAuxiliaryVariable
    ) {
        PB2CNF *pb2cnf = reinterpret_cast<PB2CNF *>(cpb2cnf);
        std::vector<int32_t> literals_vec(literals, literals + literals_len);
        std::vector< std::vector<int32_t> > formula_vec;
        firstAuxiliaryVariable = pb2cnf->encodeAtMostK(literals_vec, k, formula_vec, firstAuxiliaryVariable) + 1;
        return encode_formula(formula_vec, firstAuxiliaryVariable);
    }

    int32_t* encodeAtLeastK(
        CPB2CNF* cpb2cnf,
        int32_t* literals,
        int32_t literals_len,
        int64_t k,
        int32_t firstAuxiliaryVariable
    ) {
        PB2CNF *pb2cnf = reinterpret_cast<PB2CNF *>(cpb2cnf);
        std::vector<int32_t> literals_vec(literals, literals + literals_len);
        std::vector< std::vector<int32_t> > formula_vec;
        firstAuxiliaryVariable = pb2cnf->encodeAtLeastK(literals_vec, k, formula_vec, firstAuxiliaryVariable) + 1;
        return encode_formula(formula_vec, firstAuxiliaryVariable);
    }

    int32_t* encode_formula(std::vector< std::vector<int32_t> > formula_vec, int32_t firstAuxiliaryVariable)
    {
        int32_t formula_len = 2 + formula_vec.size();
        for(std::vector< std::vector<int32_t> >::iterator it = formula_vec.begin(); it < formula_vec.end(); it++) {
            formula_len += it->size();
        }
        int32_t* formula = (int32_t*) malloc(formula_len * sizeof(int32_t));
        int *pf = formula;
        *pf++ = formula_len;
        *pf++ = firstAuxiliaryVariable;
        for(std::vector< std::vector<int32_t> >::iterator it = formula_vec.begin(); it < formula_vec.end(); it++) {
            *pf++ = it->size();
            copy(it->begin(), it->end(), pf);
            pf += it->size();
        }
        return formula;
    }

    void deletePB2CNF(CPB2CNF* cpb2cnf)
    {
        delete reinterpret_cast<PB2CNF*>(cpb2cnf);
    }

    void freePtr(int32_t* ptr)
    {
        free(ptr);
    }
}