dumpframe.h

#ifndef DUMPFRAME_HINCLUDED
#define DUMPFRAME_HINCLUDED
/*
 ** @file dumpframe.h
 *
 * Image dumping routines for movies from PKDGRAV
 * Original author: James Wadsley, 2002
 */

#include "GravityParticle.h"
#include "cosmoType.h"

/* PST */

#ifdef USE_PNG
#include "png.h"        /* libpng header; includes zlib.h and setjmp.h */
#include "writepng.h"   /* typedefs, common macros, public prototypes */
#endif

typedef struct dfImage {
    float r,g,b;
    } DFIMAGE;

typedef struct dfColorVal {
    float fVal;
    DFIMAGE dfColor;
    } DFCOLORVAL;

const int DF_MAX_COLORENTRIES = 20;

typedef struct dfColorTable {
    int iProperty;
    int nColors;
    float fPropMin, fPropMax;
    DFCOLORVAL dfColors[DF_MAX_COLORENTRIES];
    } DFCOLORTABLE;

/* 
   Projection can be 2D or voxels

   In principle you can render voxels
   and encode them in different ways.
   I will just build a treezip every time.
   */

enum df_dimension {
    DF_2D, /* Generic 2D */
    DF_3D /* Voxel */
    };

enum df_projectstyle {
    DF_PROJECT_NULL,
    DF_PROJECT_ORTHO,
    DF_PROJECT_PERSPECTIVE
    };

/* in principle voxels can have encoding options 
   For now it will be treezip 
   */
enum df_encodestyle {
    DF_ENCODE_NULL,
    DF_ENCODE_PPM,
    DF_ENCODE_PNG,
    DF_ENCODE_RLE,
    DF_ENCODE_TREEZIP
    };

/* in principle voxels can have rendering options 
   For now this is ignored 
   */
enum df_renderstyle {
    DF_RENDER_NULL,
    DF_RENDER_POINT,
    DF_RENDER_TSC,
    DF_RENDER_SOLID,
    DF_RENDER_SHINE
    };

enum df_numbering {
    DF_NUMBERING_FRAME,
    DF_NUMBERING_STEP,
    DF_NUMBERING_TIME
    };

enum df_target {
    DF_TARGET_USER,
    DF_TARGET_COM_GAS,
    DF_TARGET_COM_DARK,
    DF_TARGET_COM_STAR,
    DF_TARGET_COM_ALL,
    DF_TARGET_OLDEST_STAR,
    DF_TARGET_PHOTOGENIC
    };

enum df_log {
    DF_LOG_NULL,
    DF_LOG_SATURATE,
    DF_LOG_COLOURSAFE
    };

enum df_star_age_colour {
    DF_STAR_AGE_BASIC,
    DF_STAR_AGE_BRIGHT,
    DF_STAR_AGE_COLOUR,
    DF_STAR_AGE_BRIGHT_COLOUR
    };

/* PST */
/* There is a common subset with framesetup that needs it's own structure -- parent class even */
typedef
struct inDumpFrame {
    double dTime;
    double dStep;
    double duTFac;
    double dExp;
    /* 2D Projection info */
    double r[3]; /* Centre */
    double x[3]; /* Projection vectors */
    double y[3];
    double z[3];
    double zClipNear,zClipFar,zEye;
    int bExpansion; /* Rescale lengths into physical units with Expansion factor? */
    int bPeriodic; /* Is it periodic? */
    double fPeriod[3];
    int nxPix,nyPix;    /* Pixmap dimensions */
    int iProject;      /* Projection */

    /* Rendering */
    double pScale1,pScale2;
    double dGasSoftMul,dDarkSoftMul,dStarSoftMul;
    double xlim,ylim,hmul;
    double dYearUnit;
    DFCOLORTABLE dfGasCT, dfDarkCT, dfStarCT;
    int bColMassWeight;
    int bGasSph;
    int iColStarAge;
    int bColLogInterp;
    int iLogScale;
    int iTarget;
    int iRender;       
    /* Render Particle interface */
    int offsetp_r,offsetp_fMass,offsetp_fSoft,offsetp_fBall2,offsetp_iActive,offsetp_fTimeForm;
    int sizeofp;
    int iTypeGas,iTypeDark,iTypeStar;
    
    double dMassGasMin, dMassGasMax;
    double dMassDarkMin,dMassDarkMax;
    double dMassStarMin,dMassStarMax;
    double dMinGasMass;

    int bNonLocal; /* Is this data going non-local? */
    int bVDetails;
    } InDumpFrame;

