CDataReader.hpp

Go to the documentation of this file.

/* -*- c++ -*- */

#ifndef __CDATAREADER_H__
#define __CDATAREADER_H__

#include <QObject>
#include <QVector>

#include <string>
#include <map>
#include <iostream>
#include <cstdlib>
#include <vector>

#include "CGlobals.hpp"
#include "enums.hpp"

using namespace std;

//______________________________________________________________________________
class CDataReader : public QObject
{
Q_OBJECT
  
  static int num_readers;

signals:
  void readError(QString);
  void readProgress(int, QString, QString);

public:

  // should be made protected. sip complains however
  CDataReader(int ifile = -1);

  // virtual destructor: needed to call member functions destructors.
  // must have a body.
  virtual ~CDataReader(); 

  virtual int    numberOfParticles() const;
  virtual int    particleType(const int & ipart) const;
  virtual int    particleID(const int & ipart) const;
  virtual int    particleIndex(const int & id) const;
  virtual double particleMass(const int & ipart) const;
  virtual double particleDensity(const int & ipart) const;
  virtual double particleSml(const int & ipart) const;
  virtual double particleSpecEnergy(const int & ipart) const;
  virtual double particleTemperature(const int & ipart) const;
  virtual double particleZ(const int & ipart) const;
  virtual double particleFe(const int & ipart) const;
  virtual double particleMg(const int & ipart) const;
  virtual double particleACEntropy(const int & ipart) const;
  virtual void   particlePosition(const int & ipart, double &x, double &y, double &z);
  virtual void   particleVelocity(const int & ipart, double &x, double &y, double &z);
  
          bool current_file_present(bool showStatus = false);
          void set_file(const int& file);
          void set_filename(const std::string & filename);
          void set_auto_filename(const bool & fn);
  virtual void convertUnits();

  virtual void        setPrefix(const std::string & prefix);
  virtual std::string getPrefix() const;

  virtual int   currentFile() const;
  virtual int   firstDump() const;
  virtual int   lastDump()  const;

  virtual std::string get_filename();
  virtual int         get_file_size();

  virtual int    getTypeStart(enums::types_t type);
  virtual int    getTypeEnd(enums::types_t type);
  virtual double getTime(void);

  /* 
   * The next routines should be overridden in you own implementation!
   * this routine should set firstdump, lastdump, ndumpspresent,
   * startfile in the all_ struct .
   * 
   * return true if files are found, false if not.
   */
  virtual bool check_files_present();
  virtual bool readFile();

  /* some Fortran convenience routines */
  virtual int  fortranTypeStart(enums::types_t type);
  virtual int  fortranTypeEnd(enums::types_t type);

protected:
  int _currentFile;
  int file_size_;

  std::string filename_;
  std::string _prefix;    
  bool auto_filename_;    
  std::vector<std::string> blocknames_;
  std::vector<int>    blocklens_;

  virtual void set_file_size();

  void setFirstDump(int dump) { _firstDump = dump; }
  void setLastDump(int dump)  { _lastDump  = dump; }

private:
  int _firstDump;  
  int _lastDump;   
};

#define NPRES 17

//______________________________________________________________________________
class CDataGadget : public CDataReader
{
Q_OBJECT

signals:
  void addBlock(QString, int);
  void sendHeaderRow(QString, double);
  void sendHeaderRow(QString, QVector<qreal>);

public:
  CDataGadget(int ifile, int snapformat=2, int inumpersnap=1);
  ~CDataGadget();

  std::string get_filename();
  bool   check_files_present();
  void   setSnapFormat(const int & format);
  void   setInumPerSnap(const int & num);

  int    getSnapFormat() const;

  bool   readFile();

  int    getTypeStart(enums::types_t type);
  int    getTypeEnd(enums::types_t type);
  double getTime(void);

private:
  void alloc_mem(void);
  void free_mem(void);

  void fixStarIDs();
  void checkStarSanity();

  void read_header(void);
  void sendTheHeader();
  bool read_info(std::string & label);

  int  getTypeIndex(const enums::types_t & type);

  bool load_snapshot(const char * fname, const int & files);
  void pd_to_arrays();
  void add_bytes_read(const int & bytes);
  void read_blocksize(int & blocksize);
  void read_label(std::string & label);

  void   add_block(std::string blockname, int blocklen);
  int    get_num_blocks();
  std::string get_blockname(const int & index);
  int    get_blocklen(const int & index);

  int    _snapFormat;     
  int    _inumPerSnap;    
  FILE * snapfile_;
  int    _present[NPRES];
};


#endif // __DATAREADER_H__

---