pGenDeck.h

Go to the documentation of this file.

/*
 * -------------------------------------------
 * Copyright (c) 2021 - 2026 Prashant K. Jha
 * -------------------------------------------
 * PeriDEM https://github.com/prashjha/PeriDEM
 *
 * Distributed under the Boost Software License, Version 1.0. (See accompanying
 * file LICENSE)
 */
 
#ifndef INP_PGENDECK_H
#define INP_PGENDECK_H
 
#include "util/io.h"
#include "util/json.h"
#include "geom/geomIncludes.h"
#include <string>
 
namespace inp {
 
  struct PGenDeck {
 
    std::string d_genMethod;
 
    json d_pGenJson;
 
    bool d_genWithRandomRotation;
 
    PGenDeck(const json &j = json({}))
      : d_genMethod("Use_Particle_Geometry"),
        d_pGenJson({}), d_genWithRandomRotation(true) {
      readFromJson(j);
    };
 
    PGenDeck(std::string genMethod, json pGenJson = json({}), bool genWithRandomRotation = true)
        : d_genMethod(genMethod),
          d_pGenJson(pGenJson), d_genWithRandomRotation(genWithRandomRotation) {};
 
    static json getExampleJson(std::string genMethod = "From_File") {
 
      auto j = json({{"Method", "From_File"}, {"Random_Rotation", true}});
      return j;
    }
 
    void readFromJson(const json &j) {
      if (j.empty())
        return;
 
      d_genMethod = j.value("Method", std::string("Use_Particle_Geometry"));
      d_genWithRandomRotation = j.value("Random_Rotation", true);
 
      if (d_genMethod == "From_File") {
        if (j.find("Data") == j.end())
          throw std::runtime_error("Need information inside key Data.");
 
        d_pGenJson = j.at("Data");
      }
    }
 
    std::string printStr(int nt = 0, int lvl = 0) const {
 
      auto tabS = util::io::getTabS(nt);
      std::ostringstream oss;
      oss << tabS << "------- PGenDeck --------" << std::endl << std::endl;
      oss << tabS << "Method = " << d_genMethod << std::endl;
      oss << tabS << "Random rotation of particles = " << d_genWithRandomRotation << std::endl;
      size_t nParticles = d_pGenJson.value("N", 0);
      oss << tabS << "Number of particles in json object = " << nParticles << std::endl;
      oss << tabS << "Data for first five particles: " << std::endl;
      if (nParticles > 0) {
        nParticles = nParticles > 5 ? 5 : nParticles;
        for (size_t i=0; i< nParticles; i++) {
          oss << tabS << "Particle number = " << i << std::endl;
          oss << tabS << d_pGenJson.at(std::to_string(i)) << std::endl;
        }
      }
      oss << tabS << std::endl;
 
      return oss.str();
    }
 
    void print(int nt = 0, int lvl = 0) const { std::cout << printStr(nt, lvl); }
  };
 
} // namespace inp
 
#endif // INP_PGENDECK_H