#
#  Created by Boyd Multerer on 24/02/19.
#  Copyright © 2019 Kry10 Industries. All rights reserved.
#

defmodule TruetypeMetrics do
  @moduledoc """
  Documentation for TruetypeMetrics.
  """

  # import IEx

  @version FontMetrics.version()

  @version_one <<0, 1, 0, 0>>
  @magic_number_bin <<0x5F, 0x0F, 0x3C, 0xF5>>

  @invalid_font {:error, :invalid_font}
  @bad_checksum {:error, :bad_checksum}

  @font_epoch {{1904, 1, 1}, {0, 0, 0}}
              |> :calendar.datetime_to_gregorian_seconds()

  @unix_epoch {{1970, 1, 1}, {0, 0, 0}}
              |> :calendar.datetime_to_gregorian_seconds()

  @font_to_unix_seconds @font_epoch - @unix_epoch

  @signature_type FontMetrics.expected_hash()

  # def go(), do: load( "test/fonts/Roboto/Roboto-Regular.ttf" )
  # def go(), do: load( "test/fonts/Bitter/Bitter-Regular.ttf" )

  # ============================================================================
  # test-only accessors
  # Yes. I know this is a debatable technique, but I really want to test the
  # checksum directly yet not expose it as a general api.
  if Mix.env() == :test do
    def test_checksum(data), do: checksum(data)
  end

  # ============================================================================
  # Main api

  # --------------------------------------------------------
  # inspect a ttf font. Is limited. Returns a map of font info
  def load(font_path) when is_bitstring(font_path) do
    case File.read(font_path) do
      {:ok, raw_font} -> parse(raw_font, Path.basename(font_path))
      err -> err
    end
  end

  # --------------------------------------------------------
  # inspect a ttf font. Is limited. Returns a map of font info
  def load!(font_path) when is_bitstring(font_path) do
    File.read!(font_path) |> parse!(Path.basename(font_path))
  end

  # --------------------------------------------------------
  # start by parsing the initial font directory table
  def parse(
        <<
          @version_one,
          table_count::unsigned-integer-size(16)-big,
          _search_range::unsigned-integer-size(16)-big,
          _entry_selector::unsigned-integer-size(16)-big,
          _range_shift::unsigned-integer-size(16)-big,
          table_data::binary
        >> = font_data,
        file_name
      ) do
    # parse out all the pieces
    with {:ok, tables} <- parse_tables(table_data, table_count),
         {:ok, head} <- parse_head(font_data, tables),
         :ok <- check_master_checksum(font_data, head.checksum_adjustment),
         {:ok, glyph_ids} <- parse_cmap_glyphs_ids(font_data, tables),
         {:ok, hhea, metrics} <- parse_metrics(font_data, tables, glyph_ids),
         {:ok, kerning} <- parse_kern(font_data, tables, glyph_ids) do
      # calculate a better sha hash of the font
      signature = :crypto.hash(@signature_type, font_data)

      # build the final FontMetrics struct
      {:ok,
       %FontMetrics{
         version: @version,
         source: %FontMetrics.Source{
           signature_type: @signature_type,
           signature: signature,
           created_at: head.created,
           modified_at: head.modified,
           font_type: :true_type,
           file: file_name
         },
         direction: head.direction,
         smallest_ppem: head.smallest_ppem,
         units_per_em: head.units_per_em,
         max_box: {head.x_min, head.y_min, head.x_max, head.y_max},
         kerning: kerning,
         ascent: hhea.ascent,
         descent: hhea.descent,
         metrics: metrics,
         line_gap: hhea.line_gap
       }}
    else
      bad_checksum when is_integer(bad_checksum) -> @invalid_font
      err -> err
    end
  end

  def parse(_d, _), do: @invalid_font

  def parse!(data, file_name) do
    {:ok, metrics} = parse(data, file_name)
    metrics
  end

  # ============================================================================
  # internal utilities

  # --------------------------------------------------------
  # The first data to parse is the table locations. This is right after the main
  # file header. Recurse through the data

  defp parse_tables(table_data, count, tables \\ %{})
  defp parse_tables(_, 0, tables), do: {:ok, tables}
  defp parse_tables("", _, _), do: {:error, :invalid_table_map}

  defp parse_tables(
         <<
           tag::binary-size(4),
           checksum::unsigned-integer-size(32)-big,
           offset::unsigned-integer-size(32)-big,
           table_size::unsigned-integer-size(32)-big,
           table_data::binary
         >>,
         count,
         tables
       ) do
    # build the header for the current table
    th = %{
      checksum: checksum,
      offset: offset,
      size: table_size
    }

    # recurse to get all the entries
    parse_tables(table_data, count - 1, Map.put(tables, tag, th))
  end

  defp parse_tables(_, _, _), do: {:error, :invalid_table_map}

  # ============================================================================
  # head table

  # --------------------------------------------------------
  defp parse_head(font_data, tables) do
    case get_table_data(font_data, tables, "head") do
      {:ok, head_data} -> do_parse_head(head_data)
      err -> err
    end
  end

  # --------------------------------------------------------
  defp do_parse_head(<<
         @version_one,
         font_revision::binary-size(4),
         checksum_adjustment::unsigned-integer-size(32)-big,
         @magic_number_bin,
         flags::binary-size(2),
         units_per_em::unsigned-integer-size(16)-big,
         created::signed-integer-size(64)-big,
         modified::signed-integer-size(64)-big,
         x_min::signed-integer-size(16)-big,
         y_min::signed-integer-size(16)-big,
         x_max::signed-integer-size(16)-big,
         y_max::signed-integer-size(16)-big,
         style::unsigned-integer-size(16)-big,
         lowest_ppem::unsigned-integer-size(16)-big,
         direction::signed-integer-size(16)-big,
         loca_format::integer-size(16)-big,
         glyph_data_format::integer-size(16)-big
       >>) do
    {:ok,
     %{
       version: @version_one,
       font_revision: font_revision,
       checksum_adjustment: checksum_adjustment,
       flags: flags,
       units_per_em: units_per_em,
       created: DateTime.from_unix!(created + @font_to_unix_seconds),
       modified: DateTime.from_unix!(modified + @font_to_unix_seconds),
       x_min: x_min,
       y_min: y_min,
       x_max: x_max,
       y_max: y_max,
       style: style,
       smallest_ppem: lowest_ppem,
       direction: direction,
       loca_format:
         case loca_format do
           0 -> :short
           1 -> :long
         end,
       glyph_data_format: glyph_data_format
     }}
  end

  # something required didn't match.
  defp do_parse_head(_), do: {:error, :invalid_table, "head"}

  # ============================================================================
  # metrics tables

  # --------------------------------------------------------
  # just gets the advancement data
  defp parse_metrics(font_data, tables, glyph_ids) do
    with {:ok, data} <- get_table_data(font_data, tables, "hhea"),
         {:ok, hhea} <- parse_hhea(data),
         {:ok, data} <- get_table_data(font_data, tables, "hmtx"),
         {:ok, hmtx} <- parse_hmtx(data, hhea.num_h_metrics) do
      # combine it all together..

      # The id points to a glyph. We want to point to codepoints
      metrics =
        Enum.reduce(glyph_ids, %{}, fn {id, codepoints}, out ->
          case hmtx[id] do
            nil -> out
            adv -> Enum.reduce(codepoints, out, &Map.put(&2, &1, adv))
          end
        end)

      # get the default advance
      default_advance = metrics[0xFFFF] || metrics[0] || hhea.advance_width_max

      # scan metrics one more time to remove dupes of default_advance
      metrics =
        Enum.reduce(metrics, metrics, fn
          {cp, ^default_advance}, m -> Map.delete(m, cp)
          _, m -> m
        end)
        |> Map.put(0, default_advance)

      {:ok, hhea, metrics}
    else
      err -> err
    end
  end

  # --------------------------------------------------------
  defp parse_hhea(<<
         @version_one,
         ascent::signed-integer-size(16)-big,
         descent::signed-integer-size(16)-big,
         line_gap::signed-integer-size(16)-big,
         advance_width_max::unsigned-integer-size(16)-big,
         min_left_side_bearing::signed-integer-size(16)-big,
         min_right_side_bearing::signed-integer-size(16)-big,
         x_max_extent::signed-integer-size(16)-big,
         caret_slope_rise::signed-integer-size(16)-big,
         caret_slope_run::signed-integer-size(16)-big,
         caret_offset::signed-integer-size(16)-big,
         # reserved space
         _::size(64),
         0::signed-integer-size(16)-big,
         num_h_metrics::unsigned-integer-size(16)-big
       >>) do
    {:ok,
     %{
       version: @version_one,
       ascent: ascent,
       descent: descent,
       line_gap: line_gap,
       advance_width_max: advance_width_max,
       min_left_side_bearing: min_left_side_bearing,
       min_right_side_bearing: min_right_side_bearing,
       x_max_extent: x_max_extent,
       caret_slope_rise: caret_slope_rise,
       caret_slope_run: caret_slope_run,
       caret_offset: caret_offset,
       num_h_metrics: num_h_metrics
     }}
  end

  # something required didn't match.
  defp parse_hhea(_), do: {:error, :invalid_table, "hhea"}

  # --------------------------------------------------------
  defp parse_hmtx(metrics, num_metrics, out \\ %{}, n \\ 0)

  defp parse_hmtx(_, 0, out, _), do: {:ok, out}

  defp parse_hmtx(
         <<
           advance_width::signed-integer-size(16)-big,
           _lsb::signed-integer-size(16)-big,
           metrics::binary
         >>,
         num_metrics,
         out,
         n
       ) do
    out = Map.put(out, n, advance_width)
    parse_hmtx(metrics, num_metrics - 1, out, n + 1)
  end

  # ============================================================================
  # cmap table - used to build a map going backwards from glyph_id to
  # one (or more) codepoints

  # --------------------------------------------------------
  defp parse_cmap_glyphs_ids(font_data, tables) do
    case get_table_data(font_data, tables, "cmap") do
      {:ok, data} -> do_parse_cmap_glyphs_ids(data)
      err -> err
    end
  end

  # --------------------------------------------------------
  defp do_parse_cmap_glyphs_ids(
         <<
           0::unsigned-integer-size(16)-big,
           num_tables::unsigned-integer-size(16)-big,
           data::binary
         >> = cmap_data
       ) do
    # parse out the encoding sub-tables
    {encoding_types, _data} = do_parse_cmap_encoding_tables(data, num_tables)

    # part 2. build the char map
    # find a sub-table type we understand ( some form of unicode )
    case do_parse_cmap_get_unicode(encoding_types) do
      {:ok, offset} ->
        <<_::binary-size(offset), cmap::binary>> = cmap_data
        do_parse_unicode_cmap_glyphs(cmap)

      _ ->
        {:error, :cmap}
    end
  end

  # --------------------------------------------------------
  # table format 0 - single byte. very uncommon now
  # defp do_parse_unicode_cmap_glyphs(
  #   <<
  #     0 :: unsigned-integer-size(16)-big,
  #     size :: unsigned-integer-size(16)-big,
  #     len :: unsigned-integer-size(16)-big,   # length of sub-table
  #     language :: unsigned-integer-size(16)-big,
  #     sub_table :: binary-size(size),
  #     _ :: binary
  #   >>
  # ) do
  #   pry()
  #   do_parse_unicode_cmap_0( sub_table )
  # end

  # type 4 - disconnected ranges. ugh. Pretty common tho
  defp do_parse_unicode_cmap_glyphs(<<
         4::unsigned-integer-size(16)-big,
         _size::unsigned-integer-size(16)-big,
         _language::unsigned-integer-size(16)-big,
         seg_count_x2::unsigned-integer-size(16)-big,
         _search_range::unsigned-integer-size(16)-big,
         _entry_selector::unsigned-integer-size(16)-big,
         _range_shift::unsigned-integer-size(16)-big,
         end_codes::binary-size(seg_count_x2),
         # reserve pad
         0::unsigned-integer-size(16)-big,
         start_codes::binary-size(seg_count_x2),
         id_deltas::binary-size(seg_count_x2),
         id_range_offsets::binary-size(seg_count_x2),
         sub_table::binary
       >>) do
    ranges = type_4_ranges(start_codes, end_codes, id_deltas, id_range_offsets, seg_count_x2)

    glyph_ids =
      Enum.reduce(ranges, %{}, fn
        {f, l, delta, 0, _}, acc ->
          # "relatively" easy case. index is the codepoint - the delta
          Enum.reduce(f..l, acc, fn codepoint, ids ->
            glyph_id = Integer.mod(codepoint + delta, 65536)
            Map.put(ids, glyph_id, [codepoint | Map.get(ids, glyph_id, [])])
          end)

        {f, l, _, offset, compensator}, acc ->
          # glyph is obtained by looking it up in the sub_table
          # unfortunately, the offset is from *the position in the offsets table*
          # which may allow for a tricky optimization in C, but totally sucks here.
          # to compensate, subtract the pre-calculated compensator
          Enum.reduce(f..l, acc, fn codepoint, ids ->
            skip = (codepoint - f) * 2 + (offset + compensator)
            glyph_id = lookup_cmap_type_4(sub_table, skip)
            Map.put(ids, glyph_id, [codepoint | Map.get(ids, glyph_id, [])])
          end)
      end)

    {:ok, glyph_ids}
  end

  defp lookup_cmap_type_4(sub_table, skip) do
    <<_::binary-size(skip), id::unsigned-size(16)-big, _::binary>> = sub_table
    id
  end

  # type 6 - densly mapped relatively easy
  # defp parse_unicode_cmap(
  # <<
  #   6 :: unsigned-integer-size(16)-big,
  #   size :: unsigned-integer-size(16)-big,
  #   language :: unsigned-integer-size(16)-big,
  #   start_code :: unsigned-integer-size(16)-big,
  #   entry_count :: unsigned-integer-size(16)-big,
  #   sub_table :: binary-size(size),
  #   _ :: binary
  # >> ) do
  #   pry()
  # end

  # defp do_parse_unicode_cmap_0( subtable, n \\ 0, out \\ %{} )
  # defp do_parse_unicode_cmap_0( _, 256, out ), do: out
  # defp do_parse_unicode_cmap_0( << i, bin :: binary >>, n, out ) do
  #   do_parse_unicode_cmap_0( bin, n + 1, Map.put( out, n, i ) )
  # end

  defp type_4_ranges(starts, ends, deltas, offsets, remaining_x2, out \\ [])
  defp type_4_ranges("", "", "", "", _, out), do: Enum.reverse(out)

  defp type_4_ranges(
         <<first::unsigned-size(16)-big, starts::binary>>,
         <<last::unsigned-size(16)-big, ends::binary>>,
         <<delta::signed-size(16)-big, deltas::binary>>,
         <<offset::unsigned-size(16)-big, offsets::binary>>,
         remaining_x2,
         out
       ) do
    out = [{first, last, delta, offset, -remaining_x2} | out]
    type_4_ranges(starts, ends, deltas, offsets, remaining_x2 - 2, out)
  end

  # ----------------------------------------------
  defp do_parse_cmap_get_unicode(%{unicode: %{offset: offset}}), do: {:ok, offset}

  defp do_parse_cmap_get_unicode(%{microsoft: %{encoding_id: 1, offset: offset}}) do
    {:ok, offset}
  end

  defp do_parse_cmap_get_unicode(%{microsoft: %{encoding_id: 10, offset: offset}}) do
    {:ok, offset}
  end

  defp do_parse_cmap_get_unicode(_), do: :error

  # ----------------------------------------------
  defp do_parse_cmap_encoding_tables(data, count, encoding_types \\ %{})
  defp do_parse_cmap_encoding_tables(data, 0, encoding_types), do: {encoding_types, data}

  defp do_parse_cmap_encoding_tables(
         <<
           0::unsigned-integer-size(16)-big,
           encoding_id::unsigned-integer-size(16)-big,
           offset::unsigned-integer-size(32)-big,
           data::binary
         >>,
         count,
         encoding_types
       ) do
    do_parse_cmap_encoding_tables(
      data,
      count - 1,
      Map.put(encoding_types, :unicode, %{encoding_id: encoding_id, offset: offset})
    )
  end

  defp do_parse_cmap_encoding_tables(
         <<
           1::unsigned-integer-size(16)-big,
           encoding_id::unsigned-integer-size(16)-big,
           offset::unsigned-integer-size(32)-big,
           data::binary
         >>,
         count,
         encoding_types
       ) do
    do_parse_cmap_encoding_tables(
      data,
      count - 1,
      Map.put(encoding_types, :macintosh, %{encoding_id: encoding_id, offset: offset})
    )
  end

  defp do_parse_cmap_encoding_tables(
         <<
           3::unsigned-integer-size(16)-big,
           encoding_id::unsigned-integer-size(16)-big,
           offset::unsigned-integer-size(32)-big,
           data::binary
         >>,
         count,
         encoding_types
       ) do
    do_parse_cmap_encoding_tables(
      data,
      count - 1,
      Map.put(encoding_types, :microsoft, %{encoding_id: encoding_id, offset: offset})
    )
  end

  # ============================================================================
  # kerning table

  # --------------------------------------------------------
  defp parse_kern(font_data, tables, glyph_ids) do
    case get_table_data(font_data, tables, "kern") do
      {:ok, data} -> do_parse_kern(data, glyph_ids)
      _ -> {:ok, %{}}
    end
  end

  # --------------------------------------------------------
  # kerning table format 0
  defp do_parse_kern(
         <<
           0::unsigned-integer-size(16)-big,
           _num_tables::unsigned-integer-size(16)-big,
           _size::unsigned-integer-size(32)-big,
           _coverage::unsigned-integer-size(16)-big,
           _pair_count::unsigned-integer-size(16)-big,
           _search_range::unsigned-integer-size(16)-big,
           _entry_selector::unsigned-integer-size(16)-big,
           _range_shift::unsigned-integer-size(16)-big,
           data::binary
         >>,
         glyph_ids
       ) do
    do_parse_kern_0_entry(data, glyph_ids)
  end

  defp do_parse_kern_0_entry(kern_data, glyph_ids, pairs \\ %{})
  defp do_parse_kern_0_entry(<<>>, _, pairs), do: {:ok, pairs}

  defp do_parse_kern_0_entry(
         <<
           left::unsigned-integer-size(16)-big,
           right::unsigned-integer-size(16)-big,
           value::signed-integer-size(16)-big,
           kern_data::binary
         >>,
         glyph_ids,
         pairs
       ) do
    # the given pair is to glyph_id. which might point to more than one
    # codepoint. We want codepoint pairs, so need to expand this into
    # codepoint space, then add each to the pairs map.
    with {:ok, left_cps} <- Map.fetch(glyph_ids, left),
         {:ok, right_cps} <- Map.fetch(glyph_ids, right) do
      pairs =
        Enum.reduce(left_cps, [], fn left_cp, acc ->
          Enum.reduce(right_cps, acc, fn right_cp, acc ->
            [{left_cp, right_cp} | acc]
          end)
        end)
        |> Enum.reduce(pairs, &Map.put(&2, &1, value))

      # pairs = Map.put( pairs, {left, right}, value)
      do_parse_kern_0_entry(kern_data, glyph_ids, pairs)
    else
      _ -> do_parse_kern_0_entry(kern_data, glyph_ids, pairs)
    end
  end

  # ============================================================================
  # utilities

  # --------------------------------------------------------
  # magic number is per the truetype spec.
  # Not a private functions so it can be tested directly.
  @doc false
  defp check_master_checksum(bin, adjustment) do
    case 0xB1B0AFBA - checksum(bin, adjustment) do
      ^adjustment ->
        :ok

      sum ->
        # in C, the above subtraction rolls over in unsigned integer math
        # simulate that here.
        case 0xFFFFFFFF + sum + 1 do
          ^adjustment -> :ok
          _ -> {:error, :checksum}
        end
    end
  end

  # --------------------------------------------------------
  # Calculate a table checksum as defined by the truetype standard. This would
  # be much faster as native code, but part of the point is to do this in pure
  # BEAM code for safety. This whole module should not live in a performance
  # critical path.
  #
  # Not a private function so it can be tested directly.
  #
  @doc false
  defp checksum(bin, adjustment \\ 0) when is_binary(bin) do
    bin =
      case rem(byte_size(bin), 4) do
        0 -> bin
        1 -> bin <> <<0, 0, 0>>
        2 -> bin <> <<0, 0>>
        3 -> bin <> <<0>>
      end

    checksum = do_checksum(bin) - adjustment
    <<clamped::unsigned-size(32)>> = <<checksum::unsigned-size(32)>>
    clamped
  end

  defp do_checksum(bin, checksum \\ 0)
  defp do_checksum("", checksum), do: checksum

  defp do_checksum(<<long::unsigned-size(32)-big, bin::binary>>, checksum) do
    do_checksum(bin, checksum + long)
  end

  defp do_checksum(_, _), do: {:error, :checksum}

  # --------------------------------------------------------
  defp get_table_data(font_data, tables, "head") when is_binary(font_data) do
    case Map.get(tables, "head") do
      nil ->
        {:error, :missing_table, "head"}

      %{offset: offset, size: table_size} ->
        <<
          _::binary-size(offset),
          table_data::binary-size(table_size),
          _::binary
        >> = font_data

        {:ok, table_data}
    end
  end

  defp get_table_data(font_data, tables, table_id) when is_binary(font_data) do
    case Map.get(tables, table_id) do
      nil -> {:error, :missing_table, table_id}
      loc -> do_get_table_data(font_data, loc)
    end
  end

  defp do_get_table_data(
         font_data,
         %{offset: offset, size: table_size, checksum: checksum}
       )
       when is_binary(font_data) do
    <<
      _::binary-size(offset),
      table_data::binary-size(table_size),
      _::binary
    >> = font_data

    # test the checksum for validity
    case checksum(table_data) do
      ^checksum -> {:ok, table_data}
      _ -> @bad_checksum
    end
  end
end