"""
.. See the NOTICE file distributed with this work for additional information
regarding copyright ownership.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""
from __future__ import print_function
import os
import pyBigWig
from dmp import dmp
from dm_generator.GenerateSampleBigWig import GenerateSampleBigWig
[docs]class bigwig_reader(object): # pylint: disable=invalid-name
"""
Class related to handling the functions for interacting directly with the
BigBed files. All required information should be passed to this class.
"""
def __init__(self, user_id, file_id, cnf_loc=''):
"""
Initialise the module and
Parameters
----------
user_id : str
Identifier to uniquely locate the users files. Can be set to
"common" if the files can be shared between users or 'test' for a
dummy file
file_id : str
Location of the file in the file system
resolution : int (Optional)
Level of resolution. This is optional, but only the functions
get_resolutions() and set_resolutions() can be called. Once the
resolution has been set then all functions are callable.
"""
# Open the bigwig file
# Open the bigbed file
if user_id == 'test':
resource_path = os.path.join(
os.path.dirname(__file__),
"../tests/data/sample.bw"
)
if os.path.isfile(resource_path) is False:
gsa = GenerateSampleBigWig()
gsa.main()
self.file_handle = pyBigWig.open(resource_path, 'r')
else:
dm_handle = dmp(cnf_loc)
file_obj = dm_handle.get_file_by_id(user_id, file_id)
self.file_handle = pyBigWig.open(file_obj['file_path'], 'r')
[docs] def get_chromosomes(self):
"""
List the chromosome names and lengths
Returns
-------
chromosomes : dict
Key value pair of chromosome name and the value is the length of the
chromosome.
"""
return self.file_handle.chroms()
[docs] def get_range(self, chr_id, start, end, file_type="wig"):
"""
Get entries in a given range
Parameters
----------
chr_id : str
Chromosome name
start : int
Start of the region to query
end : int
End of the region to query
format : string (OPTIONAL)
`wig` format returning the whole file as a string is the default
option. `list` will return the wig rows but as a list of lists.
Returns
-------
wig : str (DEFAULT)
List of strings for the rows in a wig file
wig_array : list
List of lists of each row for the wig file format
"""
bw_features = self.file_handle.intervals(str(chr_id), int(start), int(end))
if file_type == "wig":
wig_array = []
for feature in bw_features:
wig_array.append("\t".join([str(i) for i in feature]))
return "\n".join(wig_array)
return bw_features