/*=========================================================================

 Program:   vv
 Module:    $RCSfile: vvGlyphSource.cxx,v $
 Language:  C++
 Date:      $Date: 2010/01/06 13:31:57 $
 Version:   $Revision: 1.1 $
 Author :   Pierre Seroul (pierre.seroul@gmail.com)

Copyright (C) 2008
Léon Bérard cancer center http://oncora1.lyon.fnclcc.fr
CREATIS-LRMN http://www.creatis.insa-lyon.fr

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, version 3 of the License.

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, see <http://www.gnu.org/licenses/>.

=========================================================================*/
#include "vvGlyphSource.h"

#include "vtkCellArray.h"
#include "vtkCellData.h"
#include "vtkMath.h"
#include "vtkInformation.h"
#include "vtkInformationVector.h"
#include "vtkObjectFactory.h"
#include "vtkPolyData.h"
#include "vtkUnsignedCharArray.h"

vtkCxxRevisionMacro(vvGlyphSource, "$Revision: 1.1 $");
vtkStandardNewMacro(vvGlyphSource);


//----------------------------------------------------------------------------
int vvGlyphSource::RequestData(
    vtkInformation *vtkNotUsed(request),
    vtkInformationVector **vtkNotUsed(inputVector),
    vtkInformationVector *outputVector)
{
    // get the info object
    vtkInformation *outInfo = outputVector->GetInformationObject(0);

    // get the ouptut
    vtkPolyData *output = vtkPolyData::SafeDownCast(
                              outInfo->Get(vtkDataObject::DATA_OBJECT()));

    //Allocate storage
    vtkPoints *pts = vtkPoints::New();
    pts->Allocate(6,6);
    vtkCellArray *verts = vtkCellArray::New();
    verts->Allocate(verts->EstimateSize(1,1),1);
    vtkCellArray *lines = vtkCellArray::New();
    lines->Allocate(lines->EstimateSize(4,2),2);
    vtkCellArray *polys = vtkCellArray::New();
    polys->Allocate(polys->EstimateSize(1,4),4);
    vtkUnsignedCharArray *colors = vtkUnsignedCharArray::New();
    colors->SetNumberOfComponents(3);
    colors->Allocate(2,2);

    this->ConvertColor();

    //Special options
    if ( this->Dash )
    {
        int filled = this->Filled;
        this->Filled = 0;
        this->CreateDash(pts,lines,polys,colors,this->Scale2);
        this->Filled = filled;
    }
    if ( this->Cross )
    {
        int filled = this->Filled;
        this->Filled = 0;
        this->CreateCross(pts,lines,polys,colors,this->Scale2);
        this->Filled = filled;
    }

    //Call the right function
    switch (this->GlyphType)
    {
    case VTK_NO_GLYPH:
        break;
    case VTK_VERTEX_GLYPH:
        this->CreateVertex(pts,verts,colors);
        break;
    case VTK_DASH_GLYPH:
        this->CreateDash(pts,lines,polys,colors,this->Scale);
        break;
    case VTK_CROSS_GLYPH:
        this->CreateCross(pts,lines,polys,colors,this->Scale);
        break;
    case VTK_THICKCROSS_GLYPH:
        this->CreateThickCross(pts,lines,polys,colors);
        break;
    case VTK_TRIANGLE_GLYPH:
        this->CreateTriangle(pts,lines,polys,colors);
        break;
    case VTK_SQUARE_GLYPH:
        this->CreateSquare(pts,lines,polys,colors);
        break;
    case VTK_CIRCLE_GLYPH:
        this->CreateCircle(pts,lines,polys,colors);
        break;
    case VTK_DIAMOND_GLYPH:
        this->CreateDiamond(pts,lines,polys,colors);
        break;
    case VTK_ARROW_GLYPH:
        this->CreateArrow(pts,lines,polys,colors);
        break;
    case VTK_THICKARROW_GLYPH:
        this->CreateThickArrow(pts,lines,polys,colors);
        break;
    case VTK_HOOKEDARROW_GLYPH:
        this->CreateHookedArrow(pts,lines,polys,colors);
        break;
    case VTK_EDGEARROW_GLYPH:
        this->CreateEdgeArrow(pts,lines,polys,colors);
        break;
    case VTK_SPECIFICARROW_GLYPH:
        this->CreateSpecificArrow(pts,lines,polys,colors);
        break;
    }

    this->TransformGlyph(pts);

    //Clean up
    output->SetPoints(pts);
    pts->Delete();

    output->SetVerts(verts);
    verts->Delete();

    output->SetLines(lines);
    lines->Delete();

    output->SetPolys(polys);
    polys->Delete();

    output->GetCellData()->SetScalars(colors);
    colors->Delete();

    return 1;
}

void vvGlyphSource::CreateSpecificArrow(vtkPoints *pts, vtkCellArray *lines,
                                        vtkCellArray *polys, vtkUnsignedCharArray *colors)
{
    //stem
    vtkIdType ptIds[3];
    ptIds[0] = pts->InsertNextPoint( 0.0, 0.0, 0.0);
    ptIds[1] = pts->InsertNextPoint(  1.0, 0.0, 0.0);
    lines->InsertNextCell(2,ptIds);
    colors->InsertNextValue(0);
    colors->InsertNextValue(0);
    colors->InsertNextValue(1);

    //arrow head
    ptIds[0] = pts->InsertNextPoint( 0.7, -0.1, 0.0);
    ptIds[1] = pts->InsertNextPoint( 1.0,  0.0, 0.0);
    ptIds[2] = pts->InsertNextPoint( 0.7,  0.1, 0.0);
    lines->InsertNextCell(3,ptIds);
    colors->InsertNextValue(0);
    colors->InsertNextValue(1);
    colors->InsertNextValue(0);
}

void vvGlyphSource::PrintSelf(ostream& os, vtkIndent indent)
{
    this->Superclass::PrintSelf(os, indent);
}