source: python/lib/gio.h @ b8b985e

/*
 *                    Copyright (C) 2015, UChicago Argonne, LLC
 *                               All Rights Reserved
 * 
 *                               Generic IO (ANL-15-066)
 *                     Hal Finkel, Argonne National Laboratory
 * 
 *                              OPEN SOURCE LICENSE
 * 
 * Under the terms of Contract No. DE-AC02-06CH11357 with UChicago Argonne,
 * LLC, the U.S. Government retains certain rights in this software.
 * 
 * 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 names of UChicago Argonne, LLC or the Department of Energy
 *      nor the names of its contributors may be used to endorse or promote
 *      products derived from this software without specific prior written
 *      permission.
 * 
 * *****************************************************************************
 * 
 *                                  DISCLAIMER
 * THE SOFTWARE IS SUPPLIED “AS IS” WITHOUT WARRANTY OF ANY KIND.  NEITHER THE
 * UNTED STATES GOVERNMENT, NOR THE UNITED STATES DEPARTMENT OF ENERGY, NOR
 * UCHICAGO ARGONNE, LLC, NOR ANY OF THEIR EMPLOYEES, MAKES ANY WARRANTY,
 * EXPRESS OR IMPLIED, OR ASSUMES ANY LEGAL LIABILITY OR RESPONSIBILITY FOR THE
 * ACCURACY, COMPLETENESS, OR USEFULNESS OF ANY INFORMATION, DATA, APPARATUS,
 * PRODUCT, OR PROCESS DISCLOSED, OR REPRESENTS THAT ITS USE WOULD NOT INFRINGE
 * PRIVATELY OWNED RIGHTS.
 * 
 * *****************************************************************************
 */

#include <GenericIO.h>

#include <stdint.h>
#include <sstream>

template <class T>
#ifndef GENERICIO_NO_MPI
void read_gio(const MPI_Comm C, char* file_name, std::string var_name, T*& data, int field_count){
  gio::GenericIO reader(C, file_name);
  int rank;
  MPI_Comm_rank(C, &rank);
  printf("rank = %d\n", rank);
#else
void read_gio(char* file_name, std::string var_name, T*& data, int field_count){
  gio::GenericIO reader(file_name);
#endif
  reader.openAndReadHeader(gio::GenericIO::MismatchAllowed);
  int num_ranks = reader.readNRanks();
  printf("num_ranks = %d\n", num_ranks);
  uint64_t max_size = 0;
//  uint64_t rank_size[num_ranks];
  uint64_t rank_size;
//  for(int i =0;i<num_ranks;++i){
//    rank_size[i] = reader.readNumElems(i);
//    if(max_size < rank_size[i])
//      max_size = rank_size[i];
//  }
  rank_size = reader.readNumElems(rank);
  max_size = rank_size;
  printf("rank_size: %ld\nmax_size: %ld\n", rank_size, max_size);
  T* rank_data = new T[max_size*field_count+reader.requestedExtraSpace()/sizeof(T)];
  int64_t offset =0;
  reader.addScalarizedVariable(var_name,rank_data,field_count,
                               gio::GenericIO::VarHasExtraSpace);
//  for(int i=0;i<num_ranks;++i){
//    reader.readData(i,false);
//    std::copy(rank_data,rank_data+rank_size[i]*field_count,data+offset);
//    offset +=rank_size[i]*field_count;
//  }
  reader.readData(rank,false);
  std::copy(rank_data,rank_data+rank_size*field_count,data);
  delete [] rank_data;
  reader.close();
}
extern "C" int64_t get_elem_num(const MPI_Comm C, char* file_name);

extern "C" void read_gio_float (const MPI_Comm C, char* file_name, char* var_name, float* data, int field_count);
extern "C" void read_gio_double(const MPI_Comm C, char* file_name, char* var_name, double* data, int field_count);
extern "C" void read_gio_uint16 (const MPI_Comm C, char* file_name, char* var_name, uint16_t* data, int field_count); 
extern "C" void read_gio_int32 (const MPI_Comm C, char* file_name, char* var_name, int* data, int field_count); 
extern "C" void read_gio_int64 (const MPI_Comm C, char* file_name, char* var_name, int64_t* data, int field_count);
enum var_type{
  float_type=0,
  double_type=1,
  int32_type=2,
  int64_type=3,
  uint16_type=4,
  type_not_found=9,
  var_not_found=10
};
extern "C" var_type get_variable_type(const MPI_Comm C, char* file_name,char* var_name);
extern "C" int get_variable_field_count(const MPI_Comm C, char* file_name,char* var_name);
extern "C" void inspect_gio(const MPI_Comm C, char* file_name);