def __init__(self, data_np=None, metadata=None, logger=None):
Callback.Callbacks.__init__(self)
if logger != None:
self.logger = logger
else:
self.logger = logging.Logger('AstroImage')
if data_np == None:
data_np = numpy.zeros((1, 1))
self._data = data_np
self.metadata = {}
if metadata:
self.update_metadata(metadata)
self._set_minmax()
self.autocuts = AutoCuts.AutoCuts(self.logger)
# For callbacks
for name in ('modified', ):
self.enable_callback(name)
def __init__(self, logger=None, rgbmap=None, settings=None):
Callback.Callbacks.__init__(self)
if logger != None:
self.logger = logger
else:
self.logger = logging.Logger('FitsImageBase')
# RGB mapper
if rgbmap:
self.rgbmap = rgbmap
else:
rgbmap = RGBMap.RGBMapper()
self.rgbmap = rgbmap
# Object that calculates auto cut levels
self.autocuts = AutoCuts.AutoCuts(self.logger)
# Dummy 1-pixel image
self.image = AstroImage.AstroImage(numpy.zeros((1, 1)),
logger=self.logger)
# for debugging
self.name = str(self)
# Create settings and set defaults
if settings == None:
settings = Settings.SettingGroup(logger=self.logger)
self.t_ = settings
# for color mapping
self.t_.addDefaults(color_map='ramp',
intensity_map='ramp',
color_algorithm='linear',
color_hashsize=65535)
for name in ('color_map', 'intensity_map', 'color_algorithm',
'color_hashsize'):
self.t_.getSetting(name).add_callback('set', self.cmap_changed_cb)
# Initialize RGBMap
cmap_name = self.t_.get('color_map', 'ramp')
try:
cm = cmap.get_cmap(cmap_name)
except KeyError:
cm = cmap.get_cmap('ramp')
rgbmap.set_cmap(cm)
imap_name = self.t_.get('intensity_map', 'ramp')
try:
im = imap.get_imap(imap_name)
except KeyError:
im = imap.get_imap('ramp')
rgbmap.set_imap(im)
hash_size = self.t_.get('color_hashsize', 65535)
rgbmap.set_hash_size(hash_size)
hash_alg = self.t_.get('color_algorithm', 'linear')
rgbmap.set_hash_algorithm(hash_alg)
rgbmap.add_callback('changed', self.rgbmap_cb)
# for cut levels
self.t_.addDefaults(locut=0.0, hicut=0.0)
for name in ('locut', 'hicut'):
self.t_.getSetting(name).add_callback('set', self.cut_levels_cb)
# for auto cut levels
self.autocuts_options = ('on', 'override', 'off')
self.t_.addDefaults(
autocuts='override',
autocut_method='histogram',
autocut_hist_pct=AutoCuts.default_autocuts_hist_pct,
autocut_bins=AutoCuts.default_autocuts_bins)
for name in ('autocuts', 'autocut_method', 'autocut_hist_pct',
'autocut_bins'):
self.t_.getSetting(name).add_callback('set', self.auto_levels_cb)
# for zooming
self.t_.addDefaults(zoomlevel=1.0,
zoom_algorithm='step',
scale_x_base=1.0,
scale_y_base=1.0,
zoom_rate=math.sqrt(2.0))
for name in ('zoom_rate', 'zoom_algorithm', 'scale_x_base',
'scale_y_base'):
self.t_.getSetting(name).add_callback('set',
self.zoomalg_change_cb)
# max/min scaling
self.t_.addDefaults(scale_max=10000.0, scale_min=0.00001)
# autozoom options
self.autozoom_options = ('on', 'override', 'off')
self.t_.addDefaults(autozoom='on')
# for panning
self.t_makebg = False
self.t_.addDefaults(reverse_pan=False, autocenter=True)
# for transforms
self.t_.addDefaults(flip_x=False, flip_y=False, swap_xy=False)
for name in ('flip_x', 'flip_y', 'swap_xy'):
self.t_.getSetting(name).add_callback('set', self.transform_cb)
# desired rotation angle
self.t_.addDefaults(rot_deg=0.0)
self.t_.getSetting('rot_deg').add_callback('set',
self.rotation_change_cb)
# misc
self.t_.addDefaults(use_embedded_profile=True, auto_orient=False)
# PRIVATE IMPLEMENTATION STATE
# image window width and height (see set_window_dimensions())
self._imgwin_wd = 1
self._imgwin_ht = 1
self._imgwin_set = False
# center (and reference) pixel in the screen image (in pixel coords)
self._ctr_x = 1
self._ctr_y = 1
# data indexes at the reference pixel (in data coords)
self._org_x = 0
self._org_y = 0
# pan position
self._pan_x = 0.0
self._pan_y = 0.0
# Origin in the data array of what is currently displayed (LL, UR)
self._org_x1 = 0
self._org_y1 = 0
self._org_x2 = 0
self._org_y2 = 0
# offsets in the screen image for drawing (in screen coords)
self._dst_x = 0
self._dst_y = 0
self._invertY = True
# offsets in the screen image (in data coords)
self._off_x = 0
self._off_y = 0
# desired scale factors
self._scale_x = 1.0
self._scale_y = 1.0
# actual scale factors produced from desired ones
self._org_scale_x = 0
self._org_scale_y = 0
self._cutout = None
self._rotimg = None
self._prergb = None
self._rgbarr = None
self.orientMap = {
# tag: (flip_x, flip_y, swap_xy)
1: (False, True, False),
2: (True, True, False),
3: (True, False, False),
4: (False, False, False),
5: (True, False, True),
6: (True, True, True),
7: (False, True, True),
8: (False, False, True),
}
# For callbacks
# TODO: we should be able to deprecate a lot of these with the new
# settings callbacks
for name in ('cut-set', 'zoom-set', 'pan-set', 'transform', 'rotate',
'image-set', 'configure', 'autocuts', 'autozoom'):
self.enable_callback(name)
