Initial commit for Spectral.vim.

1 files changed, 250 insertions, 0 deletions

diff --git a/autoload/spectral.vim b/autoload/spectral.vim
new file mode 100644
index 0000000..0284d19
--- /dev/null
+++ b/autoload/spectral.vim
@@ -0,0 +1,250 @@
+function! s:dirname(arg) abort
+  return substitute(a:arg, '/[^/]*$', '', '')
+endfunction
+
+function! s:basename(arg) abort
+  return substitute(a:arg, '.\{-}/\([^/]*\)$', '\1', '')
+endfunction
+
+let s:cache_folder = s:dirname(expand('<sfile>')) . '/cache/'
+let s:this_file = expand('<sfile>')
+let s:this_folder = s:dirname(s:this_file)
+
+let s:source = []
+let s:cache_file = ''
+let s:loaded_python = 0
+let s:use_cache = 0
+
+function! spectral#brighter(color, amount)
+  let saved = 0
+  if exists('g:return')
+    let old_return = g:return
+    let saved = 1
+  endif
+
+  call s:loadPythonScript()
+  exec printf('python3 spectral_lighter("%s", %s)', a:color, a:amount)
+
+  let ret = g:return
+
+  if saved
+    let g:return = old_return
+  else
+    unlet g:return
+  endif
+
+  return ret
+endfunction
+
+function! spectral#desaturate(color, amount)
+  let matrix = [ spectral#normalize([1, a:amount, a:amount], 1),
+        \        spectral#normalize([a:amount, 1, a:amount], 1),
+        \        spectral#normalize([a:amount, a:amount, 1], 1) ]
+
+  let [r, g, b] = spectral#parseColor(a:color)
+
+  let nr = r * matrix[0][0] + g * matrix[0][1] + b * matrix[0][2]
+  let ng = r * matrix[1][0] + g * matrix[1][1] + b * matrix[1][2]
+  let nb = r * matrix[2][0] + g * matrix[2][1] + b * matrix[2][2]
+
+  return spectral#toColor(nr, ng, nb)
+endfunction
+
+
+function! spectral#parseColor(color) abort
+  if type(a:color) == v:t_list
+    return a:color
+  endif
+
+  return [
+        \ str2nr(a:color[1:2], 16),
+        \ str2nr(a:color[3:4], 16),
+        \ str2nr(a:color[5:6], 16) ]
+endfunction
+
+function! spectral#toColor(r, g, b) abort
+  let r = spectral#clip(a:r, 0, 255)
+  let g = spectral#clip(a:g, 0, 255)
+  let b = spectral#clip(a:b, 0, 255)
+
+  return printf("#%02x%02x%02x",
+        \ float2nr(r),
+        \ float2nr(g),
+        \ float2nr(b))
+endfunction
+
+function! spectral#clip(c, min, max) abort
+  if a:c < a:min
+    return a:min
+  elseif a:c > a:max
+    return a:max
+  else
+    return a:c
+  endif
+endfunction
+
+function! s:loadPythonScript()
+  if s:loaded_python
+    return
+  endif
+
+  exec 'py3file ' . s:this_folder . '/spectral.py'
+  let s:loaded_python = 1
+endfunction
+
+function! spectral#normalize(ls, target_mag) abort
+  let total_mag = 0
+  for l in a:ls
+    let total_mag += l * l
+  endfor
+  let total_mag = sqrt(total_mag)
+
+  let lsp = []
+  let frac = total_mag / a:target_mag
+  for l in a:ls
+    call add(lsp, l / frac)
+  endfor
+
+  return lsp
+endfunction
+
+function! spectral#mix(color1, color2, ...) abort
+  if a:0 == 1
+    let amt = a:1
+  else
+    let amt = 0.5
+  endif
+
+  let [r1, g1, b1] = spectral#parseColor(a:color1)
+  let [r2, g2, b2] = spectral#parseColor(a:color2)
+
+  return spectral#toColor(
+        \ r1 * (1 - amt) + r2 * amt,
+        \ g1 * (1 - amt) + g2 * amt,
+        \ b1 * (1 - amt) + b2 * amt)
+
+endfunction
+
+function! spectral#StartCompile()
+  if !exists('g:spectral_parameters')
+    let g:spectral_parameters = {}
+  endif
+
+  call system('mkdir -p ' . shellescape(s:cache_folder))
+
+  let basename = s:basename(g:spectral_colorscheme_file)
+  let s:cache_file = printf('%s/%s', s:cache_folder, basename)
+
+  if filereadable(s:cache_file)
+     let cache_file_time = getftime(s:cache_file)
+     let script_file_time = getftime(g:spectral_colorscheme_file)
+     let this_file_time = getftime(s:this_file)
+     let python_file_time = getftime(s:this_folder . '/spectral.py')
+
+     if cache_file_time >= script_file_time 
+           \ && cache_file_time >= this_file_time
+           \ && cache_file_time >= python_file_time
+      exec "source " . s:cache_file
+      let s:use_cache = 1
+      return 1
+    endif
+  endif
+
+  let s:use_cache = 0
+  call s:loadPythonScript()
+  exec "python3 spectral_start('" . s:cache_file . "')"
+
+  return 0
+endfunction
+
+function! s:resolve(name) abort
+  if (len(a:name) == 0 || a:name[0] != '#') && a:name != "None"
+    return eval(a:name)
+  else
+    return a:name
+  endif
+endfunction
+
+function! spectral#flip_brightness(color) abort
+  let saved = 0
+  if exists('g:return')
+    let old_return = g:return
+    let saved = 1
+  endif
+
+  call s:loadPythonScript()
+  exec printf('python3 spectral_flip_brightness("%s")', a:color)
+
+  let ret = g:return
+
+  if saved
+    let g:return = old_return
+  else
+    unlet g:return
+  endif
+
+  return ret
+endfunction
+
+function! spectral#Hi(bang, fn, name, ...)
+  if s:use_cache
+    return
+  endif
+
+  let count = a:0
+  if count == 4 
+    let fg = s:resolve(a:1)
+    let bg = s:resolve(a:2)
+    let sp = s:resolve(a:3)
+    let extra = eval(a:4)
+  elseif count == 3
+    let fg = s:resolve(a:1)
+    let bg = s:resolve(a:2)
+    let sp = "None"
+    let extra = eval(a:3)
+  elseif count == 2
+    let fg = s:resolve(a:1)
+    let bg = s:resolve(a:2)
+    let sp = "None"
+    let extra = []
+  elseif count == 1
+    let fg = s:resolve(a:1)
+    let bg = "None"
+    let sp = "None"
+    let extra = []
+  endif
+
+  if extra == []
+    let extra = ["None"]
+  endif
+
+  if a:bang
+    exec printf('python3 spectral_ctx.highlight_after("%s", "%s", "%s", "%s", %s)',
+          \ a:name, fg, bg, sp, string(extra))
+  elseif a:fn == "dark"
+    exec printf('python3 spectral_ctx.highlight_dark("%s", "%s", "%s", "%s", %s)',
+          \ a:name, fg, bg, sp, string(extra))
+  elseif a:fn == "light"
+    exec printf('python3 spectral_ctx.highlight_light("%s", "%s", "%s", "%s", %s)',
+          \ a:name, fg, bg, sp, string(extra))
+  else
+    exec printf('python3 spectral_ctx.highlight("%s", "%s", "%s", "%s", %s)',
+          \ a:name, fg, bg, sp, string(extra))
+  endif
+endfunction
+
+command! -bang -nargs=* Hi call spectral#Hi("<bang>" == "!", "", <f-args>)
+command! -bang -nargs=* HiDark call spectral#Hi("<bang>" == "!", "dark", <f-args>)
+command! -bang -nargs=* HiLight call spectral#Hi("<bang>" == "!", "light", <f-args>)
+
+function! spectral#Finish()
+  unlet g:spectral_parameters
+  unlet g:spectral_colorscheme_file
+
+  if s:use_cache
+    return
+  endif
+
+  python3 spectral_finish()
+  exec "source " . s:cache_file
+endfunction