Upload New File

brick_core.py

+import healpy as hp
+import numpy as np
+from .brick_utils import radius_exact
+
+class Config:
+    def __init__(self, nside, overlap):
+        self.nside = nside
+        self.overlap = overlap
+        
+class Brick:
+    """
+    A class representing a sky region ("brick") generated based on HEALPix.
+
+    Attributes
+    ----------
+    brick_id : int
+        Unique identifier of the brick (HEALPix pixel ID).
+    ra : float
+        Right Ascension of the brick center (degrees).
+    dec : float
+        Declination of the brick center (degrees).
+    min_radius : float
+        Minimum enclosing circle radius of the brick (degrees, without overlap).
+    radius : float
+        Enlarged radius of the brick (degrees, after applying overlap).
+    nside : int
+        HEALPix resolution parameter.
+    """
+    def __init__(self, brick_id, ra, dec, min_radius, radius, nside):
+        self.brick_id = brick_id
+        self.ra = ra      
+        self.dec = dec    
+        self.min_radius = min_radius 
+        self.radius = radius  
+        self.nside = nside  
+
+    def __repr__(self):
+        return f"Brick(brick_id={self.brick_id}, ra={self.ra}, dec={self.dec}, min_radius={self.min_radius}, radius={self.radius}, nside={self.nside})"
+
+    
+def generate_bricks(nside, brick_id=None, overlap=1.2):
+    """
+    Generate Brick objects for all HEALPix pixels or a single specified pixel.
+
+    Parameters
+    ----------
+    nside : int
+        HEALPix resolution parameter.
+    brick_id : int, optional
+        Specific HEALPix pixel (brick) ID to generate. If None (default),
+        generate bricks for all pixels.
+    overlap : float, optional
+        Overlap factor to enlarge the brick radius.
+        - 1.0 = no overlap
+        - >1.0 = enlarged radius
+        Default is 1.2.
+
+    Returns
+    -------
+    Brick or list of Brick
+        - If `brick_id` is provided: a single Brick object.
+        - If `brick_id` is None: a list of Brick objects for all pixels.
+    """
+    def _create_brick(pix, nside):
+        theta, phi = hp.pix2ang(nside, pix)
+        ra = np.degrees(phi)
+        dec = 90 - np.degrees(theta)
+        min_radius = radius_exact(nside, pix)
+        radius = min_radius * overlap
+        return Brick(brick_id=pix, ra=ra, dec=dec, min_radius=min_radius, radius=radius, nside=nside)
+
+    if brick_id is not None:
+        return _create_brick(brick_id, nside)
+    else:
+        npix = hp.nside2npix(nside)
+        return [_create_brick(pix, nside) for pix in range(npix)]
\ No newline at end of file