/*
 * Copyright (c) 1997-2006  Motoyuki Kasahara
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the project nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include "build-pre.h"
#include "eb.h"
#include "error.h"
#include "appendix.h"
#include "build-post.h"

/*
 * Unexported functions.
 */
static EB_Error_Code eb_load_appendix_subbook(EB_Appendix *appendix);
static EB_Error_Code eb_set_appendix_subbook_eb(EB_Appendix *appendix,
  EB_Subbook_Code subbook_code);
static EB_Error_Code eb_set_appendix_subbook_epwing(EB_Appendix *appendix,
  EB_Subbook_Code subbook_code);

/*
 * Initialize all subbooks in `appendix'.
 */
void
eb_initialize_appendix_subbooks(EB_Appendix *appendix)
{
  EB_Appendix_Subbook *subbook;
  int i;

  LOG(("in: eb_initialize_appendix_subbooks(appendix=%d)",
    (int)appendix->code));

  for (i = 0, subbook = appendix->subbooks; i < appendix->subbook_count;
     i++, subbook++) {
    subbook->initialized = 0;
    subbook->code = i;
    subbook->directory_name[0] = '\0';
    subbook->data_directory_name[0] = '\0';
    subbook->file_name[0] = '\0';
    subbook->character_code = EB_CHARCODE_INVALID;
    subbook->narrow_start = -1;
    subbook->wide_start = -1;
    subbook->narrow_end = -1;
    subbook->wide_end = -1;
    subbook->narrow_page = 0;
    subbook->wide_page = 0;
    subbook->stop_code0 = 0;
    subbook->stop_code1 = 0;
    zio_initialize(&subbook->zio);
  }

  LOG(("out: eb_initialize_appendix_subbooks()"));
}


/*
 * Initialize subbooks in `appendix'.
 */
void
eb_finalize_appendix_subbooks(EB_Appendix *appendix)
{
  EB_Appendix_Subbook *subbook;
  int i;

  LOG(("in: eb_finalize_appendix_subbooks(appendix=%d)",
    (int)appendix->code));

  for (i = 0, subbook = appendix->subbooks; i < appendix->subbook_count;
     i++, subbook++) {
    zio_finalize(&appendix->subbooks[i].zio);
  }

  LOG(("out: eb_finalize_appendix_subbooks()"));
}


/*
 * Load all subbooks in `appendix'.
 */
