roll_playback Namespace Reference

Functions
def	copy_next_lumi
def	sanity_check

Variables
string	calib_source = '/fff/ramdisk/playback_files/run228928'
	copy_file = True
int	current_lumi = 1
string	dat_source = '/fff/ramdisk/playback_files/run228928'
string	destination = '/fff/ramdisk'
dictionary	file_types
list	files_copied = []
int	files_copied_buffer_len = 60
tuple	j = i%len(pb_jsn_files)
tuple	k = i%len(calib_jsn_files)
int	lumi_len = 23
int	lumi_padding = 4
	lumi_skip_at = None
int	lumi_skip_length = 10
string	output_dir = destination+'/'
string	pb_source = '/fff/ramdisk/playback_files/run228928'
tuple	run_length = len(dat_jsn_files)
int	run_padding = 6
int	run_switch_interval = 90

Function Documentation

def roll_playback.copy_next_lumi	(		jsn_file,
			file,
			run_number,
			current_lumi,
			source,
			destination,
			copy_file = True
	)

Definition at line 68 of file roll_playback.py.

References reco.print(), and sanity_check().

def copy_next_lumi(jsn_file, file, run_number, current_lumi, source, destination, copy_file=True):
    index = current_lumi % len(jsn_file)

    # copy the file
    input_fn = source + '/' + file
    output_fn = file
    run_start = output_fn.find('run') + 3
    output_fn = output_fn[:run_start] + str(run_number).zfill(run_padding) + output_fn[run_start + run_padding:]
    lumi_start = output_fn.find('ls') + 2
    output_fn = destination + '/' + output_fn[:lumi_start] + str(current_lumi).zfill(lumi_padding) + output_fn[lumi_start + lumi_padding:]
    if copy_file:
        os.link(input_fn, output_fn) # instead of copying the file create a hard link
        print(input_fn + ' -> ' + output_fn)

    # load the original json contents
    input_jsn_fn = source + '/' + jsn_file
    input_jsn = open(input_jsn_fn, 'r')
    jsn_data = json.load(input_jsn)
    input_jsn.close()

    # generate the output jsn file name
    output_jsn_fn = jsn_file
    run_start = output_jsn_fn.find('run') + 3
    output_jsn_fn = output_jsn_fn[:run_start] + str(run_number).zfill(run_padding) + output_jsn_fn[run_start + run_padding:]
    lumi_start = output_jsn_fn.find('ls') + 2
    output_jsn_fn = destination + '/' + output_jsn_fn[:lumi_start] + str(current_lumi).zfill(lumi_padding) + output_jsn_fn[lumi_start + lumi_padding:]

    # modify the json file contents
    jsn_data['data'][3] = output_fn[output_fn.rfind('/')+1:]

    # create the outpuf jsn file
    output_jsn = open(output_jsn_fn, 'w')
    output_jsn.write(json.dumps(jsn_data, indent=4))
    output_jsn.close()

    print(input_jsn_fn + ' -> ' + output_jsn_fn)

    return (output_jsn_fn, output_fn) if copy_file else (output_jsn_fn, )