/* Copyright (C) 2008 Atsushi Togo */
/* All rights reserved. */

/* This file is part of spglib. */

/* Redistribution and use in source and binary forms, with or without */
/* modification, are permitted provided that the following conditions */
/* are met: */

/* * Redistributions of source code must retain the above copyright */
/*   notice, this list of conditions and the following disclaimer. */

/* * 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. */

/* * Neither the name of the spglib 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 COPYRIGHT HOLDERS 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 */
/* COPYRIGHT HOLDER 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 <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stddef.h>
#include "arithmetic.h"
#include "cell.h"
#include "debug.h"
#include "delaunay.h"
#include "determination.h"
#include "kgrid.h"
#include "kpoint.h"
#include "mathfunc.h"
#include "niggli.h"
#include "pointgroup.h"
#include "spglib.h"
#include "primitive.h"
#include "refinement.h"
#include "spacegroup.h"
#include "spg_database.h"
#include "spin.h"
#include "symmetry.h"
#include "version.h"

/*-------*/
/* error */
/*-------*/
static SpglibError spglib_error_code = SPGLIB_SUCCESS;

typedef struct {
  SpglibError error;
  char *message;
} SpglibErrorMessage;

static SpglibErrorMessage spglib_error_message[] = {
  {SPGLIB_SUCCESS, "no error"},
  {SPGERR_SPACEGROUP_SEARCH_FAILED, "spacegroup search failed"},
  {SPGERR_CELL_STANDARDIZATION_FAILED, "cell standardization failed"},
  {SPGERR_SYMMETRY_OPERATION_SEARCH_FAILED, "symmetry operation search failed"},
  {SPGERR_ATOMS_TOO_CLOSE, "too close distance between atoms"},
  {SPGERR_POINTGROUP_NOT_FOUND, "pointgroup not found"},
  {SPGERR_NIGGLI_FAILED, "Niggli reduction failed"},
  {SPGERR_DELAUNAY_FAILED, "Delaunay reduction failed"},
  {SPGERR_ARRAY_SIZE_SHORTAGE, "array size shortage"},
  {SPGERR_NONE, ""},
};

/*---------*/
/* general */
/*---------*/
static SpglibDataset * get_dataset(SPGCONST double lattice[3][3],
                                   SPGCONST double position[][3],
                                   const int types[],
                                   const int num_atom,
                                   const int hall_number,
                                   const double symprec,
                                   const double angle_tolerance);
static SpglibDataset * init_dataset(void);
static int set_dataset(SpglibDataset * dataset,
                       const Cell * cell,
                       const Primitive * primitive,
                       SPGCONST Spacegroup * spacegroup,
                       ExactStructure *exstr);
static int get_symmetry_from_dataset(int rotation[][3][3],
                                     double translation[][3],
                                     const int max_size,
                                     SPGCONST double lattice[3][3],
                                     SPGCONST double position[][3],
                                     const int types[],
                                     const int num_atom,
                                     const double symprec,
                                     const double angle_tolerance);
static int get_symmetry_with_site_tensors(int rotation[][3][3],
                                          double translation[][3],
                                          int equivalent_atoms[],
                                          double primitive_lattice[3][3],
                                          int *spin_flips,
                                          const int run_symmetry_search,
                                          const int num_operations,
                                          SPGCONST double lattice[3][3],
                                          SPGCONST double position[][3],
                                          const int types[],
                                          const double *tensors,
                                          const int tensor_rank,
                                          const int num_atom,
                                          const int is_magnetic,
                                          const double symprec,
                                          const double angle_tolerance);
static int get_multiplicity(SPGCONST double lattice[3][3],
                            SPGCONST double position[][3],
                            const int types[],
                            const int num_atom,
                            const double symprec,
                            const double angle_tolerance);
static int standardize_primitive(double lattice[3][3],
                                 double position[][3],
                                 int types[],
                                 const int num_atom,
                                 const double symprec,
                                 const double angle_tolerance);
static int standardize_cell(double lattice[3][3],
                            double position[][3],
                            int types[],
                            const int num_atom,
                            const int num_array_size,
                            const double symprec,
                            const double angle_tolerance);
static int get_standardized_cell(double lattice[3][3],
                                 double position[][3],
                                 int types[],
                                 const int num_atom,
                                 const int num_array_size,
                                 const int to_primitive,
                                 const double symprec,
                                 const double angle_tolerance);
static Centering get_centering(int hall_number);
static void set_cell(double lattice[3][3],
                     double position[][3],
                     int types[],
                     Cell * cell);
static int get_international(char symbol[11],
                             SPGCONST double lattice[3][3],
                             SPGCONST double position[][3],
                             const int types[],
                             const int num_atom,
                             const double symprec,
                             const double angle_tolerance);
static int get_schoenflies(char symbol[7],
                           SPGCONST double lattice[3][3],
                           SPGCONST double position[][3],
                           const int types[], const int num_atom,
                           const double symprec,
                           const double angle_tolerance);

/*---------*/
/* kpoints */
/*---------*/
static int get_ir_reciprocal_mesh(int grid_address[][3],
                                  int ir_mapping_table[],
                                  const int mesh[3],
                                  const int is_shift[3],
                                  const int is_time_reversal,
                                  SPGCONST double lattice[3][3],
                                  SPGCONST double position[][3],
                                  const int types[],
                                  const int num_atom,
                                  const double symprec,
                                  const double angle_tolerance);
static size_t get_dense_ir_reciprocal_mesh(int grid_address[][3],
                                           size_t ir_mapping_table[],
                                           const int mesh[3],
                                           const int is_shift[3],
                                           const int is_time_reversal,
                                           SPGCONST double lattice[3][3],
                                           SPGCONST double position[][3],
                                           const int types[],
                                           const size_t num_atom,
                                           const double symprec,
                                           const double angle_tolerance);

static int get_stabilized_reciprocal_mesh(int grid_address[][3],
                                          int ir_mapping_table[],
                                          const int mesh[3],
                                          const int is_shift[3],
                                          const int is_time_reversal,
                                          const int num_rot,
                                          SPGCONST int rotations[][3][3],
                                          const size_t num_q,
                                          SPGCONST double qpoints[][3]);
static size_t get_dense_stabilized_reciprocal_mesh(int grid_address[][3],
                                                   size_t ir_mapping_table[],
                                                   const int mesh[3],
                                                   const int is_shift[3],
                                                   const int is_time_reversal,
                                                   const size_t num_rot,
                                                   SPGCONST int rotations[][3][3],
                                                   const size_t num_q,
                                                   SPGCONST double qpoints[][3]);

/*========*/
/* global */
/*========*/

/*-----------------------------------------*/
/* Version: spglib-[major].[minor].[micro] */
/*-----------------------------------------*/
int spg_get_major_version(void)
{
  spglib_error_code = SPGLIB_SUCCESS;
  return SPGLIB_MAJOR_VERSION;
}

int spg_get_minor_version(void)
{
  spglib_error_code = SPGLIB_SUCCESS;
  return SPGLIB_MINOR_VERSION;
}

int spg_get_micro_version(void)
{
  spglib_error_code = SPGLIB_SUCCESS;
  return SPGLIB_MICRO_VERSION;
}

/*-------*/
/* error */
/*-------*/
SpglibError spg_get_error_code(void)
{
  return spglib_error_code;
}

char * spg_get_error_message(SpglibError error)
{
  int i;

  for (i = 0; i < 100; i++) {
    if (SPGERR_NONE == spglib_error_message[i].error) {
      break;
    }

    if (error == spglib_error_message[i].error) {
      return spglib_error_message[i].message;
    }
  }

  return NULL;
}

/*---------*/
/* general */
/*---------*/
/* Return NULL if failed */
SpglibDataset * spg_get_dataset(SPGCONST double lattice[3][3],
                                SPGCONST double position[][3],
                                const int types[],
                                const int num_atom,
                                const double symprec)
{
  return get_dataset(lattice,
                     position,
                     types,
                     num_atom,
                     0,
                     symprec,
                     -1.0);
}

