QPMS
/
qpms
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Nicer BaseSpec.__repr__() for BaseSpec(lMax=...)

This commit is contained in:
Marek Nečada 2022-05-25 19:16:22 +03:00
parent b9d89fb346
commit 92842d4e75
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
qpms/cybspec.pyx
View File

@ -1,4 +1,5 @@
import numpy as np import numpy as np
cimport numpy as np
import enum import enum
from .cycommon import get_mn_y, tlm2uvswfi from .cycommon import get_mn_y, tlm2uvswfi
__VSWF_norm_dict = { __VSWF_norm_dict = {
@ -140,6 +141,11 @@ cdef class BaseSpec:
return self.__ilist[key] return self.__ilist[key]
def __repr__(self): def __repr__(self):
cdef qpms_l_t lMax_candidate
lMax_candidate = qpms_nelem2lMax(len(self.__ilist) // 2)
if lMax_candidate >= 0:
if np.array_equal(self.__ilist, BaseSpec(lMax=lMax_candidate).ilist):
return 'BaseSpec(lMax=%d)' % lMax_candidate
return 'BaseSpec([' + ','.join('%d' % i for i in self.__ilist) + '])' return 'BaseSpec([' + ','.join('%d' % i for i in self.__ilist) + '])'
property ilist: property ilist:

9
qpms/indexing.h
View File

@ -63,6 +63,15 @@ static inline qpms_y_t qpms_lMax2nelem(qpms_l_t lmax){
return lmax * ((qpms_y_t)lmax + 2); return lmax * ((qpms_y_t)lmax + 2);
} }
/// Inverse of qpms_lMax2nelem(); returns -1 if there is no exact match.
static inline qpms_l_t qpms_nelem2lMax(qpms_y_t nelem) {
qpms_l_t l = qpms_y2n(nelem);
if (l == -qpms_yn2m(nelem, l))
return l - 1;
else
return -1;
}
// Scalar versions: they have a place for the 0, 0 term in the beginning // Scalar versions: they have a place for the 0, 0 term in the beginning
/// Mapping from multipole degree, order to flat index (degree ≥ 0) /// Mapping from multipole degree, order to flat index (degree ≥ 0)