#include <cpExtensions/Visualization/LineSource.h>
#include <vtkInformation.h>
#include <vtkInformationVector.h>

// -------------------------------------------------------------------------
cpExtensions::Visualization::LineSource::
Self* cpExtensions::Visualization::LineSource::
New( )
{
  return( new Self( ) );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::LineSource::
SetPoint1( float pnt[ 3 ] )
{
  double p[ 3 ];
  p[ 0 ] = pnt[ 0 ];
  p[ 1 ] = pnt[ 1 ];
  p[ 2 ] = pnt[ 2 ];
  this->SetPoint1( p );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::LineSource::
SetPoint2( float pnt[ 3 ] )
{
  double p[ 3 ];
  p[ 0 ] = pnt[ 0 ];
  p[ 1 ] = pnt[ 1 ];
  p[ 2 ] = pnt[ 2 ];
  this->SetPoint2( p );
}

// -------------------------------------------------------------------------
cpExtensions::Visualization::LineSource::
LineSource( )
  : Superclass( )
{
  this->SetNumberOfInputPorts( 0 );
  this->Point1[ 0 ] = -double( 0.5 );
  this->Point1[ 1 ] =  double( 0.0 );
  this->Point1[ 2 ] =  double( 0.0 );
  this->Point2[ 0 ] =  double( 0.5 );
  this->Point2[ 1 ] =  double( 0.0 );
  this->Point2[ 2 ] =  double( 0.0 );
}

// -------------------------------------------------------------------------
cpExtensions::Visualization::LineSource::
~LineSource( )
{
}

// -------------------------------------------------------------------------
int cpExtensions::Visualization::LineSource::
RequestData(
  vtkInformation* request,
  vtkInformationVector** inputVector,
  vtkInformationVector* outputVector
  )
{
  // Get output object
  vtkInformation* outInfo = outputVector->GetInformationObject( 0 );
  vtkPolyData* output =
    vtkPolyData::SafeDownCast(
      outInfo->Get( vtkDataObject::DATA_OBJECT( ) )
      );

  // Create points
  vtkPoints* points = vtkPoints::New( );
  points->SetDataType( VTK_FLOAT );
  points->Allocate( 2 );

  // Create cells
  vtkCellArray* verts = vtkCellArray::New( );
  vtkCellArray* lines = vtkCellArray::New( );
  vtkCellArray* faces = vtkCellArray::New( );
  vtkCellArray* strips = vtkCellArray::New( );
  lines->Allocate( lines->EstimateSize( 2, 2 ) );

  // Assign points
  points->InsertPoint( 0, this->Point1 );
  points->InsertPoint( 1, this->Point2 );

  // Assign cells
  vtkIdType cell_pts[ 2 ] = { 0, 1 };
  lines->InsertNextCell( 2, cell_pts );

  // Assign to output
  output->SetPoints( points );
  output->SetVerts( verts );
  output->SetLines( lines );
  output->SetPolys( faces );
  output->SetStrips( strips );

  // Finish and return
  points->Delete( );
  verts->Delete( );
  lines->Delete( );
  faces->Delete( );
  strips->Delete( );
  return( 1 );
}

// -------------------------------------------------------------------------
int cpExtensions::Visualization::LineSource::
RequestInformation(
  vtkInformation* request,
  vtkInformationVector** inputVector,
  vtkInformationVector* outputVector
  )
{
  vtkInformation* outInfo = outputVector->GetInformationObject( 0 );
  outInfo->Set( CAN_HANDLE_PIECE_REQUEST(), 1 );
  return( 1 );
}

// eof - $RCSfile$