static EB_Error_Code
eb_load_appendix_subbook(EB_Appendix *appendix)
{
  EB_Error_Code error_code;
  EB_Appendix_Subbook *subbook;
  char buffer[16];
  int stop_code_page;
  int character_count;

  LOG(("in: eb_load_appendix_subbook(appendix=%d)", (int)appendix->code));

  subbook = appendix->subbook_current;

  /*
   * Check for the current status.
   */
  if (subbook == NULL) {
    error_code = EB_ERR_NO_CUR_APPSUB;
    goto failed;
  }

  /*
   * If the subbook has already initialized, return immediately.
   */
  if (subbook->initialized != 0)
    goto succeeded;

  /*
   * Rewind the APPENDIX file.
   */
  if (zio_lseek(&subbook->zio, 0, SEEK_SET) < 0) {
    error_code = EB_ERR_FAIL_SEEK_APP;
    goto failed;
  }

  /*
   * Set character code used in the appendix.
   */
  if (zio_read(&subbook->zio, buffer, 16) != 16) {
    error_code = EB_ERR_FAIL_READ_APP;
    goto failed;
  }
  subbook->character_code = eb_uint2(buffer + 2);

  /*
   * Set information about alternation text of wide font.
   */
  if (zio_read(&subbook->zio, buffer, 16) != 16) {
    error_code = EB_ERR_FAIL_READ_APP;
    goto failed;
  }
  character_count = eb_uint2(buffer + 12);

  if (0 < character_count) {
    subbook->narrow_page = eb_uint4(buffer);
    subbook->narrow_start = eb_uint2(buffer + 10);

    if (subbook->character_code == EB_CHARCODE_ISO8859_1) {
      subbook->narrow_end = subbook->narrow_start
        + ((character_count / 0xfe) << 8) + (character_count % 0xfe)
        - 1;
      if (0xfe < (subbook->narrow_end & 0xff))
        subbook->narrow_end += 3;

      if ((subbook->narrow_start & 0xff) < 0x01
        || 0xfe < (subbook->narrow_start & 0xff)
        || subbook->narrow_start < 0x0001
        || 0x1efe < subbook->narrow_end) {
        error_code = EB_ERR_UNEXP_APP;
        goto failed;
      }
    } else {
      subbook->narrow_end = subbook->narrow_start
        + ((character_count / 0x5e) << 8) + (character_count % 0x5e)
        - 1;
      if (0x7e < (subbook->narrow_end & 0xff))
        subbook->narrow_end += 0xa3;

      if ((subbook->narrow_start & 0xff) < 0x21
        || 0x7e < (subbook->narrow_start & 0xff)
        || subbook->narrow_start < 0xa121
        || 0xfe7e < subbook->narrow_end) {
        error_code = EB_ERR_UNEXP_APP;
        goto failed;
      }
    }
  }

  /*
   * Set information about alternation text of wide font.
   */
  if (zio_read(&subbook->zio, buffer, 16) != 16) {
    error_code = EB_ERR_FAIL_READ_APP;
    goto failed;
  }
  character_count = eb_uint2(buffer + 12);

  if (0 < character_count) {
    subbook->wide_page = eb_uint4(buffer);
    subbook->wide_start = eb_uint2(buffer + 10);

    if (subbook->character_code == EB_CHARCODE_ISO8859_1) {
      subbook->wide_end = subbook->wide_start
        + ((character_count / 0xfe) << 8) + (character_count % 0xfe)
        - 1;
      if (0xfe < (subbook->wide_end & 0xff))
        subbook->wide_end += 3;

      if ((subbook->wide_start & 0xff) < 0x01
        || 0xfe < (subbook->wide_start & 0xff)
        || subbook->wide_start < 0x0001
        || 0x1efe < subbook->wide_end) {
        error_code = EB_ERR_UNEXP_APP;
        goto failed;
      }
    } else {
      subbook->wide_end = subbook->wide_start
        + ((character_count / 0x5e) << 8) + (character_count % 0x5e)
        - 1;
      if (0x7e < (subbook->wide_end & 0xff))
        subbook->wide_end += 0xa3;

      if ((subbook->wide_start & 0xff) < 0x21
        || 0x7e < (subbook->wide_start & 0xff)
        || subbook->wide_start < 0xa121
        || 0xfe7e < subbook->wide_end) {
        error_code = EB_ERR_UNEXP_APP;
        goto failed;
      }
    }
  }

  /*
   * Set stop-code.
   */
  if (zio_read(&subbook->zio, buffer, 16) != 16) {
    error_code = EB_ERR_FAIL_READ_APP;
    goto failed;
  }
  stop_code_page = eb_uint4(buffer);
  if (0 < stop_code_page) {
    if (zio_lseek(&subbook->zio, ((off_t) stop_code_page - 1) * EB_SIZE_PAGE,
      SEEK_SET) < 0) {
      error_code = EB_ERR_FAIL_SEEK_APP;
      goto failed;
    }
    if (zio_read(&subbook->zio, buffer, 16) != 16) {
      error_code = EB_ERR_FAIL_READ_APP;
      goto failed;
    }
    if (eb_uint2(buffer) != 0) {
      subbook->stop_code0 = eb_uint2(buffer + 2);
      subbook->stop_code1 = eb_uint2(buffer + 4);
    }
  }

  /*
   * Rewind the file descriptor, again.
   */
  if (zio_lseek(&subbook->zio, 0, SEEK_SET) < 0) {
    error_code = EB_ERR_FAIL_SEEK_APP;
    goto failed;
  }

  /*
   * Initialize the alternation text cache.
   */
  eb_initialize_alt_caches(appendix);

  succeeded:
  LOG(("out: eb_load_appendix_subbook() = %s", eb_error_string(EB_SUCCESS)));
  return EB_SUCCESS;

  /*
   * An error occurs...
   */
  failed:
  LOG(("out: eb_load_appendix_subbook() = %s", eb_error_string(error_code)));
  return error_code;
}


/*
 * Load all subbooks in the book.
 */
