Skip to content

Commit 6d0f2ea

Browse files
Fix mask optimization excluding format, version and dark module.
1 parent 456b01d commit 6d0f2ea

1 file changed

Lines changed: 13 additions & 13 deletions

File tree

qrcode/main.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -158,11 +158,11 @@ def make(self, fit=True):
158158
if fit or (self.version is None):
159159
self.best_fit(start=self.version)
160160
if self.mask_pattern is None:
161-
self.makeImpl(False, self.best_mask_pattern())
161+
self.makeImpl(self.best_mask_pattern())
162162
else:
163-
self.makeImpl(False, self.mask_pattern)
163+
self.makeImpl(self.mask_pattern)
164164

165-
def makeImpl(self, test, mask_pattern):
165+
def makeImpl(self, mask_pattern):
166166
self.modules_count = self.version * 4 + 17
167167

168168
if self.version in precomputed_qr_blanks:
@@ -179,10 +179,10 @@ def makeImpl(self, test, mask_pattern):
179179

180180
precomputed_qr_blanks[self.version] = copy_2d_array(self.modules)
181181

182-
self.setup_type_info(test, mask_pattern)
182+
self.setup_type_info(mask_pattern)
183183

184184
if self.version >= 7:
185-
self.setup_type_number(test)
185+
self.setup_type_number()
186186

187187
if self.data_cache is None:
188188
self.data_cache = util.create_data(
@@ -246,7 +246,7 @@ def best_mask_pattern(self):
246246
pattern = 0
247247

248248
for i in range(8):
249-
self.makeImpl(True, i)
249+
self.makeImpl(i)
250250

251251
lost_point = util.lost_point(self.modules)
252252

@@ -430,24 +430,24 @@ def setup_position_adjust_pattern(self):
430430
else:
431431
self.modules[row + r][col + c] = False
432432

433-
def setup_type_number(self, test):
433+
def setup_type_number(self):
434434
bits = util.BCH_type_number(self.version)
435435

436436
for i in range(18):
437-
mod = not test and ((bits >> i) & 1) == 1
437+
mod = ((bits >> i) & 1) == 1
438438
self.modules[i // 3][i % 3 + self.modules_count - 8 - 3] = mod
439439

440440
for i in range(18):
441-
mod = not test and ((bits >> i) & 1) == 1
441+
mod = ((bits >> i) & 1) == 1
442442
self.modules[i % 3 + self.modules_count - 8 - 3][i // 3] = mod
443443

444-
def setup_type_info(self, test, mask_pattern):
444+
def setup_type_info(self, mask_pattern):
445445
data = (self.error_correction << 3) | mask_pattern
446446
bits = util.BCH_type_info(data)
447447

448448
# vertical
449449
for i in range(15):
450-
mod = not test and ((bits >> i) & 1) == 1
450+
mod = ((bits >> i) & 1) == 1
451451

452452
if i < 6:
453453
self.modules[i][8] = mod
@@ -458,7 +458,7 @@ def setup_type_info(self, test, mask_pattern):
458458

459459
# horizontal
460460
for i in range(15):
461-
mod = not test and ((bits >> i) & 1) == 1
461+
mod = ((bits >> i) & 1) == 1
462462

463463
if i < 8:
464464
self.modules[8][self.modules_count - i - 1] = mod
@@ -468,7 +468,7 @@ def setup_type_info(self, test, mask_pattern):
468468
self.modules[8][15 - i - 1] = mod
469469

470470
# fixed module
471-
self.modules[self.modules_count - 8][8] = not test
471+
self.modules[self.modules_count - 8][8] = 1
472472

473473
def map_data(self, data, mask_pattern):
474474
inc = -1

0 commit comments

Comments
 (0)