QCAD::ResponseSaveField<EvalT, Traits>:: ResponseSaveField(Teuchos::ParameterList& p, const Teuchos::RCP<Albany::Layouts>& dl) : weights("Weights", dl->qp_scalar) { //! get and validate Response parameter list Teuchos::ParameterList* plist = p.get<Teuchos::ParameterList*>("Parameter List"); Teuchos::RCP<const Teuchos::ParameterList> reflist = this->getValidResponseParameters(); plist->validateParameters(*reflist,0); //! User-specified parameters if(plist->isParameter("Vector Field Name")) { fieldName = plist->get<std::string>("Vector Field Name"); isVectorField = true; } else { fieldName = plist->get<std::string>("Field Name"); isVectorField = false; } stateName = plist->get<std::string>("State Name", fieldName); outputToExodus = plist->get<bool>("Output to Exodus", true); outputCellAverage = plist->get<bool>("Output Cell Average", true); memoryHolderOnly = plist->get<bool>("Memory Placeholder Only", false); vectorOp = plist->get<std::string>("Vector Operation", "magnitude"); //! number of quad points per cell and dimension Teuchos::RCP<PHX::DataLayout> scalar_dl = dl->qp_scalar; Teuchos::RCP<PHX::DataLayout> vector_dl = dl->qp_vector; Teuchos::RCP<PHX::DataLayout> cell_dl = dl->cell_scalar; numQPs = vector_dl->dimension(1); numDims = vector_dl->dimension(2); //! add dependent fields Teuchos::RCP<PHX::DataLayout>& field_dl = isVectorField ? vector_dl : scalar_dl; PHX::MDField<ScalarT> f(fieldName, field_dl); field = f; this->addDependentField(field); this->addDependentField(weights); //! Register with state manager Albany::StateManager* pStateMgr = p.get< Albany::StateManager* >("State Manager Ptr"); if( outputCellAverage ) { pStateMgr->registerStateVariable(stateName, cell_dl, "ALL", "scalar", 0.0, false, outputToExodus); } else { pStateMgr->registerStateVariable(stateName, scalar_dl, "ALL", "scalar", 0.0, false, outputToExodus); } // Create field tag response_field_tag = Teuchos::rcp(new PHX::Tag<ScalarT>(fieldName + " Save Field Response", dl->dummy)); this->addEvaluatedField(*response_field_tag); }
AAdapt::STKAdapt<SizeField>:: STKAdapt(const Teuchos::RCP<Teuchos::ParameterList>& params_, const Teuchos::RCP<ParamLib>& paramLib_, Albany::StateManager& StateMgr_, const Teuchos::RCP<const Epetra_Comm>& comm_) : AAdapt::AbstractAdapter(params_, paramLib_, StateMgr_, comm_), remeshFileIndex(1) { disc = StateMgr_.getDiscretization(); stk_discretization = static_cast<Albany::STKDiscretization*>(disc.get()); genericMeshStruct = Teuchos::rcp_dynamic_cast<Albany::GenericSTKMeshStruct>(stk_discretization->getSTKMeshStruct()); eMesh = genericMeshStruct->getPerceptMesh(); TEUCHOS_TEST_FOR_EXCEPT(eMesh.is_null()); refinerPattern = genericMeshStruct->getRefinerPattern(); TEUCHOS_TEST_FOR_EXCEPT(refinerPattern.is_null()); num_iterations = adapt_params_->get<int>("Max Number of STK Adapt Iterations", 1); // Save the initial output file name base_exo_filename = stk_discretization->getSTKMeshStruct()->exoOutFile; }
AAdapt::AdaptiveSolutionManagerT::AdaptiveSolutionManagerT( const Teuchos::RCP<Teuchos::ParameterList>& appParams, const Teuchos::RCP<const Tpetra_Vector>& initial_guessT, const Teuchos::RCP<ParamLib>& param_lib, const Albany::StateManager& stateMgr, const Teuchos::RCP<rc::Manager>& rc_mgr, const Teuchos::RCP<const Teuchos_Comm>& commT) : out(Teuchos::VerboseObjectBase::getDefaultOStream()), appParams_(appParams), disc_(stateMgr.getDiscretization()), paramLib_(param_lib), stateMgr_(stateMgr), num_time_deriv(appParams->sublist("Discretization").get<int>("Number Of Time Derivatives")), commT_(commT) { // Create problem PL Teuchos::RCP<Teuchos::ParameterList> problemParams = Teuchos::sublist(appParams_, "Problem", true); // Note that piroParams_ is a member of Thyra_AdaptiveSolutionManager piroParams_ = Teuchos::sublist(appParams_, "Piro", true); if (problemParams->isSublist("Adaptation")) { // If the user has specified adaptation on input, grab the sublist // Note that piroParams_ and adaptiveMesh_ are members of Thyra_AdaptiveSolutionManager adaptParams_ = Teuchos::sublist(problemParams, "Adaptation", true); adaptiveMesh_ = true; buildAdapter(rc_mgr); } // Want the initial time in the parameter library to be correct // if this is a restart solution if (disc_->hasRestartSolution()) { if (paramLib_->isParameter("Time")) { double initialValue = 0.0; if(appParams->sublist("Problem"). get<std::string>("Solution Method", "Steady") == "Continuation") { initialValue = appParams->sublist("Piro").sublist("LOCA").sublist("Stepper"). get<double>("Initial Value", 0.0); } else if(appParams->sublist("Problem"). get<std::string>("Solution Method", "Steady") == "Transient") { initialValue = appParams->sublist("Piro").sublist("Trapezoid Rule"). get<double>("Initial Time", 0.0); } paramLib_->setRealValue<PHAL::AlbanyTraits::Residual>("Time", initialValue); } } const Teuchos::RCP<const Tpetra_Map> mapT = disc_->getMapT(); const Teuchos::RCP<const Tpetra_Map> overlapMapT = disc_->getOverlapMapT(); #ifdef ALBANY_AERAS //IKT, 1/20/15: the following is needed to ensure Laplace matrix is non-diagonal //for Aeras problems that have hyperviscosity and are integrated using an explicit time //integration scheme. const Teuchos::RCP<const Tpetra_CrsGraph> overlapJacGraphT = disc_ ->getImplicitOverlapJacobianGraphT(); #else const Teuchos::RCP<const Tpetra_CrsGraph> overlapJacGraphT = disc_ ->getOverlapJacobianGraphT(); #endif resizeMeshDataArrays(mapT, overlapMapT, overlapJacGraphT); { Teuchos::ArrayRCP< Teuchos::ArrayRCP<Teuchos::ArrayRCP<Teuchos::ArrayRCP<int> > > > wsElNodeEqID = disc_->getWsElNodeEqID(); Teuchos::ArrayRCP<Teuchos::ArrayRCP<Teuchos::ArrayRCP<double*> > > coords = disc_->getCoords(); Teuchos::ArrayRCP<std::string> wsEBNames = disc_->getWsEBNames(); const int numDim = disc_->getNumDim(); const int neq = disc_->getNumEq(); Teuchos::RCP<Teuchos::ParameterList> problemParams = Teuchos::sublist( appParams_, "Problem", true); if (Teuchos::nonnull(initial_guessT)) { *current_soln->getVectorNonConst(0) = *initial_guessT; } else { overlapped_soln->getVectorNonConst(0)->doImport(*current_soln->getVector(0), *importerT, Tpetra::INSERT); AAdapt::InitialConditionsT( overlapped_soln->getVectorNonConst(0), wsElNodeEqID, wsEBNames, coords, neq, numDim, problemParams->sublist("Initial Condition"), disc_->hasRestartSolution()); current_soln->getVectorNonConst(0)->doExport(*overlapped_soln->getVector(0), *exporterT, Tpetra::INSERT); if(num_time_deriv > 0){ overlapped_soln->getVectorNonConst(1)->doImport(*current_soln->getVector(1), *importerT, Tpetra::INSERT); AAdapt::InitialConditionsT( overlapped_soln->getVectorNonConst(1), wsElNodeEqID, wsEBNames, coords, neq, numDim, problemParams->sublist("Initial Condition Dot")); current_soln->getVectorNonConst(1)->doExport(*overlapped_soln->getVector(1), *exporterT, Tpetra::INSERT); } if(num_time_deriv > 1){ overlapped_soln->getVectorNonConst(2)->doImport(*current_soln->getVector(2), *importerT, Tpetra::INSERT); AAdapt::InitialConditionsT( overlapped_soln->getVectorNonConst(2), wsElNodeEqID, wsEBNames, coords, neq, numDim, problemParams->sublist("Initial Condition DotDot")); current_soln->getVectorNonConst(2)->doExport(*overlapped_soln->getVector(2), *exporterT, Tpetra::INSERT); } } } #if (defined(ALBANY_SCOREC) || defined(ALBANY_AMP)) { const Teuchos::RCP< Albany::APFDiscretization > apf_disc = Teuchos::rcp_dynamic_cast< Albany::APFDiscretization >(disc_); if ( ! apf_disc.is_null()) { apf_disc->writeSolutionMVToMeshDatabase(*overlapped_soln, 0, true); apf_disc->initTemperatureHack(); } } #endif }
int main(int ac, char* av[]) { KokkosGuard kokkos(ac, av); typedef PHX::MDField<PHAL::AlbanyTraits::Residual::ScalarT>::size_type size_type; typedef PHAL::AlbanyTraits::Residual Residual; typedef PHAL::AlbanyTraits::Residual::ScalarT ScalarT; typedef PHAL::AlbanyTraits Traits; std::cout.precision(15); // // Create a command line processor and parse command line options // Teuchos::CommandLineProcessor command_line_processor; command_line_processor.setDocString( "Material Point Simulator.\n" "For testing material models in LCM.\n"); std::string input_file = "materials.xml"; command_line_processor.setOption("input", &input_file, "Input File Name"); std::string timing_file = "timing.csv"; command_line_processor.setOption("timing", &timing_file, "Timing File Name"); int workset_size = 1; command_line_processor.setOption("wsize", &workset_size, "Workset Size"); int num_pts = 1; command_line_processor.setOption( "npoints", &num_pts, "Number of Gaussian Points"); size_t memlimit = 1024; // 1GB heap limit by default command_line_processor.setOption( "memlimit", &memlimit, "Heap memory limit in MB for CUDA kernels"); // Throw a warning and not error for unrecognized options command_line_processor.recogniseAllOptions(true); // Don't throw exceptions for errors command_line_processor.throwExceptions(false); // Parse command line Teuchos::CommandLineProcessor::EParseCommandLineReturn parse_return = command_line_processor.parse(ac, av); std::ofstream tout(timing_file.c_str()); if (parse_return == Teuchos::CommandLineProcessor::PARSE_HELP_PRINTED) { return 0; } if (parse_return != Teuchos::CommandLineProcessor::PARSE_SUCCESSFUL) { return 1; } util::TimeMonitor& tmonitor = util::PerformanceContext::instance().timeMonitor(); Teuchos::RCP<Teuchos::Time> total_time = tmonitor["MPS: Total Time"]; Teuchos::RCP<Teuchos::Time> compute_time = tmonitor["MPS: Compute Time"]; // // Process material.xml file // Read into materialDB and get material model name // // A mpi object must be instantiated before using the comm to read // material file Teuchos::GlobalMPISession mpi_session(&ac, &av); Teuchos::RCP<const Teuchos_Comm> commT = Albany::createTeuchosCommFromMpiComm(Albany_MPI_COMM_WORLD); Teuchos::RCP<Albany::MaterialDatabase> material_db; material_db = Teuchos::rcp(new Albany::MaterialDatabase(input_file, commT)); // Get the name of the material model to be used (and make sure there is one) std::string element_block_name = "Block0"; std::string material_model_name; material_model_name = material_db->getElementBlockSublist(element_block_name, "Material Model") .get<std::string>("Model Name"); TEUCHOS_TEST_FOR_EXCEPTION( material_model_name.length() == 0, std::logic_error, "A material model must be defined for block: " + element_block_name); // // Preloading stage setup // set up evaluators, create field and state managers // // Set up the data layout // const int workset_size = 1; const int num_dims = 3; const int num_vertices = 8; const int num_nodes = 8; const Teuchos::RCP<Albany::Layouts> dl = Teuchos::rcp(new Albany::Layouts( workset_size, num_vertices, num_nodes, num_pts, num_dims)); // create field name strings LCM::FieldNameMap field_name_map(false); Teuchos::RCP<std::map<std::string, std::string>> fnm = field_name_map.getMap(); //--------------------------------------------------------------------------- // Deformation gradient // initially set the deformation gradient to the identity Teuchos::ArrayRCP<ScalarT> def_grad(workset_size * num_pts * 9); for (int i = 0; i < workset_size; ++i) { for (int j = 0; j < num_pts; ++j) { int base = i * num_pts * 9 + j * 9; for (int k = 0; k < 9; ++k) def_grad[base + k] = 0.0; def_grad[base + 0] = 1.0; def_grad[base + 4] = 1.0; def_grad[base + 8] = 1.0; } } // SetField evaluator, which will be used to manually assign a value // to the def_grad field Teuchos::ParameterList setDefGradP("SetFieldDefGrad"); setDefGradP.set<std::string>("Evaluated Field Name", "F"); setDefGradP.set<Teuchos::RCP<PHX::DataLayout>>( "Evaluated Field Data Layout", dl->qp_tensor); setDefGradP.set<Teuchos::ArrayRCP<ScalarT>>("Field Values", def_grad); auto setFieldDefGrad = Teuchos::rcp(new LCM::SetField<Residual, Traits>(setDefGradP)); //--------------------------------------------------------------------------- // Det(deformation gradient) Teuchos::ArrayRCP<ScalarT> detdefgrad(workset_size * num_pts); for (int i = 0; i < workset_size * num_pts; ++i) detdefgrad[i] = 1.0; // SetField evaluator, which will be used to manually assign a value // to the detdefgrad field Teuchos::ParameterList setDetDefGradP("SetFieldDetDefGrad"); setDetDefGradP.set<std::string>("Evaluated Field Name", "J"); setDetDefGradP.set<Teuchos::RCP<PHX::DataLayout>>( "Evaluated Field Data Layout", dl->qp_scalar); setDetDefGradP.set<Teuchos::ArrayRCP<ScalarT>>("Field Values", detdefgrad); auto setFieldDetDefGrad = Teuchos::rcp(new LCM::SetField<Residual, Traits>(setDetDefGradP)); //--------------------------------------------------------------------------- // Small strain tensor // initially set the strain tensor to zeros Teuchos::ArrayRCP<ScalarT> strain(workset_size * num_pts * 9); for (int i = 0; i < workset_size; ++i) { for (int j = 0; j < num_pts; ++j) { int base = i * num_pts * 9 + j * 9; for (int k = 0; k < 9; ++k) strain[base + k] = 0.0; } } // SetField evaluator, which will be used to manually assign a value // to the strain field Teuchos::ParameterList setStrainP("SetFieldStrain"); setStrainP.set<std::string>("Evaluated Field Name", "Strain"); setStrainP.set<Teuchos::RCP<PHX::DataLayout>>( "Evaluated Field Data Layout", dl->qp_tensor); setStrainP.set<Teuchos::ArrayRCP<ScalarT>>("Field Values", strain); auto setFieldStrain = Teuchos::rcp(new LCM::SetField<Residual, Traits>(setStrainP)); //--------------------------------------------------------------------------- // Instantiate a field manager PHX::FieldManager<Traits> fieldManager; // Instantiate a field manager for States PHX::FieldManager<Traits> stateFieldManager; // Register the evaluators with the field manager fieldManager.registerEvaluator<Residual>(setFieldDefGrad); fieldManager.registerEvaluator<Residual>(setFieldDetDefGrad); fieldManager.registerEvaluator<Residual>(setFieldStrain); // Register the evaluators with the state field manager stateFieldManager.registerEvaluator<Residual>(setFieldDefGrad); stateFieldManager.registerEvaluator<Residual>(setFieldDetDefGrad); stateFieldManager.registerEvaluator<Residual>(setFieldStrain); // Instantiate a state manager Albany::StateManager stateMgr; // extract the Material ParameterList for use below std::string matName = material_db->getElementBlockParam<std::string>( element_block_name, "material"); Teuchos::ParameterList& paramList = material_db->getElementBlockSublist(element_block_name, matName); Teuchos::ParameterList& mpsParams = paramList.sublist("Material Point Simulator"); // Get loading parameters from .xml file std::string load_case = mpsParams.get<std::string>("Loading Case Name", "uniaxial"); int number_steps = mpsParams.get<int>("Number of Steps", 10); double step_size = mpsParams.get<double>("Step Size", 1.0e-2); std::cout << "Loading parameters:" << "\n number of steps: " << number_steps << "\n step_size : " << step_size << std::endl; // determine if temperature is being used bool have_temperature = mpsParams.get<bool>("Use Temperature", false); std::cout << "have_temp: " << have_temperature << std::endl; //--------------------------------------------------------------------------- // Temperature (optional) if (have_temperature) { Teuchos::ArrayRCP<ScalarT> temperature(workset_size); ScalarT temp = mpsParams.get<double>("Temperature", 1.0); for (int i = 0; i < workset_size * num_pts; ++i) temperature[0] = temp; // SetField evaluator, which will be used to manually assign a value // to the detdefgrad field Teuchos::ParameterList setTempP("SetFieldTemperature"); setTempP.set<std::string>("Evaluated Field Name", "Temperature"); setTempP.set<Teuchos::RCP<PHX::DataLayout>>( "Evaluated Field Data Layout", dl->qp_scalar); setTempP.set<Teuchos::ArrayRCP<ScalarT>>("Field Values", temperature); auto setFieldTemperature = Teuchos::rcp(new LCM::SetField<Residual, Traits>(setTempP)); fieldManager.registerEvaluator<Residual>(setFieldTemperature); stateFieldManager.registerEvaluator<Residual>(setFieldTemperature); } //--------------------------------------------------------------------------- // Time step Teuchos::ArrayRCP<ScalarT> delta_time(1); delta_time[0] = step_size; Teuchos::ParameterList setDTP("SetFieldTimeStep"); setDTP.set<std::string>("Evaluated Field Name", "Delta Time"); setDTP.set<Teuchos::RCP<PHX::DataLayout>>( "Evaluated Field Data Layout", dl->workset_scalar); setDTP.set<Teuchos::ArrayRCP<ScalarT>>("Field Values", delta_time); auto setFieldDT = Teuchos::rcp(new LCM::SetField<Residual, Traits>(setDTP)); fieldManager.registerEvaluator<Residual>(setFieldDT); stateFieldManager.registerEvaluator<Residual>(setFieldDT); // check if the material wants the tangent to be computed bool check_stability; check_stability = mpsParams.get<bool>("Check Stability", false); paramList.set<bool>("Compute Tangent", check_stability); std::cout << "Check stability = " << check_stability << std::endl; //--------------------------------------------------------------------------- // std::cout << "// Constitutive Model Parameters" //<< std::endl; Teuchos::ParameterList cmpPL; paramList.set<Teuchos::RCP<std::map<std::string, std::string>>>( "Name Map", fnm); cmpPL.set<Teuchos::ParameterList*>("Material Parameters", ¶mList); if (have_temperature) { cmpPL.set<std::string>("Temperature Name", "Temperature"); paramList.set<bool>("Have Temperature", true); } auto CMP = Teuchos::rcp( new LCM::ConstitutiveModelParameters<Residual, Traits>(cmpPL, dl)); fieldManager.registerEvaluator<Residual>(CMP); stateFieldManager.registerEvaluator<Residual>(CMP); //--------------------------------------------------------------------------- // std::cout << "// Constitutive Model Interface Evaluator" // << std::endl; Teuchos::ParameterList cmiPL; cmiPL.set<Teuchos::ParameterList*>("Material Parameters", ¶mList); if (have_temperature) { cmiPL.set<std::string>("Temperature Name", "Temperature"); } Teuchos::RCP<LCM::ConstitutiveModelInterface<Residual, Traits>> CMI = Teuchos::rcp( new LCM::ConstitutiveModelInterface<Residual, Traits>(cmiPL, dl)); fieldManager.registerEvaluator<Residual>(CMI); stateFieldManager.registerEvaluator<Residual>(CMI); // Set the evaluated fields as required for (std::vector<Teuchos::RCP<PHX::FieldTag>>::const_iterator it = CMI->evaluatedFields().begin(); it != CMI->evaluatedFields().end(); ++it) { fieldManager.requireField<Residual>(**it); } // register state variables Teuchos::RCP<Teuchos::ParameterList> p; Teuchos::RCP<PHX::Evaluator<Traits>> ev; for (int sv(0); sv < CMI->getNumStateVars(); ++sv) { CMI->fillStateVariableStruct(sv); p = stateMgr.registerStateVariable( CMI->getName(), CMI->getLayout(), dl->dummy, element_block_name, CMI->getInitType(), CMI->getInitValue(), CMI->getStateFlag(), CMI->getOutputFlag()); ev = Teuchos::rcp(new PHAL::SaveStateField<Residual, Traits>(*p)); fieldManager.registerEvaluator<Residual>(ev); stateFieldManager.registerEvaluator<Residual>(ev); } //--------------------------------------------------------------------------- if (check_stability) { std::string parametrization_type = mpsParams.get<std::string>("Parametrization Type", "Spherical"); double parametrization_interval = mpsParams.get<double>("Parametrization Interval", 0.05); std::cout << "Bifurcation Check in Material Point Simulator:" << std::endl; std::cout << "Parametrization Type: " << parametrization_type << std::endl; Teuchos::ParameterList bcPL; bcPL.set<Teuchos::ParameterList*>("Material Parameters", ¶mList); bcPL.set<std::string>("Parametrization Type Name", parametrization_type); bcPL.set<double>("Parametrization Interval Name", parametrization_interval); bcPL.set<std::string>("Material Tangent Name", "Material Tangent"); bcPL.set<std::string>("Ellipticity Flag Name", "Ellipticity_Flag"); bcPL.set<std::string>("Bifurcation Direction Name", "Direction"); bcPL.set<std::string>("Min detA Name", "Min detA"); Teuchos::RCP<LCM::BifurcationCheck<Residual, Traits>> BC = Teuchos::rcp(new LCM::BifurcationCheck<Residual, Traits>(bcPL, dl)); fieldManager.registerEvaluator<Residual>(BC); stateFieldManager.registerEvaluator<Residual>(BC); // register the ellipticity flag p = stateMgr.registerStateVariable( "Ellipticity_Flag", dl->qp_scalar, dl->dummy, element_block_name, "scalar", 0.0, false, true); ev = Teuchos::rcp(new PHAL::SaveStateField<Residual, Traits>(*p)); fieldManager.registerEvaluator<Residual>(ev); stateFieldManager.registerEvaluator<Residual>(ev); // register the direction p = stateMgr.registerStateVariable( "Direction", dl->qp_vector, dl->dummy, element_block_name, "scalar", 0.0, false, true); ev = Teuchos::rcp(new PHAL::SaveStateField<Residual, Traits>(*p)); fieldManager.registerEvaluator<Residual>(ev); stateFieldManager.registerEvaluator<Residual>(ev); // register min(det(A)) p = stateMgr.registerStateVariable( "Min detA", dl->qp_scalar, dl->dummy, element_block_name, "scalar", 0.0, false, true); ev = Teuchos::rcp(new PHAL::SaveStateField<Residual, Traits>(*p)); fieldManager.registerEvaluator<Residual>(ev); stateFieldManager.registerEvaluator<Residual>(ev); } //--------------------------------------------------------------------------- // std::cout << "// register deformation gradient" // << std::endl; p = stateMgr.registerStateVariable( "F", dl->qp_tensor, dl->dummy, element_block_name, "identity", 1.0, true, true); ev = Teuchos::rcp(new PHAL::SaveStateField<Residual, Traits>(*p)); fieldManager.registerEvaluator<Residual>(ev); stateFieldManager.registerEvaluator<Residual>(ev); //--------------------------------------------------------------------------- // std::cout << "// register small strain tensor" // << std::endl; p = stateMgr.registerStateVariable( "Strain", dl->qp_tensor, dl->dummy, element_block_name, "scalar", 0.0, false, true); ev = Teuchos::rcp(new PHAL::SaveStateField<Residual, Traits>(*p)); fieldManager.registerEvaluator<Residual>(ev); stateFieldManager.registerEvaluator<Residual>(ev); //--------------------------------------------------------------------------- // Traits::SetupData setupData = "Test String"; // std::cout << "Calling postRegistrationSetup" << std::endl; fieldManager.postRegistrationSetup(setupData); // std::cout << "// set the required fields for the state manager" //<< std::endl; Teuchos::RCP<PHX::DataLayout> dummy = Teuchos::rcp(new PHX::MDALayout<Dummy>(0)); std::vector<std::string> responseIDs = stateMgr.getResidResponseIDsToRequire(element_block_name); std::vector<std::string>::const_iterator it; for (it = responseIDs.begin(); it != responseIDs.end(); it++) { const std::string& responseID = *it; PHX::Tag<PHAL::AlbanyTraits::Residual::ScalarT> res_response_tag( responseID, dummy); stateFieldManager.requireField<PHAL::AlbanyTraits::Residual>( res_response_tag); } stateFieldManager.postRegistrationSetup(""); // std::cout << "Process using 'dot -Tpng -O <name>'\n"; fieldManager.writeGraphvizFile<Residual>("FM", true, true); stateFieldManager.writeGraphvizFile<Residual>("SFM", true, true); //--------------------------------------------------------------------------- // grab the output file name // std::string output_file = mpsParams.get<std::string>("Output File Name", "output.exo"); //--------------------------------------------------------------------------- // Create discretization, as required by the StateManager // Teuchos::RCP<Teuchos::ParameterList> discretizationParameterList = Teuchos::rcp(new Teuchos::ParameterList("Discretization")); discretizationParameterList->set<int>("1D Elements", workset_size); discretizationParameterList->set<int>("2D Elements", 1); discretizationParameterList->set<int>("3D Elements", 1); discretizationParameterList->set<std::string>("Method", "STK3D"); discretizationParameterList->set<int>("Number Of Time Derivatives", 0); discretizationParameterList->set<std::string>( "Exodus Output File Name", output_file); discretizationParameterList->set<int>("Workset Size", workset_size); Teuchos::RCP<Tpetra_Map> mapT = Teuchos::rcp(new Tpetra_Map( workset_size * num_dims * num_nodes, 0, commT, Tpetra::LocallyReplicated)); Teuchos::RCP<Tpetra_Vector> solution_vectorT = Teuchos::rcp(new Tpetra_Vector(mapT)); int numberOfEquations = 3; Albany::AbstractFieldContainer::FieldContainerRequirements req; Teuchos::RCP<Albany::AbstractSTKMeshStruct> stkMeshStruct = Teuchos::rcp(new Albany::TmplSTKMeshStruct<3>( discretizationParameterList, Teuchos::null, commT)); stkMeshStruct->setFieldAndBulkData( commT, discretizationParameterList, numberOfEquations, req, stateMgr.getStateInfoStruct(), stkMeshStruct->getMeshSpecs()[0]->worksetSize); Teuchos::RCP<Albany::AbstractDiscretization> discretization = Teuchos::rcp(new Albany::STKDiscretization( discretizationParameterList, stkMeshStruct, commT)); //--------------------------------------------------------------------------- // Associate the discretization with the StateManager // stateMgr.setupStateArrays(discretization); //--------------------------------------------------------------------------- // Create a workset // PHAL::Workset workset; workset.numCells = workset_size; workset.stateArrayPtr = &stateMgr.getStateArray(Albany::StateManager::ELEM, 0); // create MDFields PHX::MDField<ScalarT, Cell, QuadPoint, Dim, Dim> stressField( "Cauchy_Stress", dl->qp_tensor); // construct the final deformation gradient based on the loading case std::vector<ScalarT> F_vector(9, 0.0); if (load_case == "uniaxial") { F_vector[0] = 1.0 + number_steps * step_size; F_vector[4] = 1.0; F_vector[8] = 1.0; } else if (load_case == "simple-shear") { F_vector[0] = 1.0; F_vector[1] = number_steps * step_size; F_vector[4] = 1.0; F_vector[8] = 1.0; } else if (load_case == "hydrostatic") { F_vector[0] = 1.0 + number_steps * step_size; F_vector[4] = 1.0 + number_steps * step_size; F_vector[8] = 1.0 + number_steps * step_size; } else if (load_case == "general") { F_vector = mpsParams.get<Teuchos::Array<double>>("Deformation Gradient Components") .toVector(); } else { TEUCHOS_TEST_FOR_EXCEPTION( true, std::runtime_error, "Improper Loading Case in Material Point Simulator block"); } minitensor::Tensor<ScalarT> F_tensor(3, &F_vector[0]); minitensor::Tensor<ScalarT> log_F_tensor = minitensor::log(F_tensor); std::cout << "F\n" << F_tensor << std::endl; // std::cout << "log F\n" << log_F_tensor << std::endl; // // Setup loading scenario and instantiate evaluatFields // PHX::MDField<ScalarT, Cell, QuadPoint> minDetA("Min detA", dl->qp_scalar); PHX::MDField<ScalarT, Cell, QuadPoint, Dim> direction( "Direction", dl->qp_vector); // Bifurcation check parameters double mu_0 = 0; double mu_k = 0; int bifurcationTime_rough = number_steps; bool bifurcation_flag = false; for (int istep(0); istep <= number_steps; ++istep) { util::TimeGuard total_time_guard(total_time); // std::cout << "****** in MPS step " << istep << " ****** " << std::endl; // alpha \in [0,1] double alpha = double(istep) / number_steps; // std::cout << "alpha: " << alpha << std::endl; minitensor::Tensor<ScalarT> scaled_log_F_tensor = alpha * log_F_tensor; minitensor::Tensor<ScalarT> current_F = minitensor::exp(scaled_log_F_tensor); // std::cout << "scaled log F\n" << scaled_log_F_tensor << std::endl; // std::cout << "current F\n" << current_F << std::endl; for (int i = 0; i < 3; ++i) { for (int j = 0; j < 3; ++j) { def_grad[3 * i + j] = current_F(i, j); } } // jacobian detdefgrad[0] = minitensor::det(current_F); // small strain tensor minitensor::Tensor<ScalarT> current_strain; current_strain = 0.5 * (current_F + minitensor::transpose(current_F)) - minitensor::eye<ScalarT>(3); for (int i = 0; i < 3; ++i) { for (int j = 0; j < 3; ++j) { strain[3 * i + j] = current_strain(i, j); } } // std::cout << "current strain\n" << current_strain << std::endl; // Call the evaluators, evaluateFields() is the function that // computes stress based on deformation gradient compute_time->start(); fieldManager.preEvaluate<Residual>(workset); fieldManager.evaluateFields<Residual>(workset); fieldManager.postEvaluate<Residual>(workset); compute_time->stop(); stateFieldManager.getFieldData<Residual>(stressField); // Call the state field manager // std::cout << "+++ calling the stateFieldManager\n"; compute_time->start(); stateFieldManager.preEvaluate<Residual>(workset); stateFieldManager.evaluateFields<Residual>(workset); stateFieldManager.postEvaluate<Residual>(workset); compute_time->stop(); stateMgr.updateStates(); // output to the exodus file // Don't include this in timing data... total_time->stop(); discretization->writeSolutionT( *solution_vectorT, Teuchos::as<double>(istep)); // if check for bifurcation, adaptive step total_time->start(); if (check_stability) { // get current minDet(A) stateFieldManager.getFieldData<Residual>(minDetA); if (istep == 0) { mu_0 = minDetA(0, 0); } if (minDetA(0, 0) <= 0 && !bifurcation_flag) { mu_k = minDetA(0, 0); bifurcationTime_rough = istep; bifurcation_flag = true; // adaptive step begin std::cout << "\nAdaptive step begin - step " << istep << std::endl; // initialization for adaptive step double tol = 1E-8; double alpha_local = 1.0; double alpha_local_step = 0.5; int k = 1; int maxIteration = 50; // small strain tensor minitensor::Tensor<ScalarT> current_strain; // iteration begin while (((mu_k <= 0) || (std::abs(mu_k / mu_0) > tol))) { alpha = double(bifurcationTime_rough - 1 + alpha_local) / number_steps; minitensor::Tensor<ScalarT> scaled_log_F_tensor = alpha * log_F_tensor; minitensor::Tensor<ScalarT> current_F = minitensor::exp(scaled_log_F_tensor); for (int i = 0; i < 3; ++i) { for (int j = 0; j < 3; ++j) { def_grad[3 * i + j] = current_F(i, j); } } // jacobian detdefgrad[0] = minitensor::det(current_F); current_strain = 0.5 * (current_F + minitensor::transpose(current_F)) - minitensor::eye<ScalarT>(3); for (int i = 0; i < 3; ++i) { for (int j = 0; j < 3; ++j) { strain[3 * i + j] = current_strain(i, j); } } // Call the evaluators, evaluateFields() is the function that // computes stress based on deformation gradient fieldManager.preEvaluate<Residual>(workset); fieldManager.evaluateFields<Residual>(workset); fieldManager.postEvaluate<Residual>(workset); // Call the state field manager // std::cout << "+++ calling the stateFieldManager\n"; stateFieldManager.preEvaluate<Residual>(workset); stateFieldManager.evaluateFields<Residual>(workset); stateFieldManager.postEvaluate<Residual>(workset); stateFieldManager.getFieldData<Residual>(minDetA); stateFieldManager.getFieldData<Residual>(direction); mu_k = minDetA(0, 0); if (mu_k > 0) { alpha_local += alpha_local_step; } else { alpha_local -= alpha_local_step; } alpha_local_step /= 2; k = k + 1; if (k >= maxIteration) { std::cout << "Adaptive step for bifurcation check not converging after " << k << " iterations" << std::endl; break; } } // adaptive step iteration end } // end adaptive step } // end check bifurcation stateMgr.updateStates(); // if (bifurcation_flag) { // break the loading step after adaptive time step loop break; } // } // end loading steps // Summarize with AlbanyUtil performance monitors if (tout) { util::PerformanceContext::instance().timeMonitor().summarize(tout); tout.close(); } }