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 "deckField.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 const std::vector<Field<PGenDeck>> &fields() {
      static const std::vector<Field<PGenDeck>> f = {
          field(&PGenDeck::d_genMethod, "Method", std::string("From_File"),
                "Where the particle positions come from",
                {{"From_File", "Use_Particle_Geometry"}, {}, {}}),
          field(&PGenDeck::d_genWithRandomRotation, "Random_Rotation", true,
                "Give each particle a random orientation when none is given"),
      };
      return f;
    }
 
    static json getExampleJson(const json &given = json::object()) {
      return applyGiven(given, fields(), {"Data"});
    }
 
    static json getExampleJson(const std::string &) = delete;
    static json getExampleJson(const char *) = delete;
 
    void readFromJson(const json &j) {
      if (j.empty())
        return;
      readFields(*this, j, fields());
 
      // The positions themselves are a block of their own, not a field.
      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");
      }
    }
 
    json writeToJson() const {
      json j = json::object();
      writeFields(*this, j, fields());
      if (d_genMethod == "From_File")
        j["Data"] = d_pGenJson;
      return j;
    }
 
    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;
      printFields(*this, oss, fields(), tabS);
      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