Coverage for python/lsst/pipe/tasks/mocks/mockSelect.py: 33%

Shortcuts on this page

r m x p   toggle line displays

j k   next/prev highlighted chunk

0   (zero) top of page

1   (one) first highlighted chunk

16 statements  

1# 

2# LSST Data Management System 

3# Copyright 2008, 2009, 2010, 2011, 2012 LSST Corporation. 

4# 

5# This product includes software developed by the 

6# LSST Project (http://www.lsst.org/). 

7# 

8# This program is free software: you can redistribute it and/or modify 

9# it under the terms of the GNU General Public License as published by 

10# the Free Software Foundation, either version 3 of the License, or 

11# (at your option) any later version. 

12# 

13# This program is distributed in the hope that it will be useful, 

14# but WITHOUT ANY WARRANTY; without even the implied warranty of 

15# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

16# GNU General Public License for more details. 

17# 

18# You should have received a copy of the LSST License Statement and 

19# the GNU General Public License along with this program. If not, 

20# see <http://www.lsstcorp.org/LegalNotices/>. 

21# 

22import lsst.pex.config 

23import lsst.afw.table 

24import lsst.pipe.base 

25 

26 

27class MockSelectImagesTask(lsst.pipe.base.Task): 

28 """Simple select images task that just returns all the objects we simulated. 

29 """ 

30 

31 ConfigClass = lsst.pex.config.Config 

32 

33 def runDataRef(self, dataRef, coordList, makeDataRefList=True, selectDataList=[]): 

34 observations = dataRef.butlerSubset.butler.get("observations", tract=dataRef.dataId["tract"]) 

35 assert(makeDataRefList) # this is all we make, so the user better want it 

36 butler = dataRef.butlerSubset.butler 

37 visitKey = observations.getSchema().find("visit").key 

38 ccdKey = observations.getSchema().find("ccd").key 

39 dataRefList = [] 

40 for record in observations: 

41 dataId = {"visit": record.getI(visitKey), "ccd": record.getI(ccdKey)} 

42 dataRefList.append(butler.dataRef(datasetType="calexp", dataId=dataId)) 

43 return lsst.pipe.base.Struct(dataRefList=dataRefList)