fix: repair moldb-enumerate-mols workflow and correct enumeration count

data-manager/moldb.yaml

@@ -150,7 +150,7 @@ jobs:
         - POSTGRES_DATABASE
         options:
           table: enumeration
-          count: 514
+          count: 529
         checks:
           exitCode: 0
       conformer:

moldb/enumerate_mols.nf

@@ -36,7 +36,7 @@ specification = file(params.specification)
 // includes
 include { extract_need_enum } from '../nf-processes/moldb/filter.nf' addParams(output: params.file)
 include { split_txt } from '../nf-processes/file/split_txt.nf' addParams(suffix: '.smi')
-include { enumerate } from '../nf-processes/rdkit/enumerate.nf'
+include { enumerate } from '../nf-processes/moldb/enumerate.nf'
 include { load_enum } from '../nf-processes/moldb/db_load.nf'
 
 def dateFormat = new java.text.SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss'+00:00'", Locale.UK)
@@ -56,7 +56,7 @@ workflow enumerate_forms {
     extract_need_enum(specification)
     split_txt(extract_need_enum.out)
     enumerate(split_txt.out.flatten())
-    load_enum(enumerate.out)
+    load_enum(enumerate.out[0])
 
     extract_need_enum.out.subscribe {
         now = dateFormat.format(new java.util.Date())
@@ -73,7 +73,7 @@ workflow enumerate_forms {
     }
 
     enumerate_count = 0
-    enumerate.out.subscribe {
+    enumerate.out[1].subscribe {
         now = dateFormat.format(new java.util.Date())
         enumerate_count++
         log.info("$now # PROGRESS -DONE- $wrkflw:enumerate $enumerate_count")

nf-processes/moldb/enumerate.nf

@@ -0,0 +1,45 @@
+params.enumerate_charges = true
+params.enumerate_chirals = true
+params.enumerate_tautomers = true
+params.combinatorial = false
+params.interval = 100
+params.try_embedding = true
+params.add_hydrogens = false
+params.max_tautomers = null
+params.min_ph = null // default is 5
+params.max_ph = null // default is 9
+params.min_charge = null
+params.max_charge = null
+params.num_charges = null
+
+
+process enumerate {
+
+    container 'informaticsmatters/vs-moldb:stable'
+
+    input:
+    file inputs
+
+    output:
+    file "enumerated-*.cxsmi"
+    env COUNT
+
+    """
+    python -m moldb.enumerate -i $inputs -o enumerated-${inputs.name}.cxsmi --interval $params.interval\
+      ${params.enumerate_charges ? '--enumerate-charges' : ''}\
+      ${params.enumerate_chirals ? '--enumerate-chirals' : ''}\
+      ${params.enumerate_tautomers ? '--enumerate-tautomers' : ''}\
+      ${params.combinatorial ? '--combinatorial' : ''}\
+      ${params.max_tautomers ? '--max-tautomers ' + params.max_tautomers : ''}\
+      ${params.min_ph ? '--min-ph ' + params.min_ph : ''}\
+      ${params.max_ph ? '--max-ph ' + params.max_ph : ''}\
+      ${params.min_charge ? '--min-charge ' + params.min_charge : ''}\
+      ${params.max_charge ? '--max-charge ' + params.max_charge : ''}\
+      ${params.num_charges ? '--num-charges ' + params.num_charges : ''}\
+      ${params.try_embedding ? '--try-embedding' : ''}\
+      ${params.add_hydrogens ? '--add-hydrogens' : ''}
+
+    # count the number of outputs
+    COUNT=\$(wc -l < 'enumerated-${inputs.name}.cxsmi')
+    """
+}