mda.h

/****************************************************************************
 This file is part of ChainLink
 Copyright (C) 2007 Jeremy Magland (Jeremy.Magland@gmail.com)

 ChainLink is free software; you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation; either version 2 of the License, or
 (at your option) any later version.

 ChainLink is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with ChainLink; if not, write to the Free Software
 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

*****************************************************************************/

#ifndef MDA_H
#define MDA_H

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include "chainlink_complex.h"

#define MAX_MDA_DIMS 50
#define MDA_COMMAND_MASK -3

#define MDA_TYPE_COMPLEX -1
#define MDA_TYPE_BYTE -2
#define MDA_TYPE_REAL -3
#define MDA_TYPE_SHORT -4
#define MDA_TYPE_INT32 -5


class mda {
private:
        //private data
        int32 data_type;
        int32 num_dims;
        int32 dims[MAX_MDA_DIMS];
        int32 dimprod;
        unsigned char dummy_mask;
public:
        char pathname[200];

        int32 get_data_type() {
                return data_type;
        }
        int32 get_num_dims() {
                return num_dims;
        }
        int32 size(int j) {
                return dims[j];
        }
        int32 size() {
                return dimprod;
        }
        int32 *get_dims() {
                return dims;
        }

public:

        unsigned char *data_byte;
        int32 *data_int32;
        short int *data_short;
        real *data_real;
        complex *data_complex;
        unsigned char *mask;
        

        //constructors and destructor
        mda();
        mda(mda &X);
        virtual ~mda();

        void clear();
        void allocate(int32 data_type_in, 
                                  int32 N1,   int32 N2=-1,int32 N3=-1,int32 N4=-1,
                                  int32 N5=-1,int32 N6=-1,int32 N7=-1,int32 N8=-1);
        void allocate(int32 data_type,int32 num_dims,int32 *dims);
        void allocate_mask(unsigned char val=0);
        void setall(complex val);
        void setall(real val);
        void setall_mask(unsigned char val);
        complex get(int32 j0, int32 j1=-1, int32 j2=-1, int32 j3=-1, int32 j4=-1, int32 j5=-1);
        void set(complex val, int32 j0, int32 j1=-1, int32 j2=-1, int32 j3=-1, int32 j4=-1, int32 j5=-1);
        void set(real val, int32 j0, int32 j1=-1, int32 j2=-1, int32 j3=-1, int32 j4=-1, int32 j5=-1);
        complex get(int32 *inds);
        void set(complex val, int32 *inds);
        void set(real val, int32 *inds);
        unsigned char &maskat(int32 j0, int32 j1=-1, int32 j2=-1, int32 j3=-1, int32 j4=-1, int32 j5=-1);
        unsigned char &maskat(int32 *inds);
        
        complex &complexat(int32 j0, int32 j1=-1, int32 j2=-1, int32 j3=-1, int32 j4=-1, int32 j5=-1);

        real &realat(int32 j0, int32 j1=-1, int32 j2=-1, int32 j3=-1, int32 j4=-1, int32 j5=-1);

        int32 &int32at(int32 j0, int32 j1=-1, int32 j2=-1, int32 j3=-1, int32 j4=-1, int32 j5=-1);
        
        short int &intat(int32 j0, int32 j1=-1, int32 j2=-1, int32 j3=-1, int32 j4=-1, int32 j5=-1);

        unsigned char &byteat(int32 j0, int32 j1=-1, int32 j2=-1, int32 j3=-1, int32 j4=-1, int32 j5=-1);


        void operator=(mda &X);

        void tocomplex();
        void toreal();
        void toint32();
        void toshort();
        void tobyte();

        void add(real val);
        void add(complex val);
        void multiply(real val);
        void multiply(complex val);

        void crop(int32 *inds1,int32 *inds2);

        bool is_in_range(int32 *index);
        
        bool write_mda(char *fname);
        void write_mda(FILE *outf);
        bool read_mda(char *fname);
        bool read_mda(FILE *inf);
        bool read_mda_slices(char *fname,int slice_min,int slice_max);
        void read_mda_slices(FILE *inf,int slice_min,int slice_max);

        bool read_center_slice(char *fname);
        
        void squeeze();
        void shift(int32 *shifts);

        bool matches_dimensions(int32 num_dims_in,int32 *dims_in);

        bool permute(int32 *vec);

        bool is_1d();

        int32 get_1d_index(int32 j0, int32 j1=-1, int32 j2=-1, int32 j3=-1, int32 j4=-1, int32 j5=-1);
        int32 get_1d_index(int32 *inds);

        void readreal(char *fname,int32 N1,int32 N2,int32 N3); //this function will be removed at some point
private:
        void initialize();
        
};

#endif

---