Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Extended features for rsdd-ocaml #7

Open
wants to merge 15 commits into
base: main
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
points to minsungc/rsdd-dappl
  • Loading branch information
minsungc committed Sep 28, 2023
commit aba5839fd0c5e6d92f184d711909839bd96b8e71
4 changes: 2 additions & 2 deletions Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 1 addition & 1 deletion Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ crate-type = ["staticlib", "cdylib"]

[dependencies]
ocaml = { version = "^1.0.0-beta" }
rsdd = { git = "https://github.com/neuppl/rsdd", rev = "1613459" }
rsdd = { git = "https://github.com/minsungc/rsdd-dappl", rev = "a0b49e6" }

[build-dependencies]
ocaml-build = {version = "^1.0.0-beta"}
31 changes: 5 additions & 26 deletions src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -198,17 +198,19 @@ pub struct RsddWmcParamsEU(WmcParams<ExpectedUtility>);
ocaml::custom!(RsddWmcParamsEU);

#[ocaml::func]
#[ocaml::sig("rsdd_bdd_ptr -> rsdd_var_label list -> int64 -> rsdd_wmc_params_e_u -> rsdd_expected_utility * rsdd_partial_model")]
pub fn bdd_bb(
#[ocaml::sig("rsdd_bdd_ptr -> rsdd_bdd_ptr -> rsdd_var_label list -> int64 -> rsdd_wmc_params_e_u -> rsdd_expected_utility * rsdd_partial_model")]
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'm going to spin up a changelog for the package in general, but to call out - if you change the API for any existing functions, we should def mention in a PR and make this the proper semver bump!

pub fn bdd_meu(
bdd: &'static RsddBddPtr,
evidence: &'static RsddBddPtr,
join_vars: ocaml::List<RsddVarLabel>,
num_vars: u64,
wmc: &RsddWmcParamsEU,
) -> (
ocaml::Pointer<RsddExpectedUtility>,
ocaml::Pointer<RsddPartialModel>,
) {
let (eu, pm) = bdd.0.bb(
let (eu, pm) = bdd.0.meu(
evidence.0,
&join_vars
.into_linked_list()
.iter()
Expand All @@ -220,29 +222,6 @@ pub fn bdd_bb(
(RsddExpectedUtility(eu).into(), RsddPartialModel(pm).into())
}

#[ocaml::func]
#[ocaml::sig("rsdd_bdd_ptr -> rsdd_var_label list -> int64 -> rsdd_wmc_params_e_u -> rsdd_expected_utility * rsdd_partial_model")]
pub fn bdd_meu(
bdd: &'static RsddBddPtr,
decision_vars: ocaml::List<RsddVarLabel>,
num_vars: u64,
wmc: &RsddWmcParamsEU,
) -> (
ocaml::Pointer<RsddExpectedUtility>,
ocaml::Pointer<RsddPartialModel>,
) {
let (eu, pm) = bdd.0.bb(
&decision_vars
.into_linked_list()
.iter()
.map(|x| x.0)
.collect::<Vec<_>>(),
num_vars as usize,
&wmc.0,
);
(RsddExpectedUtility(eu).into(), RsddPartialModel(pm).into())
}

#[ocaml::func]
#[ocaml::sig("((float * float) * (float * float)) list -> rsdd_wmc_params_e_u")]
pub fn new_wmc_params_eu(
Expand Down
87 changes: 22 additions & 65 deletions src/rsdd.ml
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

You might need to run dune fmt to make the diff cleaner (i.e. dune build && dune fmt); I will also add this to our internal dev docs.

Original file line number Diff line number Diff line change
Expand Up @@ -12,68 +12,25 @@ type rsdd_var_label
type rsdd_wmc_params_r
type rsdd_expected_utility
type rsdd_wmc_params_e_u

external mk_bdd_builder_default_order : int64 -> rsdd_bdd_builder
= "mk_bdd_builder_default_order"

external bdd_new_var : rsdd_bdd_builder -> bool -> int64 * rsdd_bdd_ptr
= "bdd_new_var"

external bdd_ite :
rsdd_bdd_builder ->
rsdd_bdd_ptr ->
rsdd_bdd_ptr ->
rsdd_bdd_ptr ->
rsdd_bdd_ptr = "bdd_ite"

external bdd_and :
rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_and"

external bdd_or :
rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_or"

external bdd_negate : rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr
= "bdd_negate"

external bdd_true : rsdd_bdd_builder -> rsdd_bdd_ptr = "bdd_true"
external bdd_false : rsdd_bdd_builder -> rsdd_bdd_ptr = "bdd_false"
external bdd_is_true : rsdd_bdd_ptr -> bool = "bdd_is_true"
external bdd_is_false : rsdd_bdd_ptr -> bool = "bdd_is_false"
external bdd_is_const : rsdd_bdd_ptr -> bool = "bdd_is_const"

external bdd_eq : rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr -> bool
= "bdd_eq"

external bdd_topvar : rsdd_bdd_ptr -> int64 = "bdd_topvar"
external bdd_low : rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_low"
external bdd_high : rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_high"
external bdd_wmc : rsdd_bdd_ptr -> rsdd_wmc_params_r -> float = "bdd_wmc"

external new_wmc_params_r : (float * float) list -> rsdd_wmc_params_r
= "new_wmc_params_r"

external bdd_bb :
rsdd_bdd_ptr ->
rsdd_var_label list ->
int64 ->
rsdd_wmc_params_e_u ->
rsdd_expected_utility * rsdd_partial_model = "bdd_bb"

external bdd_meu :
rsdd_bdd_ptr ->
rsdd_var_label list ->
int64 ->
rsdd_wmc_params_e_u ->
rsdd_expected_utility * rsdd_partial_model = "bdd_meu"

external new_wmc_params_eu :
((float * float) * (float * float)) list -> rsdd_wmc_params_e_u
= "new_wmc_params_eu"

external cnf_from_dimacs : string -> rsdd_cnf = "cnf_from_dimacs"

external bdd_builder_compile_cnf : rsdd_bdd_builder -> rsdd_cnf -> rsdd_bdd_ptr
= "bdd_builder_compile_cnf"

external bdd_model_count : rsdd_bdd_builder -> rsdd_bdd_ptr -> int64
= "bdd_model_count"
external mk_bdd_builder_default_order: int64 -> rsdd_bdd_builder = "mk_bdd_builder_default_order"
external bdd_new_var: rsdd_bdd_builder -> bool -> (int64 * rsdd_bdd_ptr) = "bdd_new_var"
external bdd_ite: rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr -> rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_ite"
external bdd_and: rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_and"
external bdd_or: rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_or"
external bdd_negate: rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_negate"
external bdd_true: rsdd_bdd_builder -> rsdd_bdd_ptr = "bdd_true"
external bdd_false: rsdd_bdd_builder -> rsdd_bdd_ptr = "bdd_false"
external bdd_is_true: rsdd_bdd_ptr -> bool = "bdd_is_true"
external bdd_is_false: rsdd_bdd_ptr -> bool = "bdd_is_false"
external bdd_is_const: rsdd_bdd_ptr -> bool = "bdd_is_const"
external bdd_eq: rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr -> bool = "bdd_eq"
external bdd_topvar: rsdd_bdd_ptr -> int64 = "bdd_topvar"
external bdd_low: rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_low"
external bdd_high: rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_high"
external bdd_wmc: rsdd_bdd_ptr -> rsdd_wmc_params_r -> float = "bdd_wmc"
external new_wmc_params_r: (float * float) list -> rsdd_wmc_params_r = "new_wmc_params_r"
external bdd_meu: rsdd_bdd_ptr -> rsdd_bdd_ptr -> rsdd_var_label list -> int64 -> rsdd_wmc_params_e_u -> rsdd_expected_utility * rsdd_partial_model = "bdd_meu"
external new_wmc_params_eu: ((float * float) * (float * float)) list -> rsdd_wmc_params_e_u = "new_wmc_params_eu"
external cnf_from_dimacs: string -> rsdd_cnf = "cnf_from_dimacs"
external bdd_builder_compile_cnf: rsdd_bdd_builder -> rsdd_cnf -> rsdd_bdd_ptr = "bdd_builder_compile_cnf"
external bdd_model_count: rsdd_bdd_builder -> rsdd_bdd_ptr -> int64 = "bdd_model_count"
87 changes: 22 additions & 65 deletions src/rsdd.mli
Original file line number Diff line number Diff line change
Expand Up @@ -12,68 +12,25 @@ type rsdd_var_label
type rsdd_wmc_params_r
type rsdd_expected_utility
type rsdd_wmc_params_e_u

external mk_bdd_builder_default_order : int64 -> rsdd_bdd_builder
= "mk_bdd_builder_default_order"

external bdd_new_var : rsdd_bdd_builder -> bool -> int64 * rsdd_bdd_ptr
= "bdd_new_var"

external bdd_ite :
rsdd_bdd_builder ->
rsdd_bdd_ptr ->
rsdd_bdd_ptr ->
rsdd_bdd_ptr ->
rsdd_bdd_ptr = "bdd_ite"

external bdd_and :
rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_and"

external bdd_or :
rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_or"

external bdd_negate : rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr
= "bdd_negate"

external bdd_true : rsdd_bdd_builder -> rsdd_bdd_ptr = "bdd_true"
external bdd_false : rsdd_bdd_builder -> rsdd_bdd_ptr = "bdd_false"
external bdd_is_true : rsdd_bdd_ptr -> bool = "bdd_is_true"
external bdd_is_false : rsdd_bdd_ptr -> bool = "bdd_is_false"
external bdd_is_const : rsdd_bdd_ptr -> bool = "bdd_is_const"

external bdd_eq : rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr -> bool
= "bdd_eq"

external bdd_topvar : rsdd_bdd_ptr -> int64 = "bdd_topvar"
external bdd_low : rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_low"
external bdd_high : rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_high"
external bdd_wmc : rsdd_bdd_ptr -> rsdd_wmc_params_r -> float = "bdd_wmc"

external new_wmc_params_r : (float * float) list -> rsdd_wmc_params_r
= "new_wmc_params_r"

external bdd_bb :
rsdd_bdd_ptr ->
rsdd_var_label list ->
int64 ->
rsdd_wmc_params_e_u ->
rsdd_expected_utility * rsdd_partial_model = "bdd_bb"

external bdd_meu :
rsdd_bdd_ptr ->
rsdd_var_label list ->
int64 ->
rsdd_wmc_params_e_u ->
rsdd_expected_utility * rsdd_partial_model = "bdd_meu"

external new_wmc_params_eu :
((float * float) * (float * float)) list -> rsdd_wmc_params_e_u
= "new_wmc_params_eu"

external cnf_from_dimacs : string -> rsdd_cnf = "cnf_from_dimacs"

external bdd_builder_compile_cnf : rsdd_bdd_builder -> rsdd_cnf -> rsdd_bdd_ptr
= "bdd_builder_compile_cnf"

external bdd_model_count : rsdd_bdd_builder -> rsdd_bdd_ptr -> int64
= "bdd_model_count"
external mk_bdd_builder_default_order: int64 -> rsdd_bdd_builder = "mk_bdd_builder_default_order"
external bdd_new_var: rsdd_bdd_builder -> bool -> (int64 * rsdd_bdd_ptr) = "bdd_new_var"
external bdd_ite: rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr -> rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_ite"
external bdd_and: rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_and"
external bdd_or: rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_or"
external bdd_negate: rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_negate"
external bdd_true: rsdd_bdd_builder -> rsdd_bdd_ptr = "bdd_true"
external bdd_false: rsdd_bdd_builder -> rsdd_bdd_ptr = "bdd_false"
external bdd_is_true: rsdd_bdd_ptr -> bool = "bdd_is_true"
external bdd_is_false: rsdd_bdd_ptr -> bool = "bdd_is_false"
external bdd_is_const: rsdd_bdd_ptr -> bool = "bdd_is_const"
external bdd_eq: rsdd_bdd_builder -> rsdd_bdd_ptr -> rsdd_bdd_ptr -> bool = "bdd_eq"
external bdd_topvar: rsdd_bdd_ptr -> int64 = "bdd_topvar"
external bdd_low: rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_low"
external bdd_high: rsdd_bdd_ptr -> rsdd_bdd_ptr = "bdd_high"
external bdd_wmc: rsdd_bdd_ptr -> rsdd_wmc_params_r -> float = "bdd_wmc"
external new_wmc_params_r: (float * float) list -> rsdd_wmc_params_r = "new_wmc_params_r"
external bdd_meu: rsdd_bdd_ptr -> rsdd_bdd_ptr -> rsdd_var_label list -> int64 -> rsdd_wmc_params_e_u -> rsdd_expected_utility * rsdd_partial_model = "bdd_meu"
external new_wmc_params_eu: ((float * float) * (float * float)) list -> rsdd_wmc_params_e_u = "new_wmc_params_eu"
external cnf_from_dimacs: string -> rsdd_cnf = "cnf_from_dimacs"
external bdd_builder_compile_cnf: rsdd_bdd_builder -> rsdd_cnf -> rsdd_bdd_ptr = "bdd_builder_compile_cnf"
external bdd_model_count: rsdd_bdd_builder -> rsdd_bdd_ptr -> int64 = "bdd_model_count"
3 changes: 1 addition & 2 deletions test/test.ml
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@ let () =
let robdd_builder = mk_bdd_builder_default_order 6L in
let bdd =
bdd_builder_compile_cnf robdd_builder
(cnf_from_dimacs "\n p cnf 6 3\n1 2 3 4 0\n-2 -3 4 5 0\n-4 -5 6 6 0\n")
in
(cnf_from_dimacs "\n p cnf 6 3\n1 2 3 4 0\n-2 -3 4 5 0\n-4 -5 6 6 0\n") in
let model_count = bdd_model_count robdd_builder bdd in
print_endline (Int64.to_string model_count)
Loading