/* Return NULL if failed */
SpglibDataset * spgat_get_dataset(SPGCONST double lattice[3][3],
                                  SPGCONST double position[][3],
                                  const int types[],
                                  const int num_atom,
                                  const double symprec,
                                  const double angle_tolerance)
{
  return get_dataset(lattice,
                     position,
                     types,
                     num_atom,
                     0,
                     symprec,
                     angle_tolerance);
}

/* Return NULL if failed */
SpglibDataset * spg_get_dataset_with_hall_number(SPGCONST double lattice[3][3],
                                                 SPGCONST double position[][3],
                                                 const int types[],
                                                 const int num_atom,
                                                 const int hall_number,
                                                 const double symprec)
{
  return get_dataset(lattice,
                     position,
                     types,
                     num_atom,
                     hall_number,
                     symprec,
                     -1.0);
}

/* Return NULL if failed */
SpglibDataset *
spgat_get_dataset_with_hall_number(SPGCONST double lattice[3][3],
                                   SPGCONST double position[][3],
                                   const int types[],
                                   const int num_atom,
                                   const int hall_number,
                                   const double symprec,
                                   const double angle_tolerance)
{
  return get_dataset(lattice,
                     position,
                     types,
                     num_atom,
                     hall_number,
                     symprec,
                     angle_tolerance);
}

void spg_free_dataset(SpglibDataset *dataset)
{
  if (dataset->n_operations > 0) {
    free(dataset->rotations);
    dataset->rotations = NULL;
    free(dataset->translations);
    dataset->translations = NULL;
    dataset->n_operations = 0;
  }

  if (dataset->n_atoms > 0) {
    free(dataset->wyckoffs);
    dataset->wyckoffs = NULL;
    free(dataset->equivalent_atoms);
    dataset->equivalent_atoms = NULL;
    free(dataset->crystallographic_orbits);
    dataset->crystallographic_orbits = NULL;
    free(dataset->site_symmetry_symbols);
    dataset->site_symmetry_symbols = NULL;
    free(dataset->mapping_to_primitive);
    dataset->mapping_to_primitive = NULL;
    dataset->n_atoms = 0;
  }

  if (dataset->n_std_atoms > 0) {
    free(dataset->std_positions);
    dataset->std_positions = NULL;
    free(dataset->std_types);
    dataset->std_types = NULL;
    free(dataset->std_mapping_to_primitive);
    dataset->std_mapping_to_primitive = NULL;
    dataset->n_std_atoms = 0;
  }

  dataset->spacegroup_number = 0;
  dataset->hall_number = 0;
  strcpy(dataset->international_symbol, "");
  strcpy(dataset->hall_symbol, "");
  strcpy(dataset->choice, "");

  free(dataset);
}

/* Return 0 if failed */
int spg_get_symmetry(int rotation[][3][3],
                     double translation[][3],
                     const int max_size,
                     SPGCONST double lattice[3][3],
                     SPGCONST double position[][3],
                     const int types[],
                     const int num_atom,
                     const double symprec)
{
  return get_symmetry_from_dataset(rotation,
                                   translation,
                                   max_size,
                                   lattice,
                                   position,
                                   types,
                                   num_atom,
                                   symprec,
                                   -1.0);
}

/* Return 0 if failed */
int spgat_get_symmetry(int rotation[][3][3],
                       double translation[][3],
                       const int max_size,
                       SPGCONST double lattice[3][3],
                       SPGCONST double position[][3],
                       const int types[],
                       const int num_atom,
                       const double symprec,
                       const double angle_tolerance)
{
  return get_symmetry_from_dataset(rotation,
                                   translation,
                                   max_size,
                                   lattice,
                                   position,
                                   types,
                                   num_atom,
                                   symprec,
                                   angle_tolerance);
}

/* Return 0 if failed */
int spg_get_symmetry_with_collinear_spin(int rotation[][3][3],
                                         double translation[][3],
                                         int equivalent_atoms[],
                                         const int max_size,
                                         SPGCONST double lattice[3][3],
                                         SPGCONST double position[][3],
                                         const int types[],
                                         const double spins[],
                                         const int num_atom,
                                         const double symprec)
{
  int succeeded;
  double primitive_lattice[3][3];
  int *spin_flips;

  if ((spin_flips = (int*) malloc(sizeof(int) * max_size))
      == NULL) {
    warning_print("spglib: Memory could not be allocated.");
    return 0;
  }

  succeeded = get_symmetry_with_site_tensors(rotation,
                                             translation,
                                             equivalent_atoms,
                                             primitive_lattice,
                                             spin_flips,
                                             1,
                                             max_size,
                                             lattice,
                                             position,
                                             types,
                                             spins,
                                             0,
                                             num_atom,
                                             1,
                                             symprec,
                                             -1.0);
  free(spin_flips);
  spin_flips = NULL;

  return succeeded;
}

/* Return 0 if failed */
int spgat_get_symmetry_with_collinear_spin(int rotation[][3][3],
                                           double translation[][3],
                                           int equivalent_atoms[],
                                           const int max_size,
                                           SPGCONST double lattice[3][3],
                                           SPGCONST double position[][3],
                                           const int types[],
                                           const double spins[],
                                           const int num_atom,
                                           const double symprec,
                                           const double angle_tolerance)
{
  int succeeded;
  double primitive_lattice[3][3];
  int *spin_flips;

  if ((spin_flips = (int*) malloc(sizeof(int) * max_size))
      == NULL) {
    warning_print("spglib: Memory could not be allocated.");
    return 0;
  }

  succeeded = get_symmetry_with_site_tensors(rotation,
                                             translation,
                                             equivalent_atoms,
                                             primitive_lattice,
                                             spin_flips,
                                             1,
                                             max_size,
                                             lattice,
                                             position,
                                             types,
                                             spins,
                                             0,
                                             num_atom,
                                             1,
                                             symprec,
                                             angle_tolerance);

  free(spin_flips);
  spin_flips = NULL;

  return succeeded;
}

/* Return 0 if failed */
int spg_get_symmetry_with_site_tensors(int rotation[][3][3],
                                       double translation[][3],
                                       int equivalent_atoms[],
                                       double primitive_lattice[3][3],
                                       int *spin_flips,
                                       const int num_operations,
                                       SPGCONST double lattice[3][3],
                                       SPGCONST double position[][3],
                                       const int types[],
                                       const double *tensors,
                                       const int tensor_rank,
                                       const int num_atom,
                                       const int is_magnetic,
                                       const double symprec)
{
  return get_symmetry_with_site_tensors(rotation,
                                        translation,
                                        equivalent_atoms,
                                        primitive_lattice,
                                        spin_flips,
                                        0,
                                        num_operations,
                                        lattice,
                                        position,
                                        types,
                                        tensors,
                                        tensor_rank,
                                        num_atom,
                                        is_magnetic,
                                        symprec,
                                        -1.0);
}

int spgat_get_symmetry_with_site_tensors(int rotation[][3][3],
                                         double translation[][3],
                                         int equivalent_atoms[],
                                         double primitive_lattice[3][3],
                                         int *spin_flips,
                                         const int num_operations,
                                         SPGCONST double lattice[3][3],
                                         SPGCONST double position[][3],
                                         const int types[],
                                         const double *tensors,
                                         const int tensor_rank,
                                         const int num_atom,
                                         const int is_magnetic,
                                         const double symprec,
                                         const double angle_tolerance)
{
  return get_symmetry_with_site_tensors(rotation,
                                        translation,
                                        equivalent_atoms,
                                        primitive_lattice,
                                        spin_flips,
                                        0,
                                        num_operations,
                                        lattice,
                                        position,
                                        types,
                                        tensors,
                                        tensor_rank,
                                        num_atom,
                                        is_magnetic,
                                        symprec,
                                        angle_tolerance);
}

