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