Skip to content
GitLab
An error occurred while loading the file. Please try again.
Switch branch/tag
Find file BlameHistoryPermalink
PSFConfig.py 6.79 KiB
Open in Web IDE
1 
import sys
2 
import numpy as np
3 
import scipy.io
4 
from itertools import islice
5
6 
# from PSF.PSFInterp.PSFUtil import *
7 
# from PSFUtil import findNeighbors, findNeighbors_hoclist
8
9
10
11 
###加载PSF信息###
12 
def LoadPSF(iccd, iwave, ipsf, psfPath, InputMaxPixelPos=True, PSFCentroidWgt=False, PixSizeInMicrons=2.5, PSFBinning=False, PSFConvGauss=False):
13 
    """
14 
    load psf informations from psf matrix.
15
16 
    Parameters:
17 
        iccd (int): ccd number [1,30].
18 
        iwave(int): wave-index [1,4].
19 
        ipsf (int): psf number [1, 100].
20 
        psfPath (int): path to psf matrix
21 
        InputMaxPixelPos(bool-optional): only True for 30*30 psf-matrix
22 
        PSFCentroidWgt(bool-optional): True for using psfMat with libCentroid.lib
23 
    Returns:
24 
        psfInfo (dirctionary)
25 
    """
26
27 
    if iccd not in np.linspace(1, 30, 30, dtype='int'):
28 
        print('Error - iccd should be in [1, 30].')
29 
        sys.exit()
30 
    if iwave not in np.linspace(1, 4, 4, dtype='int'):
31 
        print('Error - iwave should be in [1, 4].')
32 
        sys.exit()
33 
    if ipsf not in np.linspace(1, 900, 900, dtype='int'):
34 
        print('Error - ipsf should be in [1, 900].')
35 
        sys.exit()
36
37 
    psfInfo = {}
38 
    fpath = psfPath +'/' +'ccd{:}'.format(iccd) +'/' + 'wave_{:}'.format(iwave)
39
40 
    #获取ipsf矩阵
41 
    '''
42 
    if not PSFCentroidWgt:
43 
        ##读取PSF原数据
44 
        fpathMat = fpath +'/' +'5_psf_array' +'/' +'psf_{:}.mat'.format(ipsf)
45 
        data = scipy.io.loadmat(fpathMat)
46 
        psfInfo['psfMat'] = data['psf']
47 
    if PSFCentroidWgt:
48 
        ##读取PSFCentroidWgt
49 
        ffpath = psfPath +'_proc/' +'ccd{:}'.format(iccd) +'/' + 'wave_{:}'.format(iwave)
50 
        ffpathMat = ffpath +'/' +'5_psf_array' +'/' +'psf_{:}_centroidWgt.mat'.format(ipsf)
51 
        data = scipy.io.loadmat(ffpathMat)
52 
        psfInfo['psfMat'] = data['psf']
53 
    '''
54 
    if not PSFCentroidWgt:
55 
        ffpath = fpath
56 
        ffpathMat = ffpath +'/5_psf_array/psf_{:}.mat'.format(ipsf)
57 
    if PSFCentroidWgt:
58 
        ffpath = psfPath +'_proc/ccd{:}/wave_{:}'.format(iccd, iwave)
59 
        ffpathMat = ffpath +'/5_psf_array/psf_{:}_centroidWgt.mat'.format(ipsf)
60 
        if PSFBinning and (not PSFConvGauss):
61 
            ffpath = psfPath +'_bin256x256/ccd{:}/wave_{:}'.format(iccd, iwave)
62 
            ffpathMat = ffpath +'/5_psf_array/psf_{:}_centroidWgt.mat'.format(ipsf)
63 
        if (not PSFBinning) and PSFConvGauss:
64 
            ffpath = psfPath +'_convGauss/ccd{:}/wave_{:}'.format(iccd, iwave)
65 
            ffpathMat = ffpath +'/5_psf_array/psf_{:}_centroidWgt.mat'.format(ipsf)
66 
        if PSFBinning and PSFConvGauss:
67 
            ffpath = psfPath +'_convGauss/ccd{:}/wave_{:}'.format(iccd, iwave)
68 
            ffpathMat = ffpath +'/5_psf_array/psf_{:}_centroidWgt_BC.mat'.format(ipsf)
69 
    data = scipy.io.loadmat(ffpathMat)
