import olex
import olx
from olexFunctions import OV

import inspect, os
curr_path = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))

class GuiImages:

  def __init__(self):
    if olx.HasGUI() != 'true':
      return

    img_root = OV.GetParam("gui.GuiImages.action", "%s/etc/gui/images/action" %OV.BaseDir())
    height = OV.GetParam('gui.html.input_height')
    self.img_d = {
      'root':img_root,
      'width':height,
      'height':height,
      'bg':OV.GetParam('gui.html.table_bg_colour').hexadecimal,
    }

  def make_action_button_html(self):
    link = '''
      <input
        type="button"
        image="up=%(root)s/%(which)s_off.png,down=%(root)s/%(which)s_on.png,hover=%(root)s/%(which)s_hover.png"
        hint="%(hint)s"
        width="%(width)s"
        height="%(height)s"
        onclick="%(onclick)s"
        bgcolor="%(bg)s"
      >''' %self.img_d
    return link

  def get_action_button_html(self, which, onclick, hint):
    self.img_d['which'] = which
    self.img_d['onclick'] = onclick
    self.img_d['hint'] = hint
    return self.make_action_button_html()

GI = GuiImages()
OV.registerFunction(GI.make_action_button_html,False,'gui.images')
OV.registerFunction(GI.get_action_button_html,False,'gui.images')


##Moved from olexex: Anything to do with structure image generation
def GetBitmapSize(px_changed=False):
  resolution = OV.GetParam('user.image.bitmap.resolution')
  try:
    resolution = float(resolution)
  except:
    return False, False
  width = OV.GetParam('user.image.bitmap.resolution_width')
  try:
    width = float(width)
  except:
    return False, False
  unit = OV.GetParam('user.image.bitmap.resolution_unit')
  size = OV.GetParam('user.image.bitmap.size')
  try:
    size = float(size)
  except:
    return False, False
  factor = 1
  if unit == "cm":
    factor = 0.393700787
  if px_changed:
    width = size/(resolution * factor)
    if OV.IsControl('IMAGE_BITMAP_width'):
      olx.html.SetValue('IMAGE_BITMAP_width',round(width,3))
    OV.SetParam('user.image.bitmap.resolution_width',round(width,3))
  else:
    size = resolution * width * factor
    if OV.IsControl('IMAGE_BITMAP_SIZE'):
      olx.html.SetValue('IMAGE_BITMAP_SIZE',int(round(size,0)))
    OV.SetParam('user.image.bitmap.size',int(round(size,0)))
  return int(round(resolution,0)), int(round(size,0))
OV.registerFunction(GetBitmapSize,False,'gui.images')


def MakeBitmapImage(notify_listener=True, output_folder=None):
  from ImageTools import IT
  filefull, filename, fileext = GetImageFilename(image_type = "BITMAP")
  if not filefull:
    if output_folder:
      name = OV.HKLSrc()
      _ = os.path.join(output_folder, "%s.%s" %(name, fileext))
      filefull = _
  if not filefull:
    return
  resolution, filesize = GetBitmapSize()
  if not filesize:
    return
  resolution = "-dpi=%s" %resolution
  if OV.GetParam('user.image.bitmap.transparent_background'):
    nbg = "-nbg"
  else:
    nbg = ""
  if fileext == "png/s":
    filefull = "%s.png" %filename
    filesize = 1
    pict = "a "
  else:
    pict = " -pq"
  OV.Cursor('busy','Please Wait. Making image %s.%s. This may take some time' %(filename, fileext))

  filesize = int(round(filesize,0))
  
  if OV.GetParam('user.image.bitmap.trim'):
    temp_name = '%s_temp.bmp' %(filefull.strip("'"))
    temp_name = 'temp.bmp'
    try:
      olex.m('picta "%s" 1' %(temp_name))
      new_width, old_width = IT.trim_image(im=temp_name, dry=True)
      target_size = filesize
      filesize = filesize * (old_width/new_width)
      olex.m('pict%s %s "%s" %s %s' %(pict, nbg, filefull, resolution, int(round(filesize,0))))
      if OV.GetParam('user.image.bitmap.transparent_background'):
        trimcolour=(255,255,255,0)
      else:
        trimcolour = None
      IT.trim_image(im=filefull, trimcolour=trimcolour, target_size=target_size)
    finally:
      if os.path.exists(temp_name):
        os.unlink(temp_name)
  else:
    olex.m('pict%s %s "%s" %s %s' %(pict, nbg, filefull, resolution, filesize))

  #from PIL import Image
  OV.Cursor()
  if notify_listener:
    from gui import ImageListener
    ImageListener.OnChange()
OV.registerFunction(MakeBitmapImage,False,'gui.images')