int spg_get_hall_number_from_symmetry(SPGCONST int rotation[][3][3],
                                      SPGCONST double translation[][3],
                                      const int num_operations,
                                      const double symprec)
{
  int i, hall_number;
  Symmetry *symmetry;
  Symmetry *prim_symmetry;

  symmetry = NULL;
  prim_symmetry = NULL;
  hall_number = 0;

  if ((symmetry = sym_alloc_symmetry(num_operations)) == NULL) {
    return 0;
  }

  for (i = 0; i < num_operations; i++) {
    mat_copy_matrix_i3(symmetry->rot[i], rotation[i]);
    mat_copy_vector_d3(symmetry->trans[i], translation[i]);
  }

  prim_symmetry = prm_get_primitive_symmetry(symmetry, symprec);
  sym_free_symmetry(symmetry);
  symmetry = NULL;

  if (prim_symmetry == NULL) {
    return 0;
  }

  hall_number = spa_search_spacegroup_with_symmetry(prim_symmetry, symprec);

  if (hall_number) {
    spglib_error_code = SPGLIB_SUCCESS;
  } else {
    spglib_error_code = SPGERR_SPACEGROUP_SEARCH_FAILED;
  }

  sym_free_symmetry(prim_symmetry);
  prim_symmetry = NULL;

  return hall_number;
}

/* Return 0 if failed */
int spg_get_multiplicity(SPGCONST double lattice[3][3],
                         SPGCONST double position[][3],
                         const int types[],
                         const int num_atom,
                         const double symprec)
{
  return get_multiplicity(lattice,
                          position,
                          types,
                          num_atom,
                          symprec,
                          -1.0);
}

/* Return 0 if failed */
int spgat_get_multiplicity(SPGCONST double lattice[3][3],
                           SPGCONST double position[][3],
                           const int types[],
                           const int num_atom,
                           const double symprec,
                           const double angle_tolerance)
{
  return get_multiplicity(lattice,
                          position,
                          types,
                          num_atom,
                          symprec,
                          angle_tolerance);
}

/* Return 0 if failed */
int spg_get_international(char symbol[11],
                          SPGCONST double lattice[3][3],
                          SPGCONST double position[][3],
                          const int types[],
                          const int num_atom,
                          const double symprec)
{
  return get_international(symbol,
                           lattice,
                           position,
                           types,
                           num_atom,
                           symprec,
                           -1.0);
}

/* Return 0 if failed */
int spgat_get_international(char symbol[11],
                            SPGCONST double lattice[3][3],
                            SPGCONST double position[][3],
                            const int types[],
                            const int num_atom,
                            const double symprec,
                            const double angle_tolerance)
{
  return get_international(symbol,
                           lattice,
                           position,
                           types,
                           num_atom,
                           symprec,
                           angle_tolerance);
}

/* Return 0 if failed */
int spg_get_schoenflies(char symbol[7],
                        SPGCONST double lattice[3][3],
                        SPGCONST double position[][3],
                        const int types[],
                        const int num_atom,
                        const double symprec)
{
  return get_schoenflies(symbol,
                         lattice,
                         position,
                         types,
                         num_atom,
                         symprec,
                         -1.0);
}

/* Return 0 if failed */
int spgat_get_schoenflies(char symbol[7],
                          SPGCONST double lattice[3][3],
                          SPGCONST double position[][3],
                          const int types[],
                          const int num_atom,
                          const double symprec,
                          const double angle_tolerance)
{
  return get_schoenflies(symbol,
                         lattice,
                         position,
                         types,
                         num_atom,
                         symprec,
                         angle_tolerance);
}

/* Return 0 if failed */
int spg_get_pointgroup(char symbol[6],
                       int transform_mat[3][3],
                       SPGCONST int rotations[][3][3],
                       const int num_rotations)
{
  Pointgroup pointgroup;

  pointgroup = ptg_get_transformation_matrix(transform_mat,
                                             rotations,
                                             num_rotations);

  if (pointgroup.number == 0) {
    spglib_error_code = SPGERR_POINTGROUP_NOT_FOUND;
    return 0;
  }

  memcpy(symbol, pointgroup.symbol, 6);

  spglib_error_code = SPGLIB_SUCCESS;
  return pointgroup.number;
}

/* Return 0 if failed */
int spg_get_symmetry_from_database(int rotations[192][3][3],
                                   double translations[192][3],
                                   const int hall_number)
{
  int i, size;
  Symmetry *symmetry;

  symmetry = NULL;

  if ((symmetry = spgdb_get_spacegroup_operations(hall_number)) == NULL) {
    goto err;
  }

  for (i = 0; i < symmetry->size; i++) {
    mat_copy_matrix_i3(rotations[i], symmetry->rot[i]);
    mat_copy_vector_d3(translations[i], symmetry->trans[i]);
  }
  size = symmetry->size;

  sym_free_symmetry(symmetry);
  symmetry = NULL;

  spglib_error_code = SPGLIB_SUCCESS;
  return size;

 err:
  spglib_error_code = SPGERR_SPACEGROUP_SEARCH_FAILED;
  return 0;
}

/* Return spglibtype.number = 0 if failed */
SpglibSpacegroupType spg_get_spacegroup_type(const int hall_number)
{
  SpglibSpacegroupType spglibtype;
  SpacegroupType spgtype;
  Pointgroup pointgroup;
  int arth_number;
  char arth_symbol[7];

  spglibtype.number = 0;
  strcpy(spglibtype.schoenflies, "");
  strcpy(spglibtype.hall_symbol, "");
  strcpy(spglibtype.choice, "");
  strcpy(spglibtype.international, "");
  strcpy(spglibtype.international_full, "");
  strcpy(spglibtype.international_short, "");
  strcpy(spglibtype.pointgroup_international, "");
  strcpy(spglibtype.pointgroup_schoenflies, "");
  spglibtype.arithmetic_crystal_class_number = 0;
  strcpy(spglibtype.arithmetic_crystal_class_symbol, "");

  if (0 < hall_number && hall_number < 531) {
    spgtype = spgdb_get_spacegroup_type(hall_number);
    spglibtype.number = spgtype.number;
    memcpy(spglibtype.schoenflies, spgtype.schoenflies, 7);
    memcpy(spglibtype.hall_symbol, spgtype.hall_symbol, 17);
    memcpy(spglibtype.choice, spgtype.choice, 6);
    memcpy(spglibtype.international, spgtype.international, 32);
    memcpy(spglibtype.international_full, spgtype.international_full, 20);
    memcpy(spglibtype.international_short, spgtype.international_short, 11);
    pointgroup = ptg_get_pointgroup(spgtype.pointgroup_number);
    memcpy(spglibtype.pointgroup_international, pointgroup.symbol, 6);
    memcpy(spglibtype.pointgroup_schoenflies, pointgroup.schoenflies, 4);
    arth_number = arth_get_symbol(arth_symbol, spgtype.number);
    spglibtype.arithmetic_crystal_class_number = arth_number;
    memcpy(spglibtype.arithmetic_crystal_class_symbol, arth_symbol, 7);
    spglib_error_code = SPGLIB_SUCCESS;
  } else {
    spglib_error_code = SPGERR_SPACEGROUP_SEARCH_FAILED;
  }

  return spglibtype;
}

/* Return 0 if failed */
int spg_standardize_cell(double lattice[3][3],
                         double position[][3],
                         int types[],
                         const int num_atom,
                         const int to_primitive,
                         const int no_idealize,
                         const double symprec)
{
  return spgat_standardize_cell(lattice,
                                position,
                                types,
                                num_atom,
                                to_primitive,
                                no_idealize,
                                symprec,
                                -1.0);
}

/* Return 0 if failed */
int spgat_standardize_cell(double lattice[3][3],
                           double position[][3],
                           int types[],
                           const int num_atom,
                           const int to_primitive,
                           const int no_idealize,
                           const double symprec,
                           const double angle_tolerance)
{
  if (to_primitive) {
    if (no_idealize) {
      return get_standardized_cell(lattice,
                                   position,
                                   types,
                                   num_atom,
                                   0,
                                   1,
                                   symprec,
                                   angle_tolerance);
    } else {
      return standardize_primitive(lattice,
                                   position,
                                   types,
                                   num_atom,
                                   symprec,
                                   angle_tolerance);
    }
  } else {
    if (no_idealize) {
      return get_standardized_cell(lattice,
                                   position,
                                   types,
                                   num_atom,
                                   0,
                                   0,
                                   symprec,
                                   angle_tolerance);
    } else {
      return standardize_cell(lattice,
                              position,
                              types,
                              num_atom,
                              0,
                              symprec,
                              angle_tolerance);
    }
  }
}

