VisualizationAids.cpp

Go to the documentation of this file.

/* ImLib3D
 * Copyright (c) 2001, ULP-IPB Strasbourg.
 *
 * This program 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.
 * 
 * This program 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 this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */
#include <ImLib3D/VisualizationAids.hpp>
#include<ImLib3D/Draw.hpp>

void
IP3D::TransformRectangularGrid(Image3Df& image, const Field3Df& field, const Size3D masksize, bool bAverageField)
{
    //Imagesize 
    // at the moment I (also) require image size to equal field size
    if (image.Size() != field.Size())
    {
        cout << "field size: " << field.Size() << endl;;
        cout << "image size: " << image.Size() << endl;;
        ThrowError("IP3D::TransformRectangularGrid ERROR field size must equal image size");
    }

    //center of mask 
    if (!(masksize.width%2 &&
          masksize.height%2 &&
          masksize.depth%2))
    {
        ThrowError("IP3D::TransformRectangularGrid ERROR masksize must have only odd sides (for centering)");
    }
    Vect3Df center(masksize.width/2, masksize.height/2, masksize.depth/2);

    //Create a subsampled vector field with average vectors:
    Field3Df averageField(field.Width()/masksize.width, field.Height()/masksize.height, field.Depth()/masksize.depth);
    Field3Df::iteratorXYZ pav;
    for (pav=averageField.begin(); pav!=averageField.end(); pav++)
    {
        if (bAverageField)
        {
            cout << "Averaging vectors" << endl;
            Vect3Df meanVec(0,0,0);
            RectZone3Di zone(pav.x*masksize.width, pav.y*masksize.height, pav.z*masksize.depth,
                             ((pav.x+1)*masksize.width)-1, ((pav.y+1)*masksize.height)-1, ((pav.z+1)*masksize.depth)-1);
            Field3Df::const_iteratorZone p(zone);
            for (p=field.begin();p!=field.end();++p)
            {
                meanVec += (*p);
            }
            meanVec/=(masksize.width*masksize.height*masksize.depth);
            (*pav) = meanVec;
        }
        else
        {
            (*pav) = field(rint(Vect3Df(pav.x*masksize.width, pav.y*masksize.height, pav.z*masksize.depth)+center));
        }
    }

    //Now draw lines from the endpoints of the vectors
    //This must be done in each direction (X, Y and Z)
    int x, y, z;

    //x-direction:
    cout << "Drawing x-direction connections" << endl;
    for (y=0;y<averageField.Height();y++)
    {
        for (z=0;z<averageField.Depth();z++)
        {
            Vect3Df P0(averageField(0,y,z));
            Vect3Df PB0(Vect3Df(0,y*masksize.height,z*masksize.depth)+center);
            for (x=1;x<averageField.Width();x++)
            {
                Vect3Df P1(averageField(x,y,z));
                Vect3Df PB1(Vect3Df(x*masksize.width,y*masksize.height,z*masksize.depth)+center);
                Segment3D segment(P0+PB0,P1+PB1);
                IP3D::DrawSegment3D(image, segment, Pen<float>(255), image);
                P0=P1;
                PB0=PB1;
            }
        }
    }

    //y-direction:
    cout << "Drawing y-direction connections" << endl;
    for (x=0;x<averageField.Width();x++)
    {
        for (z=0;z<averageField.Depth();z++)
        {
            Vect3Df P0(averageField(x,0,z));
            Vect3Df PB0(Vect3Df(x*masksize.width,0,z*masksize.depth)+center);
            for (y=1;y<averageField.Height();y++)
            {
                Vect3Df P1(averageField(x,y,z));
                Vect3Df PB1(Vect3Df(x*masksize.width,y*masksize.height,z*masksize.depth)+center);
                Segment3D segment(P0+PB0,P1+PB1);
                IP3D::DrawSegment3D(image, segment,Pen<float>(255), image);
                P0=P1;
                PB0=PB1;
            }
        }
    }

    //z-direction:
    cout << "Drawing z-direction connections" << endl;
    for (x=0;x<averageField.Width();x++)
    {
        for (y=0;y<averageField.Height();y++)
        {
            Vect3Df P0(averageField(x,y,0));
            Vect3Df PB0(Vect3Df(x*masksize.width,y*masksize.height,0)+center);
            for (z=1;z<averageField.Depth();z++)
            {
                Vect3Df P1(averageField(x,y,z));
                Vect3Df PB1(Vect3Df(x*masksize.width,y*masksize.height,z*masksize.depth)+center);
                Segment3D segment(P0+PB0,P1+PB1);
                IP3D::DrawSegment3D(image, segment,Pen<float>(255), image);
                P0=P1;
                PB0=PB1;
            }
        }
    }
}

---