70 
    psfInfo['psfMat'] = data['psf']
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 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140
psfInfo['pixsize'] = PixSizeInMicrons #microns if PSFBinning: psfInfo['pixsize'] = PixSizeInMicrons*2.0 #获取ipsf波长 fpathWave = fpath +'/' +'1_wavelength.txt' f = open(fpathWave, 'r') wavelength = np.float(f.readline()) f.close() psfInfo['wavelength'] = wavelength #获取ipsf位置 fpathCoordinate = fpath +'/' +'4_PSF_coordinate.txt' f = open(fpathCoordinate, 'r') header = f.readline() for line in islice(f, ipsf-1, ipsf): line = line.strip() columns = line.split() f.close() icol = 0 psfInfo['field_x'] = float(columns[icol]) #deg, 视场采样位置 icol+= 1 psfInfo['field_y'] = float(columns[icol]) #deg icol+= 1 psfInfo['centroid_x'] = float(columns[icol]) #mm, psf质心相对主光线的偏移量 icol+= 1 psfInfo['centroid_y'] = float(columns[icol]) #mm icol+= 1 if InputMaxPixelPos == True: psfInfo['max_x'] = float(columns[icol]) #mm, max pixel postion icol+= 1 psfInfo['max_y'] = float(columns[icol]) #mm icol+= 1 psfInfo['image_x'] = float(columns[icol]) #mm, 主光线位置 icol+= 1 psfInfo['image_y'] = float(columns[icol]) #mm if PSFCentroidWgt: psfInfo['centroid_x'] = data['cx'][0][0] #mm, psfCentroidWgt质心相对主光线的偏移量 psfInfo['centroid_y'] = data['cy'][0][0] #mm # if PSFBinning: # psfInfo['centroid_x'] = data['cx'][0][0] +0.0025/2 #binning引起的主光线漂移 # psfInfo['centroid_y'] = data['cy'][0][0] +0.0025/2 return psfInfo ###插值PSF图像-IDW方法### def psfMaker_IDW(px, py, PSFMat, cen_col, cen_row, IDWindex=2, OnlyNeighbors=True, hoc=None, hoclist=None, PSFCentroidWgt=False): """ psf interpolation by IDW Parameters: px, py (float, float): position of the target PSFMat (numpy.array): image cen_col, cen_row (numpy.array, numpy.array): potions of the psf centers IDWindex (int-optional): the power index of IDW OnlyNeighbors (bool-optional): only neighbors are used for psf interpolation Returns: psfMaker (numpy.array) """ minimum_psf_weight = 1e-8 ref_col = px ref_row = py
141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200
ngy, ngx = PSFMat[0, :, :].shape npsf = PSFMat[:, :, :].shape[0] psfWeight = np.zeros([npsf]) from .PSFUtil import findNeighbors, findNeighbors_hoclist if OnlyNeighbors == True: if hoc is None: neigh = findNeighbors(px, py, cen_col, cen_row, dr=5., dn=4, OnlyDistance=False) if hoc is not None: neigh = findNeighbors_hoclist(cen_col, cen_row, tx=px,ty=py, dn=4, hoc=hoc, hoclist=hoclist) neighFlag = np.zeros(npsf) neighFlag[neigh] = 1 for ipsf in range(npsf): if OnlyNeighbors == True: if neighFlag[ipsf] != 1: continue dist = np.sqrt((ref_col - cen_col[ipsf])**2 + (ref_row - cen_row[ipsf])**2) if IDWindex == 1: psfWeight[ipsf] = dist if IDWindex == 2: psfWeight[ipsf] = dist**2 if IDWindex == 3: psfWeight[ipsf] = dist**3 if IDWindex == 4: psfWeight[ipsf] = dist**4 psfWeight[ipsf] = max(psfWeight[ipsf], minimum_psf_weight) psfWeight[ipsf] = 1./psfWeight[ipsf] psfWeight /= np.sum(psfWeight) psfMaker = np.zeros([ngy, ngx], dtype=np.float32) for ipsf in range(npsf): if OnlyNeighbors == True: if neighFlag[ipsf] != 1: continue iPSFMat = PSFMat[ipsf, :, :].copy() ipsfWeight = psfWeight[ipsf] psfMaker += iPSFMat * ipsfWeight psfMaker /= np.nansum(psfMaker) return psfMaker ###TEST### if __name__ == '__main__': iccd = 1 iwave= 1 ipsf = 1 psfPath = '/data/simudata/CSSOSDataProductsSims/data/csstPSFdata/CSSOS_psf_20210108/CSST_psf_ciomp_2p5um_cycle3_ccr90' psfInfo = LoadPSF(iccd, iwave, ipsf, psfPath, InputMaxPixelPos=True, PSFCentroidWgt=True) print('psfInfo.keys:', psfInfo.keys()) print(psfInfo)