/* Return 0 if failed */
int spg_find_primitive(double lattice[3][3],
                       double position[][3],
                       int types[],
                       const int num_atom,
                       const double symprec)
{
  return standardize_primitive(lattice,
                               position,
                               types,
                               num_atom,
                               symprec,
                               -1.0);
}

/* Return 0 if failed */
int spgat_find_primitive(double lattice[3][3],
                         double position[][3],
                         int types[],
                         const int num_atom,
                         const double symprec,
                         const double angle_tolerance)
{
  return standardize_primitive(lattice,
                               position,
                               types,
                               num_atom,
                               symprec,
                               angle_tolerance);
}

/* Return 0 if failed */
int spg_refine_cell(double lattice[3][3],
                    double position[][3],
                    int types[],
                    const int num_atom,
                    const double symprec)
{
  return standardize_cell(lattice,
                          position,
                          types,
                          num_atom,
                          0,
                          symprec,
                          -1.0);
}

/* Return 0 if failed */
int spgat_refine_cell(double lattice[3][3],
                      double position[][3],
                      int types[],
                      const int num_atom,
                      const double symprec,
                      const double angle_tolerance)
{
  return standardize_cell(lattice,
                          position,
                          types,
                          num_atom,
                          0,
                          symprec,
                          angle_tolerance);
}

int spg_delaunay_reduce(double lattice[3][3], const double symprec)
{
  int i, j, succeeded;
  double red_lattice[3][3];

  succeeded = del_delaunay_reduce(red_lattice, lattice, symprec);

  if (succeeded) {
    for (i = 0; i < 3; i++) {
      for (j = 0; j < 3; j++) {
        lattice[i][j] = red_lattice[i][j];
      }
    }
    spglib_error_code = SPGLIB_SUCCESS;
  } else {
    spglib_error_code = SPGERR_DELAUNAY_FAILED;
  }

  return succeeded;
}

/*---------*/
/* kpoints */
/*---------*/
int spg_get_grid_point_from_address(const int grid_address[3],
                                    const int mesh[3])
{
  int address_double[3];
  int is_shift[3];

  is_shift[0] = 0;
  is_shift[1] = 0;
  is_shift[2] = 0;
  kgd_get_grid_address_double_mesh(address_double,
                                   grid_address,
                                   mesh,
                                   is_shift);
  return kgd_get_grid_point_double_mesh(address_double, mesh);
}

size_t spg_get_dense_grid_point_from_address(const int grid_address[3],
                                             const int mesh[3])
{
  int address_double[3];
  int is_shift[3];

  is_shift[0] = 0;
  is_shift[1] = 0;
  is_shift[2] = 0;
  kgd_get_grid_address_double_mesh(address_double,
                                   grid_address,
                                   mesh,
                                   is_shift);
  return kgd_get_dense_grid_point_double_mesh(address_double, mesh);
}

int spg_get_ir_reciprocal_mesh(int grid_address[][3],
                               int ir_mapping_table[],
                               const int mesh[3],
                               const int is_shift[3],
                               const int is_time_reversal,
                               SPGCONST double lattice[3][3],
                               SPGCONST double position[][3],
                               const int types[],
                               const int num_atom,
                               const double symprec)
{
  return get_ir_reciprocal_mesh(grid_address,
                                ir_mapping_table,
                                mesh,
                                is_shift,
                                is_time_reversal,
                                lattice,
                                position,
                                types,
                                num_atom,
                                symprec,
                                -1.0);
}

size_t spg_get_dense_ir_reciprocal_mesh(int grid_address[][3],
                                        size_t ir_mapping_table[],
                                        const int mesh[3],
                                        const int is_shift[3],
                                        const int is_time_reversal,
                                        SPGCONST double lattice[3][3],
                                        SPGCONST double position[][3],
                                        const int types[],
                                        const int num_atom,
                                        const double symprec)
{
  return get_dense_ir_reciprocal_mesh(grid_address,
                                     ir_mapping_table,
                                     mesh,
                                     is_shift,
                                     is_time_reversal,
                                     lattice,
                                     position,
                                     types,
                                     num_atom,
                                     symprec,
                                     -1.0);
}

int spg_get_stabilized_reciprocal_mesh(int grid_address[][3],
                                       int ir_mapping_table[],
                                       const int mesh[3],
                                       const int is_shift[3],
                                       const int is_time_reversal,
                                       const int num_rot,
                                       SPGCONST int rotations[][3][3],
                                       const int num_q,
                                       SPGCONST double qpoints[][3])
{
  return get_stabilized_reciprocal_mesh(grid_address,
                                        ir_mapping_table,
                                        mesh,
                                        is_shift,
                                        is_time_reversal,
                                        num_rot,
                                        rotations,
                                        num_q,
                                        qpoints);
}

size_t spg_get_dense_stabilized_reciprocal_mesh(int grid_address[][3],
                                                size_t ir_mapping_table[],
                                                const int mesh[3],
                                                const int is_shift[3],
                                                const int is_time_reversal,
                                                const int num_rot,
                                                SPGCONST int rotations[][3][3],
                                                const int num_q,
                                                SPGCONST double qpoints[][3])
{
  return get_dense_stabilized_reciprocal_mesh(grid_address,
                                              ir_mapping_table,
                                              mesh,
                                              is_shift,
                                              is_time_reversal,
                                              num_rot,
                                              rotations,
                                              num_q,
                                              qpoints);
}

void spg_get_dense_grid_points_by_rotations(size_t rot_grid_points[],
                                            const int address_orig[3],
                                            const int num_rot,
                                            SPGCONST int rot_reciprocal[][3][3],
                                            const int mesh[3],
                                            const int is_shift[3])
{
  kpt_get_dense_grid_points_by_rotations(rot_grid_points,
                                         address_orig,
                                         rot_reciprocal,
                                         num_rot,
                                         mesh,
                                         is_shift);
}

void spg_get_dense_BZ_grid_points_by_rotations(size_t rot_grid_points[],
                                               const int address_orig[3],
                                               const int num_rot,
                                               SPGCONST int rot_reciprocal[][3][3],
                                               const int mesh[3],
                                               const int is_shift[3],
                                               const size_t bz_map[])
{
  kpt_get_dense_BZ_grid_points_by_rotations(rot_grid_points,
                                            address_orig,
                                            rot_reciprocal,
                                            num_rot,
                                            mesh,
                                            is_shift,
                                            bz_map);
}

int spg_relocate_BZ_grid_address(int bz_grid_address[][3],
                                 int bz_map[],
                                 SPGCONST int grid_address[][3],
                                 const int mesh[3],
                                 SPGCONST double rec_lattice[3][3],
                                 const int is_shift[3])
{
  return kpt_relocate_BZ_grid_address(bz_grid_address,
                                      bz_map,
                                      grid_address,
                                      mesh,
                                      rec_lattice,
                                      is_shift);
}

size_t spg_relocate_dense_BZ_grid_address(int bz_grid_address[][3],
                                          size_t bz_map[],
                                          SPGCONST int grid_address[][3],
                                          const int mesh[3],
                                          SPGCONST double rec_lattice[3][3],
                                          const int is_shift[3])
{
  return kpt_relocate_dense_BZ_grid_address(bz_grid_address,
                                            bz_map,
                                            grid_address,
                                            mesh,
                                            rec_lattice,
                                            is_shift);
}

