//=========================================================================
//  Copyright (c) Kitware, Inc.
//  All rights reserved.
//  See LICENSE.txt for details.
//
//  This software is distributed WITHOUT ANY WARRANTY; without even
//  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
//  PURPOSE.  See the above copyright notice for more information.
//=========================================================================

#include "vtkCMBMesh.h"

#include "vtkCMBModelEdgeMesh.h"
#include "vtkCMBModelFaceMesh.h"
#include "vtkCMBModelVertexMesh.h"
#include "vtkCollection.h"
#include <vtkDiscreteModel.h>
#include <vtkDiscreteModelGeometricEntity.h>
#include <vtkIdList.h>
#include <vtkMath.h>
#include <vtkMergeEventData.h>
#include <vtkModel.h>
#include <vtkModelEdge.h>
#include <vtkModelEntity.h>
#include <vtkModelFace.h>
#include <vtkModelGeometricEntity.h>
#include <vtkModelItemIterator.h>
#include <vtkModelVertex.h>
#include <vtkObjectFactory.h>
#include <vtkPolyData.h>
#include <vtkSmartPointer.h>
#include <vtkSplitEventData.h>
#include <vtkWeakPointer.h>

vtkCMBMesh::vtkCMBMesh()
{
  this->Visible = true;
  this->GlobalLength = 0;
  this->GlobalMinimumAngle = 0;
  this->Model = NULL;
}

vtkCMBMesh::~vtkCMBMesh()
{
}

void vtkCMBMesh::Reset()
{
  this->GlobalLength = 0;
  this->GlobalMinimumAngle = 0;
  this->Model = NULL;
  this->Modified();
}

vtkModel* vtkCMBMesh::GetModel()
{
  return this->Model;
}

void vtkCMBMesh::ModelGeometricEntityChanged(
  vtkObject* /*caller*/, unsigned long event, void* cData, void* callData)
{
  vtkCMBMesh* cmbMesh = static_cast<vtkCMBMesh*>(cData);
  vtkModel* model = cmbMesh->Model;
  if (event == ModelGeometricEntitySplit)
  {
    vtkSplitEventData* splitEventData = static_cast<vtkSplitEventData*>(callData);
    if (model->GetModelDimension() == 2)
    {
      cmbMesh->ModelEdgeSplit(splitEventData);
    }
    else
    {
      vtkGenericWarningMacro("Model face split not implemented yet.")
    }
  }
  else if (event == ModelGeometricEntitiesAboutToMerge)
  {
    vtkMergeEventData* mergeEventData = static_cast<vtkMergeEventData*>(callData);
    if (model->GetModelDimension() == 2)
    {
      cmbMesh->ModelEdgeMerge(mergeEventData);
    }
    else
    {
      vtkGenericWarningMacro("Model face merge not implemented yet.")
    }
  }
  else if (event == ModelGeometricEntityBoundaryModified)
  {
    cmbMesh->ModelEntityBoundaryModified(static_cast<vtkModelGeometricEntity*>(callData));
  }
}

bool vtkCMBMesh::SetLocalMeshLength(vtkCollection* selectedMeshEntities, double localLen)
{
  if (!selectedMeshEntities || selectedMeshEntities->GetNumberOfItems() == 0)
  {
    return false;
  }
  bool res = true;
  for (int i = 0; i < selectedMeshEntities->GetNumberOfItems(); i++)
  {
    vtkSmartPointer<vtkCMBModelEntityMesh> meshEntity =
      vtkCMBModelEntityMesh::SafeDownCast(selectedMeshEntities->GetItemAsObject(i));
    if (meshEntity)
    {
      res = res && meshEntity->SetLocalLength(localLen);
    }
  }
  return res;
}

bool vtkCMBMesh::SetLocalMeshMinimumAngle(vtkCollection* selectedMeshEntities, double localMinAngle)
{
  if (!selectedMeshEntities || selectedMeshEntities->GetNumberOfItems() == 0)
  {
    return false;
  }
  bool res = true;
  for (int i = 0; i < selectedMeshEntities->GetNumberOfItems(); i++)
  {
    vtkSmartPointer<vtkCMBModelFaceMesh> faceMesh =
      vtkCMBModelFaceMesh::SafeDownCast(selectedMeshEntities->GetItemAsObject(i));
    if (faceMesh)
    {
      res = res && faceMesh->SetLocalMinimumAngle(localMinAngle);
    }
  }
  return res;
}

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

  os << indent << "Visible: " << this->Visible << "\n";
  os << indent << "GlobalLength: " << this->GlobalLength << "\n";
  os << indent << "GlobalMinimumAngle: " << this->GlobalMinimumAngle << "\n";
  if (this->Model)
  {
    os << indent << "Model: " << this->Model << "\n";
  }
  else
  {
    os << indent << "Model: (NULL)\n";
  }
}
