|
5 | 5 | import sys |
6 | 6 | from warnings import warn |
7 | 7 |
|
8 | | -import ele |
| 8 | +from ele import element_from_symbol |
9 | 9 | import freud |
10 | 10 | from openbabel import openbabel |
11 | 11 | from openbabel import pybel |
@@ -229,7 +229,7 @@ def get_chromo_ids_smiles(snap, smarts_str, conversion_dict=None): |
229 | 229 | if conversion_dict is not None: |
230 | 230 | element = conversion_dict[snap.particles.types[typeid]] |
231 | 231 | else: |
232 | | - element = ele.element_from_symbol(snap.particles.types[typeid]) |
| 232 | + element = element_from_symbol(snap.particles.types[typeid]) |
233 | 233 | a.SetAtomicNum(element.atomic_number) |
234 | 234 | a.SetVector(*[float(x) for x in unwrapped_positions[i]]) |
235 | 235 |
|
@@ -335,90 +335,90 @@ def set_neighbors_voronoi(chromo_list, snap, conversion_dict=None, d_cut=10): |
335 | 335 |
|
336 | 336 |
|
337 | 337 | conversion_dict = { |
338 | | - "S1": ele.element_from_symbol("S"), |
339 | | - "H1": ele.element_from_symbol("H"), |
340 | | - "C5": ele.element_from_symbol("C"), |
341 | | - "C1": ele.element_from_symbol("C"), |
342 | | - "C4": ele.element_from_symbol("C"), |
343 | | - "C6": ele.element_from_symbol("C"), |
344 | | - "C8": ele.element_from_symbol("C"), |
345 | | - "C9": ele.element_from_symbol("C"), |
346 | | - "C3": ele.element_from_symbol("C"), |
347 | | - "C7": ele.element_from_symbol("C"), |
348 | | - "C2": ele.element_from_symbol("C"), |
349 | | - "C10": ele.element_from_symbol("C"), |
| 338 | + "S1": element_from_symbol("S"), |
| 339 | + "H1": element_from_symbol("H"), |
| 340 | + "C5": element_from_symbol("C"), |
| 341 | + "C1": element_from_symbol("C"), |
| 342 | + "C4": element_from_symbol("C"), |
| 343 | + "C6": element_from_symbol("C"), |
| 344 | + "C8": element_from_symbol("C"), |
| 345 | + "C9": element_from_symbol("C"), |
| 346 | + "C3": element_from_symbol("C"), |
| 347 | + "C7": element_from_symbol("C"), |
| 348 | + "C2": element_from_symbol("C"), |
| 349 | + "C10":element_from_symbol("C"), |
350 | 350 | } |
351 | 351 |
|
352 | 352 |
|
353 | 353 | amber_dict = { |
354 | | - "c": ele.element_from_symbol("C"), |
355 | | - "c1": ele.element_from_symbol("C"), |
356 | | - "c2": ele.element_from_symbol("C"), |
357 | | - "c3": ele.element_from_symbol("C"), |
358 | | - "ca": ele.element_from_symbol("C"), |
359 | | - "cp": ele.element_from_symbol("C"), |
360 | | - "cq": ele.element_from_symbol("C"), |
361 | | - "cc": ele.element_from_symbol("C"), |
362 | | - "cd": ele.element_from_symbol("C"), |
363 | | - "ce": ele.element_from_symbol("C"), |
364 | | - "cf": ele.element_from_symbol("C"), |
365 | | - "cg": ele.element_from_symbol("C"), |
366 | | - "ch": ele.element_from_symbol("C"), |
367 | | - "cx": ele.element_from_symbol("C"), |
368 | | - "cy": ele.element_from_symbol("C"), |
369 | | - "cu": ele.element_from_symbol("C"), |
370 | | - "cv": ele.element_from_symbol("C"), |
371 | | - "h1": ele.element_from_symbol("H"), |
372 | | - "h2": ele.element_from_symbol("H"), |
373 | | - "h3": ele.element_from_symbol("H"), |
374 | | - "h4": ele.element_from_symbol("H"), |
375 | | - "h5": ele.element_from_symbol("H"), |
376 | | - "ha": ele.element_from_symbol("H"), |
377 | | - "hc": ele.element_from_symbol("H"), |
378 | | - "hn": ele.element_from_symbol("H"), |
379 | | - "ho": ele.element_from_symbol("H"), |
380 | | - "hp": ele.element_from_symbol("H"), |
381 | | - "hs": ele.element_from_symbol("H"), |
382 | | - "hw": ele.element_from_symbol("H"), |
383 | | - "hx": ele.element_from_symbol("H"), |
384 | | - "f": ele.element_from_symbol("F"), |
385 | | - "cl": ele.element_from_symbol("Cl"), |
386 | | - "br": ele.element_from_symbol("Br"), |
387 | | - "i": ele.element_from_symbol("I"), |
388 | | - "n": ele.element_from_symbol("N"), |
389 | | - "n1": ele.element_from_symbol("N"), |
390 | | - "n2": ele.element_from_symbol("N"), |
391 | | - "n3": ele.element_from_symbol("N"), |
392 | | - "n4": ele.element_from_symbol("N"), |
393 | | - "na": ele.element_from_symbol("N"), |
394 | | - "nb": ele.element_from_symbol("N"), |
395 | | - "nc": ele.element_from_symbol("N"), |
396 | | - "nd": ele.element_from_symbol("N"), |
397 | | - "ne": ele.element_from_symbol("N"), |
398 | | - "nf": ele.element_from_symbol("N"), |
399 | | - "nh": ele.element_from_symbol("N"), |
400 | | - "no": ele.element_from_symbol("N"), |
401 | | - "o": ele.element_from_symbol("O"), |
402 | | - "oh": ele.element_from_symbol("O"), |
403 | | - "os": ele.element_from_symbol("O"), |
404 | | - "ow": ele.element_from_symbol("O"), |
405 | | - "p2": ele.element_from_symbol("P"), |
406 | | - "p3": ele.element_from_symbol("P"), |
407 | | - "p4": ele.element_from_symbol("P"), |
408 | | - "p5": ele.element_from_symbol("P"), |
409 | | - "pb": ele.element_from_symbol("P"), |
410 | | - "pc": ele.element_from_symbol("P"), |
411 | | - "pd": ele.element_from_symbol("P"), |
412 | | - "pe": ele.element_from_symbol("P"), |
413 | | - "pf": ele.element_from_symbol("P"), |
414 | | - "px": ele.element_from_symbol("P"), |
415 | | - "py": ele.element_from_symbol("P"), |
416 | | - "s": ele.element_from_symbol("S"), |
417 | | - "s2": ele.element_from_symbol("S"), |
418 | | - "s4": ele.element_from_symbol("S"), |
419 | | - "s6": ele.element_from_symbol("S"), |
420 | | - "sh": ele.element_from_symbol("S"), |
421 | | - "ss": ele.element_from_symbol("S"), |
422 | | - "sx": ele.element_from_symbol("S"), |
423 | | - "sy": ele.element_from_symbol("S"), |
| 354 | + "c": element_from_symbol("C"), |
| 355 | + "c1": element_from_symbol("C"), |
| 356 | + "c2": element_from_symbol("C"), |
| 357 | + "c3": element_from_symbol("C"), |
| 358 | + "ca": element_from_symbol("C"), |
| 359 | + "cp": element_from_symbol("C"), |
| 360 | + "cq": element_from_symbol("C"), |
| 361 | + "cc": element_from_symbol("C"), |
| 362 | + "cd": element_from_symbol("C"), |
| 363 | + "ce": element_from_symbol("C"), |
| 364 | + "cf": element_from_symbol("C"), |
| 365 | + "cg": element_from_symbol("C"), |
| 366 | + "ch": element_from_symbol("C"), |
| 367 | + "cx": element_from_symbol("C"), |
| 368 | + "cy": element_from_symbol("C"), |
| 369 | + "cu": element_from_symbol("C"), |
| 370 | + "cv": element_from_symbol("C"), |
| 371 | + "h1": element_from_symbol("H"), |
| 372 | + "h2": element_from_symbol("H"), |
| 373 | + "h3": element_from_symbol("H"), |
| 374 | + "h4": element_from_symbol("H"), |
| 375 | + "h5": element_from_symbol("H"), |
| 376 | + "ha": element_from_symbol("H"), |
| 377 | + "hc": element_from_symbol("H"), |
| 378 | + "hn": element_from_symbol("H"), |
| 379 | + "ho": element_from_symbol("H"), |
| 380 | + "hp": element_from_symbol("H"), |
| 381 | + "hs": element_from_symbol("H"), |
| 382 | + "hw": element_from_symbol("H"), |
| 383 | + "hx": element_from_symbol("H"), |
| 384 | + "f": element_from_symbol("F"), |
| 385 | + "cl": element_from_symbol("Cl"), |
| 386 | + "br": element_from_symbol("Br"), |
| 387 | + "i": element_from_symbol("I"), |
| 388 | + "n": element_from_symbol("N"), |
| 389 | + "n1":element_from_symbol("N"), |
| 390 | + "n2":element_from_symbol("N"), |
| 391 | + "n3":element_from_symbol("N"), |
| 392 | + "n4": element_from_symbol("N"), |
| 393 | + "na": element_from_symbol("N"), |
| 394 | + "nb": element_from_symbol("N"), |
| 395 | + "nc": element_from_symbol("N"), |
| 396 | + "nd": element_from_symbol("N"), |
| 397 | + "ne": element_from_symbol("N"), |
| 398 | + "nf": element_from_symbol("N"), |
| 399 | + "nh": element_from_symbol("N"), |
| 400 | + "no": element_from_symbol("N"), |
| 401 | + "o": element_from_symbol("O"), |
| 402 | + "oh": element_from_symbol("O"), |
| 403 | + "os": element_from_symbol("O"), |
| 404 | + "ow": element_from_symbol("O"), |
| 405 | + "p2": element_from_symbol("P"), |
| 406 | + "p3": element_from_symbol("P"), |
| 407 | + "p4": element_from_symbol("P"), |
| 408 | + "p5": element_from_symbol("P"), |
| 409 | + "pb": element_from_symbol("P"), |
| 410 | + "pc": element_from_symbol("P"), |
| 411 | + "pd": element_from_symbol("P"), |
| 412 | + "pe": element_from_symbol("P"), |
| 413 | + "pf": element_from_symbol("P"), |
| 414 | + "px": element_from_symbol("P"), |
| 415 | + "py": element_from_symbol("P"), |
| 416 | + "s": element_from_symbol("S"), |
| 417 | + "s2": element_from_symbol("S"), |
| 418 | + "s4": element_from_symbol("S"), |
| 419 | + "s6": element_from_symbol("S"), |
| 420 | + "sh": element_from_symbol("S"), |
| 421 | + "ss": element_from_symbol("S"), |
| 422 | + "sx": element_from_symbol("S"), |
| 423 | + "sy": element_from_symbol("S"), |
424 | 424 | } |
0 commit comments