io_readsnap.h

Go to the documentation of this file.

/*******************************************************************************
 * \copyright   This file is part of the GADGET4 N-body/SPH code developed
 * \copyright   by Volker Springel. Copyright (C) 2014-2020 by Volker Springel
 * \copyright   (vspringel@mpa-garching.mpg.de) and all contributing authors.
 *******************************************************************************/
 
#ifndef READSNAP_IO_H
#define READSNAP_IO_H
 
#include "gadgetconfig.h"
 
#ifdef MERGERTREE
 
#include "../data/allvars.h"
#include "../data/dtypes.h"
#include "../data/mymalloc.h"
#include "../fof/fof.h"
#include "../io/hdf5_util.h"
#include "../io/io.h"
#include "../logs/timer.h"
#include "../main/simulation.h"
#include "../mergertree/mergertree.h"
#include "../mpi_utils/mpi_utils.h"
#include "../sort/parallel_sort.h"
#include "../subfind/subfind.h"
#include "../system/system.h"
 
class readsnap_io : public IO_Def
{
 private:
  mergertree *MergerTree;
 
 public:
  readsnap_io(mergertree *MergerTree_ptr, MPI_Comm comm, int format);
 
  void mergertree_read_snap_ids(int num);
 
  /* supplied virtual functions */
  void fill_file_header(int writeTask, int lastTask, long long *nloc_part, long long *npart);
  void read_file_header(const char *fname, int filenr, int readTask, int lastTask, long long *nloc_part, long long *npart,
                        int *nstart);
  void get_datagroup_name(int grnr, char *gname);
  void write_header_fields(hid_t);
  void read_header_fields(const char *fname);
  void read_increase_numbers(int type, int n_for_this_task);
  int get_filenr_from_header(void);
  void set_filenr_in_header(int);
  void *get_base_address_of_structure(enum arrays array, int index);
  int get_type_of_element(int index);
  void set_type_of_element(int index, int type);
 
#ifdef GADGET2_HEADER
  struct io_header
  {
    int npart[NTYPES_HEADER];                       
    double mass[NTYPES_HEADER];                     
    double time;                                    
    double redshift;                                
    int flag_sfr;                                   
    int flag_feedback;                              
    unsigned int npartTotalLowWord[NTYPES_HEADER];  
    int flag_cooling;                               
    int num_files;                                  
    double BoxSize;                                 
    double Omega0;                                  
    double OmegaLambda;                             
    double HubbleParam;                             
    double Hubble;                                  
    unsigned int npartTotalHighWord[NTYPES_HEADER]; 
    int flag_entropy_instead_u;                     
    int flag_doubleprecision;                       
    int flag_ic_info;                    
    float lpt_scalingfactor;             
    long long npartTotal[NTYPES_HEADER]; 
  };
  io_header header; 
#else
 
  /* new simplified header */
  struct io_header
  {
    long long npart[NTYPES_HEADER]; 
    long long npartTotal[NTYPES_HEADER];
    double mass[NTYPES_HEADER]; 
    double time;                
    double redshift;            
    double BoxSize;             
    int num_files;              
  };
  io_header header; 
#endif
};
 
#endif
 
#endif /* READSNAP_IO_H */