EB_Error_Code
eb_load_all_appendix_subbooks(EB_Appendix *appendix)
{
  EB_Error_Code error_code;
  EB_Subbook_Code current_subbook_code;
  EB_Appendix_Subbook *subbook;
  int i;

  eb_lock(&appendix->lock);
  LOG(("in: eb_load_all_appendix_subbooks(appendix=%d)",
    (int)appendix->code));

  /*
   * The appendix must have been bound.
   */
  if (appendix->path == NULL) {
    error_code = EB_ERR_UNBOUND_APP;
    goto failed;
  }

  /*
   * Get the current subbook.
   */
  if (appendix->subbook_current != NULL)
    current_subbook_code = appendix->subbook_current->code;
  else
    current_subbook_code = -1;

  /*
   * Initialize each subbook.
   */
  for (i = 0, subbook = appendix->subbooks;
     i < appendix->subbook_count; i++, subbook++) {
    error_code = eb_set_appendix_subbook(appendix, subbook->code);
    if (error_code != EB_SUCCESS)
      goto failed;
  }

  /*
   * Restore the current subbook.
   */
  if (current_subbook_code < 0)
    eb_unset_appendix_subbook(appendix);
  else {
    error_code = eb_set_appendix_subbook(appendix, current_subbook_code);
    if (error_code != EB_SUCCESS)
      goto failed;
  }

  LOG(("out: eb_load_all_appendix_subbooks() = %s",
    eb_error_string(EB_SUCCESS)));
  eb_unlock(&appendix->lock);

  return EB_SUCCESS;

  /*
   * An error occurs...
   */
  failed:
  LOG(("out: eb_load_all_appendix_subbooks() = %s",
    eb_error_string(error_code)));
  eb_unlock(&appendix->lock);
  return error_code;
}


/*
 * Get a subbook list in `appendix'.
 */
EB_Error_Code
eb_appendix_subbook_list(EB_Appendix *appendix, EB_Subbook_Code *subbook_list,
  int *subbook_count)
{
  EB_Error_Code error_code;
  EB_Subbook_Code *list_p;
  int i;

  eb_lock(&appendix->lock);
  LOG(("in: eb_appendix_subbook_list(appendix=%d)", (int)appendix->code));

  /*
   * Check for the current status.
   */
  if (appendix->path == NULL) {
    error_code = EB_ERR_UNBOUND_APP;
    goto failed;
  }

  /*
   * Make a subbook list.
   */
  for (i = 0, list_p = subbook_list; i < appendix->subbook_count;
     i++, list_p++)
    *list_p = i;
  *subbook_count = appendix->subbook_count;

  LOG(("out: eb_appendix_subbook_list(subbook_count=%d) = %s",
    *subbook_count, eb_error_string(EB_SUCCESS)));
  eb_unlock(&appendix->lock);

  return EB_SUCCESS;

  /*
   * An error occurs...
   */
  failed:
  *subbook_count = 0;
  LOG(("out: eb_appendix_subbook_list() = %s", eb_error_string(error_code)));
  eb_unlock(&appendix->lock);
  return error_code;
}


/*
 * Get the subbook-code of the current subbook in `appendix'.
 */
EB_Error_Code
eb_appendix_subbook(EB_Appendix *appendix, EB_Subbook_Code *subbook_code)
{
  EB_Error_Code error_code;

  eb_lock(&appendix->lock);
  LOG(("in: eb_appendix_subbook(appendix=%d)", (int)appendix->code));

  /*
   * Check for the current status.
   */
  if (appendix->subbook_current == NULL) {
    error_code = EB_ERR_NO_CUR_APPSUB;
    goto failed;
  }

  /*
   * Copy the current subbook code to `subbook_code'.
   */
  *subbook_code = appendix->subbook_current->code;

  LOG(("out: eb_appendix_subbook(subbook=%d) = %s", (int)*subbook_code,
    eb_error_string(EB_SUCCESS)));
  eb_unlock(&appendix->lock);

  return EB_SUCCESS;

  /*
   * An error occurs...
   */
  failed:
  *subbook_code = EB_SUBBOOK_INVALID;
  LOG(("out: eb_appendix_subbook() = %s", eb_error_string(error_code)));
  eb_unlock(&appendix->lock);
  return error_code;
}


/*
 * Get the directory name of the current subbook in `appendix'.
 */
