Lines Matching refs:Ns

834 PetscErrorCode DMSwarmGetNumSpecies(DM sw, PetscInt *Ns)  in DMSwarmGetNumSpecies()  argument
839 *Ns = swarm->Ns; in DMSwarmGetNumSpecies()
856 PetscErrorCode DMSwarmSetNumSpecies(DM sw, PetscInt Ns) in DMSwarmSetNumSpecies() argument
861 swarm->Ns = Ns; in DMSwarmSetNumSpecies()
991 PetscInt Ns, cStart, cEnd, c, dim, d, Nq, q, Np = 0, p, s; in DMSwarmComputeLocalSize() local
996 PetscCall(DMSwarmGetNumSpecies(sw, &Ns)); in DMSwarmComputeLocalSize()
1006 PetscCall(PetscCalloc2(Ns, &n_int, (cEnd - cStart) * Ns, &npc_s)); in DMSwarmComputeLocalSize()
1015 PetscCall(PetscArrayzero(n_int, Ns)); in DMSwarmComputeLocalSize()
1021 for (s = 0; s < Ns; ++s) { in DMSwarmComputeLocalSize()
1023 n_int[s] += (detJp * den) * (detJ * wq[q]) / (PetscReal)Ns; in DMSwarmComputeLocalSize()
1026 for (s = 0; s < Ns; ++s) { in DMSwarmComputeLocalSize()
1028 npc_s[c * Ns + s] += (PetscInt)(Ni * n_int[s] + 0.5); // TODO Wish we wrapped round() in DMSwarmComputeLocalSize()
1029 Np += npc_s[c * Ns + s]; in DMSwarmComputeLocalSize()
1038 for (s = 0; s < Ns; ++s) { in DMSwarmComputeLocalSize()
1039 for (q = 0; q < npc_s[c * Ns + s]; ++q, ++p) swarm_cellid[p] = c; in DMSwarmComputeLocalSize()
1064 PetscInt *N, Ns, dim, n; in DMSwarmComputeLocalSizeFromOptions() local
1075 PetscCall(DMSwarmGetNumSpecies(sw, &Ns)); in DMSwarmComputeLocalSizeFromOptions()
1076 …tionsInt("-dm_swarm_num_species", "The number of species", "DMSwarmSetNumSpecies", Ns, &Ns, &flg)); in DMSwarmComputeLocalSizeFromOptions()
1077 if (flg) PetscCall(DMSwarmSetNumSpecies(sw, Ns)); in DMSwarmComputeLocalSizeFromOptions()
1083 PetscCall(DMSwarmGetNumSpecies(sw, &Ns)); in DMSwarmComputeLocalSizeFromOptions()
1086 PetscCall(DMSwarmGetNumSpecies(sw, &Ns)); in DMSwarmComputeLocalSizeFromOptions()
1087 PetscCall(DMSwarmSetLocalSizes(sw, N[rank] * Ns, 0)); in DMSwarmComputeLocalSizeFromOptions()
1124 PetscInt Nfc, Np, p, Ns, dim, d, bs; in DMSwarmInitializeCoordinates() local
1130 PetscCall(DMSwarmGetNumSpecies(sw, &Ns)); in DMSwarmInitializeCoordinates()
1148 species[p] = p % Ns; in DMSwarmInitializeCoordinates()
1180 species[p] = p % Ns; in DMSwarmInitializeCoordinates()