/*--------*/
/* Niggli */
/*--------*/
/* Return 0 if failed */
int spg_niggli_reduce(double lattice[3][3], const double symprec)
{
  int i, j, succeeded;
  double vals[9];

  for (i = 0; i < 3; i++) {
    for (j = 0; j < 3; j++) {
      vals[i * 3 + j] = lattice[i][j];
    }
  }

  succeeded = niggli_reduce(vals, symprec);

  if (succeeded) {
    for (i = 0; i < 3; i++) {
      for (j = 0; j < 3; j++) {
        lattice[i][j] = vals[i * 3 + j];
      }
    }
    spglib_error_code = SPGLIB_SUCCESS;
  } else {
    spglib_error_code = SPGERR_NIGGLI_FAILED;
  }

  return succeeded;
}




/*=======*/
/* local */
/*=======*/

/*---------*/
/* general */
/*---------*/
/* Return NULL if failed */
static SpglibDataset * get_dataset(SPGCONST double lattice[3][3],
                                   SPGCONST double position[][3],
                                   const int types[],
                                   const int num_atom,
                                   const int hall_number,
                                   const double symprec,
                                   const double angle_tolerance)
{
  SpglibDataset *dataset;
  Cell *cell;
  DataContainer *container;

  dataset = NULL;
  cell = NULL;
  container = NULL;

  if ((dataset = init_dataset()) == NULL) {
    goto not_found;
  }

  if ((cell = cel_alloc_cell(num_atom)) == NULL) {
    free(dataset);
    dataset = NULL;
    goto not_found;
  }

  cel_set_cell(cell, lattice, position, types);
  if (cel_any_overlap_with_same_type(cell, symprec)) {
    cel_free_cell(cell);
    cell = NULL;
    free(dataset);
    dataset = NULL;
    goto atoms_too_close;
  }

  if ((container = det_determine_all(cell,
                                     hall_number,
                                     symprec,
                                     angle_tolerance))
      != NULL) {
    if (set_dataset(dataset,
                    cell,
                    container->primitive,
                    container->spacegroup,
                    container->exact_structure)) {
      det_free_container(container);
      container = NULL;
      cel_free_cell(cell);
      cell = NULL;
      goto found;
    }
    det_free_container(container);
    container = NULL;
  }

  cel_free_cell(cell);
  cell = NULL;
  free(dataset);
  dataset = NULL;

 not_found:
  spglib_error_code = SPGERR_SPACEGROUP_SEARCH_FAILED;
  return NULL;

 atoms_too_close:
  spglib_error_code = SPGERR_ATOMS_TOO_CLOSE;
  return NULL;

 found:

  spglib_error_code = SPGLIB_SUCCESS;
  return dataset;
}

static SpglibDataset * init_dataset(void)
{
  int i, j;
  SpglibDataset *dataset;

  dataset = NULL;

  if ((dataset = (SpglibDataset*) malloc(sizeof(SpglibDataset))) == NULL) {
    warning_print("spglib: Memory could not be allocated.");
    return NULL;
  }

  dataset->spacegroup_number = 0;
  dataset->hall_number = 0;
  strcpy(dataset->international_symbol, "");
  strcpy(dataset->hall_symbol, "");
  strcpy(dataset->choice, "");
  dataset->origin_shift[0] = 0;
  dataset->origin_shift[1] = 0;
  dataset->origin_shift[2] = 0;
  dataset->n_atoms = 0;
  dataset->wyckoffs = NULL;
  dataset->equivalent_atoms = NULL;
  dataset->crystallographic_orbits = NULL;
  dataset->mapping_to_primitive = NULL;
  dataset->n_operations = 0;
  dataset->rotations = NULL;
  dataset->translations = NULL;
  dataset->n_std_atoms = 0;
  dataset->std_positions = NULL;
  dataset->std_types = NULL;
  for (i = 0; i < 3; i++) {
    for (j = 0; j < 3; j++) {
      dataset->std_rotation_matrix[i][j] = 0;
    }
  }
  dataset->std_mapping_to_primitive = NULL;
  /* dataset->pointgroup_number = 0; */
  strcpy(dataset->pointgroup_symbol, "");

  return dataset;
}

/* Return 0 if failed */
static int set_dataset(SpglibDataset * dataset,
                       const Cell * cell,
                       const Primitive * primitive,
                       SPGCONST Spacegroup * spacegroup,
                       ExactStructure *exstr)
{
  int i, j;
  double inv_lat[3][3];
  Pointgroup pointgroup;

  /* Spacegroup type, transformation matrix, origin shift */
  dataset->n_atoms = cell->size;
  dataset->spacegroup_number = spacegroup->number;
  dataset->hall_number = spacegroup->hall_number;
  memcpy(dataset->international_symbol, spacegroup->international_short, 11);
  memcpy(dataset->hall_symbol, spacegroup->hall_symbol, 17);
  memcpy(dataset->choice, spacegroup->choice, 6);
  mat_inverse_matrix_d3(inv_lat, spacegroup->bravais_lattice, 0);
  mat_multiply_matrix_d3(dataset->transformation_matrix,
                         inv_lat, cell->lattice);
  mat_copy_vector_d3(dataset->origin_shift, spacegroup->origin_shift);

  dataset->n_operations = exstr->symmetry->size;

  if ((dataset->rotations =
       (int (*)[3][3]) malloc(sizeof(int[3][3]) * dataset->n_operations))
      == NULL) {
    warning_print("spglib: Memory could not be allocated.");
    goto err;
  }

  if ((dataset->translations =
       (double (*)[3]) malloc(sizeof(double[3]) * dataset->n_operations))
      == NULL) {
    warning_print("spglib: Memory could not be allocated.");
    goto err;
  }

  for (i = 0; i < exstr->symmetry->size; i++) {
    mat_copy_matrix_i3(dataset->rotations[i], exstr->symmetry->rot[i]);
    mat_copy_vector_d3(dataset->translations[i], exstr->symmetry->trans[i]);
  }

  /* Wyckoff positions */
  if ((dataset->wyckoffs = (int*) malloc(sizeof(int) * dataset->n_atoms))
      == NULL) {
    warning_print("spglib: Memory could not be allocated.");
    goto err;
  }

  if ((dataset->site_symmetry_symbols =
       (char(*)[7]) malloc(sizeof(char[7]) * dataset->n_atoms)) == NULL) {
    warning_print("spglib: Memory could not be allocated.");
    goto err;
  }

  if ((dataset->equivalent_atoms =
       (int*) malloc(sizeof(int) * dataset->n_atoms)) == NULL) {
    warning_print("spglib: Memory could not be allocated.");
    goto err;
  }

  if ((dataset->crystallographic_orbits =
       (int*) malloc(sizeof(int) * dataset->n_atoms)) == NULL) {
    warning_print("spglib: Memory could not be allocated.");
    goto err;
  }

  for (i = 0; i < dataset->n_atoms; i++) {
    dataset->wyckoffs[i] = exstr->wyckoffs[i];
    for (j = 0; j < 7; j++) {
      dataset->site_symmetry_symbols[i][j] = exstr->site_symmetry_symbols[i][j];
    }
    dataset->equivalent_atoms[i] = exstr->equivalent_atoms[i];
    dataset->crystallographic_orbits[i] = exstr->crystallographic_orbits[i];
  }

  if ((dataset->mapping_to_primitive =
       (int*) malloc(sizeof(int) * dataset->n_atoms)) == NULL) {
    warning_print("spglib: Memory could not be allocated.");
    goto err;
  }

  debug_print("[line %d, %s]\n", __LINE__, __FILE__);
  debug_print("refined cell after ref_get_Wyckoff_positions\n");
  debug_print_matrix_d3(exstr->bravais->lattice);
#ifdef SPGDEBUG
  for (i = 0; i < exstr->bravais->size; i++) {
    printf("%d: %f %f %f\n",
           exstr->bravais->types[i],
           exstr->bravais->position[i][0],
           exstr->bravais->position[i][1],
           exstr->bravais->position[i][2]);
  }
#endif

  mat_copy_matrix_d3(dataset->primitive_lattice, primitive->cell->lattice);
  for (i = 0; i < dataset->n_atoms; i++) {
    dataset->mapping_to_primitive[i] = primitive->mapping_table[i];
  }

  dataset->n_std_atoms = exstr->bravais->size;
  mat_copy_matrix_d3(dataset->std_lattice, exstr->bravais->lattice);

  if ((dataset->std_positions =
       (double (*)[3]) malloc(sizeof(double[3]) * dataset->n_std_atoms))
      == NULL) {
    warning_print("spglib: Memory could not be allocated.");
    goto err;
  }

  if ((dataset->std_types = (int*) malloc(sizeof(int) * dataset->n_std_atoms))
      == NULL) {
    warning_print("spglib: Memory could not be allocated.");
    goto err;
  }

  if ((dataset->std_mapping_to_primitive =
       (int*) malloc(sizeof(int) * dataset->n_std_atoms)) == NULL) {
    warning_print("spglib: Memory could not be allocated.");
    goto err;
  }

  for (i = 0; i < dataset->n_std_atoms; i++) {
    mat_copy_vector_d3(dataset->std_positions[i], exstr->bravais->position[i]);
    dataset->std_types[i] = exstr->bravais->types[i];
    dataset->std_mapping_to_primitive[i] = exstr->std_mapping_to_primitive[i];
  }

  mat_copy_matrix_d3(dataset->std_rotation_matrix, exstr->rotation);

  /* dataset->pointgroup_number = spacegroup->pointgroup_number; */
  pointgroup = ptg_get_pointgroup(spacegroup->pointgroup_number);
  memcpy(dataset->pointgroup_symbol, pointgroup.symbol, 6);

  return 1;

 err:
  if (dataset->std_positions != NULL) {
    free(dataset->std_positions);
    dataset->std_positions = NULL;
  }
  if (dataset->std_mapping_to_primitive != NULL) {
    free(dataset->std_mapping_to_primitive);
    dataset->std_mapping_to_primitive = NULL;
  }
  if (dataset->equivalent_atoms != NULL) {
    free(dataset->equivalent_atoms);
    dataset->equivalent_atoms = NULL;
  }
  if (dataset->crystallographic_orbits != NULL) {
    free(dataset->crystallographic_orbits);
    dataset->crystallographic_orbits = NULL;
  }
  if (dataset->mapping_to_primitive != NULL) {
    free(dataset->mapping_to_primitive);
    dataset->mapping_to_primitive = NULL;
  }
  if (dataset->site_symmetry_symbols != NULL) {
    free(dataset->site_symmetry_symbols);
    dataset->site_symmetry_symbols = NULL;
  }
  if (dataset->wyckoffs != NULL) {
    free(dataset->wyckoffs);
    dataset->wyckoffs = NULL;
  }
  if (dataset->translations != NULL) {
    free(dataset->translations);
    dataset->translations = NULL;
  }
  if (dataset->rotations != NULL) {
    free(dataset->rotations);
    dataset->rotations = NULL;
  }

  return 0;
}

