Skip to content

Commit e6fb3e9

Browse files
committed
Add new mask parameters to all modules that mask
1 parent 4f04599 commit e6fb3e9

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

Diff for: src/commons/Parameters.cpp

+4
Original file line numberDiff line numberDiff line change
@@ -767,7 +767,9 @@ Parameters::Parameters():
767767
splitsequence.push_back(&PARAM_V);
768768
769769
// mask sequence
770+
masksequence.push_back(&PARAM_MASK_RESIDUES);
770771
masksequence.push_back(&PARAM_MASK_PROBABILTY);
772+
masksequence.push_back(&PARAM_MASK_LOWER_CASE);
771773
masksequence.push_back(&PARAM_MASK_N_REPEAT);
772774
masksequence.push_back(&PARAM_THREADS);
773775
masksequence.push_back(&PARAM_COMPRESSED);
@@ -841,6 +843,8 @@ Parameters::Parameters():
841843
makepaddedseqdb.push_back(&PARAM_SCORE_BIAS);
842844
makepaddedseqdb.push_back(&PARAM_MASK_RESIDUES);
843845
makepaddedseqdb.push_back(&PARAM_MASK_PROBABILTY);
846+
makepaddedseqdb.push_back(&PARAM_MASK_LOWER_CASE);
847+
makepaddedseqdb.push_back(&PARAM_MASK_N_REPEAT);
844848
makepaddedseqdb.push_back(&PARAM_WRITE_LOOKUP);
845849
makepaddedseqdb.push_back(&PARAM_THREADS);
846850
makepaddedseqdb.push_back(&PARAM_V);

Diff for: src/util/masksequence.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ int masksequence(int argc, const char **argv, const Command& command) {
4545
#pragma omp for schedule(dynamic, 1)
4646
for (size_t id = 0; id < reader.getSize(); ++id) {
4747
seq.mapSequence(id, reader.getDbKey(id), reader.getData(id, thread_idx), reader.getSeqLen(id));
48-
masker.maskSequence(seq, true, par.maskProb, par.maskLowerCaseMode, par.maskNrepeats);
48+
masker.maskSequence(seq, par.mask, par.maskProb, par.maskLowerCaseMode, par.maskNrepeats);
4949
memcpy(charSequence, seq.getSeqData(), seq.L * sizeof(char));
5050
masker.applySoftmasking(charSequence, seq.numSequence, seq.L);
5151
charSequence[seq.L] = '\n';

0 commit comments

Comments
 (0)