Lines Matching refs:colour

56   for (PetscInt colour = 0; colour < p; colour++) {  in GenerateSeedMatrix()  local
57 PetscCall(ISGetLocalSize(is[colour], &size)); in GenerateSeedMatrix()
58 PetscCall(ISGetIndices(is[colour], &indices)); in GenerateSeedMatrix()
59 for (PetscInt j = 0; j < size; j++) S[indices[j]][colour] = 1.; in GenerateSeedMatrix()
60 PetscCall(ISRestoreIndices(is[colour], &indices)); in GenerateSeedMatrix()
81 PetscInt p, size, colour, j; in GenerateSeedMatrixPlusRecovery() local
86 for (colour = 0; colour < p; colour++) { in GenerateSeedMatrixPlusRecovery()
87 PetscCall(ISGetLocalSize(is[colour], &size)); in GenerateSeedMatrixPlusRecovery()
88 PetscCall(ISGetIndices(is[colour], &indices)); in GenerateSeedMatrixPlusRecovery()
90 S[indices[j]][colour] = 1.; in GenerateSeedMatrixPlusRecovery()
91 R[indices[j]] = colour; in GenerateSeedMatrixPlusRecovery()
93 PetscCall(ISRestoreIndices(is[colour], &indices)); in GenerateSeedMatrixPlusRecovery()
115 PetscInt i, j, k, colour; in GetRecoveryMatrix() local
119 for (colour = 0; colour < p; colour++) { in GetRecoveryMatrix()
120 R[i][colour] = -1.; in GetRecoveryMatrix()
123 if (S[j][colour] == 1.) { in GetRecoveryMatrix()
124 R[i][colour] = j; in GetRecoveryMatrix()
151 for (PetscInt colour = 0; colour < p; colour++) { in RecoverJacobian() local
152 PetscInt j = (PetscInt)R[i][colour]; in RecoverJacobian()
154 if (a) C[i][colour] *= *a; in RecoverJacobian()
155 PetscCall(MatSetValues(A, 1, &i, 1, &j, &C[i][colour], mode)); in RecoverJacobian()
181 for (PetscInt colour = 0; colour < p; colour++) { in RecoverJacobianLocal() local
182 PetscInt j = (PetscInt)R[i][colour]; in RecoverJacobianLocal()
184 if (a) C[i][colour] *= *a; in RecoverJacobianLocal()
185 PetscCall(MatSetValuesLocal(A, 1, &i, 1, &j, &C[i][colour], mode)); in RecoverJacobianLocal()
209 PetscInt colour = (PetscInt)R[i]; in RecoverDiagonal() local
210 if (a) C[i][colour] *= *a; in RecoverDiagonal()
211 PetscCall(VecSetValues(diag, 1, &i, &C[i][colour], mode)); in RecoverDiagonal()
233 PetscInt colour = (PetscInt)R[i]; in RecoverDiagonalLocal() local
234 if (a) C[i][colour] *= *a; in RecoverDiagonalLocal()
235 PetscCall(VecSetValuesLocal(diag, 1, &i, &C[i][colour], mode)); in RecoverDiagonalLocal()