coreTest_Module.h

// Copyright 2009-2025 NTESS. Under the terms
// of Contract DE-NA0003525 with NTESS, the U.S.
// Government retains certain rights in this software.
//
// Copyright (c) 2009-2025, NTESS
// All rights reserved.
//
// This file is part of the SST software package. For license
// information, see the LICENSE file in the top level directory of the
// distribution.

#ifndef SST_CORE_CORETEST_MODULE_H
#define SST_CORE_CORETEST_MODULE_H

#include "sst/core/component.h"
#include "sst/core/link.h"
#include "sst/core/module.h"
#include "sst/core/rng/rng.h"

#include <cstdint>
#include <string>
#include <vector>

using namespace SST;
using namespace SST::RNG;

namespace SST::CoreTestModule {

class CoreTestModuleExample : public SST::Module
{

public:
    SST_ELI_REGISTER_MODULE_API(SST::CoreTestModule::CoreTestModuleExample) // API & module in one

    SST_ELI_REGISTER_MODULE(
        CoreTestModuleExample, "coreTestElement", "CoreTestModule", SST_ELI_ELEMENT_VERSION(1, 0, 0),
        "CoreTest module to demonstrate interface.", SST::CoreTestModule::CoreTestModuleExample)

    SST_ELI_DOCUMENT_PARAMS(
        { "rng",     "The random number generator to use (Marsaglia or Mersenne), default is Mersenne", "Mersenne"},
        { "seed_w",  "The seed to use for the random number generator", "7" },
        { "seed_z",  "The seed to use for the random number generator", "5" },
        { "seed",    "The seed to use for the random number generator.", "11" },
    )

    SST_ELI_IS_CHECKPOINTABLE()

    explicit CoreTestModuleExample(SST::Params& params);
    ~CoreTestModuleExample();
    std::string getRNGType() const;
    uint32_t    getNext();

    CoreTestModuleExample() {}
    void serialize_order(SST::Core::Serialization::serializer& ser) override;
    ImplementSerializable(SST::CoreTestModule::CoreTestModuleExample)

private:
    std::string rng_type;
    Random*     rng;
};


class coreTestModuleLoader : public SST::Component
{
public:
    // REGISTER THIS COMPONENT INTO THE ELEMENT LIBRARY
    SST_ELI_REGISTER_COMPONENT(
        coreTestModuleLoader,
        "coreTestElement",
        "coreTestModuleLoader",
        SST_ELI_ELEMENT_VERSION(1,0,0),
        "Component that loads an RNG module",
        COMPONENT_CATEGORY_UNCATEGORIZED
    )

    SST_ELI_DOCUMENT_PARAMS(
        { "seed_w",  "The seed to use for the random number generator", "7" },
        { "seed_z",  "The seed to use for the random number generator", "5" },
        { "seed",    "The seed to use for the random number generator.", "11" },
        { "rng",     "The random number generator to use (Marsaglia or Mersenne), default is Mersenne", "Mersenne"},
        { "count",   "The number of random numbers to generate, default is 1000", "1000" },
        { "verbose", "Sets the output verbosity of the component", "0" }
    )

    // Optional since there is nothing to document
    SST_ELI_DOCUMENT_STATISTICS(
    )

    // Optional since there is nothing to document
    SST_ELI_DOCUMENT_PORTS(
    )

    // Optional since there is nothing to document
    SST_ELI_DOCUMENT_SUBCOMPONENT_SLOTS(
    )

    SST_ELI_IS_CHECKPOINTABLE()

    coreTestModuleLoader(SST::ComponentId_t id, SST::Params& params);
    ~coreTestModuleLoader();
    void setup() override;
    void finish() override;
    void serialize_order(SST::Core::Serialization::serializer& ser) override;
    ImplementSerializable(SST::CoreTestModule::coreTestModuleLoader)

private:
    coreTestModuleLoader() = default; // for serialization only

    virtual bool tick(SST::Cycle_t);

    Output*                output;
    int                    rng_max_count;
    int                    rng_count;
    CoreTestModuleExample* rng_module;
};

} // namespace SST::CoreTestModule

#endif // SST_CORE_CORETEST_MODULE_H