Coverage for examples/suprime.py: 0%

34 statements  

« prev     ^ index     » next       coverage.py v6.5.0, created at 2023-02-16 03:17 -0800

1import os.path 

2 

3import lsst.daf.persistence as dafPersist 

4import lsst.obs.suprimecam as obsSc 

5from .utils import WrapperMapper 

6 

7 

8def _getSuprimeMapper(rootdir=None, calibdir=None, outrootdir=None): 

9 if rootdir is None: 

10 rootdir = os.path.join(os.environ['HOME'], 'lsst', 'ACT-data') 

11 if calibdir is None: 

12 calibdir = os.path.join(rootdir, 'CALIB') 

13 mapperArgs = dict(root=rootdir, calibRoot=calibdir, outputRoot=outrootdir) 

14 mapper = obsSc.SuprimecamMapper(**mapperArgs) 

15 # return mapper 

16 wrap = WrapperMapper(mapper) 

17 return wrap 

18 

19 

20def _getSuprimeButler(rootdir=None, calibdir=None, outrootdir=None): 

21 mapper = _getSuprimeMapper(rootdir, calibdir, outrootdir) 

22 butlerFactory = dafPersist.ButlerFactory(mapper=mapper) 

23 butler = butlerFactory.create() 

24 return butler 

25 

26 

27def getSuprimeDataref(visit, ccd, single=True, rootdir=None, calibdir=None, outrootdir=None): 

28 butler = _getSuprimeButler(rootdir=rootdir, calibdir=calibdir, outrootdir=outrootdir) 

29 print('Butler', butler) 

30 dataRef = butler.subset('raw', dataId=dict(visit=visit, ccd=ccd)) 

31 print('dataRef:', dataRef) 

32 print('len(dataRef):', len(dataRef)) 

33 for dr in dataRef: 

34 print(' ', dr) 

35 if single: 

36 assert(len(dataRef) == 1) 

37 # dataRef doesn't support indexing, but it does support iteration? 

38 dr = None 

39 for dr in dataRef: 

40 break 

41 assert(dr) 

42 return dr 

43 return dataRef