module_selector_from_json Namespace Reference

Functions
def	find (bec=None, layer_disk=None, phi=None, eta=None, side=None, asdec=False, input_data="geometry.json", output_file=None)
def	merge (file_1="selected_modules_1.json", file_2="selected_modules_2.json", output_file=None)
def	select_random (frac=None, input_data=None, output_file=None)

Variables
def	data = find(bec = 0, layer_disk = 0, phi = [0,1,2,3,4], asdec = True, output_file="test_selected_modules.json")

Function Documentation

find()

def module_selector_from_json.find	(		bec = None,
			layer_disk = None,
			phi = None,
			eta = None,
			side = None,
			asdec = False,
			input_data = "geometry.json",
			output_file = None
	)

Definition at line 12 of file module_selector_from_json.py.

def find(bec=None, layer_disk=None, phi=None, eta=None, side=None, asdec=False, input_data="geometry.json", output_file=None):
    data = {}
 
    with open(input_data) as f:
        if input_data.endswith(".json"):
            data = json.load(f)
        else:
            print("Unexpected input file or format.")
            return []
 
    IDs = []
 
    for ID, info in data.items():
        if bec is not None and info["BEC"] != str(bec):
            continue
 
        if layer_disk is not None and info["LayerDisk"] != str(layer_disk):
            continue
 
        if phi is not None and info["PhiModule"] not in str([i for i in phi]):
            continue
 
        if eta is not None and info["EtaModule"] > str(eta):
            continue
 
        if side is not None and info["Side"] != str(side):
            continue
 
        if asdec:
            info["Decimal_ID"] = str(int(ID, 16))
 
        IDs.append(ID)
 
    with open(output_file, "w") as f:
        selected_data = {ID: data[ID] for ID in IDs}
        json.dump(selected_data, f, indent=4)
 
    return IDs
 
 

merge()

def module_selector_from_json.merge	(		file_1 = "selected_modules_1.json",
			file_2 = "selected_modules_2.json",
			output_file = None
	)

Definition at line 52 of file module_selector_from_json.py.

def merge(file_1="selected_modules_1.json", file_2="selected_modules_2.json", output_file=None):
    data1 = {}
    data2 = {}
    IDs = []
 
    with open(file_1) as f1:
        if file_1.endswith(".json"):
            data1 = json.load(f1)
        else:
            print("Unexpected input file_1 or format.")
            return []
 
    IDs.extend(data1.keys())
 
    with open(file_2) as f2:
        if file_2.endswith(".json"):
            data2 = json.load(f2)
        else:
            print("Unexpected input file_2 or format.")
            return []
 
    IDs.extend(data2.keys())
 
    with open(output_file, "w") as f3:
        selected_data = {}
        for ID in IDs:
            if ID in data1:
                selected_data[ID] = data1[ID]
            elif ID in data2:
                selected_data[ID] = data2[ID]
        json.dump(selected_data, f3, indent=4)
 
    return IDs
 

select_random()

def module_selector_from_json.select_random	(		frac = None,
			input_data = None,
			output_file = None
	)

Definition at line 86 of file module_selector_from_json.py.

def select_random(frac = None, input_data=None, output_file=None):
    data = {}
 
    with open(input_data) as f:
        if input_data.endswith(".json"):
            data = json.load(f)
        else:
            print("Unexpected input file or format.")
            return []
 
    IDs = []
 
    num = int(round(frac * len(data), 0))
    IDs = random.choices(list(data.keys()), k=num)
 
    for ID, info in data.items():
        info["Decimal_ID"] = str(int(ID, 16))
 
    with open(output_file, "w") as f:
        selected_data = {ID: data[ID] for ID in IDs}
        json.dump(selected_data, f, indent=4)
 
    return IDs      
 
 

Variable Documentation

data

def module_selector_from_json.data = find(bec = 0, layer_disk = 0, phi = [0,1,2,3,4], asdec = True, output_file="test_selected_modules.json")

Definition at line 113 of file module_selector_from_json.py.