vtkMaterialInterfacePieceTransactionMatrix.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Kitware Inc.
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkMaterialInterfacePieceTransactionMatrix_h
#define vtkMaterialInterfacePieceTransactionMatrix_h

#include "vtkMaterialInterfacePieceTransaction.h" // for vtkMaterialInterfacePieceTransaction
#include "vtkPVVTKExtensionsFiltersMaterialInterfaceModule.h" //needed for exports
#include "vtkType.h"                                          // for vtkIdType
#include <vector>                                             // for std::vector

class vtkCommunicator;

class VTKPVVTKEXTENSIONSFILTERSMATERIALINTERFACE_EXPORT vtkMaterialInterfacePieceTransactionMatrix
{
public:
  vtkMaterialInterfacePieceTransactionMatrix();
  vtkMaterialInterfacePieceTransactionMatrix(int nFragments, int nProcs);

  ~vtkMaterialInterfacePieceTransactionMatrix();
  void Initialize(int nFragments, int nProcs);

  void Clear();
  vtkIdType GetNumberOfTransactions(int procId);
  std::vector<vtkMaterialInterfacePieceTransaction>& GetTransactions(int fragmentId, int procId);
  void PushTransaction(
    int fragmentId, int procId, vtkMaterialInterfacePieceTransaction& transaction);

  void Broadcast(vtkCommunicator* comm, int srcProc);
  //
  void Print();

  vtkIdType Capacity()
  {
    return this->FlatMatrixSize +
      this->NumberOfTransactions * sizeof(vtkMaterialInterfacePieceTransaction);
  }

private:
  vtkIdType Pack(int*& buffer);
  vtkIdType PackPartial(int*& buffer, int* rows, int nRows);

  int UnPack(int* buffer);
  int UnPackPartial(int* buffer);
  int NProcs;
  int NFragments;
  std::vector<vtkMaterialInterfacePieceTransaction>* Matrix;
  vtkIdType FlatMatrixSize;
  vtkIdType NumberOfTransactions;
};
//
inline std::vector<vtkMaterialInterfacePieceTransaction>&
vtkMaterialInterfacePieceTransactionMatrix::GetTransactions(int fragmentId, int procId)
{
  int idx = fragmentId + procId * this->NFragments;
  return this->Matrix[idx];
}
//
inline vtkIdType vtkMaterialInterfacePieceTransactionMatrix::GetNumberOfTransactions(int procId)
{
  size_t nTransactions = 0;

  for (int fragmentId = 0; fragmentId < this->NFragments; ++fragmentId)
  {
    nTransactions += this->GetTransactions(fragmentId, procId).size();
  }

  return static_cast<vtkIdType>(nTransactions);
}
#endif