/** \file
 * \brief Label Control.
 *
 * See Copyright Notice in "iup.h"
 */

#include <stdio.h>
#include <stdlib.h>

#include "iup.h"

#include "iup_object.h"
#include "iup_attrib.h"
#include "iup_str.h"
#include "iup_drvfont.h"
#include "iup_stdcontrols.h"
#include "iup_label.h"
#include "iup_image.h"
#include "iup_class.h"
#include "iup_markup.h"


static int iLabelSetSeparatorAttrib(Ihandle* ih, const char* value)
{
  /* valid only before map */
  if (ih->handle)
    return 0;

  if (value)
  {
    if (iupStrEqualNoCase(value, "HORIZONTAL"))
    {
      if (!(ih->expand & IUP_EXPAND_WIDTH))
        ih->expand = IUP_EXPAND_WFREE;
    }
    else if (iupStrEqualNoCase(value, "VERTICAL"))
    {
      if (!(ih->expand & IUP_EXPAND_HEIGHT))
        ih->expand = IUP_EXPAND_HFREE;
    }
    else
      return 0;
  }

  return 1;
}

static char* iLabelGetSeparatorAttrib(Ihandle* ih)
{
  if (ih->handle)
  {
    if (ih->data->type == IUP_LABEL_SEP_HORIZ)
      return "HORIZONTAL";
    else if (ih->data->type == IUP_LABEL_SEP_VERT)
      return "VERTICAL";
  }
  return NULL;
}

int iupLabelGetTypeBeforeMap(Ihandle* ih)
{
  int type = ih->data->type;

  if (!ih->handle)
  {
    /* if not mapped must initialize the internal values */
    char* value = iupAttribGet(ih, "SEPARATOR");
    if (value)
    {
      if (iupStrEqualNoCase(value, "HORIZONTAL"))
        type = IUP_LABEL_SEP_HORIZ;
      else /* "VERTICAL" */
        type = IUP_LABEL_SEP_VERT;
    }
    else
    {
      value = iupAttribGet(ih, "IMAGE");
      if (value)
        type = IUP_LABEL_IMAGE;
      else
        type = IUP_LABEL_TEXT;
    }
  }

  return type;
}

char* iupLabelGetPaddingAttrib(Ihandle* ih)
{
  /* this method can be called before map */
  int type = iupLabelGetTypeBeforeMap(ih);
  if (type != IUP_LABEL_SEP_HORIZ && type != IUP_LABEL_SEP_VERT)
    return iupStrReturnIntInt(ih->data->horiz_padding, ih->data->vert_padding, 'x');
  else
    return NULL;
}

/**************************************************************************************/

static int iLabelCreateMethod(Ihandle* ih, void** params)
{
  if (params && params[0])
  {
    int flags = 0;
    iupClassRegisterGetAttribute(ih->iclass, "MARKUP", NULL, NULL, NULL, NULL, &flags);
    if (flags & IUPAF_NOT_SUPPORTED)
    {
      char* stripped = iupMarkupStripTags((char*)(params[0]));
      iupAttribSetStr(ih, "TITLE", stripped);
      free(stripped);
    }
    else
      iupAttribSetStr(ih, "TITLE", (char*)(params[0]));
  }

  ih->data = iupALLOCCTRLDATA();

  /* used only by the Windows driver */
  ih->data->horiz_alignment = IUP_ALIGN_ALEFT;
  ih->data->vert_alignment = IUP_ALIGN_ACENTER;
  return IUP_NOERROR;
}

static void iLabelComputeNaturalSizeMethod(Ihandle* ih, int *w, int *h, int *children_expand)
{
  int natural_w = 0,
      natural_h = 0,
      type = iupLabelGetTypeBeforeMap(ih);
  (void)children_expand; /* unset if not a container */

  if (type == IUP_LABEL_SEP_HORIZ)
    natural_h = 2;
  else if (type == IUP_LABEL_SEP_VERT)
    natural_w = 2;
  else if (type == IUP_LABEL_IMAGE)
  {
    iupImageGetInfo(iupAttribGet(ih, "IMAGE"), &natural_w, &natural_h, NULL);

    natural_w += 2*ih->data->horiz_padding;
    natural_h += 2*ih->data->vert_padding;
  }
  else /* IUP_LABEL_TEXT */
  {
    /* must use IupGetAttribute to check from the native implementation */
    char* title = IupGetAttribute(ih, "TITLE");
    char* str = iupStrProcessMnemonic(title, NULL, 0);   /* remove & */
    iupdrvFontGetMultiLineStringSize(ih, str, &natural_w, &natural_h);
    if (str && str!=title) free(str);

    natural_w += 2*ih->data->horiz_padding;
    natural_h += 2*ih->data->vert_padding;
  }

  iupdrvLabelAddExtraPadding(ih, &natural_w, &natural_h);

  *w = natural_w;
  *h = natural_h;
}

/******************************************************************************/

IUP_API Ihandle* IupLabel(const char* title)
{
  void *params[2];
  params[0] = (void*)title;
  params[1] = NULL;
  return IupCreatev("label", params);
}

Iclass* iupLabelNewClass(void)
{
  Iclass* ic = iupClassNew(NULL);

  ic->name = "label";
  ic->format = "s"; /* one string */
  ic->format_attr = "TITLE";
  ic->nativetype = IUP_TYPECONTROL;
  ic->childtype = IUP_CHILDNONE;
  ic->is_interactive = 0;

  /* Class functions */
  ic->New = iupLabelNewClass;
  ic->Create = iLabelCreateMethod;
  ic->ComputeNaturalSize = iLabelComputeNaturalSizeMethod;

  ic->LayoutUpdate = iupdrvBaseLayoutUpdateMethod;
  ic->UnMap = iupdrvBaseUnMapMethod;

  /* Manually register the Common Callbacks */
  iupClassRegisterCallback(ic, "MAP_CB", "");
  iupClassRegisterCallback(ic, "UNMAP_CB", "");

  iupClassRegisterCallback(ic, "BUTTON_CB", "iiiis");
  iupClassRegisterCallback(ic, "MOTION_CB", "iis");
  iupClassRegisterCallback(ic, "ENTERWINDOW_CB", "");
  iupClassRegisterCallback(ic, "LEAVEWINDOW_CB", "");

  /* Common */
  iupBaseRegisterCommonAttrib(ic);

  /* Visual */
  iupBaseRegisterVisualAttrib(ic);

  /* Drag&Drop */
  iupdrvRegisterDragDropAttrib(ic);

  /* IupLabel only */
  iupClassRegisterAttribute(ic, "SEPARATOR", iLabelGetSeparatorAttrib, iLabelSetSeparatorAttrib, NULL, NULL, IUPAF_NOT_MAPPED|IUPAF_NO_INHERIT);
  iupClassRegisterAttribute(ic, "CPADDING", iupBaseGetCPaddingAttrib, iupBaseSetCPaddingAttrib, NULL, NULL, IUPAF_NO_SAVE | IUPAF_NOT_MAPPED);

  iupdrvLabelInitClass(ic);

  return ic;
}