EB_Error_Code
eb_appendix_subbook_directory(EB_Appendix *appendix, char *directory)
{
  EB_Error_Code error_code;

  eb_lock(&appendix->lock);
  LOG(("in: eb_appendix_subbook_directory(appendix=%d)",
    (int)appendix->code));

  /*
   * Check for the current status.
   */
  if (appendix->subbook_current == NULL) {
    error_code = EB_ERR_NO_CUR_APPSUB;
    goto failed;
  }

  /*
   * Copy the directory name to `directory'.
   */
  strcpy(directory, appendix->subbook_current->directory_name);

  LOG(("out: eb_appendix_subbook_directory(directory=%s) = %s",
    directory, eb_error_string(EB_SUCCESS)));
  eb_unlock(&appendix->lock);

  return EB_SUCCESS;

  /*
   * An error occurs...
   */
  failed:
  *directory = '\0';
  LOG(("out: eb_appendix_subbook_directory() = %s",
    eb_error_string(error_code)));
  eb_unlock(&appendix->lock);
  return error_code;
}


/*
 * Get the directory name of the subbook `subbook_code' in `appendix'.
 */
EB_Error_Code
eb_appendix_subbook_directory2(EB_Appendix *appendix,
  EB_Subbook_Code subbook_code, char *directory)
{
  EB_Error_Code error_code;

  eb_lock(&appendix->lock);
  LOG(("in: eb_appendix_subbook_directory2(appendix=%d, subbook=%d)",
    (int)appendix->code, (int)subbook_code));

  /*
   * Check for the current status.
   */
  if (appendix->path == NULL) {
    error_code = EB_ERR_UNBOUND_APP;
    goto failed;
  }

  /*
   * Check for `subbook_code'.
   */
  if (subbook_code < 0 || appendix->subbook_count <= subbook_code) {
    error_code = EB_ERR_NO_SUCH_APPSUB;
    goto failed;
  }

  /*
   * Copy the directory name to `directory'.
   */
  strcpy(directory, (appendix->subbooks + subbook_code)->directory_name);

  LOG(("out: eb_appendix_subbook_directory2(directory=%s) = %s",
    directory, eb_error_string(EB_SUCCESS)));
  eb_unlock(&appendix->lock);

  return EB_SUCCESS;

  /*
   * An error occurs...
   */
  failed:
  *directory = '\0';
  LOG(("out: eb_appendix_subbook_directory2() = %s",
    eb_error_string(error_code)));
  eb_unlock(&appendix->lock);
  return error_code;
}


/*
 * Set the subbook `subbook_code' as the current subbook.
 */
EB_Error_Code
eb_set_appendix_subbook(EB_Appendix *appendix, EB_Subbook_Code subbook_code)
{
  EB_Error_Code error_code;

  eb_lock(&appendix->lock);
  LOG(("in: eb_set_appendix_subbook(appendix=%d, subbook=%d)",
    (int)appendix->code, (int)subbook_code));

  /*
   * Check for the current status.
   */
  if (appendix->path == NULL) {
    error_code = EB_ERR_UNBOUND_APP;
    goto failed;
  }

  /*
   * Check for `subbook_code'.
   */
  if (subbook_code < 0 || appendix->subbook_count <= subbook_code) {
    error_code = EB_ERR_NO_SUCH_APPSUB;
    goto failed;
  }

  /*
   * If the current subbook is `subbook_code', return immediately.
   * Otherwise close the current subbook and continue.
   */
  if (appendix->subbook_current != NULL) {
    if (appendix->subbook_current->code == subbook_code)
      goto succeeded;
    eb_unset_appendix_subbook(appendix);
  }

  /*
   * Disc type specific section.
   */
  if (appendix->disc_code == EB_DISC_EB)
    error_code = eb_set_appendix_subbook_eb(appendix, subbook_code);
  else
    error_code = eb_set_appendix_subbook_epwing(appendix, subbook_code);

  if (error_code != EB_SUCCESS)
    goto failed;

  /*
   * Load the subbook.
   */
  error_code = eb_load_appendix_subbook(appendix);
  if (error_code != EB_SUCCESS)
    goto failed;

  succeeded:
  LOG(("out: eb_set_appendix_subbook() = %s", eb_error_string(EB_SUCCESS)));
  eb_unlock(&appendix->lock);

  return EB_SUCCESS;

  /*
   * An error occurs...
   */
  failed:
  if (appendix->subbook_current != NULL)
    zio_close(&appendix->subbook_current->zio);
  appendix->subbook_current = NULL;
  LOG(("out: eb_set_appendix_subbook() = %s", eb_error_string(error_code)));
  eb_unlock(&appendix->lock);
  return error_code;
}


/*
 * EB* specific section of eb_set_appendix_subbook().
 */