/* Return 0 if failed */
static int get_symmetry_from_dataset(int rotation[][3][3],
                                     double translation[][3],
                                     const int max_size,
                                     SPGCONST double lattice[3][3],
                                     SPGCONST double position[][3],
                                     const int types[],
                                     const int num_atom,
                                     const double symprec,
                                     const double angle_tolerance)
{
  int i, num_sym;
  SpglibDataset *dataset;

  num_sym = 0;
  dataset = NULL;

  if ((dataset = get_dataset(lattice,
                             position,
                             types,
                             num_atom,
                             0,
                             symprec,
                             angle_tolerance)) == NULL) {
    return 0;
  }

  if (dataset->n_operations > max_size) {
    fprintf(stderr,
            "spglib: Indicated max size(=%d) is less than number ", max_size);
    fprintf(stderr,
            "spglib: of symmetry operations(=%d).\n", dataset->n_operations);
    goto err;
  }

  num_sym = dataset->n_operations;
  for (i = 0; i < num_sym; i++) {
    mat_copy_matrix_i3(rotation[i], dataset->rotations[i]);
    mat_copy_vector_d3(translation[i], dataset->translations[i]);
  }

  spg_free_dataset(dataset);
  dataset = NULL;
  return num_sym;

 err:
  spg_free_dataset(dataset);
  dataset = NULL;
  spglib_error_code = SPGERR_ARRAY_SIZE_SHORTAGE;
  return 0;
}

/* Return 0 if failed */
static int get_symmetry_with_site_tensors(int rotation[][3][3],
                                          double translation[][3],
                                          int equivalent_atoms[],
                                          double primitive_lattice[3][3],
                                          int *spin_flips,
                                          const int run_symmetry_search,
                                          const int num_operations,
                                          SPGCONST double lattice[3][3],
                                          SPGCONST double position[][3],
                                          const int types[],
                                          const double *tensors,
                                          const int tensor_rank,
                                          const int num_atom,
                                          const int is_magnetic,
                                          const double symprec,
                                          const double angle_tolerance)
{
  int i, size;
  Symmetry *symmetry, *sym_nonspin;
  Cell *cell;
  SpglibDataset *dataset;

  size = 0;
  symmetry = NULL;
  sym_nonspin = NULL;
  cell = NULL;
  dataset = NULL;

  if (run_symmetry_search) {
    if ((dataset = get_dataset(lattice,
                               position,
                               types,
                               num_atom,
                               0,
                               symprec,
                               angle_tolerance)) == NULL) {
      goto err;
    }

    if ((sym_nonspin = sym_alloc_symmetry(dataset->n_operations)) == NULL) {
      spg_free_dataset(dataset);
      dataset = NULL;
      goto err;
    }

    for (i = 0; i < dataset->n_operations; i++) {
      mat_copy_matrix_i3(sym_nonspin->rot[i], dataset->rotations[i]);
      mat_copy_vector_d3(sym_nonspin->trans[i], dataset->translations[i]);
    }
    spg_free_dataset(dataset);
    dataset = NULL;
  } else {
    if ((sym_nonspin = sym_alloc_symmetry(num_operations)) == NULL) {
      goto err;
    }
    for (i = 0; i < num_operations; i++) {
      mat_copy_matrix_i3(sym_nonspin->rot[i], rotation[i]);
      mat_copy_vector_d3(sym_nonspin->trans[i], translation[i]);
    }
  }

  if ((cell = cel_alloc_cell(num_atom)) == NULL) {
    goto err;
  }
  cel_set_cell(cell, lattice, position, types);
  symmetry = spn_get_operations_with_site_tensors(equivalent_atoms,
                                                  primitive_lattice,
                                                  spin_flips,
                                                  sym_nonspin,
                                                  cell,
                                                  tensors,
                                                  tensor_rank,
                                                  is_magnetic,
                                                  symprec,
                                                  angle_tolerance);

  sym_free_symmetry(sym_nonspin);
  sym_nonspin = NULL;
  cel_free_cell(cell);
  cell = NULL;

  if (symmetry == NULL) {
    goto err;
  }

  if (symmetry->size > num_operations) {
    fprintf(stderr, "spglib: Indicated max size(=%d) is less than number ",
            num_operations);
    fprintf(stderr, "spglib: of symmetry operations(=%d).\n", symmetry->size);
    sym_free_symmetry(symmetry);
    symmetry = NULL;
    goto array_size_shortage_err;
  }

  for (i = 0; i < symmetry->size; i++) {
    mat_copy_matrix_i3(rotation[i], symmetry->rot[i]);
    mat_copy_vector_d3(translation[i], symmetry->trans[i]);
  }
  size = symmetry->size;

  sym_free_symmetry(symmetry);
  symmetry = NULL;
  spglib_error_code = SPGLIB_SUCCESS;

  return size;

 err:
  spglib_error_code = SPGERR_SYMMETRY_OPERATION_SEARCH_FAILED;
  return 0;

array_size_shortage_err:
  spglib_error_code = SPGERR_ARRAY_SIZE_SHORTAGE;
  return 0;

}

