-
Notifications
You must be signed in to change notification settings - Fork 6
/
increment_global.py
30 lines (26 loc) · 1.18 KB
/
increment_global.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import argparse
from larpix import Configuration_v2
def main(*files, global_threshold_inc=0, dry_run=False, **kwargs):
for file in files:
config = Configuration_v2()
config.load(file)
if config.threshold_global + global_threshold_inc >= 255:
config.threshold_global = 255
else:
config.threshold_global += global_threshold_inc
if not dry_run:
config.write(file, force=True)
else:
print('loaded {}'.format(file))
print('set global threshold to {}'.format(config.threshold_global))
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument('input_files', nargs='+', help='''files to modify''')
parser.add_argument('--inc', type=int, default=0, help='''amount to change global threshold by''')
parser.add_argument('--dry_run', action='store_true', help='''print stuff but don't do anything''')
args = parser.parse_args()
main(
*args.input_files,
global_threshold_inc=args.inc,
dry_run=args.dry_run
)