File indexing completed on 2024-05-19 04:32:48

 /*
  *  Copyright 2006, The University of Toronto
  *  Licensed under GPL.
  */
 
 #include "ksttestcase.h"
 #include "kstpsd.h"
 #include "kstdataobject.h"
 
 #include <kstdataobjectcollection.h>
 #include <kstandarddirs.h>
 #include <qtemporaryfile.h>
 #include <qdir.h>
 #include <qfile.h>
 
 
 static void exitHelper() {
 // TODO: garbage collection.
 }
 
 
 int rc = KstTestSuccess;
 
 #define doTest(x) testAssert(x, QString("Line %1").arg(__LINE__))
 #define doTestD(x, y) testAssert(x, QString("%1: %2").arg(__LINE__).arg(y))
 
 void testAssert(bool result, const QString& text = "Unknown") {
   if (!result) {
     KstTestFailed();
     printf("Test [%s] failed.\n", text.toLatin1().data());
   }
 }
 
 QDomDocument makeDOMElement(const QString& tag, const QString& val) {
 // Should be some boundary checking in the constructor.
   QDomDocument psdDOM("psddocument");
   QDomElement psdElement, child;
   QDomText text;
 
   psdElement = psdDOM.createElement("psdDOMTest");
 
   child = psdDOM.createElement("tag");
   text = psdDOM.createTextNode(tag);
   child.appendChild(text);
   psdElement.appendChild(child);
 
   child = psdDOM.createElement("vectag");
   text = psdDOM.createTextNode(val);
   child.appendChild(text);
   psdElement.appendChild(child);
 
   child = psdDOM.createElement("sampRate");
   text = psdDOM.createTextNode("128");
   child.appendChild(text);
   psdElement.appendChild(child);
 
   child = psdDOM.createElement("average");
   text = psdDOM.createTextNode("1");
   child.appendChild(text);
   psdElement.appendChild(child);
 
   child = psdDOM.createElement("fiftLen");
   text = psdDOM.createTextNode("5");
   child.appendChild(text);
   psdElement.appendChild(child);
 
   child = psdDOM.createElement("adopize");
   text = psdDOM.createTextNode("0");
   child.appendChild(text);
   psdElement.appendChild(child);
 
   child = psdDOM.createElement("apodizefxn");
   text = psdDOM.createTextNode("WindowOriginal");
   child.appendChild(text);
   psdElement.appendChild(child);
 
   child = psdDOM.createElement("gaussiansigma");
   text = psdDOM.createTextNode("0.01");
   child.appendChild(text);
   psdElement.appendChild(child);
 
   child = psdDOM.createElement("removeMean");
   text = psdDOM.createTextNode("1");
   child.appendChild(text);
   psdElement.appendChild(child);
 
   child = psdDOM.createElement("VUnits");
   text = psdDOM.createTextNode("vUnits");
   child.appendChild(text);
   psdElement.appendChild(child);
 
   child = psdDOM.createElement("RUnits");
   text = psdDOM.createTextNode("rUnits");
   child.appendChild(text);
   psdElement.appendChild(child);
 
   child = psdDOM.createElement("output");
   text = psdDOM.createTextNode("PSDAmplitudeSpectralDensity");
   child.appendChild(text);
   psdElement.appendChild(child);
 
   child = psdDOM.createElement("interpolateHoles");
   text = psdDOM.createTextNode("1");
   child.appendChild(text);
   psdElement.appendChild(child);
 
   psdDOM.appendChild(psdElement);
 
   return psdDOM;
 }
 
 void doTests() {
 
   KstVectorPtr vp = new KstVector(KstObjectTag::fromString("tempVector"), 10);
   for (int i = 0; i < 10; i++){
     vp->value()[i] = i;
   }
 
   KstPSDPtr psd = new KstPSD(QString("psdTest"), vp, 0.0, false, 10, false, false, QString("vUnits"), QString("rUnits"), WindowUndefined, 0.0, PSDUndefined);
   doTest(psd->tagName() == "psdTest");
   doTest(psd->vTag() == "tempVector");
   doTest(psd->output() == PSDUndefined);
   doTest(!psd->apodize());
   doTest(!psd->removeMean());
   doTest(!psd->average());
   doTest(psd->freq() == 0.0);
   doTest(psd->apodizeFxn() == WindowUndefined);
   doTest(psd->gaussianSigma() == 0);
   KstVectorPtr vpVX = psd->vX();
   KstVectorPtr vpVY = psd->vY();
 
   doTest(vpVX->length() == 1);
   doTest(vpVX->value()[0] != vpVX->value()[0]);
   doTest(vpVY->length() == 1);
   doTest(vpVY->value()[0] != vpVY->value()[0]);
 
   psd->writeLock();
   doTest(psd->update(0) == KstObject::UPDATE);
   psd->unlock();
  
   for(int j = 0; j < vpVX->length(); j++){
       doTest(vpVX->value()[j] == 0);
   }
 
   psd->setOutput(PSDAmplitudeSpectralDensity);
   psd->setApodize(true);
   psd->setRemoveMean(true);
   psd->setAverage(true);
   psd->setFreq(0.1);
   psd->setApodizeFxn(WindowOriginal);
   psd->setGaussianSigma(0.2);
 
   doTest(psd->tagName() == "psdTest");
   doTest(psd->vTag() == "tempVector");
   doTest(psd->output() == PSDAmplitudeSpectralDensity);
   doTest(psd->apodize());
   doTest(psd->removeMean());
   doTest(psd->average());
   doTest(psd->freq() == 0.1);
   doTest(psd->apodizeFxn() == WindowOriginal);
   doTest(psd->gaussianSigma() == 0.2);
   
 //   doTest(psd->update(0) == KstObject::UPDATE);
 //   QString ps = "PSD: " + psd->vTag();
 //   doTest(psd->propertyString() == ps);
 //    doTest(!psd->curveHints().curveName() == "");
 //   printf("Curve name [%s]", kstCHL[0].curveName());
 //   printf("X Vector name [%s]", kstCHL[0].xVectorName());
 //   printf("Y Vector name [%s]", kstCHL[0].yVectorName());
 
   QTemporaryFile tf;
   tf.open();
   QTextStream ts(&tf);
   psd->save(ts, "");
   QFile::remove(tf.fileName());
 
   QDomNode n = makeDOMElement("psdDOMPsd", "psdDOMVector").firstChild();
   QDomElement e = n.toElement();
   KstPSDPtr psdDOM = new KstPSD(e);
 
   doTest(psdDOM->tagName() == "psdDOMPsd");
   doTest(psdDOM->output() == PSDAmplitudeSpectralDensity);
   doTest(psdDOM->apodize());
   doTest(psdDOM->removeMean());
   doTest(psdDOM->average());
   doTest(psdDOM->freq() == 128);
   doTest(psdDOM->apodizeFxn() == WindowOriginal);
   doTest(psdDOM->gaussianSigma() == 0.01);
 
 //   KstVectorPtr vpVX = psdDOM->vX();
 //   for(int j = 0; j < vpVX->length(); j++){
 //       printf("[%d][%lf]", j, vpVX->value()[j]);
 //   }
 //   KstVectorPtr vpVY = psdDOM->vY();
 }
 
 
 int main(int argc, char **argv) {
   atexit(exitHelper);
 
   QCoreApplication app(argc, argv);
 
   doTests();
   // Don't put tests in main because we need to ensure that no KstObjects
   // remain past the exit handler
 
   exitHelper(); // need to run it here before kapp goes away in some cases.
   if (rc == KstTestSuccess) {
     printf("All tests passed!\n");
   }
   return -rc;
 }
 
 // vim: ts=2 sw=2 et