/** * rofi * * MIT/X11 License * Modified 2016 Qball Cow * * Permission is hereby granted, free of charge, to any person obtaining * a copy of this software and associated documentation files (the * "Software"), to deal in the Software without restriction, including * without limitation the rights to use, copy, modify, merge, publish, * distribute, sublicense, and/or sell copies of the Software, and to * permit persons to whom the Software is furnished to do so, subject to * the following conditions: * * The above copyright notice and this permission notice shall be * included in all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #define SN_API_NOT_YET_FROZEN #include #include "settings.h" #include "rofi.h" #include "mode.h" #include "xcb-internal.h" #include "xkb-internal.h" #include "helper.h" #include "textbox.h" #include "scrollbar.h" #include "x11-helper.h" #include "xrmoptions.h" #include "dialogs/dialogs.h" #include "view.h" #include "view-internal.h" #include "xcb.h" const char *const separator_style_none = "none"; const char *const separator_style_dash = "dash"; GThreadPool *tpool = NULL; RofiViewState *current_active_menu = NULL; static void rofi_view_resize ( RofiViewState *state ); struct { xcb_window_t main_window; cairo_surface_t *surface; cairo_surface_t *fake_bg; int fake_bgrel; cairo_t *draw; MenuFlags flags; GQueue views; int width, height; int x, y; workarea mon; } CacheState = { .main_window = XCB_WINDOW_NONE, .surface = NULL, .fake_bg = NULL, .fake_bgrel = FALSE, .draw = NULL, .flags = MENU_NORMAL, .views = G_QUEUE_INIT, .width = 0, .height = 0, .x = 0, .y = 0, }; static char * get_matching_state ( void ) { if ( config.case_sensitive ) { if ( config.levenshtein_sort ) { return "±"; } else { return "-"; } } else{ if ( config.levenshtein_sort ) { return "+"; } } return " "; } /** * Levenshtein Sorting. */ static int lev_sort ( const void *p1, const void *p2, void *arg ) { const int *a = p1; const int *b = p2; int *distances = arg; return distances[*a] - distances[*b]; } /** * Stores a screenshot of Rofi at that point in time. */ static void menu_capture_screenshot ( void ) { const char *outp = g_getenv ( "ROFI_PNG_OUTPUT" ); if ( CacheState.surface == NULL ) { // Nothing to store. fprintf ( stderr, "There is no rofi surface to store\n" ); return; } const char *xdg_pict_dir = g_get_user_special_dir ( G_USER_DIRECTORY_PICTURES ); if ( outp == NULL && xdg_pict_dir == NULL ) { fprintf ( stderr, "XDG user picture directory or ROFI_PNG_OUTPUT is not set. Cannot store screenshot.\n" ); return; } // Get current time. GDateTime *now = g_date_time_new_now_local (); // Format filename. char *timestmp = g_date_time_format ( now, "rofi-%Y-%m-%d-%H%M" ); char *filename = g_strdup_printf ( "%s.png", timestmp ); // Build full path char *fpath = NULL; if ( outp == NULL ) { int index = 0; fpath = g_build_filename ( xdg_pict_dir, filename, NULL ); while ( g_file_test ( fpath, G_FILE_TEST_EXISTS ) && index < 99 ) { g_free ( fpath ); g_free ( filename ); // Try the next index. index++; // Format filename. filename = g_strdup_printf ( "%s-%d.png", timestmp, index ); // Build full path fpath = g_build_filename ( xdg_pict_dir, filename, NULL ); } } else { fpath = g_strdup ( outp ); } fprintf ( stderr, color_green "Storing screenshot %s\n"color_reset, fpath ); cairo_status_t status = cairo_surface_write_to_png ( CacheState.surface, fpath ); if ( status != CAIRO_STATUS_SUCCESS ) { fprintf ( stderr, "Failed to produce screenshot '%s', got error: '%s'\n", filename, cairo_status_to_string ( status ) ); } g_free ( fpath ); g_free ( filename ); g_free ( timestmp ); g_date_time_unref ( now ); } /** * Calculates the window position */ static void calculate_window_position ( void ) { if ( config.fullscreen ) { CacheState.x = CacheState.mon.x; CacheState.y = CacheState.mon.y; return; } // Default location is center. CacheState.y = CacheState.mon.y + ( CacheState.mon.h - CacheState.height ) / 2; CacheState.x = CacheState.mon.x + ( CacheState.mon.w - CacheState.width ) / 2; // Determine window location switch ( config.location ) { case WL_NORTH_WEST: CacheState.x = CacheState.mon.x; case WL_NORTH: CacheState.y = CacheState.mon.y; break; case WL_NORTH_EAST: CacheState.y = CacheState.mon.y; case WL_EAST: CacheState.x = CacheState.mon.x + CacheState.mon.w - CacheState.width; break; case WL_EAST_SOUTH: CacheState.x = CacheState.mon.x + CacheState.mon.w - CacheState.width; case WL_SOUTH: CacheState.y = CacheState.mon.y + CacheState.mon.h - CacheState.height; break; case WL_SOUTH_WEST: CacheState.y = CacheState.mon.y + CacheState.mon.h - CacheState.height; case WL_WEST: CacheState.x = CacheState.mon.x; break; case WL_CENTER: default: break; } // Apply offset. CacheState.x += config.x_offset; CacheState.y += config.y_offset; } void rofi_view_queue_redraw ( void ) { if ( current_active_menu ) { current_active_menu->update = TRUE; xcb_clear_area ( xcb->connection, CacheState.main_window, 1, 0, 0, 1, 1 ); xcb_flush ( xcb->connection ); } } void rofi_view_restart ( RofiViewState *state ) { state->quit = FALSE; state->retv = MENU_CANCEL; } RofiViewState * rofi_view_get_active ( void ) { return current_active_menu; } void rofi_view_set_active ( RofiViewState *state ) { if ( current_active_menu != NULL && state != NULL ) { g_queue_push_head ( &( CacheState.views ), state ); printf ( "stack view.\n" ); rofi_view_resize ( current_active_menu ); rofi_view_queue_redraw (); return; } else if ( state == NULL && !g_queue_is_empty ( &( CacheState.views ) ) ) { printf ( "pop view\n" ); current_active_menu = g_queue_pop_head ( &( CacheState.views ) ); rofi_view_resize ( current_active_menu ); rofi_view_queue_redraw (); return; } g_assert ( ( current_active_menu == NULL && state != NULL ) || ( current_active_menu != NULL && state == NULL ) ); current_active_menu = state; } void rofi_view_set_selected_line ( RofiViewState *state, unsigned int selected_line ) { state->selected_line = selected_line; // Find the line. state->selected = 0; for ( unsigned int i = 0; ( ( state->selected_line ) ) < UINT32_MAX && !state->selected && i < state->filtered_lines; i++ ) { if ( state->line_map[i] == ( state->selected_line ) ) { state->selected = i; break; } } state->update = TRUE; xcb_clear_area ( xcb->connection, CacheState.main_window, 1, 0, 0, 1, 1 ); xcb_flush ( xcb->connection ); } void rofi_view_free ( RofiViewState *state ) { // Do this here? // Wait for final release? textbox_free ( state->text ); textbox_free ( state->prompt_tb ); textbox_free ( state->case_indicator ); scrollbar_free ( state->scrollbar ); if ( state->overlay ) { textbox_free ( state->overlay ); } for ( unsigned int i = 0; i < state->max_elements; i++ ) { textbox_free ( state->boxes[i] ); } g_free ( state->boxes ); g_free ( state->line_map ); g_free ( state->distance ); // Free the switcher boxes. // When state is free'ed we should no longer need these. if ( config.sidebar_mode == TRUE ) { for ( unsigned int j = 0; j < state->num_modi; j++ ) { textbox_free ( state->modi[j] ); state->modi[j] = NULL; } g_free ( state->modi ); state->num_modi = 0; } g_free ( state ); } MenuReturn rofi_view_get_return_value ( const RofiViewState *state ) { return state->retv; } unsigned int rofi_view_get_selected_line ( const RofiViewState *state ) { return state->selected_line; } unsigned int rofi_view_get_next_position ( const RofiViewState *state ) { unsigned int next_pos = state->selected_line; if ( ( state->selected + 1 ) < state->num_lines ) { ( next_pos ) = state->line_map[state->selected + 1]; } return next_pos; } unsigned int rofi_view_get_completed ( const RofiViewState *state ) { return state->quit; } static void rofi_view_resize ( RofiViewState *state ) { if ( ( state->menu_flags & MENU_ERROR_DIALOG ) == MENU_ERROR_DIALOG ) { // Resize of error dialog. int entrybox_width = CacheState.width - ( 2 * ( state->border ) ); textbox_moveresize ( state->text, state->text->widget.x, state->text->widget.y, entrybox_width, state->line_height ); state->rchanged = TRUE; state->update = TRUE; return; } unsigned int sbw = config.line_margin + 8; widget_move ( WIDGET ( state->scrollbar ), CacheState.width - state->border - sbw, state->top_offset ); if ( config.sidebar_mode == TRUE ) { int width = ( CacheState.width - ( 2 * ( state->border ) + ( state->num_modi - 1 ) * config.line_margin ) ) / state->num_modi; for ( unsigned int j = 0; j < state->num_modi; j++ ) { textbox_moveresize ( state->modi[j], state->border + j * ( width + config.line_margin ), CacheState.height - state->line_height - state->border, width, state->line_height ); textbox_draw ( state->modi[j], CacheState.draw ); } } int entrybox_width = CacheState.width - ( 2 * ( state->border ) ); int offset = 0; int width = textbox_get_width ( state->case_indicator ); entrybox_width -= width + textbox_get_width ( state->prompt_tb ); offset = width; textbox_moveresize ( state->text, state->text->widget.x, state->text->widget.y, entrybox_width, state->line_height ); widget_move ( WIDGET ( state->case_indicator ), CacheState.width - state->border - offset, state->border ); /** * Resize in Height */ if ( state->num_lines > 0 ) { unsigned int last_length = state->max_elements; int element_height = state->line_height * config.element_height + config.line_margin; // Calculated new number of boxes. int h = ( CacheState.height - state->top_offset - config.padding ); if ( config.sidebar_mode == TRUE ) { h -= state->line_height + config.line_margin; } if ( h < 0 ) { fprintf ( stderr, "Current padding %u (on each side) does not fit within visible window %u.\n", config.padding, CacheState.height ); h = ( CacheState.height - state->top_offset - CacheState.height / 3 ); if ( config.sidebar_mode == TRUE ) { h -= state->line_height + config.line_margin; } } state->max_rows = MAX ( 1, ( h / element_height ) ); state->menu_lines = state->max_rows; state->max_elements = state->max_rows * config.menu_columns; // Free boxes no longer needed. for ( unsigned int i = state->max_elements; i < last_length; i++ ) { textbox_free ( state->boxes[i] ); } // resize array. state->boxes = g_realloc ( state->boxes, state->max_elements * sizeof ( textbox* ) ); int y_offset = state->top_offset; int x_offset = state->border; // Add newly added boxes. for ( unsigned int i = last_length; i < state->max_elements; i++ ) { state->boxes[i] = textbox_create ( ( state->menu_flags & MENU_INDICATOR ) ? TB_INDICATOR : 0, x_offset, y_offset, state->element_width, element_height, NORMAL, "" ); } scrollbar_resize ( state->scrollbar, -1, ( state->max_rows ) * ( element_height ) - config.line_margin ); } state->rchanged = TRUE; state->update = TRUE; } void rofi_view_itterrate ( RofiViewState *state, xcb_generic_event_t *event, xkb_stuff *xkb ) { uint8_t type = event->response_type & ~0x80; switch ( type ) { case XCB_EXPOSE: state->update = TRUE; break; case XCB_CONFIGURE_NOTIFY: { xcb_configure_notify_event_t *xce = (xcb_configure_notify_event_t *) event; if ( xce->window == CacheState.main_window ) { if ( CacheState.x != xce->x || CacheState.y != xce->y ) { CacheState.x = xce->x; CacheState.y = xce->y; state->update = TRUE; } if ( CacheState.width != xce->width || CacheState.height != xce->height ) { CacheState.width = xce->width; CacheState.height = xce->height; cairo_xcb_surface_set_size ( CacheState.surface, CacheState.width, CacheState.height ); rofi_view_resize ( state ); } } break; } default: state->x11_event_loop ( state, event, xkb ); } rofi_view_update ( state ); } const char * rofi_view_get_user_input ( const RofiViewState *state ) { if ( state->text ) { return state->text->text; } return NULL; } /** * Create a new, 0 initialized RofiViewState structure. * * @returns a new 0 initialized RofiViewState */ static RofiViewState * __rofi_view_state_create ( void ) { return g_malloc0 ( sizeof ( RofiViewState ) ); } typedef struct _thread_state { RofiViewState *state; GRegex **tokens; unsigned int start; unsigned int stop; unsigned int count; GCond *cond; GMutex *mutex; unsigned int *acount; void ( *callback )( struct _thread_state *t, gpointer data ); }thread_state; /** * @param data A thread_state object. * @param user_data User data to pass to thread_state callback * * Small wrapper function that is internally used to pass a job to a worker. */ static void rofi_view_call_thread ( gpointer data, gpointer user_data ) { thread_state *t = (thread_state *) data; t->callback ( t, user_data ); g_mutex_lock ( t->mutex ); ( *( t->acount ) )--; g_cond_signal ( t->cond ); g_mutex_unlock ( t->mutex ); } static void filter_elements ( thread_state *t, G_GNUC_UNUSED gpointer user_data ) { // input changed for ( unsigned int i = t->start; i < t->stop; i++ ) { int match = mode_token_match ( t->state->sw, t->tokens, i ); // If each token was matched, add it to list. if ( match ) { t->state->line_map[t->start + t->count] = i; if ( config.levenshtein_sort ) { // This is inefficient, need to fix it. char * str = mode_get_completion ( t->state->sw, i ); char * input = mode_preprocess_input ( t->state->sw, t->state->text->text ); t->state->distance[i] = levenshtein ( input, str ); g_free ( input ); g_free ( str ); } t->count++; } } } static void rofi_view_setup_fake_transparency ( void ) { if ( CacheState.fake_bg == NULL ) { cairo_surface_t *s = NULL; /** * Select Background to use for fake transparency. * Current options: 'screenshot','background' */ if ( g_strcmp0 ( config.fake_background, "screenshot" ) == 0 ) { s = cairo_xcb_surface_create ( xcb->connection, xcb_stuff_get_root_window ( xcb ), root_visual, xcb->screen->width_in_pixels, xcb->screen->height_in_pixels ); } else if ( g_strcmp0 ( config.fake_background, "background" ) == 0 ) { s = x11_helper_get_bg_surface (); } else { char *fpath = rofi_expand_path ( config.fake_background ); s = cairo_image_surface_create_from_png ( fpath ); CacheState.fake_bgrel = TRUE; g_free ( fpath ); } if ( s != NULL ) { if ( cairo_surface_status ( s ) != CAIRO_STATUS_SUCCESS ) { cairo_surface_destroy ( s ); s = NULL; } else { CacheState.fake_bg = cairo_image_surface_create ( CAIRO_FORMAT_ARGB32, CacheState.mon.w, CacheState.mon.h ); cairo_t *dr = cairo_create ( CacheState.fake_bg ); if ( CacheState.fake_bgrel ) { cairo_set_source_surface ( dr, s, 0, 0 ); } else { cairo_set_source_surface ( dr, s, -CacheState.mon.x, -CacheState.mon.y ); } cairo_paint ( dr ); cairo_destroy ( dr ); cairo_surface_destroy ( s ); } } TICK_N ( "Fake transparency" ); } } void __create_window ( MenuFlags menu_flags ) { uint32_t selmask = XCB_CW_BACK_PIXEL | XCB_CW_BORDER_PIXEL | XCB_CW_EVENT_MASK | XCB_CW_COLORMAP; uint32_t selval[] = { 0, 0, XCB_EVENT_MASK_EXPOSURE | XCB_EVENT_MASK_BUTTON_PRESS | XCB_EVENT_MASK_BUTTON_RELEASE | XCB_EVENT_MASK_KEY_PRESS | XCB_EVENT_MASK_KEY_RELEASE | XCB_EVENT_MASK_KEYMAP_STATE | XCB_EVENT_MASK_STRUCTURE_NOTIFY | XCB_EVENT_MASK_FOCUS_CHANGE | XCB_EVENT_MASK_BUTTON_1_MOTION, map }; xcb_window_t box = xcb_generate_id ( xcb->connection ); xcb_create_window ( xcb->connection, depth->depth, box, xcb_stuff_get_root_window ( xcb ), 0, 0, 200, 100, 0, XCB_WINDOW_CLASS_INPUT_OUTPUT, visual->visual_id, selmask, selval ); CacheState.surface = cairo_xcb_surface_create ( xcb->connection, box, visual, 200, 100 ); // Create a drawable. CacheState.draw = cairo_create ( CacheState.surface ); g_assert ( CacheState.draw != NULL ); cairo_set_operator ( CacheState.draw, CAIRO_OPERATOR_SOURCE ); // Set up pango context. cairo_font_options_t *fo = cairo_font_options_create (); // Take font description from xlib surface cairo_surface_get_font_options ( CacheState.surface, fo ); PangoContext *p = pango_cairo_create_context ( CacheState.draw ); // Set the font options from the xlib surface pango_cairo_context_set_font_options ( p, fo ); // Setup dpi if ( config.dpi > 0 ) { PangoFontMap *font_map = pango_cairo_font_map_get_default (); pango_cairo_font_map_set_resolution ( (PangoCairoFontMap *) font_map, (double) config.dpi ); } // Setup font. if ( config.menu_font ) { PangoFontDescription *pfd = pango_font_description_from_string ( config.menu_font ); pango_context_set_font_description ( p, pfd ); pango_font_description_free ( pfd ); } // Tell textbox to use this context. textbox_set_pango_context ( p ); // cleanup g_object_unref ( p ); cairo_font_options_destroy ( fo ); // // make it an unmanaged window if ( ( ( menu_flags & MENU_NORMAL_WINDOW ) == 0 ) ) { window_set_atom_prop ( box, xcb->ewmh._NET_WM_STATE, &( xcb->ewmh._NET_WM_STATE_ABOVE ), 1 ); uint32_t values[] = { 1 }; xcb_change_window_attributes ( xcb->connection, box, XCB_CW_OVERRIDE_REDIRECT, values ); } else{ window_set_atom_prop ( box, xcb->ewmh._NET_WM_WINDOW_TYPE, &( xcb->ewmh._NET_WM_WINDOW_TYPE_NORMAL ), 1 ); } if ( config.fullscreen ) { xcb_atom_t atoms[] = { xcb->ewmh._NET_WM_STATE_FULLSCREEN, xcb->ewmh._NET_WM_STATE_ABOVE }; window_set_atom_prop ( box, xcb->ewmh._NET_WM_STATE, atoms, sizeof ( atoms ) / sizeof ( xcb_atom_t ) ); } // Set the WM_NAME xcb_change_property ( xcb->connection, XCB_PROP_MODE_REPLACE, box, xcb->ewmh._NET_WM_NAME, xcb->ewmh.UTF8_STRING, 8, 4, "rofi" ); xcb_change_property ( xcb->connection, XCB_PROP_MODE_REPLACE, box, XCB_ATOM_WM_NAME, XCB_ATOM_STRING, 8, 4, "rofi" ); x11_set_window_opacity ( box, config.window_opacity ); CacheState.main_window = box; CacheState.flags = menu_flags; monitor_active ( &( CacheState.mon ) ); if ( config.fake_transparency ) { rofi_view_setup_fake_transparency (); } if ( xcb->sncontext != NULL ) { sn_launchee_context_setup_window ( xcb->sncontext, CacheState.main_window ); } } /** * @param state Internal state of the menu. * * Calculate the number of rows, columns and elements to display based on the * configuration and available data. */ static void rofi_view_calculate_rows_columns ( RofiViewState *state ) { state->columns = config.menu_columns; state->max_elements = MIN ( state->menu_lines * state->columns, state->num_lines ); // Calculate the number or rows. We do this by getting the num_lines rounded up to X columns // (num elements is better name) then dividing by columns. state->max_rows = MIN ( state->menu_lines, (unsigned int) ( ( state->num_lines + ( state->columns - state->num_lines % state->columns ) % state->columns ) / ( state->columns ) ) ); // Always have at least one row. state->max_rows = MAX ( 1, state->max_rows ); if ( config.fixed_num_lines == TRUE ) { state->max_elements = state->menu_lines * state->columns; state->max_rows = state->menu_lines; // If it would fit in one column, only use one column. if ( state->num_lines < state->max_elements ) { state->columns = ( state->num_lines + ( state->max_rows - state->num_lines % state->max_rows ) % state->max_rows ) / state->max_rows; state->max_elements = state->menu_lines * state->columns; } // Sanitize. if ( state->columns == 0 ) { state->columns = 1; } } } /** * @param state Internal state of the menu. * * Calculate the width of the window and the width of an element. */ static void rofi_view_calculate_window_and_element_width ( RofiViewState *state ) { if ( config.fullscreen ) { CacheState.width = CacheState.mon.w; } else if ( config.menu_width < 0 ) { double fw = textbox_get_estimated_char_width ( ); CacheState.width = -( fw * config.menu_width ); CacheState.width += 2 * state->border + 4; // 4 = 2*SIDE_MARGIN } else{ // Calculate as float to stop silly, big rounding down errors. CacheState.width = config.menu_width < 101 ? ( CacheState.mon.w / 100.0f ) * ( float ) config.menu_width : config.menu_width; } if ( state->columns > 0 ) { state->element_width = CacheState.width - ( 2 * ( state->border ) ); // Divide by the # columns state->element_width = ( state->element_width - ( state->columns - 1 ) * config.line_margin ) / state->columns; } } /** * Nav helper functions, to avoid duplicate code. */ /** * @param state The current RofiViewState * * Move the selection one page down. * - No wrap around. * - Clip at top/bottom */ inline static void rofi_view_nav_page_next ( RofiViewState *state ) { // If no lines, do nothing. if ( state->filtered_lines == 0 ) { return; } state->selected += ( state->max_elements ); if ( state->selected >= state->filtered_lines ) { state->selected = state->filtered_lines - 1; } state->update = TRUE; } /** * @param state The current RofiViewState * * Move the selection one page up. * - No wrap around. * - Clip at top/bottom */ inline static void rofi_view_nav_page_prev ( RofiViewState * state ) { if ( state->selected < state->max_elements ) { state->selected = 0; } else{ state->selected -= ( state->max_elements ); } state->update = TRUE; } /** * @param state The current RofiViewState * * Move the selection one column to the right. * - No wrap around. * - Do not move to top row when at start. */ inline static void rofi_view_nav_right ( RofiViewState *state ) { // If no lines, do nothing. if ( state->filtered_lines == 0 ) { return; } if ( ( state->selected + state->max_rows ) < state->filtered_lines ) { state->selected += state->max_rows; state->update = TRUE; } else if ( state->selected < ( state->filtered_lines - 1 ) ) { // We do not want to move to last item, UNLESS the last column is only // partially filled, then we still want to move column and select last entry. // First check the column we are currently in. int col = state->selected / state->max_rows; // Check total number of columns. int ncol = state->filtered_lines / state->max_rows; // If there is an extra column, move. if ( col != ncol ) { state->selected = state->filtered_lines - 1; state->update = TRUE; } } } /** * @param state The current RofiViewState * * Move the selection one column to the left. * - No wrap around. */ inline static void rofi_view_nav_left ( RofiViewState *state ) { if ( state->selected >= state->max_rows ) { state->selected -= state->max_rows; state->update = TRUE; } } /** * @param state The current RofiViewState * * Move the selection one row up. * - Wrap around. */ inline static void rofi_view_nav_up ( RofiViewState *state ) { // If no lines or don't cycle, do nothing. if ( state->filtered_lines == 0 || ( state->selected == 0 && !config.cycle ) ) { return; } // Wrap around. if ( state->selected == 0 ) { state->selected = state->filtered_lines; } state->selected--; state->update = TRUE; } /** * @param state The current RofiViewState * * Move the selection one row down. * - Wrap around. */ inline static void rofi_view_nav_down ( RofiViewState *state ) { // If no lines or don't cycle, do nothing. if ( state->filtered_lines == 0 || ( state->selected == state->filtered_lines - 1 && !config.cycle ) ) { return; } state->selected = state->selected < state->filtered_lines - 1 ? MIN ( state->filtered_lines - 1, state->selected + 1 ) : 0; state->update = TRUE; } /** * @param state The current RofiViewState * * Tab handling. */ static void rofi_view_nav_row_tab ( RofiViewState *state ) { if ( state->filtered_lines == 1 ) { state->retv = MENU_OK; ( state->selected_line ) = state->line_map[state->selected]; state->quit = 1; return; } // Double tab! if ( state->filtered_lines == 0 && ROW_TAB == state->prev_action ) { state->retv = MENU_NEXT; ( state->selected_line ) = 0; state->quit = TRUE; } else { rofi_view_nav_down ( state ); } state->prev_action = ROW_TAB; } /** * @param state The current RofiViewState * * complete current row. */ inline static void rofi_view_nav_row_select ( RofiViewState *state ) { // If a valid item is selected, return that.. if ( state->selected < state->filtered_lines ) { char *str = mode_get_completion ( state->sw, state->line_map[state->selected] ); textbox_text ( state->text, str ); g_free ( str ); textbox_cursor_end ( state->text ); state->update = TRUE; state->refilter = TRUE; } } /** * @param state The current RofiViewState * * Move the selection to first row. */ inline static void rofi_view_nav_first ( RofiViewState * state ) { state->selected = 0; state->update = TRUE; } /** * @param state The current RofiViewState * * Move the selection to last row. */ inline static void rofi_view_nav_last ( RofiViewState * state ) { // If no lines, do nothing. if ( state->filtered_lines == 0 ) { return; } state->selected = state->filtered_lines - 1; state->update = TRUE; } static unsigned int rofi_scroll_per_page ( RofiViewState * state ) { int offset = 0; // selected row is always visible. // If selected is visible do not scroll. if ( ( ( state->selected - ( state->last_offset ) ) < ( state->max_elements ) ) && ( state->selected >= ( state->last_offset ) ) ) { offset = state->last_offset; } else{ // Do paginating unsigned int page = ( state->max_elements > 0 ) ? ( state->selected / state->max_elements ) : 0; offset = page * state->max_elements; state->last_offset = offset; if ( page != state->cur_page ) { state->cur_page = page; state->rchanged = TRUE; } // Set the position scrollbar_set_handle ( state->scrollbar, page * state->max_elements ); } return offset; } static unsigned int rofi_scroll_continious ( RofiViewState * state ) { unsigned int middle = ( state->menu_lines - ( ( state->menu_lines & 1 ) == 0 ) ) / 2; unsigned int offset = 0; if ( state->selected > middle ) { if ( state->selected < ( state->filtered_lines - ( state->menu_lines - middle ) ) ) { offset = state->selected - middle; } // Don't go below zero. else if ( state->filtered_lines > state->menu_lines ) { offset = state->filtered_lines - state->menu_lines; } } if ( offset != state->cur_page ) { state->rchanged = TRUE; scrollbar_set_handle ( state->scrollbar, offset ); state->cur_page = offset; } return offset; } static void rofi_view_draw ( RofiViewState *state, cairo_t *d ) { unsigned int i, offset = 0; if ( config.scroll_method == 1 ) { offset = rofi_scroll_continious ( state ); } else { offset = rofi_scroll_per_page ( state ); } // Re calculate the boxes and sizes, see if we can move this in the menu_calc*rowscolumns // Get number of remaining lines to display. unsigned int a_lines = MIN ( ( state->filtered_lines - offset ), state->max_elements ); // Calculate number of columns unsigned int columns = ( a_lines + ( state->max_rows - a_lines % state->max_rows ) % state->max_rows ) / state->max_rows; columns = MIN ( columns, state->columns ); // Update the handle length. // Calculate number of visible rows. unsigned int max_elements = MIN ( a_lines, state->max_rows * columns ); scrollbar_set_handle_length ( state->scrollbar, columns * state->max_rows ); scrollbar_draw ( state->scrollbar, d ); // Element width. unsigned int element_width = CacheState.width - ( 2 * ( state->border ) ); if ( state->scrollbar != NULL ) { element_width -= state->scrollbar->widget.w; } if ( columns > 0 ) { element_width = ( element_width - ( columns - 1 ) * config.line_margin ) / columns; } int element_height = state->line_height * config.element_height; int y_offset = state->top_offset; int x_offset = state->border; if ( state->rchanged ) { char *input = mode_preprocess_input ( state->sw, state->text->text ); GRegex **tokens = tokenize ( input, config.case_sensitive ); g_free ( input ); // Move, resize visible boxes and show them. for ( i = 0; i < max_elements && ( i + offset ) < state->filtered_lines; i++ ) { unsigned int ex = ( ( i ) / state->max_rows ) * ( element_width + config.line_margin ); unsigned int ey = ( ( i ) % state->max_rows ) * ( element_height + config.line_margin ); // Move it around. textbox_moveresize ( state->boxes[i], ex + x_offset, ey + y_offset, element_width, element_height ); { TextBoxFontType type = ( ( ( i % state->max_rows ) & 1 ) == 0 ) ? NORMAL : ALT; int fstate = 0; char *text = mode_get_display_value ( state->sw, state->line_map[i + offset], &fstate, TRUE ); TextBoxFontType tbft = fstate | ( ( i + offset ) == state->selected ? HIGHLIGHT : type ); textbox_font ( state->boxes[i], tbft ); textbox_text ( state->boxes[i], text ); PangoAttrList *list = textbox_get_pango_attributes ( state->boxes[i] ); if ( list != NULL ) { pango_attr_list_ref ( list ); } else{ list = pango_attr_list_new (); } token_match_get_pango_attr ( tokens, textbox_get_visible_text ( state->boxes[i] ), list ); textbox_set_pango_attributes ( state->boxes[i], list ); pango_attr_list_unref ( list ); g_free ( text ); } textbox_draw ( state->boxes[i], d ); } tokenize_free ( tokens ); state->rchanged = FALSE; } else{ // Only do basic redrawing + highlight of row. for ( i = 0; i < max_elements && ( i + offset ) < state->filtered_lines; i++ ) { TextBoxFontType type = ( ( ( i % state->max_rows ) & 1 ) == 0 ) ? NORMAL : ALT; int fstate = 0; mode_get_display_value ( state->sw, state->line_map[i + offset], &fstate, FALSE ); TextBoxFontType tbft = fstate | ( ( i + offset ) == state->selected ? HIGHLIGHT : type ); textbox_font ( state->boxes[i], tbft ); textbox_draw ( state->boxes[i], d ); } } } void rofi_view_update ( RofiViewState *state ) { if ( !state->update ) { return; } TICK (); cairo_surface_t * surf = cairo_image_surface_create ( CAIRO_FORMAT_ARGB32, CacheState.width, CacheState.height ); cairo_t *d = cairo_create ( surf ); cairo_set_operator ( d, CAIRO_OPERATOR_SOURCE ); if ( config.fake_transparency ) { if ( CacheState.fake_bg != NULL ) { if ( CacheState.fake_bgrel ) { cairo_set_source_surface ( d, CacheState.fake_bg, 0.0, 0.0 ); } else { cairo_set_source_surface ( d, CacheState.fake_bg, -(double) ( CacheState.x - CacheState.mon.x ), -(double) ( CacheState.y - CacheState.mon.y ) ); } cairo_paint ( d ); cairo_set_operator ( d, CAIRO_OPERATOR_OVER ); color_background ( d ); cairo_paint ( d ); } } else { // Paint the background. color_background ( d ); cairo_paint ( d ); } TICK_N ( "Background" ); color_border ( d ); if ( config.menu_bw > 0 ) { cairo_save ( d ); cairo_set_line_width ( d, config.menu_bw ); cairo_rectangle ( d, config.menu_bw / 2.0, config.menu_bw / 2.0, CacheState.width - config.menu_bw, CacheState.height - config.menu_bw ); cairo_stroke ( d ); cairo_restore ( d ); } // Always paint as overlay over the background. cairo_set_operator ( d, CAIRO_OPERATOR_OVER ); if ( state->max_elements > 0 ) { rofi_view_draw ( state, d ); } if ( state->prompt_tb ) { textbox_draw ( state->prompt_tb, d ); } if ( state->text ) { textbox_draw ( state->text, d ); } if ( state->case_indicator ) { textbox_draw ( state->case_indicator, d ); } if ( state->message_tb ) { textbox_draw ( state->message_tb, d ); } color_separator ( d ); if ( strcmp ( config.separator_style, separator_style_none ) ) { if ( strcmp ( config.separator_style, separator_style_dash ) == 0 ) { const double dashes[1] = { 4 }; cairo_set_dash ( d, dashes, 1, 0.0 ); } cairo_move_to ( d, state->border, state->line_height + ( state->border ) * 1 + config.line_margin + 1 ); cairo_line_to ( d, CacheState.width - state->border, state->line_height + ( state->border ) * 1 + config.line_margin + 1 ); cairo_stroke ( d ); if ( state->message_tb ) { cairo_move_to ( d, state->border, state->top_offset - ( config.line_margin ) - 1 ); cairo_line_to ( d, CacheState.width - state->border, state->top_offset - ( config.line_margin ) - 1 ); cairo_stroke ( d ); } if ( config.sidebar_mode == TRUE ) { cairo_move_to ( d, state->border, CacheState.height - state->line_height - ( state->border ) * 1 - 1 - config.line_margin ); cairo_line_to ( d, CacheState.width - state->border, CacheState.height - state->line_height - ( state->border ) * 1 - 1 - config.line_margin ); cairo_stroke ( d ); } } if ( config.sidebar_mode == TRUE ) { for ( unsigned int j = 0; j < state->num_modi; j++ ) { if ( state->modi[j] != NULL ) { textbox_draw ( state->modi[j], d ); } } } if ( state->overlay ) { textbox_draw ( state->overlay, d ); } state->update = FALSE; // Draw to actual window. cairo_set_source_surface ( CacheState.draw, surf, 0, 0 ); cairo_paint ( CacheState.draw ); // Cleanup cairo_destroy ( d ); cairo_surface_destroy ( surf ); // Flush the surface. cairo_surface_flush ( CacheState.surface ); xcb_flush ( xcb->connection ); TICK (); } /** * @param state Internal state of the menu. * @param xse X selection event. * * Handle paste event. */ static void rofi_view_paste ( RofiViewState *state, xcb_selection_notify_event_t *xse ) { if ( xse->property == XCB_ATOM_NONE ) { fprintf ( stderr, "Failed to convert selection\n" ); } else if ( xse->property == xcb->ewmh.UTF8_STRING ) { gchar *text = window_get_text_prop ( CacheState.main_window, xcb->ewmh.UTF8_STRING ); if ( text != NULL && text[0] != '\0' ) { unsigned int dl = strlen ( text ); // Strip new line while ( dl > 0 && text[dl] == '\n' ) { text[dl] = '\0'; dl--; } // Insert string move cursor. textbox_insert ( state->text, state->text->cursor, text, dl ); textbox_cursor ( state->text, state->text->cursor + g_utf8_strlen ( text, -1 ) ); // Force a redraw and refiltering of the text. state->update = TRUE; state->refilter = TRUE; } g_free ( text ); } else { fprintf ( stderr, "Failed\n" ); } } static void rofi_view_mouse_navigation ( RofiViewState *state, xcb_button_press_event_t *xbe ) { // Scroll event if ( xbe->detail > 3 ) { if ( xbe->detail == 4 ) { rofi_view_nav_up ( state ); } else if ( xbe->detail == 5 ) { rofi_view_nav_down ( state ); } else if ( xbe->detail == 6 ) { rofi_view_nav_left ( state ); } else if ( xbe->detail == 7 ) { rofi_view_nav_right ( state ); } return; } else { if ( state->scrollbar && widget_intersect ( &( state->scrollbar->widget ), xbe->event_x, xbe->event_y ) ) { state->selected = scrollbar_clicked ( state->scrollbar, xbe->event_y ); state->update = TRUE; return; } for ( unsigned int i = 0; config.sidebar_mode == TRUE && i < state->num_modi; i++ ) { if ( widget_intersect ( &( state->modi[i]->widget ), xbe->event_x, xbe->event_y ) ) { ( state->selected_line ) = 0; state->retv = MENU_QUICK_SWITCH | ( i & MENU_LOWER_MASK ); state->quit = TRUE; state->skip_absorb = TRUE; return; } } for ( unsigned int i = 0; i < state->max_elements; i++ ) { if ( widget_intersect ( &( state->boxes[i]->widget ), xbe->event_x, xbe->event_y ) ) { // Only allow items that are visible to be selected. if ( ( state->last_offset + i ) >= state->filtered_lines ) { break; } // state->selected = state->last_offset + i; state->update = TRUE; if ( ( xbe->time - state->last_button_press ) < 200 ) { state->retv = MENU_OK; ( state->selected_line ) = state->line_map[state->selected]; // Quit state->quit = TRUE; state->skip_absorb = TRUE; } state->last_button_press = xbe->time; break; } } } } static void rofi_view_refilter ( RofiViewState *state ) { TICK_N ( "Filter start" ); if ( strlen ( state->text->text ) > 0 ) { unsigned int j = 0; gchar *input = mode_preprocess_input ( state->sw, state->text->text ); GRegex **tokens = tokenize ( input, config.case_sensitive ); g_free ( input ); /** * On long lists it can be beneficial to parallelize. * If number of threads is 1, no thread is spawn. * If number of threads > 1 and there are enough (> 1000) items, spawn jobs for the thread pool. * For large lists with 8 threads I see a factor three speedup of the whole function. */ unsigned int nt = MAX ( 1, state->num_lines / 500 ); thread_state states[nt]; GCond cond; GMutex mutex; g_mutex_init ( &mutex ); g_cond_init ( &cond ); unsigned int count = nt; unsigned int steps = ( state->num_lines + nt ) / nt; for ( unsigned int i = 0; i < nt; i++ ) { states[i].state = state; states[i].tokens = tokens; states[i].start = i * steps; states[i].stop = MIN ( state->num_lines, ( i + 1 ) * steps ); states[i].count = 0; states[i].cond = &cond; states[i].mutex = &mutex; states[i].acount = &count; states[i].callback = filter_elements; if ( i > 0 ) { g_thread_pool_push ( tpool, &states[i], NULL ); } } // Run one in this thread. rofi_view_call_thread ( &states[0], NULL ); // No need to do this with only one thread. if ( nt > 1 ) { g_mutex_lock ( &mutex ); while ( count > 0 ) { g_cond_wait ( &cond, &mutex ); } g_mutex_unlock ( &mutex ); } g_cond_clear ( &cond ); g_mutex_clear ( &mutex ); for ( unsigned int i = 0; i < nt; i++ ) { if ( j != states[i].start ) { memmove ( &( state->line_map[j] ), &( state->line_map[states[i].start] ), sizeof ( unsigned int ) * ( states[i].count ) ); } j += states[i].count; } if ( config.levenshtein_sort ) { g_qsort_with_data ( state->line_map, j, sizeof ( int ), lev_sort, state->distance ); } // Cleanup + bookkeeping. state->filtered_lines = j; tokenize_free ( tokens ); } else{ for ( unsigned int i = 0; i < state->num_lines; i++ ) { state->line_map[i] = i; } state->filtered_lines = state->num_lines; } if ( state->filtered_lines > 0 ) { state->selected = MIN ( state->selected, state->filtered_lines - 1 ); } else { state->selected = 0; } if ( config.auto_select == TRUE && state->filtered_lines == 1 && state->num_lines > 1 ) { ( state->selected_line ) = state->line_map[state->selected]; state->retv = MENU_OK; state->quit = TRUE; } scrollbar_set_max_value ( state->scrollbar, state->filtered_lines ); state->refilter = FALSE; state->rchanged = TRUE; state->update = TRUE; TICK_N ( "Filter done" ); } /** * @param state The Menu Handle * * Check if a finalize function is set, and if sets executes it. */ void process_result ( RofiViewState *state ); void rofi_view_finalize ( RofiViewState *state ) { if ( state && state->finalize != NULL ) { state->finalize ( state ); } } gboolean rofi_view_trigger_action ( RofiViewState *state, KeyBindingAction action ) { gboolean ret = TRUE; switch ( action ) { // Handling of paste case PASTE_PRIMARY: xcb_convert_selection ( xcb->connection, CacheState.main_window, XCB_ATOM_PRIMARY, xcb->ewmh.UTF8_STRING, xcb->ewmh.UTF8_STRING, XCB_CURRENT_TIME ); xcb_flush ( xcb->connection ); break; case PASTE_SECONDARY: xcb_convert_selection ( xcb->connection, CacheState.main_window, XCB_ATOM_SECONDARY, xcb->ewmh.UTF8_STRING, xcb->ewmh.UTF8_STRING, XCB_CURRENT_TIME ); xcb_flush ( xcb->connection ); break; case SCREENSHOT: menu_capture_screenshot ( ); break; case TOGGLE_SORT: config.levenshtein_sort = !config.levenshtein_sort; state->refilter = TRUE; state->update = TRUE; textbox_text ( state->case_indicator, get_matching_state () ); break; case MODE_PREVIOUS: state->retv = MENU_PREVIOUS; ( state->selected_line ) = 0; state->quit = TRUE; break; // Menu navigation. case MODE_NEXT: state->retv = MENU_NEXT; ( state->selected_line ) = 0; state->quit = TRUE; break; // Toggle case sensitivity. case TOGGLE_CASE_SENSITIVITY: config.case_sensitive = !config.case_sensitive; ( state->selected_line ) = 0; state->refilter = TRUE; state->update = TRUE; textbox_text ( state->case_indicator, get_matching_state () ); break; // Special delete entry command. case DELETE_ENTRY: if ( state->selected < state->filtered_lines ) { ( state->selected_line ) = state->line_map[state->selected]; state->retv = MENU_ENTRY_DELETE; state->quit = TRUE; } else { ret = FALSE; } break; case CUSTOM_1: case CUSTOM_2: case CUSTOM_3: case CUSTOM_4: case CUSTOM_5: case CUSTOM_6: case CUSTOM_7: case CUSTOM_8: case CUSTOM_9: case CUSTOM_10: case CUSTOM_11: case CUSTOM_12: case CUSTOM_13: case CUSTOM_14: case CUSTOM_15: case CUSTOM_16: case CUSTOM_17: case CUSTOM_18: case CUSTOM_19: state->selected_line = UINT32_MAX; if ( state->selected < state->filtered_lines ) { ( state->selected_line ) = state->line_map[state->selected]; } state->retv = MENU_QUICK_SWITCH | ( ( action - CUSTOM_1 ) & MENU_LOWER_MASK ); state->quit = TRUE; break; // If you add a binding here, make sure to add it to rofi_view_keyboard_navigation too case CANCEL: state->retv = MENU_CANCEL; state->quit = TRUE; break; case ROW_UP: rofi_view_nav_up ( state ); break; case ROW_TAB: rofi_view_nav_row_tab ( state ); break; case ROW_DOWN: rofi_view_nav_down ( state ); break; case ROW_LEFT: rofi_view_nav_left ( state ); break; case ROW_RIGHT: rofi_view_nav_right ( state ); break; case PAGE_PREV: rofi_view_nav_page_prev ( state ); break; case PAGE_NEXT: rofi_view_nav_page_next ( state ); break; case ROW_FIRST: rofi_view_nav_first ( state ); break; case ROW_LAST: rofi_view_nav_last ( state ); break; case ROW_SELECT: rofi_view_nav_row_select ( state ); break; // If you add a binding here, make sure to add it to textbox_keybinding too case MOVE_CHAR_BACK: case MOVE_CHAR_FORWARD: case CLEAR_LINE: case MOVE_FRONT: case MOVE_END: case REMOVE_TO_EOL: case REMOVE_TO_SOL: case REMOVE_WORD_BACK: case REMOVE_WORD_FORWARD: case REMOVE_CHAR_FORWARD: case MOVE_WORD_BACK: case MOVE_WORD_FORWARD: case REMOVE_CHAR_BACK: { int rc = textbox_keybinding ( state->text, action ); if ( rc == 1 ) { // Entry changed. state->refilter = TRUE; state->update = TRUE; } else if ( rc == 2 ) { // Movement. state->update = TRUE; } break; } case ACCEPT_ALT: { state->selected_line = UINT32_MAX; if ( state->selected < state->filtered_lines ) { ( state->selected_line ) = state->line_map[state->selected]; state->retv = MENU_OK; } else { // Nothing entered and nothing selected. state->retv = MENU_CUSTOM_INPUT; } state->retv |= MENU_CUSTOM_ACTION; state->quit = TRUE; break; } case ACCEPT_CUSTOM: { state->selected_line = UINT32_MAX; state->retv = MENU_CUSTOM_INPUT; state->quit = TRUE; break; } case ACCEPT_ENTRY: { // If a valid item is selected, return that.. state->selected_line = UINT32_MAX; if ( state->selected < state->filtered_lines ) { ( state->selected_line ) = state->line_map[state->selected]; state->retv = MENU_OK; } else { // Nothing entered and nothing selected. state->retv = MENU_CUSTOM_INPUT; } state->quit = TRUE; break; } case NUM_ABE: ret = FALSE; break; } return ret; } static void rofi_view_handle_keypress ( RofiViewState *state, xkb_stuff *xkb, xcb_key_press_event_t *xkpe ) { xcb_keysym_t key; char pad[32]; int len = 0; key = xkb_state_key_get_one_sym ( xkb->state, xkpe->detail ); if ( xkb->compose.state != NULL ) { if ( ( key != XKB_KEY_NoSymbol ) && ( xkb_compose_state_feed ( xkb->compose.state, key ) == XKB_COMPOSE_FEED_ACCEPTED ) ) { switch ( xkb_compose_state_get_status ( xkb->compose.state ) ) { case XKB_COMPOSE_CANCELLED: /* Eat the keysym that cancelled the compose sequence. * This is default behaviour with Xlib */ case XKB_COMPOSE_COMPOSING: key = XKB_KEY_NoSymbol; break; case XKB_COMPOSE_COMPOSED: key = xkb_compose_state_get_one_sym ( xkb->compose.state ); len = xkb_compose_state_get_utf8 ( xkb->compose.state, pad, sizeof ( pad ) ); break; case XKB_COMPOSE_NOTHING: break; } if ( ( key == XKB_KEY_NoSymbol ) && ( len == 0 ) ) { return; } } } if ( len == 0 ) { len = xkb_state_key_get_utf8 ( xkb->state, xkpe->detail, pad, sizeof ( pad ) ); } unsigned int modstate = x11_canonalize_mask ( xkpe->state ); if ( key != XKB_KEY_NoSymbol ) { KeyBindingAction action; action = abe_find_action ( modstate, key ); if ( rofi_view_trigger_action ( state, action ) ) { return; } } if ( ( len > 0 ) && ( textbox_append_char ( state->text, pad, len ) ) ) { state->refilter = TRUE; state->update = TRUE; return; } } static void rofi_view_mainloop_iter ( RofiViewState *state, xcb_generic_event_t *ev, xkb_stuff *xkb ) { switch ( ev->response_type & ~0x80 ) { case XCB_FOCUS_IN: if ( ( state->menu_flags & MENU_NORMAL_WINDOW ) == 0 ) { take_keyboard ( CacheState.main_window ); } break; case XCB_FOCUS_OUT: if ( ( state->menu_flags & MENU_NORMAL_WINDOW ) == 0 ) { release_keyboard ( ); } break; case XCB_MOTION_NOTIFY: { xcb_motion_notify_event_t *xme = (xcb_motion_notify_event_t *) ev; if ( state->scrollbar != NULL && xme->event_x >= state->scrollbar->widget.x && xme->event_x < ( state->scrollbar->widget.x + state->scrollbar->widget.w ) ) { state->selected = scrollbar_clicked ( state->scrollbar, xme->event_y ); state->update = TRUE; } break; } case XCB_BUTTON_PRESS: rofi_view_mouse_navigation ( state, (xcb_button_press_event_t *) ev ); break; // Paste event. case XCB_SELECTION_NOTIFY: rofi_view_paste ( state, (xcb_selection_notify_event_t *) ev ); break; case XCB_KEYMAP_NOTIFY: { xcb_keymap_notify_event_t *kne = (xcb_keymap_notify_event_t *) ev; guint modstate = x11_get_current_mask ( xkb ); for ( gint32 by = 0; by < 32; ++by ) { for ( gint8 bi = 0; bi < 7; ++bi ) { if ( kne->keys[by] & ( 1 << bi ) ) { // X11 keycodes starts at 8 xkb_keysym_t key = xkb_state_key_get_one_sym ( xkb->state, ( 8 * by + bi ) + 8 ); abe_find_action ( modstate, key ); } } } break; } case XCB_KEY_PRESS: rofi_view_handle_keypress ( state, xkb, (xcb_key_press_event_t *) ev ); break; case XCB_KEY_RELEASE: { xcb_key_release_event_t *xkre = (xcb_key_release_event_t *) ev; unsigned int modstate = x11_canonalize_mask ( xkre->state ); if ( modstate == 0 ) { abe_trigger_release ( ); } break; } } // Update if requested. if ( state->refilter ) { rofi_view_refilter ( state ); } rofi_view_update ( state ); } RofiViewState *rofi_view_create ( Mode *sw, const char *input, char *prompt, const char *message, MenuFlags menu_flags, void ( *finalize )( RofiViewState * ) ) { TICK (); RofiViewState *state = __rofi_view_state_create (); state->menu_flags = menu_flags; state->sw = sw; state->selected_line = UINT32_MAX; state->retv = MENU_CANCEL; state->distance = NULL; state->quit = FALSE; state->skip_absorb = FALSE; //We want to filter on the first run. state->refilter = TRUE; state->update = FALSE; state->rchanged = TRUE; state->cur_page = -1; state->border = config.padding + config.menu_bw; state->x11_event_loop = rofi_view_mainloop_iter; state->finalize = finalize; // Request the lines to show. state->num_lines = mode_get_num_entries ( sw ); TICK_N ( "Startup notification" ); // Try to grab the keyboard as early as possible. // We grab this using the rootwindow (as dmenu does it). // this seems to result in the smallest delay for most people. if ( ( CacheState.flags & MENU_NORMAL_WINDOW ) == 0 ) { int has_keyboard = take_keyboard ( xcb_stuff_get_root_window ( xcb ) ); if ( !has_keyboard ) { fprintf ( stderr, "Failed to grab keyboard, even after %d uS.", 500 * 1000 ); // Break off. rofi_view_free ( state ); return NULL; } take_pointer ( xcb_stuff_get_root_window ( xcb ) ); } TICK_N ( "Grab keyboard" ); // Get active monitor size. TICK_N ( "Get active monitor" ); // we need this at this point so we can get height. state->line_height = textbox_get_estimated_char_height (); int element_height = state->line_height * config.element_height; state->case_indicator = textbox_create ( TB_AUTOWIDTH, ( state->border ), ( state->border ), 0, state->line_height, NORMAL, "*" ); state->top_offset = state->border * 1 + state->line_height + 2 + config.line_margin * 2; // Height of a row. if ( config.menu_lines == 0 || config.fullscreen ) { // Autosize it. int h = CacheState.mon.h - state->top_offset - config.padding; if ( config.sidebar_mode == TRUE ) { h -= state->line_height + config.line_margin; } int r = MAX ( 1, ( h ) / ( element_height + config.line_margin ) ); state->menu_lines = r; } else { state->menu_lines = config.menu_lines; } rofi_view_calculate_rows_columns ( state ); rofi_view_calculate_window_and_element_width ( state ); // Prompt box. state->prompt_tb = textbox_create ( TB_AUTOWIDTH, ( state->border ), ( state->border ), 0, state->line_height, NORMAL, prompt ); // Entry box int entrybox_width = CacheState.width - ( 2 * ( state->border ) ) - textbox_get_width ( state->prompt_tb ) - textbox_get_width ( state->case_indicator ); TextboxFlags tfl = TB_EDITABLE; tfl |= ( ( menu_flags & MENU_PASSWORD ) == MENU_PASSWORD ) ? TB_PASSWORD : 0; state->text = textbox_create ( tfl, ( state->border ) + textbox_get_width ( state->prompt_tb ), ( state->border ), entrybox_width, state->line_height, NORMAL, input ); // Move indicator to end. widget_move ( WIDGET ( state->case_indicator ), state->border + textbox_get_width ( state->prompt_tb ) + entrybox_width, state->border ); textbox_text ( state->case_indicator, get_matching_state () ); state->message_tb = NULL; if ( message ) { state->message_tb = textbox_create ( TB_AUTOHEIGHT | TB_MARKUP | TB_WRAP, ( state->border ), state->top_offset, CacheState.width - ( 2 * ( state->border ) ), -1, NORMAL, message ); state->top_offset += textbox_get_height ( state->message_tb ); state->top_offset += config.line_margin * 2 + 2; } state->overlay = textbox_create ( TB_AUTOWIDTH, 0, 0, 20, state->line_height, URGENT, "blaat" ); widget_disable ( WIDGET ( state->overlay ) ); // filtered list display state->boxes = g_malloc0_n ( state->max_elements, sizeof ( textbox* ) ); int y_offset = state->top_offset; int x_offset = state->border; for ( unsigned int i = 0; i < state->max_elements; i++ ) { state->boxes[i] = textbox_create ( ( state->menu_flags & MENU_INDICATOR ) ? TB_INDICATOR : 0, x_offset, y_offset, state->element_width, element_height, NORMAL, "" ); } if ( !config.hide_scrollbar ) { unsigned int sbw = config.line_margin + config.scrollbar_width; state->scrollbar = scrollbar_create ( CacheState.width - state->border - sbw, state->top_offset, sbw, ( state->max_rows - 1 ) * ( element_height + config.line_margin ) + element_height ); } scrollbar_set_max_value ( state->scrollbar, state->num_lines ); // filtered list state->line_map = g_malloc0_n ( state->num_lines, sizeof ( unsigned int ) ); state->distance = (int *) g_malloc0_n ( state->num_lines, sizeof ( int ) ); // resize window vertically to suit // Subtract the margin of the last row. CacheState.height = state->top_offset + ( element_height + config.line_margin ) * ( state->max_rows ) - config.line_margin; CacheState.height += state->border; CacheState.height += 0; // Add entry if ( config.sidebar_mode == TRUE ) { CacheState.height += state->line_height + 2 * config.line_margin + 2; } // Sidebar or fullscreen mode. if ( config.menu_lines == 0 || config.fullscreen ) { CacheState.height = CacheState.mon.h; } // Move the window to the correct x,y position. calculate_window_position ( ); if ( config.sidebar_mode == TRUE ) { state->num_modi = rofi_get_num_enabled_modi (); int width = ( CacheState.width - ( 2 * ( state->border ) + ( state->num_modi - 1 ) * config.line_margin ) ) / state->num_modi; state->modi = g_malloc0 ( state->num_modi * sizeof ( textbox * ) ); for ( unsigned int j = 0; j < state->num_modi; j++ ) { const Mode * mode = rofi_get_mode ( j ); state->modi[j] = textbox_create ( TB_CENTER, state->border + j * ( width + config.line_margin ), CacheState.height - state->line_height - state->border, width, state->line_height, ( mode == state->sw ) ? HIGHLIGHT : NORMAL, mode_get_display_name ( mode ) ); } } uint16_t mask = XCB_CONFIG_WINDOW_X | XCB_CONFIG_WINDOW_Y | XCB_CONFIG_WINDOW_WIDTH | XCB_CONFIG_WINDOW_HEIGHT; uint32_t vals[] = { CacheState.x, CacheState.y, CacheState.width, CacheState.height }; // Display it. xcb_configure_window ( xcb->connection, CacheState.main_window, mask, vals ); cairo_xcb_surface_set_size ( CacheState.surface, CacheState.width, CacheState.height ); // if grabbing keyboard failed, fall through state->selected = 0; state->quit = FALSE; state->update = TRUE; rofi_view_refilter ( state ); rofi_view_update ( state ); xcb_map_window ( xcb->connection, CacheState.main_window ); xcb_flush ( xcb->connection ); if ( xcb->sncontext != NULL ) { sn_launchee_context_complete ( xcb->sncontext ); } // TODO move resize window into the 'active window' part. return state; } static void __error_dialog_event_loop ( RofiViewState *state, xcb_generic_event_t *ev, G_GNUC_UNUSED xkb_stuff *xkb ) { // Handle event. switch ( ev->response_type & ~0x80 ) { // Key press event. case XCB_KEY_PRESS: state->quit = TRUE; } rofi_view_update ( state ); } int rofi_view_error_dialog ( const char *msg, int markup ) { RofiViewState *state = __rofi_view_state_create (); state->retv = MENU_CANCEL; state->update = TRUE; state->border = config.padding + config.menu_bw; state->x11_event_loop = __error_dialog_event_loop; state->menu_flags = MENU_ERROR_DIALOG; state->finalize = process_result; // Try to grab the keyboard as early as possible. // We grab this using the rootwindow (as dmenu does it). // this seems to result in the smallest delay for most people. if ( ( CacheState.flags & MENU_NORMAL_WINDOW ) == 0 ) { int has_keyboard = take_keyboard ( xcb_stuff_get_root_window ( xcb ) ); if ( !has_keyboard ) { fprintf ( stderr, "Failed to grab keyboard, even after %d uS.", 500 * 1000 ); return FALSE; } take_pointer ( xcb_stuff_get_root_window ( xcb ) ); } rofi_view_calculate_window_and_element_width ( state ); state->max_elements = 0; state->text = textbox_create ( ( TB_AUTOHEIGHT | TB_WRAP ) + ( ( markup ) ? TB_MARKUP : 0 ), ( state->border ), ( state->border ), ( CacheState.width - ( 2 * ( state->border ) ) ), 1, NORMAL, ( msg != NULL ) ? msg : "" ); state->line_height = textbox_get_height ( state->text ); // resize window vertically to suit CacheState.height = state->line_height + ( state->border ) * 2; // Calculte window position. calculate_window_position ( ); // Move the window to the correct x,y position. uint16_t mask = XCB_CONFIG_WINDOW_X | XCB_CONFIG_WINDOW_Y | XCB_CONFIG_WINDOW_WIDTH | XCB_CONFIG_WINDOW_HEIGHT; uint32_t vals[] = { CacheState.x, CacheState.y, CacheState.width, CacheState.height }; xcb_configure_window ( xcb->connection, CacheState.main_window, mask, vals ); cairo_xcb_surface_set_size ( CacheState.surface, CacheState.width, CacheState.height ); // Display it. xcb_map_window ( xcb->connection, CacheState.main_window ); if ( xcb->sncontext != NULL ) { sn_launchee_context_complete ( xcb->sncontext ); } // Set it has current window. rofi_view_set_active ( state ); // TODO move resize window into the 'active window' part. return TRUE; } void rofi_view_hide ( void ) { if ( CacheState.main_window != XCB_WINDOW_NONE ) { xcb_unmap_window ( xcb->connection, CacheState.main_window ); release_keyboard ( ); xcb_flush ( xcb->connection ); } } void rofi_view_cleanup () { if ( CacheState.fake_bg ) { cairo_surface_destroy ( CacheState.fake_bg ); CacheState.fake_bg = NULL; } if ( CacheState.draw ) { cairo_destroy ( CacheState.draw ); CacheState.draw = NULL; } if ( CacheState.surface ) { cairo_surface_destroy ( CacheState.surface ); CacheState.surface = NULL; } if ( CacheState.main_window != XCB_WINDOW_NONE ) { xcb_unmap_window ( xcb->connection, CacheState.main_window ); xcb_destroy_window ( xcb->connection, CacheState.main_window ); CacheState.main_window = XCB_WINDOW_NONE; } if ( map != XCB_COLORMAP_NONE ) { xcb_free_colormap ( xcb->connection, map ); map = XCB_COLORMAP_NONE; } g_assert ( g_queue_is_empty ( &( CacheState.views ) ) ); } void rofi_view_workers_initialize ( void ) { TICK_N ( "Setup Threadpool, start" ); if ( config.threads == 0 ) { config.threads = 1; long procs = sysconf ( _SC_NPROCESSORS_CONF ); if ( procs > 0 ) { config.threads = MIN ( procs, 128l ); } } // Create thread pool GError *error = NULL; tpool = g_thread_pool_new ( rofi_view_call_thread, NULL, config.threads, FALSE, &error ); if ( error == NULL ) { // Idle threads should stick around for a max of 60 seconds. g_thread_pool_set_max_idle_time ( 60000 ); // We are allowed to have g_thread_pool_set_max_threads ( tpool, config.threads, &error ); } // If error occured during setup of pool, tell user and exit. if ( error != NULL ) { fprintf ( stderr, "Failed to setup thread pool: '%s'", error->message ); g_error_free ( error ); exit ( EXIT_FAILURE ); } TICK_N ( "Setup Threadpool, done" ); } void rofi_view_workers_finalize ( void ) { if ( tpool ) { g_thread_pool_free ( tpool, TRUE, FALSE ); tpool = NULL; } } Mode * rofi_view_get_mode ( RofiViewState *state ) { return state->sw; } void rofi_view_set_overlay ( RofiViewState *state, const char *text ) { if ( state->overlay == NULL ) { return; } if ( text == NULL ) { widget_disable ( WIDGET ( state->overlay ) ); state->update = TRUE; return; } widget_enable ( WIDGET ( state->overlay ) ); textbox_text ( state->overlay, text ); unsigned int x_offset = CacheState.width - ( 2 * state->border ) - textbox_get_width ( state->case_indicator ); x_offset -= textbox_get_width ( state->overlay ); widget_move ( WIDGET ( state->overlay ), x_offset, state->border ); state->update = TRUE; }