def setup(self):
"""Setup main parameters"""
pd.set_option('display.width', 120)
base.iedbmhc1path = self.iedbmhc1_path
base.iedbmhc2path = self.iedbmhc2_path
self.sequences = get_sequences(self.sequence_file)
self.mhc1_alleles = self.mhc1_alleles.split(',')
self.mhc2_alleles = self.mhc2_alleles.split(',')
if len(self.mhc1_alleles) == 0 and len(self.mhc2_alleles) == 0:
return False
self.predictors = self.predictors.split(',')
for p in self.predictors:
if p not in base.predictors:
print('unknown predictor in config file. Use:')
show_predictors()
return False
if self.mhc1_alleles[0] in base.mhc1_presets:
self.mhc1_alleles = base.get_preset_alleles(self.mhc1_alleles[0])
elif self.mhc2_alleles[0] in base.mhc2_presets:
self.mhc2_alleles = base.get_preset_alleles(self.mhc2_alleles[0])
self.cutoff = float(self.cutoff)
self.names = self.names.split(',')
if self.names == ['']: self.names = None
if not os.path.exists(self.path) and self.path != '':
os.mkdir(self.path)
return True
def setup(self):
"""Setup main parameters"""
if base.check_snap() == True:
print ('running in a snap')
cmd = base.set_netmhcpan_cmd()
print ('netmhcpan cmd set to:', cmd)
pd.set_option('display.width', 120)
#override base.defaults entries if provided in conf
set_defaults(self.__dict__)
self.sequences = None
self.peptides = None
if self.sequence_file is not '':
self.sequences = get_sequences(self.sequence_file, header_sep=self.fasta_header_sep)
print ('input is %s protein sequences' %len(self.sequences))
elif self.peptide_file is not '':
self.peptides = read_names(self.peptide_file)
print ('input is %s peptides' %len(self.peptides))
if self.sequences is None and self.peptides is None:
print ('no valid inputs found')
return False
self.mhc1_alleles = get_alleles(self.mhc1_alleles)
self.mhc2_alleles = get_alleles(self.mhc2_alleles)
if len(self.mhc1_alleles)==0 and len(self.mhc2_alleles)==0:
return False
self.predictors = self.predictors.split(',')
for p in self.predictors:
if p not in base.predictors:
print ('unknown predictor in config file. Use:')
show_predictors()
return False
if self.mhc1_alleles[0] in base.mhc1_presets:
self.mhc1_alleles = base.get_preset_alleles(self.mhc1_alleles[0])
if self.mhc2_alleles[0] in base.mhc2_presets:
self.mhc2_alleles = base.get_preset_alleles(self.mhc2_alleles[0])
if type(self.cutoffs) is int:
self.cutoffs = [self.cutoffs]
else:
self.cutoffs = [float(i) for i in self.cutoffs.split(',')]
if os.path.exists(self.names):
self.names = read_names(self.names)
elif self.names == '':
self.names=None
else:
self.names = self.names.split(',')
if self.names is not None:
print ('selected sequences:', self.names)
if not os.path.exists(self.path) and self.path != '':
os.mkdir(self.path)
#copy input seqs to path
if self.sequences is not None:
self.sequences.to_csv(os.path.join(self.path, 'input.csv'),index=False)
return True
def setup(self):
"""Setup main parameters"""
if check_imports() == False:
return
#check_ensembl()
pd.set_option('display.width', 120)
base.iedbmhc1path = self.iedbmhc1_path
base.iedbmhc2path = self.iedbmhc2_path
self.vcf_files = self.vcf_files.split(',')
f = self.vcf_files[0]
fileext = os.path.splitext(f)[1]
if fileext == '.txt' and os.path.exists(f):
print('found file list')
self.vcf_files = read_names(f)
self.mhc1_alleles = self.mhc1_alleles.split(',')
self.mhc2_alleles = self.mhc2_alleles.split(',')
if len(self.mhc1_alleles) == 0 and len(self.mhc2_alleles) == 0:
return False
self.predictors = self.predictors.split(',')
for p in self.predictors:
if p not in base.predictors:
print('unknown predictor in config file. Use:')
show_predictors()
return False
if self.mhc1_alleles[0] in base.mhc1_presets:
self.mhc1_alleles = base.get_preset_alleles(self.mhc1_alleles[0])
elif self.mhc2_alleles[0] in base.mhc2_presets:
self.mhc2_alleles = base.get_preset_alleles(self.mhc2_alleles[0])
if type(self.cutoffs) is int or type(self.cutoffs) is float:
self.cutoffs = [self.cutoffs]
else:
self.cutoffs = [float(i) for i in self.cutoffs.split(',')]
self.names = None
if not os.path.exists(self.path) and self.path != '':
os.mkdir(self.path)
return True
def show_preset_alleles():
print ('preset allele list ids:')
for i in base.mhc1_presets+base.mhc2_presets:
print (i, len( base.get_preset_alleles(i)))