Lines Matching refs:printCtx

373   PrintCtx  *printCtx;  in PostStep()  local
378 PetscCall(TSGetApplicationContext(ts, &printCtx)); in PostStep()
379 if (!printCtx->print && !printCtx->print_entropy) PetscFunctionReturn(PETSC_SUCCESS); in PostStep()
380 ctx = printCtx->ctx; in PostStep()
381 …if (printCtx->v_target < printCtx->global_vertex_id_0 || printCtx->v_target >= printCtx->global_ve… in PostStep()
384 v_id = printCtx->v_target % ctx->batch_sz; in PostStep()
391 PetscCall(DMCompositeGetAccessArray(pack, X, nDMs, NULL, printCtx->globXArray)); in PostStep()
392 if (printCtx->print_entropy && printCtx->v_target >= 0 && 0) { in PostStep()
396 DM sw = printCtx->globSwarmArray[LAND_PACK_IDX(v_id, grid)]; in PostStep()
397 Vec work, subX = printCtx->globXArray[LAND_PACK_IDX(v_id, grid)]; in PostStep()
401 …PetscCall(gridToParticles(printCtx->grid_dm[grid], sw, subX, work, printCtx->globMpArray[LAND_PACK… in PostStep()
423 DM sw = printCtx->globSwarmArray[LAND_PACK_IDX(v_id, grid)]; in PostStep()
439 if (printCtx->print && printCtx->g_target >= 0) { in PostStep()
440 PetscInt grid = printCtx->g_target, id; in PostStep()
447 …PetscCall(VecViewFromOptions(printCtx->globXArray[LAND_PACK_IDX(v_id % ctx->batch_sz, grid)], NULL… in PostStep()
450 …PetscCall(VecViewFromOptions(printCtx->globXArray[LAND_PACK_IDX(v_id % ctx->batch_sz, grid + 1)], … in PostStep()
455 PetscCall(DMCompositeRestoreAccessArray(pack, X, nDMs, NULL, printCtx->globXArray)); in PostStep()
475 PrintCtx *printCtx; in go() local
489 PetscCall(PetscNew(&printCtx)); in go()
490 PetscCall(TSSetApplicationContext(ts, printCtx)); in go()
491 printCtx->v_target = v_target; in go()
492 printCtx->g_target = g_target; in go()
493 printCtx->ctx = ctx; in go()
494 printCtx->globSwarmArray = globSwarmArray; in go()
495 printCtx->grid_dm = grid_dm; in go()
496 printCtx->globMpArray = globMpArray; in go()
497 printCtx->g_Mass = g_Mass; in go()
498 printCtx->globXArray = globXArray; in go()
499 printCtx->print_entropy = PETSC_FALSE; in go()
501 …, "Print entropy and moments at each time step", "ex30.c", printCtx->print_entropy, &printCtx->pri… in go()
539 printCtx->global_vertex_id_0 = global_vertex_id_0; in go()
542 printCtx->print = PETSC_TRUE; in go()
543 } else printCtx->print = PETSC_FALSE; in go()
774 …if (v_target >= global_vertex_id_0 && v_target < global_vertex_id_0 + ctx->batch_sz && printCtx->p… in go()
996 …if (v_target >= global_vertex_id_0 && v_target < global_vertex_id_0 + ctx->batch_sz && printCtx->p… in go()
1026 PetscCall(PetscFree(printCtx)); in go()