static EB_Error_Code
eb_set_appendix_subbook_eb(EB_Appendix *appendix, EB_Subbook_Code subbook_code)
{
  EB_Error_Code error_code;
  EB_Appendix_Subbook *subbook;
  char appendix_path_name[EB_MAX_PATH_LENGTH + 1];
  Zio_Code zio_code;

  LOG(("in: eb_set_appendix_subbook_eb(appendix=%d, subbook=%d)",
    (int)appendix->code, (int)subbook_code));

  /*
   * Set the current subbook.
   */
  appendix->subbook_current = appendix->subbooks + subbook_code;
  subbook = appendix->subbook_current;

  /*
   * Open an appendix file.
   */
  if (eb_find_file_name2(appendix->path, subbook->directory_name,
    EB_FILE_NAME_APPENDIX, subbook->file_name) != EB_SUCCESS) {
    error_code = EB_ERR_FAIL_OPEN_APP;
    goto failed;
  }

  eb_compose_path_name2(appendix->path, subbook->directory_name,
    subbook->file_name, appendix_path_name);
  eb_path_name_zio_code(appendix_path_name, ZIO_PLAIN, &zio_code);

  if (zio_open(&subbook->zio, appendix_path_name, zio_code) < 0) {
    error_code = EB_ERR_FAIL_OPEN_APP;
    goto failed;
  }

  LOG(("out: eb_set_appendix_subbook_eb() = %s",
    eb_error_string(EB_SUCCESS)));
  return EB_SUCCESS;

  /*
   * An error occurs...
   */
  failed:
  LOG(("out: eb_set_appendix_subbook_eb() = %s",
    eb_error_string(error_code)));
  return error_code;
}


/*
 * EPWING specific section of eb_set_appendix_subbook().
 */
static EB_Error_Code
eb_set_appendix_subbook_epwing(EB_Appendix *appendix,
  EB_Subbook_Code subbook_code)
{
  EB_Error_Code error_code;
  EB_Appendix_Subbook *subbook;
  char appendix_path_name[EB_MAX_PATH_LENGTH + 1];
  Zio_Code zio_code;

  LOG(("in: eb_set_appendix_subbook_epwing(appendix=%d, subbook=%d)",
    (int)appendix->code, (int)subbook_code));

  /*
   * Set the current subbook.
   */
  appendix->subbook_current = appendix->subbooks + subbook_code;
  subbook = appendix->subbook_current;

  zio_initialize(&subbook->zio);

  /*
   * Adjust a directory name.
   */
  strcpy(subbook->data_directory_name, EB_DIRECTORY_NAME_DATA);
  eb_fix_directory_name2(appendix->path, subbook->directory_name,
    subbook->data_directory_name);

  /*
   * Open an appendix file.
   */
  if (eb_find_file_name3(appendix->path, subbook->directory_name,
    subbook->data_directory_name, EB_FILE_NAME_FUROKU, subbook->file_name)
    != EB_SUCCESS) {
    error_code = EB_ERR_FAIL_OPEN_APP;
    goto failed;
  }

  eb_compose_path_name3(appendix->path, subbook->directory_name,
    subbook->data_directory_name, subbook->file_name,
    appendix_path_name);
  eb_path_name_zio_code(appendix_path_name, ZIO_PLAIN, &zio_code);

  if (zio_open(&subbook->zio, appendix_path_name, zio_code) < 0) {
    subbook = NULL;
    error_code = EB_ERR_FAIL_OPEN_APP;
    goto failed;
  }

  LOG(("out: eb_set_appendix_subbook_epwing() = %s",
    eb_error_string(EB_SUCCESS)));
  return EB_SUCCESS;

  /*
   * An error occurs...
   */
  failed:
  LOG(("out: eb_set_appendix_subbook_epwing() = %s",
    eb_error_string(error_code)));
  return error_code;
}


/*
 * Unset the current subbook.
 */
void
eb_unset_appendix_subbook(EB_Appendix *appendix)
{
  eb_lock(&appendix->lock);
  LOG(("in: eb_unset_appendix_subbook(appendix=%d)", (int)appendix->code));

  /*
   * Close a file for the current subbook.
   */
  if (appendix->subbook_current != NULL) {
    zio_close(&appendix->subbook_current->zio);
    appendix->subbook_current = NULL;
  }

  LOG(("out: eb_unset_appendix_subbook()"));
  eb_unlock(&appendix->lock);
}


