Hide keyboard shortcuts

Hot-keys 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

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

# 

# LSST Data Management System 

# Copyright 2008, 2009, 2010 LSST Corporation. 

# 

# This product includes software developed by the 

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

# 

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

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

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

# (at your option) any later version. 

# 

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

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

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

# GNU General Public License for more details. 

# 

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

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

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

# 

import unittest 

import os 

 

import lsst.meas.algorithms as measAlg 

from lsst.meas.extensions.astrometryNet import ANetAstrometryTask, ANetAstrometryConfig 

from lsst.meas.base import SingleFrameMeasurementTask, SingleFrameMeasurementConfig 

import lsst.utils.tests 

import lsst.afw.detection as afwDetection 

import lsst.afw.image as afwImage 

import lsst.afw.table as afwTable 

 

import test_findAstrometryNetDataDir as helper 

 

 

class TestForceWcs(unittest.TestCase): 

 

def setUp(self): 

helper.setupAstrometryNetDataDir('t2155') 

 

def test1(self): 

fn = os.path.join(os.path.dirname(__file__), 'data', 'mini-r-3-113,0.fits.gz') 

print('Reading image', fn) 

exposure = afwImage.ExposureF(fn) 

exposure.setPsf(afwDetection.GaussianPsf(15, 15, 3)) 

schema = afwTable.SourceTable.makeMinimalSchema() 

idFactory = afwTable.IdFactory.makeSimple() 

 

dconf = measAlg.SourceDetectionConfig() 

dconf.reEstimateBackground = False 

dconf.includeThresholdMultiplier = 5. 

 

mconf = SingleFrameMeasurementConfig() 

 

aconf = ANetAstrometryConfig() 

aconf.forceKnownWcs = True 

 

det = measAlg.SourceDetectionTask(schema=schema, config=dconf) 

meas = SingleFrameMeasurementTask(schema, config=mconf) 

astrom = ANetAstrometryTask(schema, config=aconf, name='astrom') 

 

astrom.log.setLevel(astrom.log.TRACE) 

 

inwcs = exposure.getWcs() 

print('inwcs:', inwcs) 

instr = inwcs.getFitsMetadata().toString() 

print('inwcs:', instr) 

table = afwTable.SourceTable.make(schema, idFactory) 

sources = det.makeSourceCatalog(table, exposure, sigma=1).sources 

meas.measure(sources, exposure) 

 

for dosip in [False, True]: 

aconf.solver.calculateSip = dosip 

ast = astrom.run(sourceCat=sources, exposure=exposure) 

outwcs = exposure.getWcs() 

outstr = outwcs.getFitsMetadata().toString() 

if not dosip: 

self.assertEqual(inwcs, outwcs) 

self.assertEqual(instr, outstr) 

print('inwcs:', instr) 

print('outwcs:', outstr) 

print(len(ast.matches), 'matches') 

self.assertGreater(len(ast.matches), 10) 

 

 

class MyMemoryTestCase(lsst.utils.tests.MemoryTestCase): 

pass 

 

 

def setup_module(module): 

lsst.utils.tests.init() 

 

 

94 ↛ 95line 94 didn't jump to line 95, because the condition on line 94 was never trueif __name__ == "__main__": 

lsst.utils.tests.init() 

unittest.main()