Merge branch 'master' into weight_pol_type

CMB-S4 · Jan 23, 2024 · 884ae48 · 884ae48
2 parents 52d287b + 7052169
commit 884ae48
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/maps/python/map_modules.py b/maps/python/map_modules.py
@@ -125,7 +125,7 @@ def FlattenPol(frame, invert=False):
         return
 
     ValidateMaps(frame)
-    qmap, umap = frame.pop("Q"), frame.pop("U")
+    tmap, qmap, umap = frame.pop("T"), frame.pop("Q"), frame.pop("U")
 
     if "Wpol" in frame:
         wmap = frame.pop("Wpol")
@@ -134,6 +134,9 @@ def FlattenPol(frame, invert=False):
     else:
         maps.flatten_pol(qmap, umap, invert=invert)
 
+    tmap.flat_pol = not invert
+
+    frame["T"] = tmap
     frame["Q"] = qmap
     frame["U"] = umap
 

diff --git a/maps/src/maputils.cxx b/maps/src/maputils.cxx
@@ -263,6 +263,8 @@ void FlattenPol(FlatSkyMapPtr Q, FlatSkyMapPtr U, G3SkyMapWeightsPtr W, double h
 	U->SetFlatPol(!invert);
 
 	if (!!W) {
+		flatptr = boost::dynamic_pointer_cast<FlatSkyMap>(W->TT);
+		flatptr->SetFlatPol(!invert);
 		flatptr = boost::dynamic_pointer_cast<FlatSkyMap>(W->TQ);
 		flatptr->SetFlatPol(!invert);
 		flatptr = boost::dynamic_pointer_cast<FlatSkyMap>(W->TU);