InitGaussianRandomWeights.cpp

/*
 * InitGaussianRandomWeights.cpp
 *
 *  Created on: Aug 9, 2011
 *      Author: kpeterson
 */

#include "InitGaussianRandomWeights.hpp"

namespace PV {

InitGaussianRandomWeights::InitGaussianRandomWeights(char const *name, HyPerCol *hc) {
   initialize(name, hc);
}

InitGaussianRandomWeights::InitGaussianRandomWeights() {}

InitGaussianRandomWeights::~InitGaussianRandomWeights() {
   pvAssert(dynamic_cast<Random *>(mGaussianRandState) == mRandState);
   delete mGaussianRandState;
   mRandState = nullptr; // Prevents InitRandomWeights destructor from double-deleting
}

int InitGaussianRandomWeights::initialize(char const *name, HyPerCol *hc) {
   int status = InitRandomWeights::initialize(name, hc);
   return status;
}

int InitGaussianRandomWeights::ioParamsFillGroup(enum ParamsIOFlag ioFlag) {
   int status = InitRandomWeights::ioParamsFillGroup(ioFlag);
   ioParam_wGaussMean(ioFlag);
   ioParam_wGaussStdev(ioFlag);
   return status;
}

void InitGaussianRandomWeights::ioParam_wGaussMean(enum ParamsIOFlag ioFlag) {
   parent->parameters()->ioParamValue(ioFlag, name, "wGaussMean", &mWGaussMean, mWGaussMean);
}

void InitGaussianRandomWeights::ioParam_wGaussStdev(enum ParamsIOFlag ioFlag) {
   parent->parameters()->ioParamValue(ioFlag, name, "wGaussStdev", &mWGaussStdev, mWGaussStdev);
}

int InitGaussianRandomWeights::initRNGs(bool isKernel) {
   pvAssert(mRandState == nullptr && mGaussianRandState == nullptr);
   int status = PV_SUCCESS;
   if (isKernel) {
      mGaussianRandState = new GaussianRandom(mWeights->getNumDataPatches());
   }
   else {
      mGaussianRandState =
            new GaussianRandom(&mWeights->getGeometry()->getPreLoc(), true /*isExtended*/);
   }

   if (mGaussianRandState == nullptr) {
      Fatal().printf(
            "InitRandomWeights error in rank %d process: unable to create object of class "
            "Random.\n",
            parent->getCommunicator()->globalCommRank());
   }
   mRandState = (Random *)mGaussianRandState;
   return status;
}

void InitGaussianRandomWeights::randomWeights(float *patchDataStart, int patchIndex) {
   const int patchSize = mWeights->getPatchSizeOverall();
   for (int n = 0; n < patchSize; n++) {
      patchDataStart[n] = mGaussianRandState->gaussianDist(patchIndex, mWGaussMean, mWGaussStdev);
   }
}

} /* namespace PV */