def load_qa_data(self, remove=False, multi_root=None):
""" Load the QA data files for a given exposure (currently yaml)
Args:
remove: bool, optional
Remove QA frame files
multi_root: str, optional
If provided, load the QA from a slurped file (preferred because it is fast)
"""
if multi_root is None:
qafiles = desiio.get_files(filetype='qa_' + self.type,
night=self.night,
expid=self.expid,
specprod_dir=self.specprod_dir)
# Load into frames
for camera, qadata_path in qafiles.items():
qa_frame = desiio.load_qa_frame(qadata_path)
# Remove?
if remove:
os.remove(qadata_path)
# Test
for key in ['expid', 'night']:
assert getattr(qa_frame, key) == getattr(self, key)
# Save
self.data['frames'][camera] = qa_frame.qa_data
else:
# Load
mdict = load_qa_multiexp(os.path.join(self.qaprod_dir, multi_root))
self.parse_multi_qa_dict(mdict)
def load_qa_data(self, remove=False):
""" Load the QA data files for a given exposure (currently yaml)
Args:
remove: bool, optional
Remove QA frame files
"""
from desispec import io as desiio
qafiles = desiio.get_files(filetype='qa_' + self.type,
night=self.night,
expid=self.expid,
specprod_dir=self.specprod_dir)
#import pdb; pdb.set_trace()
# Load into frames
for camera, qadata_path in qafiles.iteritems():
qa_frame = desiio.load_qa_frame(qadata_path)
# Remove?
if remove:
#import pdb; pdb.set_trace()
os.remove(qadata_path)
# Test
for key in ['expid', 'night']:
assert getattr(qa_frame, key) == getattr(self, key)
# Save
self.data['frames'][camera] = qa_frame.qa_data
def test_qa_frame_write_load_data(self):
# Write
frm0 = self._make_frame()
qafrm0 = QA_Frame(frm0)
write_qa_frame(self.qafile_b0, qafrm0)
# Load
qafrm2 = load_qa_frame(self.qafile_b0, frm0)
assert qafrm2.night == qafrm0.night
def test_qa_frame_write_load_data(self):
# Write
frm0 = self._make_frame()
qafrm0 = QA_Frame(frm0)
write_qa_frame(self.qafile_b0, qafrm0)
# Load
qafrm2 = load_qa_frame(self.qafile_b0, frm0)
assert qafrm2.night == qafrm0.night
def test_qa_frame_write_load_data(self):
# Write
frm0 = self._make_frame()
qafrm0 = QA_Frame(frm0)
# Write
outfile = findfile('qa_data', night=self.nights[0], expid=self.expids[0],
specprod_dir=self.testDir, camera='b0')
write_qa_frame(outfile, qafrm0)
self.files_written.append(outfile)
# Load
qafrm2 = load_qa_frame(outfile, frame_meta=frm0.meta)
assert qafrm2.night == qafrm0.night
def test_qa_frame_write_load_data(self):
# Write
frm0 = self._make_frame()
qafrm0 = QA_Frame(frm0)
# Write
outfile = findfile('qa_data', night=self.nights[0], expid=self.expids[0],
specprod_dir=self.testDir, camera='b0')
write_qa_frame(outfile, qafrm0)
self.files_written.append(outfile)
# Load
qafrm2 = load_qa_frame(outfile, frm0)
assert qafrm2.night == qafrm0.night
def load_qa_data(self, remove=False):
""" Load the QA data files for a given exposure (currently yaml)
Args:
remove: bool, optional
Remove QA frame files
"""
from desispec import io as desiio
qafiles = desiio.get_files(filetype='qa_'+self.type, night=self.night,
expid=self.expid,
specprod_dir=self.specprod_dir)
#import pdb; pdb.set_trace()
# Load into frames
for camera,qadata_path in qafiles.iteritems():
qa_frame = desiio.load_qa_frame(qadata_path)
# Remove?
if remove:
#import pdb; pdb.set_trace()
os.remove(qadata_path)
# Test
for key in ['expid','night']:
assert getattr(qa_frame,key) == getattr(self, key)
# Save
self.data['frames'][camera] = qa_frame.qa_data