def MakePovRayImage():
  filefull, filename, fileext = GetImageFilename(image_type = "PR")
  if not filefull:
    return
  OV.Cursor('busy','Please Wait. Making image %s.%s. This may take some time' %(filename, fileext))
  olex.m('pictPR "%s"' %filefull)
  print('Image %s created' %filefull)
  OV.Cursor()
  from gui import ImageListener
  ImageListener.OnChange()
OV.registerFunction(MakePovRayImage,False,'gui.images')

def MakePostScriptImage(notify_listener=True, output_folder=None):
  filefull, filename, fileext = GetImageFilename(image_type = "PS")
  if output_folder:
    filefull = "%s/%s.%s" %(output_folder, filename, fileext)
  if not filefull:
    return
  OV.Cursor('busy','Please Wait. Making image %s.%s. This may take some time' %(filename, fileext))

  colour_line = OV.GetParam('snum.image.ps.colour_line')
  colour_bond = OV.GetParam('snum.image.ps.colour_bond')
  colour_fill = OV.GetParam('snum.image.ps.colour_fill')
  image_perspective = OV.GetParam('snum.image.ps.perspective')
  lw_ellipse = str(OV.GetParam('snum.image.ps.outline_width'))
  lw_octant = str(OV.GetParam('snum.image.ps.octant_width'))
  lw_pie = str(OV.GetParam('snum.image.ps.pie_width'))
  lw_font = str(OV.GetParam('snum.image.ps.font_weight'))
  div_pie = str(OV.GetParam('snum.image.ps.octant_count'))
  scale_hb = str(OV.GetParam('snum.image.ps.h_bond_width'))
  octant_atoms = '-$C'
  import olex_gui
  if olex_gui.IsControl('IMAGE_PS_OCTANTS_ATOMS'):
    octant_atoms = olx.html.GetValue('IMAGE_PS_OCTANTS_ATOMS')

  if image_perspective.lower() == 'true':
    image_perspective = "-p"

  olex.m("pictps" + \
         " '" + filefull + \
         "' " + colour_line + \
         " " + colour_bond + \
         " " + image_perspective + \
         " " + colour_fill + \
         " " + "-bond_outline_oversize=" + "20" + \
         " " + "-atom_outline_color=" + "16777215" + \
         " " + "-bond_outline_color=" + "15592941" + \
         " " + "-lw_ellipse=" + lw_ellipse + \
         " " + "-lw_octant=" + lw_octant + \
         " " + "-lw_pie=" + lw_pie + \
         " " + "-lw_font=" + lw_font + \
         " " + "-div_pie=" + div_pie + \
         " " + "-octants=" + octant_atoms + \
         " " + "-scale_hb=" + scale_hb)

  print('Image %s created' %filefull)
  OV.Cursor()
  if notify_listener:
    from gui import ImageListener
    ImageListener.OnChange()
OV.registerFunction(MakePostScriptImage,False,'gui.images')

def GetImageFilename(image_type):
  filename = OV.GetParam('snum.image.name')
  OV.SetParam('snum.image.name', None)
  if image_type == "PS":
    fileext = "eps"
  elif image_type == "PR":
    fileext = "pov"
  elif image_type == "BITMAP":
    fileext = OV.GetParam('user.image.bitmap.type')
  else:
    fileext = OV.GetParam('snum.image.%s.type' %image_type.lower())
  if not filename:
    try:
      current = int(olx.xf.CurrentData())
      filename = olx.xf.DataName(current)
    except:
      filename = OV.FileName()
    if not filename:
      import gui
      filename = gui.FileSave("Choose Filename", "*.%s" %fileext, OV.FilePath())
    if not filename:
      return None, None, None
  if_exists = OV.GetParam('snum.image.if_file_exists')
  if os.path.exists("%s.%s" %(filename, fileext)):
    if if_exists == 'increment':
      fp = olx.FilePath()
      fn = filename
      inc = 1
      while True:
        tf = os.path.normpath('%s/%s%d.%s' %(fp, fn, inc, fileext))
        if not os.path.exists(tf):
          filename = '%s%d' %(fn, inc)
          filefull = "'%s.%s'" %(filename, fileext)
          break
        inc += 1
    elif if_exists == 'ask':
      import gui
      filename = gui.FileSave("Choose Filename", "*.%s" %fileext, OV.FilePath(),
                              default_name=filename)
      if not filename:
        return None, None, None
  if filename.endswith(".%s" %fileext):
    filefull = "%s" %filename
  else:
    filefull = "%s.%s" %(filename, fileext)
  OV.SetParam('snum.image.%s.name' %image_type.lower(),None)
  return filefull, filename, fileext