/* Return 0 if failed */
static int get_multiplicity(SPGCONST double lattice[3][3],
                            SPGCONST double position[][3],
                            const int types[],
                            const int num_atom,
                            const double symprec,
                            const double angle_tolerance)
{
  int size;
  SpglibDataset *dataset;

  size = 0;
  dataset = NULL;

  if ((dataset = get_dataset(lattice,
                             position,
                             types,
                             num_atom,
                             0,
                             symprec,
                             angle_tolerance)) == NULL) {
    return 0;
  }

  size = dataset->n_operations;
  spg_free_dataset(dataset);
  dataset = NULL;

  return size;
}

static int standardize_primitive(double lattice[3][3],
                                 double position[][3],
                                 int types[],
                                 const int num_atom,
                                 const double symprec,
                                 const double angle_tolerance)
{
  int i, num_prim_atom;
  int *mapping_table;
  Centering centering;
  SpglibDataset *dataset;
  Cell *primitive, *bravais;

  double identity[3][3] = {{ 1, 0, 0 },
                           { 0, 1, 0 },
                           { 0, 0, 1 }};

  num_prim_atom = 0;
  mapping_table = NULL;
  dataset = NULL;
  primitive = NULL;
  bravais = NULL;

  if ((dataset = get_dataset(lattice,
                             position,
                             types,
                             num_atom,
                             0,
                             symprec,
                             angle_tolerance)) == NULL) {
    return 0;
  }

  if ((centering = get_centering(dataset->hall_number)) == CENTERING_ERROR) {
    spg_free_dataset(dataset);
    dataset = NULL;
    goto err;
  }

  if ((bravais = cel_alloc_cell(dataset->n_std_atoms)) == NULL) {
    spg_free_dataset(dataset);
    dataset = NULL;
    goto err;
  }

  cel_set_cell(bravais,
               dataset->std_lattice,
               dataset->std_positions,
               dataset->std_types);

  spg_free_dataset(dataset);
  dataset = NULL;

  if ((mapping_table = (int*) malloc(sizeof(int) * bravais->size)) == NULL) {
    warning_print("spglib: Memory could not be allocated ");
    cel_free_cell(bravais);
    bravais = NULL;
    goto err;
  }

  primitive = spa_transform_to_primitive(mapping_table,
                                         bravais,
                                         identity,
                                         centering,
                                         symprec);

  for (i = 0; i < primitive->size; i++) {
    /* This is an assertion. */
    if (mapping_table[i] != i) {
      warning_print("spglib: spa_transform_to_primitive failed.");
      warning_print("Unexpected atom index mapping to primitive (%d != %d).\n",
                    mapping_table[i], i);
      warning_print(" (line %d, %s).\n", __LINE__, __FILE__);
      free(mapping_table);
      mapping_table = NULL;
      cel_free_cell(bravais);
      bravais = NULL;
      goto err;
    }
  }

  free(mapping_table);
  mapping_table = NULL;
  cel_free_cell(bravais);
  bravais = NULL;

  if (primitive == NULL) {
    goto err;
  }

  set_cell(lattice, position, types, primitive);
  num_prim_atom = primitive->size;

  cel_free_cell(primitive);
  primitive = NULL;

  return num_prim_atom;

 err:
  spglib_error_code = SPGERR_CELL_STANDARDIZATION_FAILED;
  return 0;
}

static int standardize_cell(double lattice[3][3],
                            double position[][3],
                            int types[],
                            const int num_atom,
                            const int num_array_size,
                            const double symprec,
                            const double angle_tolerance)
{
  int i, n_std_atoms;
  SpglibDataset *dataset;

  n_std_atoms = 0;
  dataset = NULL;

  if ((dataset = get_dataset(lattice,
                             position,
                             types,
                             num_atom,
                             0,
                             symprec,
                             angle_tolerance)) == NULL) {
    goto err;
  }

  if (num_array_size > 0) {
    if (num_atom < dataset->n_std_atoms) {
      goto array_size_shortage_err;
    }
  }

  n_std_atoms = dataset->n_std_atoms;
  mat_copy_matrix_d3(lattice, dataset->std_lattice);
  for (i = 0; i < dataset->n_std_atoms; i++) {
    types[i] = dataset->std_types[i];
    mat_copy_vector_d3(position[i], dataset->std_positions[i]);
  }

  spg_free_dataset(dataset);
  dataset = NULL;

  return n_std_atoms;

err:
  spglib_error_code = SPGERR_CELL_STANDARDIZATION_FAILED;
  return 0;

array_size_shortage_err:
  spglib_error_code = SPGERR_ARRAY_SIZE_SHORTAGE;
  return 0;
}

static int get_standardized_cell(double lattice[3][3],
                                 double position[][3],
                                 int types[],
                                 const int num_atom,
                                 const int num_array_size,
                                 const int to_primitive,
                                 const double symprec,
                                 const double angle_tolerance)
{
  int i, num_std_atom, num_prim_atom;
  int *mapping_table;
  SpglibDataset *dataset;
  Cell *std_cell, *cell, *primitive;
  Centering centering;

  num_std_atom = 0;
  mapping_table = NULL;
  dataset = NULL;
  std_cell = NULL;
  cell = NULL;
  primitive = NULL;

  if ((dataset = get_dataset(lattice,
                             position,
                             types,
                             num_atom,
                             0,
                             symprec,
                             angle_tolerance)) == NULL) {
    goto err;
  }

  if ((centering = get_centering(dataset->hall_number)) == CENTERING_ERROR) {
    goto err;
  }

  if ((cell = cel_alloc_cell(num_atom)) == NULL) {
    spg_free_dataset(dataset);
    dataset = NULL;
    goto err;
  }

  cel_set_cell(cell, lattice, position, types);

  if ((mapping_table = (int*) malloc(sizeof(int) * cell->size)) == NULL) {
    warning_print("spglib: Memory could not be allocated ");
    cel_free_cell(cell);
    cell = NULL;
    spg_free_dataset(dataset);
    dataset = NULL;
    goto err;
  }

  if ((primitive = spa_transform_to_primitive(mapping_table,
                                              cell,
                                              dataset->transformation_matrix,
                                              centering,
                                              symprec)) == NULL) {
    warning_print("spglib: spa_transform_to_primitive failed.");
    warning_print(" (line %d, %s).\n", __LINE__, __FILE__);
  }

  for (i = 0; i < cell->size; i++) {
    /* This is an assertion. */
    if (mapping_table[i] != dataset->mapping_to_primitive[i]) {
      warning_print("spglib: spa_transform_to_primitive failed.");
      warning_print("Unexpected atom index mapping to primitive (%d != %d).\n",
                    mapping_table[i], dataset->mapping_to_primitive[i]);
      warning_print(" (line %d, %s).\n", __LINE__, __FILE__);
      free(mapping_table);
      mapping_table = NULL;
      cel_free_cell(cell);
      cell = NULL;
      spg_free_dataset(dataset);
      dataset = NULL;
      goto err;
    }
  }

  free(mapping_table);
  mapping_table = NULL;
  cel_free_cell(cell);
  cell = NULL;
  spg_free_dataset(dataset);
  dataset = NULL;

  if (primitive == NULL) {
    goto err;
  }

  if (to_primitive || centering == PRIMITIVE) {
    set_cell(lattice, position, types, primitive);
    num_prim_atom = primitive->size;
    cel_free_cell(primitive);
    primitive = NULL;
    return num_prim_atom;
  }

  if ((std_cell = spa_transform_from_primitive(primitive, centering, symprec))
      == NULL) {
    warning_print("spglib: spa_transform_from_primitive failed.");
    warning_print(" (line %d, %s).\n", __LINE__, __FILE__);
  }
  cel_free_cell(primitive);
  primitive = NULL;

  if (std_cell == NULL) {
    goto err;
  }

  if (num_array_size > 0) {
    if (num_array_size < std_cell->size) {
      cel_free_cell(std_cell);
      std_cell = NULL;
      goto array_size_shortage_err;
    }
  }

  num_std_atom = std_cell->size;
  set_cell(lattice, position, types, std_cell);
  cel_free_cell(std_cell);
  std_cell = NULL;
  return num_std_atom;

err:
  spglib_error_code = SPGERR_CELL_STANDARDIZATION_FAILED;
  return 0;

array_size_shortage_err:
  spglib_error_code = SPGERR_ARRAY_SIZE_SHORTAGE;
  return 0;
}

