aij/seq/seqhashmatsetvalues.h

26cec326SBarry Smith/*
26cec326SBarry Smith   used by SEQAIJ, BAIJ and SBAIJ to reduce code duplication
26cec326SBarry Smith
26cec326SBarry Smith     define TYPE to AIJ BAIJ or SBAIJ
26cec326SBarry Smith            TYPE_BS_ON for BAIJ and SBAIJ and bs > 1
26cec326SBarry Smith            TYPE_SBAIJ for SBAIJ
26cec326SBarry Smith
26cec326SBarry Smith*/
26cec326SBarry Smithstatic PetscErrorCode PetscConcat(MatSetValues_Seq_Hash, TYPE_BS)(Mat A, PetscInt m, const PetscInt *rows, PetscInt n, const PetscInt *cols, const PetscScalar *values, InsertMode addv)
26cec326SBarry Smith{
26cec326SBarry Smith  PetscConcat(Mat_Seq, TYPE) *a = (PetscConcat(Mat_Seq, TYPE) *)A->data;
26cec326SBarry Smith#if defined(TYPE_BS_ON)
26cec326SBarry Smith  PetscInt bs;
26cec326SBarry Smith#endif
26cec326SBarry Smith
26cec326SBarry Smith  PetscFunctionBegin;
26cec326SBarry Smith#if defined(TYPE_BS_ON)
26cec326SBarry Smith  PetscCall(MatGetBlockSize(A, &bs));
26cec326SBarry Smith#endif
26cec326SBarry Smith  for (PetscInt r = 0; r < m; ++r) {
26cec326SBarry Smith    PetscHashIJKey key;
26cec326SBarry Smith    PetscBool      missing;
26cec326SBarry Smith    PetscScalar    value;
26cec326SBarry Smith#if defined(TYPE_BS_ON)
26cec326SBarry Smith    PetscHashIJKey bkey;
26cec326SBarry Smith#endif
26cec326SBarry Smith
26cec326SBarry Smith    key.i = rows[r];
26cec326SBarry Smith#if defined(TYPE_BS_ON)
26cec326SBarry Smith    bkey.i = key.i / bs;
26cec326SBarry Smith#endif
26cec326SBarry Smith    if (key.i < 0) continue;
26cec326SBarry Smith    for (PetscInt c = 0; c < n; ++c) {
26cec326SBarry Smith      key.j = cols[c];
26cec326SBarry Smith#if defined(TYPE_BS_ON)
26cec326SBarry Smith      bkey.j = key.j / bs;
26cec326SBarry Smith  #if defined(TYPE_SBAIJ)
26cec326SBarry Smith      if (bkey.j < bkey.i) continue;
26cec326SBarry Smith  #else
26cec326SBarry Smith      if (key.j < 0) continue;
26cec326SBarry Smith  #endif
26cec326SBarry Smith#else
26cec326SBarry Smith  #if defined(TYPE_SBAIJ)
26cec326SBarry Smith      if (key.j < key.i) continue;
26cec326SBarry Smith  #else
26cec326SBarry Smith      if (key.j < 0) continue;
26cec326SBarry Smith  #endif
26cec326SBarry Smith#endif
*f4f49eeaSPierre Jolivet      value = values ? (a->roworiented ? values[r * n + c] : values[r + m * c]) : 0;
26cec326SBarry Smith      switch (addv) {
26cec326SBarry Smith      case INSERT_VALUES:
26cec326SBarry Smith        PetscCall(PetscHMapIJVQuerySet(a->ht, key, value, &missing));
26cec326SBarry Smith        break;
26cec326SBarry Smith      case ADD_VALUES:
26cec326SBarry Smith        PetscCall(PetscHMapIJVQueryAdd(a->ht, key, value, &missing));
26cec326SBarry Smith        break;
26cec326SBarry Smith      default:
26cec326SBarry Smith        SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "InsertMode not supported");
26cec326SBarry Smith      }
26cec326SBarry Smith      if (missing) ++a->dnz[key.i];
26cec326SBarry Smith#if defined(TYPE_BS_ON)
26cec326SBarry Smith      PetscCall(PetscHSetIJQueryAdd(a->bht, bkey, &missing));
26cec326SBarry Smith      if (missing) ++a->bdnz[bkey.i];
26cec326SBarry Smith#endif
26cec326SBarry Smith    }
26cec326SBarry Smith  }
26cec326SBarry Smith  PetscFunctionReturn(PETSC_SUCCESS);
26cec326SBarry Smith}