EMM_VTK.hpp

Go to the documentation of this file.

#ifndef EMM_VTK_HPP
#define EMM_VTK_HPP

#include "EMM_Exception.h"

template<class T>
tBoolean EMM_VTK::writeField(ofstream& file,
                             const tString& name,
                             const tBoolean& isUniform,
                             const tUIndex& n,
                             const tUSInt& dim,
                             const tUIndex& startIndex,
                             const tUIndex& endIndex,
                             const tUIndex& inc,
                             const T* array,
                             const tUIndex& nWeight,
                             const T* weight) const {
    
    // get the type of T
    tString typeName=CORE_Object::getTypeName<T>();
    
    //translate in vtk type
    tString vtkType=getVTKType(typeName);
    //if no vtk type
    if (vtkType.compare("")==0) return false;
    
    
    //write the header
    file << "<DataArray type=\""+vtkType+"\" Name=\""<<name<<"\" ";
    //write the dimension of the field
    if (dim>1) file <<"NumberOfComponents=\""<<dim<<"\" ";
    //write the format
    file <<"format=\"ascii\">"<<endl;
    
    //get the start index & end index
    tUIndex i,end=endIndex,start=startIndex;
    //end index null: write all the data
    if (end==0) end=n;
    
    const T *pArray=null;
    //weight
    const T *pWeight=weight;
    
    //take the uniform weight
    T w=(T)1;
    if (pWeight!=null) w=(*pWeight);
    tBoolean isWeightUniform=((pWeight==null) || (nWeight<=1));
    if ((!isWeightUniform) && (nWeight!=n))
        throw EMM_Exception("core",
                            "EMM_VTK::writeField",
                            "weight has incompatible size");
    
    //loop on coordinate
    tUSInt k;
    if (isUniform) {//field is uniform && weight is uniform
        if (isWeightUniform) {
            //get only the field at element 0
            pArray=&array[0];
            for(i=start;i<end;i+=inc){//loop on cell
                for (k=0;k<dim;k++) {//loo pon coordinate
                    file<<toVTKNumber(vtkType,(w * (*pArray)))<<" ";
                    pArray++;
                }
                file<<endl;
                pArray=&array[0];
            }
        } else {//uniform field && not uniform weight
            pArray=&array[0];
            pWeight=&weight[start];
            for(i=start;i<end;i+=inc){//loop on cell
                for (k=0;k<dim;k++) {//loop on coordinate
                    file<<toVTKNumber(vtkType,( (*pWeight) * (*pArray)))<<" ";
                    pArray++;
                }
                file<<endl;
                
                //next cell
                pArray=&array[0];
                pWeight++;
            }
        }
    } else {//field is not uniform
        pArray=&array[dim*start];
        if (inc==1) {//increment  between 2 elements = 1
            if (isWeightUniform) {//weight is uniform
                for(i=start;i<end;i++){//loop on cell
                    for (k=0;k<dim;k++) {//loop on coordinate
                        file<<toVTKNumber(vtkType,w*(*pArray))<<" ";
                        pArray++;
                    }//end loop on coordinate
                    file<<endl;
                }//end loop on cell
            } else { //weight is not uniform
                pWeight=&weight[start];
                for(i=start;i<end;i++){ //loop on cell
                    for (k=0;k<dim;k++) {//loop on ccoordinate
                        file<<toVTKNumber(vtkType,((*pWeight)*(*pArray)))<<" ";
                        pArray++;
                    }//end loop on coordinate
                    pWeight++;
                    file<<endl;
                }//end loop on cell
            }
        } else if (inc>1) { //increment between 2 elements > 1
            tUIndex ld=(inc-1)*dim;
            if (isWeightUniform) {//field is not uniform && inc> 1 && weight is uniform
                for(i=start;i<end;i+=inc){  
                    for (k=0;k<dim;k++) {
                        file<<toVTKNumber(vtkType,(w*(*pArray)))<<" ";
                        pArray++;
                    }//end loop on coordinate
                    //next element 
                    pArray+=ld;
                    file<<endl;
                }//end loop on cell
            } else {//field is not uniform && inc> 1 && weight is not uniform
                pWeight=&weight[start];
                for(i=start;i<end;i+=inc){  
                    for (k=0;k<dim;k++) {
                        file<<toVTKNumber(vtkType,((*pWeight)*(*pArray)))<<" ";
                        pArray++;
                    }//end loop on coordinate

                    //next element 
                    pArray+=ld;
                    pWeight+=inc;
                    file<<endl;
                }//end loop on cell
            }
        }//end inc>1
    }//end field is not uniform
        
    file<<"</DataArray>"<<endl;
        
    return true;
} //end write field method

    
#endif