コード例 #1
0
ファイル: ana_ModifyIteration.py プロジェクト: enzyx/hdWE
parser.add_argument('-f',
                    '--overwrite',
                    dest="overwrite",
                    default=False,
                    action='store_true',
                    help="Overwrite output files.")

args = parser.parse_args()
logger = Logger(args.logdir)
try:
    iteration = logger.loadIteration(args.iteration_index)
except IOError:
    print("Could not find file for iteration {}!".format(args.iteration_index))
    sys.exit(-1)

if not args.overwrite:
    print("Running dry! Use -f to save changes.")

# Change target number of segments if required
if args.bins_target:
    for binId, targetNumberOfSegments in zip(*[iter(args.bins_target)] * 2):
        print("bin: {} targetNumberOfSegments: {} --> {}".format(
            binId, iteration.bins[binId].getTargetNumberOfSegments(),
            targetNumberOfSegments))
        iteration.bins[
            binId].target_number_of_segments = targetNumberOfSegments

if args.overwrite:
    print("Overwriting iteration file: ", args.iteration_index)
    logger.logIteration(iteration)