Skip to content

Commit

Permalink
[IN PROGRESS] optimizing feature extraction
Browse files Browse the repository at this point in the history
  • Loading branch information
khb7840 committed Oct 2, 2024
1 parent 6444b99 commit d3948ab
Show file tree
Hide file tree
Showing 4 changed files with 143 additions and 122 deletions.
119 changes: 29 additions & 90 deletions Cargo.lock

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

5 changes: 3 additions & 2 deletions Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -24,14 +24,15 @@ pico-args = { version = "0.5.0", features = ["combined-flags"], path = "lib/pico
rayon = "1.8.0"
byteorder = "1.5.0"
memmap2 = "0.9.0"
dashmap = { version = "5.5.3", features = ["rayon"] }
rustc-hash = "1.1.0"
peak_alloc = "0.2.1"
flate2 = { version = "1.0.28" }
toml = "0.8.12"
regex = "1.10.4"
petgraph = "0.6.4"
libc = "0.2.155"
lazy_static = "1.5.0"
dashmap = { version = "6.1.0", features = ["rayon"] }


[build-dependencies]
Expand All @@ -40,4 +41,4 @@ cmake = { version = "0.1.50", optional = true }


[features]
foldcomp = ["bindgen", "cmake"]
foldcomp = ["bindgen", "cmake"]
7 changes: 4 additions & 3 deletions src/structure/core.rs
Original file line number Diff line number Diff line change
Expand Up @@ -383,12 +383,13 @@ impl CompactStructure {
(ca1, ca2, cb1, cb2, n1, n2)
{

let ca_dist = self.get_ca_distance(idx1, idx2).unwrap();
let ca_dist = ca1.calc_distance(&ca2);
if ca_dist > dist_cutoff {
return None;
}
let cb_dist = self.get_cb_distance(idx1, idx2).unwrap();
let ca_cb_angle = self.get_ca_cb_angle(idx1, idx2, true).unwrap();
let cb_dist = cb1.calc_distance(&cb2);
// let ca_cb_angle = self.get_ca_cb_angle(idx1, idx2, true).unwrap();
let ca_cb_angle = ca1.calc_angle(&cb1, &ca2, &cb2, true);
let theta1 = calc_torsion_radian(&n1, &ca1, &cb1, &cb2);
let theta2 = calc_torsion_radian(&cb1, &cb2, &ca2, &n2);
Some((ca_dist, cb_dist, ca_cb_angle, theta1, theta2))
Expand Down
Loading

0 comments on commit d3948ab

Please sign in to comment.