struct dfFrameSetup {
    double dTime;
    double duTFac;
    int bCooling;
    double dYearUnit;
    /* Projection info */
    double target[3]; /* Centre */
    double eye[3]; /* Eye Position */
    double up[3]; /* up vector */
    double FOV;
    double zEye1,zEye2,zEye;
    double zClipNear,zClipFar; /* clipping */
    double eye2[3]; /* a second vector to add to the eye vector */
    int bEye2;
    int bzClipFrac; /* Is z clipping a fraction of eye to target distance? */
    int bzEye,bzEye1,bzEye2; /* Is zEye a fixed distance? */
    int bEyeAbsolute; /* Eye position is in absolute coordinates (default relative to target point) */
    int bAnchor;
    int nxPix,nyPix;    /* Pixmap dimensions */
    int bExpansion; /* Rescale lengths into physical units with Expansion factor? */
    int bPeriodic; /* Is it periodic? */
    double fPeriod[3];
    int iProject;      /* Projection */

    /* Rendering controllers */
    double pScale1,pScale2;
    DFCOLORTABLE dfGasCT, dfDarkCT, dfStarCT;
    int bColMassWeight;
    int bGasSph;
    int iColStarAge;
    int bColLogInterp;
    int iLogScale;
    int iTarget;
    double dGasSoftMul,dDarkSoftMul,dStarSoftMul;
    int iRender;       /* Rendering */
        


    int bNonLocal; /* Is this data going non-local? */
    };

/* MSR level in PKDGRAV */

struct DumpFrameContext {
    int bAllocated; /* Was this malloc'ed? */

    int nFrame;
    int iMaxRung;   
    double dStep;
    double dTime;
    double duTFac;
    int bCooling;
    double dDumpFrameStep;
    double dDumpFrameTime;
    double dYearUnit;
    /* Particle Filters */
    double dMassGasMin, dMassGasMax;
    double dMassDarkMin,dMassDarkMax;
    double dMassStarMin,dMassStarMax;

    /* Time dependent Frame setup data */
    int iFrameSetup;
    int nFrameSetup;
    struct dfFrameSetup *fs;
    struct dfFrameSetup a;
    struct dfFrameSetup b;
    struct dfFrameSetup c;
    struct dfFrameSetup d;
    double rdt;
    double dTimeMod;
    double dTimeLoop,dPeriodLoop;
    int bLoop;
    int bGetCentreOfMass;
    int bGetOldestStar;
    int bGetPhotogenic;

    int iDimension; /* 2D Pixel or 3D Voxel */
    int iEncode;
    int iNumbering;
    int bVDetails;
    char FileName[161];
    };

enum arraytypes {
    OUT_TEMP_ARRAY,
    OUT_DENSITY_ARRAY,
    OUT_UDOT_ARRAY,
    OUT_U_ARRAY,
    OUT_METALS_ARRAY,
    OUT_TIMEFORM_ARRAY,
    OUT_GROUP_ARRAY,
    OUT_AGE_ARRAY
};
std::string VecFilename(int iType);

void dfInitialize( struct DumpFrameContext **pdf, double dYearUnit, double dTime, 
                  double dDumpFrameTime, double dStep, double dDumpFrameStep,
                   double dDelta, int iMaxRung, int bVDetails, char*,
                   int bPeriodic, Vector3D<double> vPeriod);
void dfFinalize( struct DumpFrameContext *df );

void *dfAllocateImage( int nxPix, int nyPix );
void dfFreeImage( void *Image );

void dfParseOptions( struct DumpFrameContext *df, char * filename );

void dfParseCameraDirections( struct DumpFrameContext *df, char * filename );

void dfSetupFrame( struct DumpFrameContext *df, double dTime, double dStep, double dExp, double *com, struct inDumpFrame *in, int nxPix, int nyPix);

void dfMergeImage( struct inDumpFrame *in, void *Image1, int *nImage1, void *Image2, int *nImage2 );
void dfClearImage( struct inDumpFrame *in, void *Image, int *nImage );

class DataManager;
void dfRenderParticlePoint( struct inDumpFrame *in, void *vImage,
                GravityParticle *p, DataManager *dm);
void dfRenderParticleTSC( struct inDumpFrame *in, void *vImage, 
              GravityParticle *p, DataManager *dm);
void dfRenderParticleSolid( struct inDumpFrame *in, void *vImage, 
                GravityParticle *p, DataManager *dm);

void dfFinishFrame( struct DumpFrameContext *df, double dTime, double dStep, struct inDumpFrame *in, void *Image, int liveViz, unsigned char **outgray);

/* Interpolation Functions */

void dfGetCoeff4( struct DumpFrameContext *df, int ifs );
void dfGetCoeff3L( struct DumpFrameContext *df, int ifs );
void dfGetCoeff3R( struct DumpFrameContext *df, int ifs );
void dfGetCoeff2( struct DumpFrameContext *df, int ifs );

void dfGetCoeff( struct DumpFrameContext *df, int ifs );

void dfInterp( struct DumpFrameContext *df, struct dfFrameSetup *pfs, double x );

/* #include "dumpvoxel.h" */

#endif