generate_datasets.py

####################################################################################################
# ### generate_datasets.py
#
# Script for mass generating datasets. This script does not contain fancy conditions nor settings.
# Datasets tend to be generated only once and need an easy method for automatic generation.
# Generation of specific datasets are handled in blocks and can be enabled by removing the comments.
# Datasets are automatically generated in the /datasets/ sub-directory.
#
####################################################################################################

import os
from SiFiCCNN.root import RootParser, RootFiles

# get current path, go two subdirectories higher
path = os.getcwd()
while True:
    if os.path.basename(path) == "SiFiCC-SplitNeuralNetwork":
        break
    path = os.path.abspath(os.path.join(path, os.pardir))

path_main = path
path_root = path + "/root_files/"
path_datasets = path + "/datasets/"

####################################################################################################
# Graph SiPM
####################################################################################################
"""
from analysis.EdgeConvResNetSiPM import downloader

n = 2000
root_parser_0mm = RootParser.RootParser(path_root + RootFiles.fourtoone_BP0mm_simv4)
root_parser_5mm = RootParser.RootParser(path_root + RootFiles.fourtoone_BP5mm_simv4)
root_parser_cont = RootParser.RootParser(path_root + RootFiles.fourtoone_CONT_simv4)
root_parser_m5mm = RootParser.RootParser(path_root + RootFiles.fourtoone_BPm5mm_simv4)
root_parser_10mm = RootParser.RootParser(path_root + RootFiles.fourtoone_BP10mm_simv4)
downloader.load(root_parser_0mm, path=path_datasets, n=n)
downloader.load(root_parser_5mm, path=path_datasets, n=n)
downloader.load(root_parser_cont, path=path_datasets, n=100000)
downloader.load(root_parser_m5mm, path=path_datasets, n=n)
downloader.load(root_parser_10mm, path=path_datasets, n=n)
"""
####################################################################################################
# Graph Cluster
####################################################################################################

from analysis.EdgeConvResNetCluster import downloader

n = 100000
files = [RootFiles.onetoone_CONT_simV2,
         RootFiles.onetoone_BP0mm_simV2,
         RootFiles.onetoone_BP5mm_simV2,
         RootFiles.onetoone_BPminus5mm_simV2]

for file in files:
    root_parser = RootParser.RootParser(path_root + file)
    downloader.load(root_parser, path=path_datasets, n=n)