/*=========================================================================
 *
 *  Copyright Insight Software Consortium
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0.txt
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 *=========================================================================*/
#ifndef itkGPUInPlaceImageFilter_hxx
#define itkGPUInPlaceImageFilter_hxx

#include "itkGPUInPlaceImageFilter.h"

namespace itk
{
template< typename TInputImage, typename TOutputImage, typename TParentImageFilter >
GPUInPlaceImageFilter< TInputImage, TOutputImage, TParentImageFilter >
::GPUInPlaceImageFilter()
{
}

template< typename TInputImage, typename TOutputImage, typename TParentImageFilter  >
GPUInPlaceImageFilter< TInputImage, TOutputImage, TParentImageFilter  >
::~GPUInPlaceImageFilter()
{
}

template< typename TInputImage, typename TOutputImage, typename TParentImageFilter >
void
GPUInPlaceImageFilter< TInputImage, TOutputImage, TParentImageFilter >
::PrintSelf(std::ostream & os, Indent indent) const
{
  GPUSuperclass::PrintSelf(os, indent);
}

template< typename TInputImage, typename TOutputImage, typename TParentImageFilter >
void
GPUInPlaceImageFilter< TInputImage, TOutputImage, TParentImageFilter >
::ReleaseInputs()
{
  CPUSuperclass::ReleaseInputs();
/*
  if( this->GetGPUEnabled() )
  {
    // do something
    std::cout << "ToDo: GPUInPlaceImageFilter::ReleaseInputs()" << std::endl;
  }
  else
  {
    CPUSuperclass::ReleaseInputs();
  }
*/
}

template< typename TInputImage, typename TOutputImage, typename TParentImageFilter  >
void
GPUInPlaceImageFilter< TInputImage, TOutputImage, TParentImageFilter >
::AllocateOutputs()
{
  if( this->GetGPUEnabled() )
    {
    // if told to run in place and the types support it,
    if ( this->GetInPlace() && this->CanRunInPlace() )
      {
      // Graft this first input to the output.  Later, we'll need to
      // remove the input's hold on the bulk data.
      //
      OutputImagePointer inputAsOutput =
        dynamic_cast< TOutputImage * >( const_cast< TInputImage * >( this->GetInput() ) );
      if ( inputAsOutput )
        {
          this->GraftOutput(inputAsOutput);
        }
      else
        {
        // if we cannot cast the input to an output type, then allocate
        // an output usual.
          OutputImagePointer outputPtr;

          outputPtr = this->GetOutput(0);
          outputPtr->SetBufferedRegion( outputPtr->GetRequestedRegion() );
          outputPtr->Allocate();
        }

      using ImageBaseType = ImageBase< OutputImageDimension >;
      typename ImageBaseType::Pointer outputPtr;

      // If there are more than one outputs, allocate the remaining outputs
      for ( unsigned int i = 1; i < this->GetNumberOfOutputs(); i++ )
        {
        // Check whether the output is an image of the appropriate
        // dimension (use ProcessObject's version of the GetInput()
        // method since it returns the input as a pointer to a
        // DataObject as opposed to the subclass version which
        // static_casts the input to an TInputImage).
        outputPtr = dynamic_cast< ImageBaseType * >( this->ProcessObject::GetOutput(i) );

        if ( outputPtr )
          {
          outputPtr->SetBufferedRegion( outputPtr->GetRequestedRegion() );
          outputPtr->Allocate();
          }
        // if the output is not of simular type then it is assumed the
        // the derived class allocated the output if needed.
        }

      }
    else
      {
      CPUSuperclass::AllocateOutputs();
      }
    }
  else
    {
    CPUSuperclass::AllocateOutputs();
    }
}

} // end of namespace itk

#endif
