int KernelProbe::outputState(double timed) {
InterColComm * icComm = parent->icCommunicator();
const int rank = icComm->commRank();
if( rank != 0 ) return PV_SUCCESS;
assert(getTargetConn()!=NULL);
int nxp = getTargetHyPerConn()->xPatchSize();
int nyp = getTargetHyPerConn()->yPatchSize();
int nfp = getTargetHyPerConn()->fPatchSize();
int patchSize = nxp*nyp*nfp;
const pvwdata_t * wdata = getTargetHyPerConn()->get_wDataStart(arborID)+patchSize*kernelIndex;
const pvwdata_t * dwdata = outputPlasticIncr ?
getTargetHyPerConn()->get_dwDataStart(arborID)+patchSize*kernelIndex : NULL;
fprintf(outputstream->fp, "Time %f, Conn \"%s\", nxp=%d, nyp=%d, nfp=%d\n",
timed, getTargetConn()->getName(),nxp, nyp, nfp);
for(int f=0; f<nfp; f++) {
for(int y=0; y<nyp; y++) {
for(int x=0; x<nxp; x++) {
int k = kIndex(x,y,f,nxp,nyp,nfp);
fprintf(outputstream->fp, " x=%d, y=%d, f=%d (index %d):", x, y, f, k);
if(getOutputWeights()) {
fprintf(outputstream->fp, " weight=%f", (float)wdata[k]);
}
if(getOutputPlasticIncr()) {
fprintf(outputstream->fp, " dw=%f", (float)dwdata[k]);
}
fprintf(outputstream->fp,"\n");
}
}
}
return PV_SUCCESS;
}
int KernelProbe::allocateDataStructures() {
int status = PV_SUCCESS;
assert(getTargetConn());
if (getKernelIndex()<0 || getKernelIndex()>=getTargetHyPerConn()->getNumDataPatches()) {
fprintf(stderr, "KernelProbe \"%s\": kernelIndex %d is out of bounds. (min 0, max %d)\n", name, getKernelIndex(), getTargetHyPerConn()->getNumDataPatches()-1);
exit(EXIT_FAILURE);
}
if (getArbor()<0 || getArbor()>=getTargetConn()->numberOfAxonalArborLists()) {
fprintf(stderr, "KernelProbe \"%s\": arborId %d is out of bounds. (min 0, max %d)\n", name, getArbor(), getTargetConn()->numberOfAxonalArborLists()-1);
exit(EXIT_FAILURE);
}
if(outputstream) {
fprintf(outputstream->fp, "Probe \"%s\", kernel index %d, arbor index %d.\n", name, getKernelIndex(), getArbor());
}
if(getOutputPatchIndices()) {
patchIndices(getTargetHyPerConn());
}
return status;
}
/**
* @timef
*/
int PlasticConnTestProbe::outputState(double timed) {
HyPerConn * c = getTargetHyPerConn();
InterColComm * icComm = c->getParent()->icCommunicator();
const int rcvProc = 0;
if( icComm->commRank() != rcvProc ) {
return PV_SUCCESS;
}
assert(getTargetConn()!=NULL);
outputStream->printf(" Time %f, connection \"%s\":\n", timed, getTargetName());
const pvwdata_t * w = c->get_wDataHead(getArbor(), getKernelIndex());
const pvdata_t * dw = c->get_dwDataHead(getArbor(), getKernelIndex());
if( getOutputPlasticIncr() && dw == NULL ) {
pvError().printf("PlasticConnTestProbe \"%s\": connection \"%s\" has dKernelData(%d,%d) set to null.\n", getName(), getTargetName(), getKernelIndex(), getArbor());
}
int nxp = c->xPatchSize();
int nyp = c->yPatchSize();
int nfp = c->fPatchSize();
int status = PV_SUCCESS;
for( int k=0; k<nxp*nyp*nfp; k++ ) {
int x=kxPos(k,nxp,nyp,nfp);
int wx = (nxp-1)/2 - x; // assumes connection is one-to-one
if(getOutputWeights()) {
pvdata_t wCorrect = timed*wx;
pvdata_t wObserved = w[k];
if( fabs( ((double) (wObserved - wCorrect))/timed ) > 1e-4 ) {
int y=kyPos(k,nxp,nyp,nfp);
int f=featureIndex(k,nxp,nyp,nfp);
outputStream->printf(" index %d (x=%d, y=%d, f=%d: w = %f, should be %f\n", k, x, y, f, wObserved, wCorrect);
}
}
if(timed > 0 && getOutputPlasticIncr() && dw != NULL) {
pvdata_t dwCorrect = wx;
pvdata_t dwObserved = dw[k];
if( dwObserved != dwCorrect ) {
int y=kyPos(k,nxp,nyp,nfp);
int f=featureIndex(k,nxp,nyp,nfp);
outputStream->printf(" index %d (x=%d, y=%d, f=%d: dw = %f, should be %f\n", k, x, y, f, dwObserved, dwCorrect);
}
}
}
assert(status==PV_SUCCESS);
if( status == PV_SUCCESS ) {
if (getOutputWeights()) { outputStream->printf(" All weights are correct.\n"); }
if (getOutputPlasticIncr()) { outputStream->printf(" All plastic increments are correct.\n"); }
}
if(getOutputPatchIndices()) {
patchIndices(c);
}
return PV_SUCCESS;
}
/**
* @timef
*/
int MomentumConnTestProbe::outputState(double timed) {
HyPerConn * c = getTargetHyPerConn();
InterColComm * icComm = c->getParent()->icCommunicator();
const int rcvProc = 0;
if( icComm->commRank() != rcvProc ) {
return PV_SUCCESS;
}
assert(getTargetConn()!=NULL);
FILE * fp = getStream()->fp;
fprintf(fp, " Time %f, connection \"%s\":\n", timed, getTargetName());
const pvwdata_t * w = c->get_wDataHead(getArbor(), getKernelIndex());
const pvdata_t * dw = c->get_dwDataHead(getArbor(), getKernelIndex());
if( getOutputPlasticIncr() && dw == NULL ) {
fprintf(stderr, "MomentumConnTestProbe \"%s\": connection \"%s\" has dKernelData(%d,%d) set to null.\n", getName(), getTargetName(), getKernelIndex(), getArbor());
assert(false);
}
int nxp = c->xPatchSize();
int nyp = c->yPatchSize();
int nfp = c->fPatchSize();
int status = PV_SUCCESS;
for( int k=0; k<nxp*nyp*nfp; k++ ) {
pvdata_t wObserved = w[k];
//Pulse happens at time 3
pvdata_t wCorrect;
if(timed < 3){
wCorrect = 0;
}
else{
if(isViscosity){
wCorrect = 1;
for(int i = 0; i < (timed - 3); i++){
wCorrect += exp(-(2*(i+1)));
}
}
else{
wCorrect = 2 - pow(2, -(timed - 3));
}
}
if( fabs( ((double) (wObserved - wCorrect))/timed ) > 1e-4 ) {
int y=kyPos(k,nxp,nyp,nfp);
int f=featureIndex(k,nxp,nyp,nfp);
fprintf(fp, " w = %f, should be %f\n", wObserved, wCorrect);
exit(-1);
}
}
return PV_SUCCESS;
}
int OjaKernelSpikeRateProbe::allocateDataStructures() {
targetOjaKernelConn = dynamic_cast<OjaKernelConn *>(getTargetConn());
if (targetOjaKernelConn == NULL) {
if (getParent()->columnId()==0) {
fprintf(stderr, "LCATraceProbe error: connection \"%s\" must be an LCALIFLateralConn.\n", getTargetConn()->getName());
}
abort();
}
HyPerLayer * targetLayer = NULL;
if (isInputRate) {
targetLayer = targetOjaKernelConn->preSynapticLayer();
}
else {
targetLayer = targetOjaKernelConn->postSynapticLayer();
}
const PVLayerLoc * loc = targetLayer->getLayerLoc();
int x_local = xg - loc->kx0;
int y_local = yg - loc->ky0;
bool inBounds = (x_local >= 0 && x_local < loc->nx && y_local >= 0 && y_local < loc->ny);
if(inBounds ) { // if inBounds
int krestricted = kIndex(x_local, y_local, feature, loc->nx, loc->ny, loc->nf);
if (isInputRate) {
int kextended = kIndexExtended(krestricted, loc->nx, loc->ny, loc->nf, loc->halo.lt, loc->halo.rt, loc->halo.dn, loc->halo.up);
spikeRate = &targetOjaKernelConn->getInputFiringRate(arbor)[kextended];
}
else {
spikeRate = &targetOjaKernelConn->getOutputFiringRate()[krestricted];
}
}
else {
outputstream = NULL;
}
//This is now being done in BaseConnectionProbe
//getTargetConn()->insertProbe(this);
return PV_SUCCESS;
}
