# -*- coding: utf-8 -*-
# Color picker plugin
# This file is part of gedit-plugins
#
# Copyright (C) 2006 Jesse van den Kieboom
# Copyright (C) 2012 Ignacio Casal Quinteiro
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor,
# Boston, MA 02110-1301, USA.
import gi
gi.require_version('Gtk', '3.0')
gi.require_version('Gedit', '3.0')
from gi.repository import GObject, Gio, Gtk, Gdk, Gedit
import re
from gpdefs import *
try:
import gettext
gettext.bindtextdomain('gedit-plugins')
gettext.textdomain('gedit-plugins')
_ = gettext.gettext
except:
_ = lambda s: s
class ColorHelper:
def scale_color_component(self, component):
return min(max(int(round(component * 255.)), 0), 255)
def skip_hex(self, buf, iter, next_char):
while True:
char = iter.get_char()
if not char:
return
if char.lower() not in \
('0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'a', 'b', 'c', 'd', 'e', 'f'):
return
if not next_char(iter):
return
def get_rgba_position(self, buf, use_bounds):
bounds = buf.get_selection_bounds()
if bounds == ():
if use_bounds:
return None
# No selection, find color in the current cursor position
start = buf.get_iter_at_mark(buf.get_insert())
end = start.copy()
start.backward_char()
self.skip_hex(buf, start, lambda iter: iter.backward_char())
self.skip_hex(buf, end, lambda iter: iter.forward_char())
else:
start, end = bounds
text = buf.get_text(start, end, False)
if not re.match('#?[0-9a-zA-Z]+', text):
return None
if text[0] != '#':
start.backward_char()
if start.get_char() != '#':
return None
return start, end
def insert_color(self, view, text):
if not view or not view.get_editable():
return
doc = view.get_buffer()
if not doc:
return
doc.begin_user_action()
# Get the color
bounds = self.get_rgba_position(doc, False)
if not bounds:
doc.delete_selection(False, True)
else:
doc.delete(bounds[0], bounds[1])
doc.insert_at_cursor('#' + text)
doc.end_user_action()
def get_current_color(self, doc, use_bounds):
if not doc:
return None
bounds = self.get_rgba_position(doc, use_bounds)
if bounds:
return doc.get_text(bounds[0], bounds[1], False)
else:
return None
class ColorPickerAppActivatable(GObject.Object, Gedit.AppActivatable):
app = GObject.Property(type=Gedit.App)
def __init__(self):
GObject.Object.__init__(self)
def do_activate(self):
self.menu_ext = self.extend_menu("tools-section")
item = Gio.MenuItem.new(_("Pick _Color…"), "win.colorpicker")
self.menu_ext.prepend_menu_item(item)
def do_deactivate(self):
self.menu_ext = None
class ColorPickerWindowActivatable(GObject.Object, Gedit.WindowActivatable):
window = GObject.Property(type=Gedit.Window)
def __init__(self):
GObject.Object.__init__(self)
self._dialog = None
self._color_helper = ColorHelper()
def do_activate(self):
action = Gio.SimpleAction(name="colorpicker")
action.connect('activate', lambda a, p: self.on_color_picker_activate())
self.window.add_action(action)
self._update()
def do_deactivate(self):
self.window.remove_action("colorpicker")
def do_update_state(self):
self._update()
def _update(self):
tab = self.window.get_active_tab()
self.window.lookup_action("colorpicker").set_enabled(tab is not None)
if not tab and self._dialog and \
self._dialog.get_transient_for() == self.window:
self._dialog.response(Gtk.ResponseType.CLOSE)
# Signal handlers
def on_color_picker_activate(self):
if not self._dialog:
self._dialog = Gtk.ColorChooserDialog.new(_('Pick Color'), self.window)
self._dialog.connect_after('response', self.on_dialog_response)
rgba_str = self._color_helper.get_current_color(self.window.get_active_document(), False)
if rgba_str:
rgba = Gdk.RGBA()
parsed = rgba.parse(rgba_str)
if parsed:
self._dialog.set_rgba(rgba)
self._dialog.present()
def on_dialog_response(self, dialog, response):
if response == Gtk.ResponseType.OK:
rgba = dialog.get_rgba()
self._color_helper.insert_color(self.window.get_active_view(),
"%02x%02x%02x" % (self._color_helper.scale_color_component(rgba.red),
self._color_helper.scale_color_component(rgba.green),
self._color_helper.scale_color_component(rgba.blue)))
else:
self._dialog.destroy()
self._dialog = None
class ColorPickerViewActivatable(GObject.Object, Gedit.ViewActivatable):
view = GObject.Property(type=Gedit.View)
def __init__(self):
GObject.Object.__init__(self)
self._rgba_str = None
self._color_button = None
self._color_helper = ColorHelper()
def do_activate(self):
buf = self.view.get_buffer()
buf.connect_after('mark-set', self.on_buffer_mark_set)
def do_deactivate(self):
if self._color_button is not None:
self._color_button.destroy()
self._color_button = None
def on_buffer_mark_set(self, buf, location, mark):
if not buf.get_has_selection():
if self._color_button:
self._color_button.destroy()
self._color_button = None
return
if mark != buf.get_insert() and mark != buf.get_selection_bound():
return
rgba_str = self._color_helper.get_current_color(self.view.get_buffer(), True)
if rgba_str is not None and rgba_str != self._rgba_str and self._color_button is not None:
rgba = Gdk.RGBA()
parsed = rgba.parse(rgba_str)
if parsed:
self._rgba_str = rgba_str
self._color_button.set_rgba(rgba)
elif rgba_str is not None and self._color_button is None:
rgba = Gdk.RGBA()
parsed = rgba.parse(rgba_str)
if parsed:
self._rgba_str = rgba_str
bounds = buf.get_selection_bounds()
if bounds != ():
self._color_button = Gtk.ColorButton.new_with_rgba(rgba)
self._color_button.set_halign(Gtk.Align.START)
self._color_button.set_valign(Gtk.Align.START)
self._color_button.show()
self._color_button.connect('color-set', self.on_color_set)
start, end = bounds
location = self.view.get_iter_location(start)
min_width, nat_width = self._color_button.get_preferred_width()
min_height, nat_height = self._color_button.get_preferred_height()
x = location.x
if location.y - nat_height > 0:
y = location.y - nat_height
else:
y = location.y + location.height
self.view.add_child_in_window(self._color_button, Gtk.TextWindowType.TEXT, x, y)
elif not rgba_str and self._color_button is not None:
self._color_button.destroy()
self._color_button = None
def on_color_set(self, color_button):
rgba = color_button.get_rgba()
self._color_helper.insert_color(self.view,
"%02x%02x%02x" % (self._color_helper.scale_color_component(rgba.red),
self._color_helper.scale_color_component(rgba.green),
self._color_helper.scale_color_component(rgba.blue)))
# ex:ts=4:et: