//=========================================================================
//  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.
//=========================================================================

#ifndef vtk_aeva_ext_vtkSideSetsToScalars_h
#define vtk_aeva_ext_vtkSideSetsToScalars_h

#include "vtk/aeva/ext/AEVAExtModule.h"
#include "vtkMultiBlockDataSetAlgorithm.h"

#include <set>
#include <string>
#include <unordered_map>

// Given two multi blocks, which have the following structure:
// - The first block is a set of vtkUnstructuredGrid, the entire geometry separated by cell type
//  They should contain global ids.
// - They second block is a set of vtkUnstructuredGrid, a vtkUnstructuredGrid per "group", which
//  represents a grouping from the "master" geometry (first block). They should contain global ids.
//  per cell or point and vtkCompositeName for the block.
// - They all share the same points
// This filter then produces, as output, a shallow copy of the first block, with each vtkUnstructuredGrid
// recieving a scalar array that represents the flatten families from the groups.
class AEVAEXT_EXPORT vtkSideSetsToScalars : public vtkMultiBlockDataSetAlgorithm
{
public:
  static vtkSideSetsToScalars* New();
  vtkTypeMacro(vtkSideSetsToScalars, vtkMultiBlockDataSetAlgorithm);

  vtkSideSetsToScalars()
  {
    SetNumberOfInputPorts(2);
    SetNumberOfOutputPorts(1);
  }

public:
  //@{
  /**
      * The name of the array in Scalar array to create
      */
  vtkSetMacro(ScalarArrayName, std::string);
  vtkGetMacro(ScalarArrayName, std::string);
  //@}

  /**
      * Given a family id, returns all the group ids of the family
      */
  std::set<vtkIdType> getCellGroupSet(const int familyId) const
  {
    if (familyIdToCellGroupSet.count(familyId) != 0)
      return familyIdToCellGroupSet.at(familyId);
    else
      return std::set<vtkIdType>();
  }

  /**
      * Given a family id, returns all the group ids of the family
      */
  std::set<vtkIdType> getVertexGroupSet(const int familyId) const
  {
    if (familyIdToVertexGroupSet.count(familyId) != 0)
      return familyIdToVertexGroupSet.at(familyId);
    else
      return std::set<vtkIdType>();
  }

  /**
      * Returns the familyId map for cells
      */
  const std::unordered_map<int, std::set<vtkIdType> >& getFamilyIdToCellGroupSetMap() const
  {
    return familyIdToCellGroupSet;
  }

  /**
      * Returns the familyId map for vertices
      */
  const std::unordered_map<int, std::set<vtkIdType> >& getFamilyIdToVertexGroupSetMap() const
  {
    return familyIdToVertexGroupSet;
  }

protected:
  int RequestData(vtkInformation* request,
    vtkInformationVector** inputVec,
    vtkInformationVector* outputVec) override;

private:
  vtkSideSetsToScalars(const vtkSideSetsToScalars&) = delete;
  void operator=(const vtkSideSetsToScalars&) = delete;

  std::string ScalarArrayName = "FAM";
  std::unordered_map<int, std::set<vtkIdType> > familyIdToCellGroupSet;
  std::unordered_map<int, std::set<vtkIdType> > familyIdToVertexGroupSet;
};

#endif
