Stress.h

Go to the documentation of this file.

/*
 * Stress.h
 *
 *  Created on: Nov 5, 2019
 *      Author: Nikhil
 */

#ifndef STRESS_H_
#define STRESS_H_

#include <vector>
#include "Grid.h"
#include "typedef.h"
#include "SpatialHash.h"
#include "Method.h"
#include <string>
#include <fstream>
#include <iostream>

template<typename TMethod,
         StressType stressType,
         typename TGrid = typename std::conditional<stressType==Piola,Grid<Reference>,Grid<Current>>::type>
class Stress {
public:
    std::vector<Matrix3d> field;
    TGrid* pgrid;
    const Method<TMethod>& method;
    std::string name;

    Stress(std::string name,
           const Method<TMethod>& method,
           TGrid* pgrid): name(name),pgrid(pgrid),method(method)
    {
        field.resize(pgrid->ngrid);
        for(auto& matrix : field)
            matrix= Matrix3d::Zero();
    }
    Stress(const Method<TMethod>& method,
           TGrid* pgrid): pgrid(pgrid),method(method)
    {
        field.resize(pgrid->ngrid);
        for(auto& matrix : field)
            matrix= Matrix3d::Zero();
    }

    void write()
    {
        if (name.empty())
            MY_ERROR("Stress object created without specifying a name. Use write(filename) instead of write()");
        std::ofstream file(name+".stress");

        file << field.size() << "\n";
        file << "\n";
        int index= 0;
        Eigen::IOFormat fmt(Eigen::FullPrecision, 0, "      ", "\n", "", "", "");
        for (auto& stress : field)
        {
            Eigen::Map<Eigen::Matrix<double,1,DIM*DIM>> stressRow(stress.data(), stress.size());
            file << pgrid->coordinates[index].format(fmt) << std::setw(5) << stressRow.format(fmt) << std::endl;
            index++;
        }
    }

    void write(const std::string& filename)
    {
        if (name.empty())
            name= filename;
        else
            std::cout << "Stress object created with name " << name << ". Ignoring the filename: " << filename << "." << std::endl; 
        write();
    }

    ~Stress()
    {
        // TODO Auto-generated destructor stub
    }


};

#endif /* STRESS_H_ */