static void set_cell(double lattice[3][3],
                     double position[][3],
                     int types[],
                     Cell * cell)
{
  int i;

  mat_copy_matrix_d3(lattice, cell->lattice);
  for (i = 0; i < cell->size; i++) {
    types[i] = cell->types[i];
    mat_copy_vector_d3(position[i], cell->position[i]);
  }
}

static Centering get_centering(int hall_number)
{
  SpacegroupType spgtype;

  spgtype = spgdb_get_spacegroup_type(hall_number);

  return spgtype.centering;
}

static int get_international(char symbol[11],
                             SPGCONST double lattice[3][3],
                             SPGCONST double position[][3],
                             const int types[],
                             const int num_atom,
                             const double symprec,
                             const double angle_tolerance)
{
  SpglibDataset *dataset;
  int number;

  dataset = NULL;

  if ((dataset = get_dataset(lattice,
                             position,
                             types,
                             num_atom,
                             0,
                             symprec,
                             angle_tolerance)) == NULL) {
    goto err;
  }

  if (dataset->spacegroup_number > 0) {
    number = dataset->spacegroup_number;
    memcpy(symbol, dataset->international_symbol, 11);

    spg_free_dataset(dataset);
    dataset = NULL;
  } else {
    spg_free_dataset(dataset);
    dataset = NULL;
    goto err;
  }

  spglib_error_code = SPGLIB_SUCCESS;
  return number;

 err:
  spglib_error_code = SPGERR_SPACEGROUP_SEARCH_FAILED;
  return 0;
}

static int get_schoenflies(char symbol[7],
                           SPGCONST double lattice[3][3],
                           SPGCONST double position[][3],
                           const int types[],
                           const int num_atom,
                           const double symprec,
                           const double angle_tolerance)
{
  SpglibDataset *dataset;
  SpglibSpacegroupType spgtype;
  int number;

  dataset = NULL;

  if ((dataset = get_dataset(lattice,
                             position,
                             types,
                             num_atom,
                             0,
                             symprec,
                             angle_tolerance)) == NULL) {
    goto err;
  }

  if (dataset->spacegroup_number > 0) {
    number = dataset->spacegroup_number;
    spgtype = spg_get_spacegroup_type(dataset->hall_number);
    memcpy(symbol, spgtype.schoenflies, 7);
    spg_free_dataset(dataset);
    dataset = NULL;
  } else {
    spg_free_dataset(dataset);
    dataset = NULL;
    goto err;
  }

  spglib_error_code = SPGLIB_SUCCESS;
  return number;

 err:
  spglib_error_code = SPGERR_SPACEGROUP_SEARCH_FAILED;
  return 0;
}


/*---------*/
/* kpoints */
/*---------*/
static int get_ir_reciprocal_mesh(int grid_address[][3],
                                  int ir_mapping_table[],
                                  const int mesh[3],
                                  const int is_shift[3],
                                  const int is_time_reversal,
                                  SPGCONST double lattice[3][3],
                                  SPGCONST double position[][3],
                                  const int types[],
                                  const int num_atom,
                                  const double symprec,
                                  const double angle_tolerance)
{
  SpglibDataset *dataset;
  int num_ir, i;
  MatINT *rotations, *rot_reciprocal;

  if ((dataset = get_dataset(lattice,
                             position,
                             types,
                             num_atom,
                             0,
                             symprec,
                             angle_tolerance)) == NULL) {
    return 0;
  }

  if ((rotations = mat_alloc_MatINT(dataset->n_operations)) == NULL) {
    spg_free_dataset(dataset);
    dataset = NULL;
    return 0;
  }

  for (i = 0; i < dataset->n_operations; i++) {
    mat_copy_matrix_i3(rotations->mat[i], dataset->rotations[i]);
  }
  rot_reciprocal = kpt_get_point_group_reciprocal(rotations, is_time_reversal);
  num_ir = kpt_get_irreducible_reciprocal_mesh(grid_address,
                                               ir_mapping_table,
                                               mesh,
                                               is_shift,
                                               rot_reciprocal);
  mat_free_MatINT(rot_reciprocal);
  rot_reciprocal = NULL;
  mat_free_MatINT(rotations);
  rotations = NULL;
  spg_free_dataset(dataset);
  dataset = NULL;
  return num_ir;
}

static size_t get_dense_ir_reciprocal_mesh(int grid_address[][3],
                                           size_t ir_mapping_table[],
                                           const int mesh[3],
                                           const int is_shift[3],
                                           const int is_time_reversal,
                                           SPGCONST double lattice[3][3],
                                           SPGCONST double position[][3],
                                           const int types[],
                                           const size_t num_atom,
                                           const double symprec,
                                           const double angle_tolerance)
{
  SpglibDataset *dataset;
  int i;
  size_t num_ir;
  MatINT *rotations, *rot_reciprocal;

  if ((dataset = get_dataset(lattice,
                             position,
                             types,
                             num_atom,
                             0,
                             symprec,
                             angle_tolerance)) == NULL) {
    return 0;
  }

  if ((rotations = mat_alloc_MatINT(dataset->n_operations)) == NULL) {
    spg_free_dataset(dataset);
    dataset = NULL;
    return 0;
  }

  for (i = 0; i < dataset->n_operations; i++) {
    mat_copy_matrix_i3(rotations->mat[i], dataset->rotations[i]);
  }
  rot_reciprocal = kpt_get_point_group_reciprocal(rotations, is_time_reversal);
  num_ir = kpt_get_dense_irreducible_reciprocal_mesh(grid_address,
                                                     ir_mapping_table,
                                                     mesh,
                                                     is_shift,
                                                     rot_reciprocal);
  mat_free_MatINT(rot_reciprocal);
  rot_reciprocal = NULL;
  mat_free_MatINT(rotations);
  rotations = NULL;
  spg_free_dataset(dataset);
  dataset = NULL;
  return num_ir;
}

static int get_stabilized_reciprocal_mesh(int grid_address[][3],
                                          int map[],
                                          const int mesh[3],
                                          const int is_shift[3],
                                          const int is_time_reversal,
                                          const int num_rot,
                                          SPGCONST int rotations[][3][3],
                                          const size_t num_q,
                                          SPGCONST double qpoints[][3])
{
  MatINT *rot_real;
  int i, num_ir;

  rot_real = NULL;

  if ((rot_real = mat_alloc_MatINT(num_rot)) == NULL) {
    return 0;
  }

  for (i = 0; i < num_rot; i++) {
    mat_copy_matrix_i3(rot_real->mat[i], rotations[i]);
  }

  num_ir = kpt_get_stabilized_reciprocal_mesh(grid_address,
                                              map,
                                              mesh,
                                              is_shift,
                                              is_time_reversal,
                                              rot_real,
                                              num_q,
                                              qpoints);

  mat_free_MatINT(rot_real);
  rot_real = NULL;

  return num_ir;
}

static size_t get_dense_stabilized_reciprocal_mesh(int grid_address[][3],
                                                   size_t ir_mapping_table[],
                                                   const int mesh[3],
                                                   const int is_shift[3],
                                                   const int is_time_reversal,
                                                   const size_t num_rot,
                                                   SPGCONST int rotations[][3][3],
                                                   const size_t num_q,
                                                   SPGCONST double qpoints[][3])
{
  MatINT *rot_real;
  size_t i, num_ir;

  rot_real = NULL;

  if ((rot_real = mat_alloc_MatINT(num_rot)) == NULL) {
    return 0;
  }

  for (i = 0; i < num_rot; i++) {
    mat_copy_matrix_i3(rot_real->mat[i], rotations[i]);
  }

  num_ir = kpt_get_dense_stabilized_reciprocal_mesh(grid_address,
                                                    ir_mapping_table,
                                                    mesh,
                                                    is_shift,
                                                    is_time_reversal,
                                                    rot_real,
                                                    num_q,
                                                    qpoints);

  mat_free_MatINT(rot_real);
  rot_real = NULL;

  return num_ir;
}
