KJS::Value KstBindPowerSpectrum::rUnits(KJS::ExecState *exec) const {
Q_UNUSED(exec)
KstPSDPtr d = makePSD(_d);
if (d) {
KstReadLocker rl(d);
return KJS::String(d->rUnits());
}
return KJS::String();
}
KJS::Value KstBindPowerSpectrum::length(KJS::ExecState *exec) const {
Q_UNUSED(exec)
KstPSDPtr d = makePSD(_d);
if (d) {
KstReadLocker rl(d);
return KJS::Number(d->len());
}
return KJS::Number(0);
}
KJS::Value KstBindPowerSpectrum::average(KJS::ExecState *exec) const {
Q_UNUSED(exec)
KstPSDPtr d = makePSD(_d);
if (d) {
KstReadLocker rl(d);
return KJS::Boolean(d->average());
}
return KJS::Boolean(false);
}
KJS::Value KstBindPowerSpectrum::yVector(KJS::ExecState *exec) const {
KstPSDPtr d = makePSD(_d);
if (d) {
KstReadLocker rl(d);
KstVectorPtr vp = d->vY();
if (vp) {
return KJS::Object(new KstBindVector(exec, vp));
}
}
return KJS::Undefined();
}
QStringList KstIfaceImpl::createPowerSpectrum(const QString& name,
const QString& vector,
bool appodize,
bool removeMean,
int fftLength,
const QString& rateUnits,
double sampleRate,
const QString& vectorUnits)
{
//get the vector
KST::vectorList.lock().readLock();
KstVectorList::Iterator iter = KST::vectorList.findTag(vector);
KST::vectorList.lock().unlock();
if (iter == KST::vectorList.end()) {
return QStringList();
}
//suggest a name if not supplied
QString pstag_end;
if (name.isEmpty())
pstag_end = vector;
else
pstag_end = name;
//count number of power spectra and make a unique name
KstPSDList pslist = kstObjectSubList<KstDataObject,KstPSD>(KST::dataObjectList);
int i = pslist.count() + 1;
QString stringnum = QString::number(i);
QString pstag = stringnum + "-" + pstag_end;
while (KstData::self()->dataTagNameNotUnique(pstag, false)) {
stringnum.setNum(++i);
pstag = stringnum + "-" + pstag_end;
}
KstPSDPtr powerspectrum = new KstPSD(pstag, *iter, sampleRate, true, fftLength,
appodize, removeMean, vectorUnits, rateUnits, WindowOriginal);
KST::dataObjectList.lock().writeLock();
KST::dataObjectList.append(KstDataObjectPtr(powerspectrum));
KST::dataObjectList.lock().unlock();
QStringList returnList;
returnList.push_back(powerspectrum->tagName());
returnList.push_back(powerspectrum->xVTag());
returnList.push_back(powerspectrum->yVTag());
_doc->forceUpdate();
_doc->setModified();
return returnList;
}
void KstBindPowerSpectrum::setRUnits(KJS::ExecState *exec, const KJS::Value& value) {
if (value.type() != KJS::StringType) {
KJS::Object eobj = KJS::Error::create(exec, KJS::TypeError);
exec->setException(eobj);
return;
}
KstPSDPtr d = makePSD(_d);
if (d) {
KstWriteLocker wl(d);
d->setRUnits(value.toString(exec).qstring());
}
}
void KstBindPowerSpectrum::setFrequency(KJS::ExecState *exec, const KJS::Value& value) {
if (value.type() != KJS::NumberType) {
KJS::Object eobj = KJS::Error::create(exec, KJS::TypeError);
exec->setException(eobj);
return;
}
KstPSDPtr d = makePSD(_d);
if (d) {
KstWriteLocker wl(d);
d->setFreq(value.toNumber(exec));
}
}
void KstBindPowerSpectrum::setLength(KJS::ExecState *exec, const KJS::Value& value) {
unsigned val;
if (value.type() != KJS::NumberType || !value.toUInt32(val)) {
KJS::Object eobj = KJS::Error::create(exec, KJS::TypeError);
exec->setException(eobj);
return;
}
KstPSDPtr d = makePSD(_d);
if (d) {
KstWriteLocker wl(d);
d->setLen(val);
}
}
bool KstPsdDialog::newObject() {
QString tag_name = _tagName->text();
bool retVal = false;
if (tag_name == defaultTag) {
tag_name = KST::suggestPSDName(KstObjectTag::fromString(_w->_vector->selectedVector()));
}
if (KstData::self()->dataTagNameNotUnique(tag_name)) {
_tagName->setFocus();
return false;
}
if (_w->_vector->selectedVector().isEmpty()) {
QMessageBox::warning(this, QObject::tr("Kst"), QObject::tr("New spectrum not made: define vectors first."));
return false;
}
KST::vectorList.lock().readLock();
KstVectorPtr p = *KST::vectorList.findTag(_w->_vector->selectedVector());
KST::vectorList.lock().unlock();
if (_w->_kstFFTOptions->checkValues()) {
if (p) {
KstVCurvePtr vc;
KstPSDPtr psd;
QColor color;
p->readLock();
psd = new KstPSD(tag_name, p,
_w->_kstFFTOptions->SampRate->text().toDouble(),
_w->_kstFFTOptions->Interleaved->isChecked(),
_w->_kstFFTOptions->FFTLen->text().toInt(),
_w->_kstFFTOptions->Apodize->isChecked(),
_w->_kstFFTOptions->RemoveMean->isChecked(),
_w->_kstFFTOptions->VectorUnits->text(),
_w->_kstFFTOptions->RateUnits->text(),
ApodizeFunction(_w->_kstFFTOptions->ApodizeFxn->currentIndex()),
_w->_kstFFTOptions->Sigma->value(),
PSDType(_w->_kstFFTOptions->Output->currentIndex()));
psd->setInterpolateHoles(_w->_kstFFTOptions->InterpolateHoles->isChecked());
p->unlock();
// xxx color = KstApp::inst()->chooseColorDlg()->getColorForCurve(psd->vX(), psd->vY());
if (!color.isValid()) {
color = _w->_curveAppearance->color();
}
vc = new KstVCurve(KST::suggestCurveName(psd->tag(),true), psd->vX(), psd->vY(), KstVectorPtr(), KstVectorPtr(), KstVectorPtr(), KstVectorPtr(), color);
vc->setHasPoints(_w->_curveAppearance->showPoints());
vc->setHasLines(_w->_curveAppearance->showLines());
vc->setHasBars(_w->_curveAppearance->showBars());
vc->setPointStyle(_w->_curveAppearance->pointType());
vc->setLineWidth(_w->_curveAppearance->lineWidth());
vc->setLineStyle(_w->_curveAppearance->lineStyle());
vc->setBarStyle(_w->_curveAppearance->barStyle());
vc->setPointDensity(_w->_curveAppearance->pointDensity());
QString legend_text = _legendText->text();
if (legend_text == defaultTag) {
vc->setLegendText(QString::null);
} else {
vc->setLegendText(legend_text);
}
Kst2DPlotPtr plot;
KstViewWindow *w;
/* xxx
w = dynamic_cast<KstViewWindow*>(KstApp::inst()->findWindow(_w->_curvePlacement->_plotWindow->currentText()));
if (!w) {
QString n = KstApp::inst()->newWindow(KST::suggestWinName());
w = static_cast<KstViewWindow*>(KstApp::inst()->findWindow(n));
}
*/
if (w) {
if (_w->_curvePlacement->existingPlot()) {
plot = kst_cast<Kst2DPlot>(w->view()->findChild(_w->_curvePlacement->plotName()));
if (plot) {
plot->addCurve(vc);
}
}
if (_w->_curvePlacement->newPlot()) {
QString name = w->createPlot(KST::suggestPlotName());
if (_w->_curvePlacement->reGrid()) {
w->view()->cleanup(_w->_curvePlacement->columns());
}
plot = kst_cast<Kst2DPlot>(w->view()->findChild(name));
if (plot) {
plot->setXAxisInterpretation(false, KstAxisInterpretation(), KstAxisDisplay());
plot->setYAxisInterpretation(false, KstAxisInterpretation(), KstAxisDisplay());
_w->_curvePlacement->update();
_w->_curvePlacement->setCurrentPlot(plot->tagName());
plot->addCurve(vc);
plot->generateDefaultLabels();
}
}
}
KST::dataObjectList.lock().writeLock();
KST::dataObjectList.append(psd);
KST::dataObjectList.append(vc);
KST::dataObjectList.lock().unlock();
psd = 0L;
vc = 0L;
emit modified();
retVal = true;
}
}
return retVal;
}
bool KstPsdDialog::editSingleObject(KstPSDPtr psPtr) {
psPtr->writeLock();
KST::vectorList.lock().readLock();
KstVectorPtr v = *KST::vectorList.findTag(_w->_vector->selectedVector());
KST::vectorList.lock().unlock();
if (v) { // Can be null if edit multiple and it wasn't changed
psPtr->setVector(v);
}
// get the values that need to be checked for consistency
double pSampRate;
int pFFTLen;
if (_sampRateDirty) {
pSampRate = _w->_kstFFTOptions->SampRate->text().toDouble();
} else {
pSampRate = psPtr->freq();
}
if (_fFTLenDirty) {
pFFTLen = _w->_kstFFTOptions->FFTLen->text().toInt();
} else {
pFFTLen = psPtr->len();
}
if (!_w->_kstFFTOptions->checkGivenValues(pSampRate, pFFTLen)) {
psPtr->unlock();
return false;
}
if (_sampRateDirty) {
psPtr->setFreq(_w->_kstFFTOptions->SampRate->text().toDouble());
}
if (_fFTLenDirty) {
psPtr->setLen(_w->_kstFFTOptions->FFTLen->text().toInt());
}
if (_vectorUnitsDirty) {
psPtr->setVUnits(_w->_kstFFTOptions->VectorUnits->text());
}
if (_rateUnitsDirty) {
psPtr->setRUnits(_w->_kstFFTOptions->RateUnits->text());
}
if (_apodizeDirty) {
psPtr->setApodize(_w->_kstFFTOptions->Apodize->isChecked());
}
if (_apodizeFxnDirty) {
if (_editMultipleMode) {
psPtr->setApodizeFxn(ApodizeFunction(_w->_kstFFTOptions->ApodizeFxn->currentIndex()-1));
} else {
psPtr->setApodizeFxn(ApodizeFunction(_w->_kstFFTOptions->ApodizeFxn->currentIndex()));
}
}
if (_gaussianSigmaDirty) {
psPtr->setGaussianSigma(_editMultipleMode ? _w->_kstFFTOptions->Sigma->value() - 1 :
_w->_kstFFTOptions->Sigma->value());
}
if (_removeMeanDirty) {
psPtr->setRemoveMean(_w->_kstFFTOptions->RemoveMean->isChecked());
}
if (_interleavedDirty) {
psPtr->setAverage(_w->_kstFFTOptions->Interleaved->isChecked());
}
if (_outputDirty) {
if (_editMultipleMode) {
psPtr->setOutput(PSDType(_w->_kstFFTOptions->Output->currentIndex()-1));
} else {
psPtr->setOutput(PSDType(_w->_kstFFTOptions->Output->currentIndex()));
}
}
if (_interpolateHolesDirty) {
psPtr->setInterpolateHoles(_w->_kstFFTOptions->InterpolateHoles->isChecked());
}
psPtr->setRecursed(false);
if (psPtr->recursion()) {
psPtr->setRecursed(true);
psPtr->unlock();
QMessageBox::critical(this, QObject::tr("Kst"), QObject::tr("There is a recursion resulting from the spectrum you entered."));
return false;
}
psPtr->unlock();
return true;
}
// returns true if succesful
bool KstPsdDialogI::editObject() {
// if the user selected no vector, treat it as non-dirty
_vectorDirty = _w->_vector->_vector->currentItem() != 0;
_apodizeDirty = _w->_kstFFTOptions->ApodizeFxn->currentItem() != 0;
_fFTLenDirty = _w->_kstFFTOptions->FFTLen->text() != " ";
_sampRateDirty = !_w->_kstFFTOptions->SampRate->text().isEmpty();
_vectorUnitsDirty = !_w->_kstFFTOptions->VectorUnits->text().isEmpty();
_rateUnitsDirty = !_w->_kstFFTOptions->RateUnits->text().isEmpty();
_outputDirty = !_w->_kstFFTOptions->Output->currentItem() != 0;
KstPSDList psList = kstObjectSubList<KstDataObject,KstPSD>(KST::dataObjectList);
// if editing multiple objects, edit each one
if (_editMultipleMode) {
bool didEdit = false;
for (uint i = 0; i < _editMultipleWidget->_objectList->count(); i++) {
if (_editMultipleWidget->_objectList->isSelected(i)) {
// get the pointer to the object
KstPSDList::Iterator psIter = psList.findTag(_editMultipleWidget->_objectList->text(i));
if (psIter == psList.end()) {
return false;
}
KstPSDPtr psPtr = *psIter;
if (!editSingleObject(psPtr)) {
return false;
}
didEdit = true;
}
}
if (!didEdit) {
KMessageBox::sorry(this, i18n("Select one or more objects to edit."));
return false;
}
} else {
KstPSDPtr pp = kst_cast<KstPSD>(_dp);
// verify that the name is unique
QString tag_name = _tagName->text();
if (!pp || (tag_name != pp->tagName() && KstData::self()->dataTagNameNotUnique(tag_name))) {
_tagName->setFocus();
return false;
}
pp->writeLock();
pp->setTagName(KstObjectTag(tag_name, pp->tag().context())); // FIXME: doesn't allow changing tag context
pp->unlock();
// then edit the object
_vectorDirty = true;
_apodizeDirty = true;
_apodizeFxnDirty = true;
_gaussianSigmaDirty = true;
_removeMeanDirty = true;
_interleavedDirty = true;
_sampRateDirty = true;
_vectorUnitsDirty = true;
_rateUnitsDirty = true;
_fFTLenDirty = true;
_outputDirty = true;
_interpolateHolesDirty = true;
if (!editSingleObject(pp)) {
return false;
}
}
emit modified();
return true;
}
void doTests() {
KstVectorPtr vp = new KstVector(KstObjectTag("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();
// until we call update the x and y vectors will be uninitialised and
// and so they should be of length 1 and the value of vpVX[0] and
// vpVX[0] should be NAN...
doTestV(QString("vpVX->length()"), vpVX->length(), 1);
doTestV(QString("vpVY->length()"), vpVY->length(), 1);
doTestV(QString("vpVX->length()"), isnan(vpVX->value()[0]), 1);
doTestV(QString("vpVY->length()"), isnan(vpVY->value()[0]), 1);
doTest(psd->update(0) == KstObject::UPDATE);
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());
KTempFile tf(locateLocal("tmp", "kst-csd"), "txt");
QFile *qf = tf.file();
QTextStream ts(qf);
psd->save(ts, "");
QFile::remove(tf.name());
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();
}
/* returns true if succesful */
bool KstPsdDialogI::newObject() {
QString tag_name = _tagName->text();
if (tag_name == defaultTag) {
tag_name = KST::suggestPSDName(KstObjectTag::fromString(_w->_vector->selectedVector()));
}
// verify that the curve name is unique
if (KstData::self()->dataTagNameNotUnique(tag_name)) {
_tagName->setFocus();
return false;
}
if (_w->_vector->selectedVector().isEmpty()) {
KMessageBox::sorry(this, i18n("New PSD not made: define vectors first."));
return false;
}
KST::vectorList.lock().readLock();
KstVectorPtr p = *KST::vectorList.findTag(_w->_vector->selectedVector());
KST::vectorList.lock().unlock();
if (!p) {
kstdFatal() << "Bug in kst: the vector field (PSD) refers to "
<< "a non existant vector...." << endl;
}
// create the psd curve
if (!_w->_kstFFTOptions->checkValues()) {
return false;
} else {
p->readLock();
KstPSDPtr psd = new KstPSD(tag_name, p,
_w->_kstFFTOptions->SampRate->text().toDouble(),
_w->_kstFFTOptions->Interleaved->isChecked(),
_w->_kstFFTOptions->FFTLen->text().toInt(),
_w->_kstFFTOptions->Apodize->isChecked(),
_w->_kstFFTOptions->RemoveMean->isChecked(),
_w->_kstFFTOptions->VectorUnits->text(),
_w->_kstFFTOptions->RateUnits->text(),
ApodizeFunction(_w->_kstFFTOptions->ApodizeFxn->currentItem()),
_w->_kstFFTOptions->Sigma->value(),
PSDType(_w->_kstFFTOptions->Output->currentItem()));
psd->setInterpolateHoles(_w->_kstFFTOptions->InterpolateHoles->isChecked());
p->unlock();
KstVCurvePtr vc = new KstVCurve(KST::suggestCurveName(psd->tag(),true), psd->vX(), psd->vY(), 0L, 0L, 0L, 0L, _w->_curveAppearance->color());
vc->setHasPoints(_w->_curveAppearance->showPoints());
vc->setHasLines(_w->_curveAppearance->showLines());
vc->setHasBars(_w->_curveAppearance->showBars());
vc->pointType = _w->_curveAppearance->pointType();
vc->setLineWidth(_w->_curveAppearance->lineWidth());
vc->setLineStyle(_w->_curveAppearance->lineStyle());
vc->setBarStyle(_w->_curveAppearance->barStyle());
vc->setPointDensity(_w->_curveAppearance->pointDensity());
QString legend_text = _legendText->text();
if (legend_text == defaultTag) {
vc->setLegendText(QString::null);
} else {
vc->setLegendText(legend_text);
}
Kst2DPlotPtr plot;
KstViewWindow *w = dynamic_cast<KstViewWindow*>(KstApp::inst()->findWindow(_w->_curvePlacement->_plotWindow->currentText()));
if (!w) {
QString n = KstApp::inst()->newWindow(KST::suggestWinName());
w = static_cast<KstViewWindow*>(KstApp::inst()->findWindow(n));
}
if (w) {
if (_w->_curvePlacement->existingPlot()) {
// assign curve to plot
plot = kst_cast<Kst2DPlot>(w->view()->findChild(_w->_curvePlacement->plotName()));
if (plot) {
plot->addCurve(vc.data());
}
}
if (_w->_curvePlacement->newPlot()) {
// assign curve to plot
QString name = w->createObject<Kst2DPlot>(KST::suggestPlotName());
if (_w->_curvePlacement->reGrid()) {
w->view()->cleanup(_w->_curvePlacement->columns());
}
plot = kst_cast<Kst2DPlot>(w->view()->findChild(name));
if (plot) {
plot->setXAxisInterpretation(false, KstAxisInterpretation(), KstAxisDisplay());
plot->setYAxisInterpretation(false, KstAxisInterpretation(), KstAxisDisplay());
_w->_curvePlacement->update();
_w->_curvePlacement->setCurrentPlot(plot->tagName());
plot->addCurve(vc.data());
plot->generateDefaultLabels();
}
}
}
KST::dataObjectList.lock().writeLock();
KST::dataObjectList.append(psd.data());
KST::dataObjectList.append(vc.data());
KST::dataObjectList.lock().unlock();
psd = 0L;
vc = 0L;
emit modified();
}
return true;
}
KJS::Object KstBindPowerSpectrum::construct(KJS::ExecState *exec, const KJS::List& args) {
if (args.size() < 2) {
KJS::Object eobj = KJS::Error::create(exec, KJS::SyntaxError);
exec->setException(eobj);
return KJS::Object();
}
if (args[1].type() != KJS::NumberType) {
KJS::Object eobj = KJS::Error::create(exec, KJS::TypeError);
exec->setException(eobj);
return KJS::Object();
}
double gaussianSigma = 0.0;
double freq = args[1].toNumber(exec);
bool average = true;
unsigned len = 16;
bool apodize = true;
bool removeMean = true;
int apodizeFxn = WindowOriginal;
int output = PSDAmplitudeSpectralDensity;
QString vunits = "V";
QString runits = "Hz";
KstVectorPtr v = extractVector(exec, args[0]);
if (!v) {
KJS::Object eobj = KJS::Error::create(exec, KJS::TypeError);
exec->setException(eobj);
return KJS::Object();
}
if (args.size() > 2) {
if (args[2].type() != KJS::BooleanType) {
KJS::Object eobj = KJS::Error::create(exec, KJS::TypeError);
exec->setException(eobj);
return KJS::Object();
}
average = args[2].toBoolean(exec);
}
if (args.size() > 3) {
if (args[3].type() != KJS::NumberType || !args[3].toUInt32(len)) {
KJS::Object eobj = KJS::Error::create(exec, KJS::TypeError);
exec->setException(eobj);
return KJS::Object();
}
}
if (args.size() > 4) {
if (args[4].type() != KJS::BooleanType) {
KJS::Object eobj = KJS::Error::create(exec, KJS::TypeError);
exec->setException(eobj);
return KJS::Object();
}
apodize = args[4].toBoolean(exec);
}
if (args.size() > 5) {
if (args[5].type() != KJS::BooleanType) {
KJS::Object eobj = KJS::Error::create(exec, KJS::TypeError);
exec->setException(eobj);
return KJS::Object();
}
removeMean = args[5].toBoolean(exec);
}
if (args.size() > 6) {
if (args[6].type() != KJS::StringType) {
KJS::Object eobj = KJS::Error::create(exec, KJS::TypeError);
exec->setException(eobj);
return KJS::Object();
}
vunits = args[6].toString(exec).qstring();
}
if (args.size() > 7) {
if (args[7].type() != KJS::StringType) {
KJS::Object eobj = KJS::Error::create(exec, KJS::TypeError);
exec->setException(eobj);
return KJS::Object();
}
runits = args[7].toString(exec).qstring();
}
if (args.size() > 8) {
if (args[8].type() != KJS::NumberType) {
KJS::Object eobj = KJS::Error::create(exec, KJS::TypeError);
exec->setException(eobj);
return KJS::Object();
}
apodizeFxn = args[8].toInt32(exec);
}
if (args.size() > 9) {
if (args[9].type() != KJS::NumberType) {
KJS::Object eobj = KJS::Error::create(exec, KJS::TypeError);
exec->setException(eobj);
return KJS::Object();
}
gaussianSigma = args[9].toNumber(exec);
}
if (args.size() > 10) {
if (args[10].type() != KJS::NumberType) {
KJS::Object eobj = KJS::Error::create(exec, KJS::TypeError);
exec->setException(eobj);
return KJS::Object();
}
output = args[10].toInt32(exec);
}
if (args.size() > 11) {
KJS::Object eobj = KJS::Error::create(exec, KJS::SyntaxError);
exec->setException(eobj);
return KJS::Object();
}
KstPSDPtr d = new KstPSD(QString::null, v, freq, average, len, apodize, removeMean,
vunits, runits, (ApodizeFunction)apodizeFxn, gaussianSigma,
(PSDType)output);
KST::dataObjectList.lock().writeLock();
KST::dataObjectList.append(d.data());
KST::dataObjectList.lock().unlock();
return KJS::Object(new KstBindPowerSpectrum(exec, d));
}
KstBindPowerSpectrum::KstBindPowerSpectrum(KJS::ExecState *exec, KstPSDPtr d)
: KstBindDataObject(exec, d.data(), "PowerSpectrum") {
KJS::Object o(this);
addBindings(exec, o);
}
int main(int argc, char *argv[]) {
int i_file, i_v, i_curve;
int i_plot;
QString fullPath;
KAboutData aboutData("kst", I18N_NOOP("Kst"),
KSTVERSION, description, KAboutData::License_GPL,
I18N_NOOP("(c) 2000-2007 Barth Netterfield"),
0,
"http://kst.kde.org/");
aboutData.addAuthor("Barth Netterfield",
I18N_NOOP("Original author and maintainer."),
"*****@*****.**",
"http://omega.astro.utoronto.ca/");
aboutData.addAuthor("Staikos Computing Services Inc.",
I18N_NOOP("Developed for the University of Toronto."),
"*****@*****.**",
"http://www.staikos.net/");
aboutData.addAuthor("Sumus Technology Limited",
I18N_NOOP("Developed for the University of British Columbia"),
"*****@*****.**",
"http://www.sumusltd.com/");
aboutData.addAuthor("Rick Chern",
I18N_NOOP("University of British Columbia"),
"",
"");
aboutData.addAuthor("Duncan Hanson",
I18N_NOOP("University of British Columbia"),
"",
"");
aboutData.addAuthor("Nicolas Brisset",
"",
"",
"");
aboutData.addAuthor("Matthew Truch",
"",
"http://matt.truch.net/",
"*****@*****.**");
aboutData.addAuthor("Theodore Kisner",
"",
"*****@*****.**",
"");
aboutData.setTranslator(I18N_NOOP("_: NAME OF TRANSLATORS\nYour names"),
I18N_NOOP("_: EMAIL OF TRANSLATORS\nYour emails"));
KCmdLineArgs::init( argc, argv, &aboutData );
KCmdLineArgs::addCmdLineOptions( options ); // Add our own options.
KApplication app;
KImageIO::registerFormats();
KstDialogs::replaceSelf(new KstGuiDialogs);
KstData::replaceSelf(new KstGuiData);
KstApp::initialize();
atexit(exitHelper);
if (app.isRestored()) {
RESTORE(KstApp)
} else {
KstApp *kst = new KstApp;
InType in;
QColor color;
QCStringList ycolList;
QCStringList matrixList;
QCStringList yEqList;
QCStringList psdList;
QCStringList hsList;
QCStringList errorList;
unsigned int i_ycol;
QCStringList::Iterator hs_string;
QCStringList::Iterator eq_i;
QCStringList::Iterator mat_i;
bool showQuickStart = false;
bool showDataWizard = false;
bool nOK;
int n_y = 0;
KCmdLineArgs *args = KCmdLineArgs::parsedArgs();
CheckForCMDErrors(args);
QString wizardfile = args->getOption("w");
QString printfile = args->getOption("print");
QString pngfile = args->getOption("png");
bool print_and_exit = false;
if (printfile != "<none>") {
print_and_exit = true;
}
if (pngfile != "<none>") {
print_and_exit = true;
}
if (!print_and_exit) {
app.setMainWidget(kst);
QRect rect = KGlobalSettings::desktopGeometry(kst);
kst->resize(5 * rect.width() / 6, 5 * rect.height() / 6);
kst->show();
}
// get Y axis columns
ycolList = args->getOptionList("y");
matrixList = args->getOptionList("z");
yEqList = args->getOptionList("ye");
psdList = args->getOptionList("p");
hsList = args->getOptionList("h");
errorList = args->getOptionList("e");
// y axis or PSD specified, so the files are data files, not kst files.
n_y = ycolList.count() + psdList.count() + hsList.count() + yEqList.count() + matrixList.count();
if (n_y > 0) {
QString creatingEquations = i18n("Creating equations");
QString creatingCurves = i18n("Creating curves");
QString creatingPlots = i18n("Creating plots");
int count;
int handled;
kst->slotUpdateProgress( 0, 0, QString::null );
SetCMDOptions(args, in, n_y);
KstTopLevelViewPtr tlv = kst->activeView();
if (!tlv) {
// if there was no active view then we create one...
kst->newWindow(false);
tlv = kst->activeView();
}
if (!tlv) {
kstdError() << i18n("Can't create a view.") << endl;
return 0;
}
CreatePlots(in, tlv);
Kst2DPlotList plist = kstObjectSubList<KstViewObject, Kst2DPlot>(tlv->children());
i_plot = 0;
Kst2DPlotPtr plot = *plist.at(i_plot);
KstVCurveList vcurves = kstObjectSubList<KstBaseCurve,KstVCurve>(plot->Curves);
// make stand alone equations if there are no files
if (args->count() < 1) {
if (!yEqList.isEmpty()) {
QString eqS;
double max, min;
int n;
bool xeq;
SetEqXRanges(args->getOption("xe"), &min, &max, &n, &xeq);
if (xeq) {
count = yEqList.size();
handled = 0;
kst->slotUpdateProgress( count, handled, creatingEquations );
for (eq_i = yEqList.begin(); eq_i != yEqList.end(); ++eq_i) {
eqS = *eq_i;
if (NoVectorEq(eqS)) {
KstEquationPtr eq = new KstEquation(KST::suggestEQName(eqS), eqS, min, max, n);
KstVCurvePtr vc = new KstVCurve(KST::suggestCurveName(eq->tag(), true),
eq->vX(), eq->vY(), 0L, 0L, 0L, 0L,
KstColorSequence::next(vcurves,plot->backgroundColor()));
KST::dataObjectList.lock().writeLock();
KST::dataObjectList.append(eq.data());
KST::dataObjectList.append(vc.data());
KST::dataObjectList.lock().unlock();
plot->addCurve(vc.data());
if (in.sep_plots) {
i_plot++;
if (i_plot < in.n_plots) {
plot = *plist.at(i_plot);
}
}
}
handled++;
kst->slotUpdateProgress( count, handled, creatingEquations );
}
}
}
}
// make the requested curves for each data file
count = args->count();
handled = 0;
kst->slotUpdateProgress( count, handled, creatingCurves );
for (i_curve = i_v = 0, i_file = 0; i_file < args->count(); i_file++) {
// make the file
if (QFile::exists(args->arg(i_file))) {
fullPath = QFileInfo(args->arg(i_file)).absFilePath();
} else {
fullPath = args->arg(i_file);
}
KstDataSourcePtr file = KstDataSource::loadSource(fullPath);
if (file) {
if (!file->isValid() || file->isEmpty()) {
kstdError() << i18n("No data in file %1. Trying to continue...").arg(args->arg(i_file)) << endl;
// The file might get data later!
}
KST::dataObjectList.lock().writeLock();
KST::dataSourceList.append(file);
KST::dataObjectList.lock().unlock();
KstRVectorPtr yvector;
KstRVectorPtr evector;
KstVCurvePtr curve;
KstPSDPtr psd;
KstHistogramPtr hs;
KstRVectorPtr xvector;
if (!ycolList.isEmpty()) { // if there are some xy plots
// make the x axis vector
xvector = GetOrCreateVector(args->getOption("x"), file, in);
if (xvector) {
// make the y axis vectors
for (i_ycol = 0; i_ycol < ycolList.count(); ++i_ycol ) {
yvector = GetOrCreateVector(*(ycolList.at(i_ycol)), file, in);
if (yvector) {
// make the curves
color = KstColorSequence::next(vcurves,plot->backgroundColor());
curve = new KstVCurve(KST::suggestCurveName(yvector->tag(), false),
KstVectorPtr(xvector), KstVectorPtr(yvector),
0L, 0L, 0L, 0L, color);
if (in.has_points) {
curve->setHasPoints(true);
curve->setHasLines(false);
}
if (i_ycol<errorList.count()) {
evector = GetOrCreateVector(*(errorList.at(i_ycol)), file, in);
if (evector) {
curve->setYError(KstVectorPtr(evector));
curve->setYMinusError(KstVectorPtr(evector));
}
}
KST::dataObjectList.lock().writeLock();
KST::dataObjectList.append(curve.data());
KST::dataObjectList.lock().unlock();
plot->addCurve(curve.data());
if (in.sep_plots) {
plot->setTagName(curve->tag());
i_plot++;
if (i_plot < in.n_plots) {
plot = *plist.at(i_plot);
}
} // end (if they are separate plots)
}
} // next y col
}
} // end (if there are some xy plots)
if (!yEqList.isEmpty()) {
QString eqS;
double max, min;
int n;
bool xeq, eq_ok;
SetEqXRanges(args->getOption("xe"), &min, &max, &n, &xeq);
for (eq_i = yEqList.begin(); eq_i != yEqList.end(); ++eq_i) {
KstEquationPtr eq;
eqS = *eq_i;
ProcessEq(eqS, file, in, &eq_ok);
if (xeq) {
eq = new KstEquation(KST::suggestEQName(eqS), eqS, min,max,n);
} else {
if (!xvector) {
xvector = GetOrCreateVector(args->getOption("x"), file, in);
}
if (xvector) {
eq = new KstEquation(KST::suggestEQName(eqS), eqS, KstVectorPtr(xvector), true);
}
}
if (eq) {
KstVCurvePtr vc = new KstVCurve(KST::suggestCurveName(eq->tag(), true),
eq->vX(), eq->vY(), 0L, 0L, 0L, 0L,
KstColorSequence::next(vcurves,plot->backgroundColor()));
KST::dataObjectList.lock().writeLock();
KST::dataObjectList.append(eq.data());
KST::dataObjectList.append(vc.data());
KST::dataObjectList.lock().unlock();
plot->addCurve(vc.data());
if (in.sep_plots) {
plot->setTagName(eq->tag());
i_plot++;
if (i_plot <in.n_plots) {
plot = *plist.at(i_plot);
}
}
}
}
}
if (psdList.count() > 0) { // if there are some psd plots
KstRVectorList rvl = kstObjectSubList<KstVector,KstRVector>(KST::vectorList);
for (QCStringList::ConstIterator it = psdList.begin(); it != psdList.end(); ++it) {
yvector = GetOrCreateVector(*it, file, in);
if (yvector) {
color = KstColorSequence::next(vcurves,plot->backgroundColor());
psd = new KstPSD( KST::suggestPSDName(yvector->tag()), // FIXME: this was yvector->field(), is this right?
KstVectorPtr(yvector), in.rate, true, in.len,
true, true, in.VUnits, in.RUnits, WindowOriginal);
KstVCurvePtr vc = new KstVCurve(KST::suggestCurveName(psd->tag(), true),
psd->vX(), psd->vY(), 0L, 0L, 0L, 0L, color);
if (in.has_points) {
vc->setHasPoints(true);
vc->setHasLines(false);
}
KST::dataObjectList.lock().writeLock();
KST::dataObjectList.append(psd.data());
KST::dataObjectList.append(vc.data());
KST::dataObjectList.lock().unlock();
plot->addCurve(vc.data());
if (in.sep_plots) {
plot->setTagName(psd->tag());
i_plot++;
if (i_plot <in.n_plots) {
plot = *plist.at(i_plot);
}
}
}
} // next psd
} // end (if there are some psds)
if (hsList.count() > 0) { // if there are some histograms
double max, min;
int N;
KstRVectorList rvl = kstObjectSubList<KstVector,KstRVector>(KST::vectorList);
for (hs_string = hsList.begin(); hs_string != hsList.end(); ++hs_string) {
yvector = GetOrCreateVector(*hs_string, file, in);
if (yvector) {
color = KstColorSequence::next(vcurves,plot->backgroundColor());
KstHistogram::AutoBin(KstVectorPtr(yvector), &N, &max, &min);
hs = new KstHistogram(KST::suggestHistogramName(yvector->tag()),
KstVectorPtr(yvector), min, max, N, KST_HS_NUMBER);
KstVCurvePtr vc = new KstVCurve(KST::suggestCurveName(hs->tag(), true),
hs->vX(), hs->vY(),
0L, 0L, 0L, 0L, color);
vc->setHasPoints(false);
vc->setHasLines(false);
vc->setHasBars(true);
vc->setBarStyle(1);
KST::dataObjectList.lock().writeLock();
KST::dataObjectList.append(hs.data());
KST::dataObjectList.append(vc.data());
KST::dataObjectList.lock().unlock();
plot->addCurve(vc.data());
if (in.sep_plots) {
plot->setTagName(hs->tag());
i_plot++;
if (i_plot < in.n_plots) {
plot = *plist.at(i_plot);
}
}
}
} // next histogram
} // end (if there are some histograms)
if (matrixList.count() > 0) { // if there are some matrixes
for (mat_i = matrixList.begin(); mat_i != matrixList.end(); ++mat_i) {
QString tag_name = KST::suggestMatrixName(*mat_i);
if (!file->isValidMatrix(*mat_i)) {
startupErrors.append(i18n("Failed to create matrix '%1' from file '%2'.").arg(*mat_i).arg(file->fileName()));
}
KstRMatrixPtr matrix = new KstRMatrix(file, *mat_i,
KstObjectTag(tag_name, file->tag()),
0,0,-1,-1,false,false,0);
// xStart, yStart, xNumSteps, yNumSteps,
//doAve, doSkip, skip);
// Time to create the image from the matrix
tag_name = KST::suggestImageName(matrix->tag());
QStringList palList = KPalette::getPaletteList();
QString pal;
if (palList.contains("IDL 13 RAINBOW")) {
pal = QString("IDL 13 RAINBOW");
} else {
pal = QString(*palList.at(0));
}
KPalette* newPal = new KPalette(pal);
KstImagePtr image = new KstImage(tag_name, KstMatrixPtr(matrix), 0.0, 1.0,
true, newPal);
plot->addCurve(KstBaseCurvePtr(image));
KST::dataObjectList.lock().writeLock();
KST::dataObjectList.append(image.data());
KST::dataObjectList.lock().unlock();
image = 0L; // drop the reference
if (in.sep_plots) {
plot->setTagName(matrix->tag());
i_plot++;
if (i_plot < in.n_plots) {
plot = *plist.at(i_plot);
}
}
}
}
} else {
startupErrors.append(i18n("Failed to load file '%1'.").arg(args->arg(i_file)));
}
handled++;
kst->slotUpdateProgress( count, handled, creatingCurves );
} // next data file
count = in.n_plots;
handled = 0;
kst->slotUpdateProgress( count, handled, creatingPlots );
for (i_plot = 0; i_plot < in.n_plots; i_plot++) {
plot = *plist.at(i_plot);
plot->generateDefaultLabels();
// if we have only images in a plot then set the scale mode to AUTO (instead of AUTOBORDER)
KstImageList images = kstObjectSubList<KstBaseCurve,KstImage>(plot->Curves);
if (images.count() == plot->Curves.count()) {
plot->setXScaleMode(AUTO);
plot->setYScaleMode(AUTO);
}
if (plot->Curves.count() > 3 || in.dolegend) {
KstViewLegendPtr vl = plot->getOrCreateLegend();
vl->resizeFromAspect(0.1, 0.1, 0.2, 0.1);
vl->setBorderWidth(2);
}
handled++;
kst->slotUpdateProgress( count, handled, creatingPlots );
}
kst->slotUpdateProgress( 0, 0, QString::null );
} else if (args->count() > 0) { // open a kst file
// some of the options can be overridden
kst->openDocumentFile(args->arg(0),
args->getOption("F"), // override FileName
args->getOption("n").toInt(&nOK), // override number of frames
args->getOption("f").toInt(&nOK), // override starting frame
args->getOption("s").toInt(&nOK), // override skip
args->isSet("a"), // add averaging
!print_and_exit); // delayed
} else {
//kst->openDocumentFile();
showQuickStart = true;
}
if (args->isSet("nq")) {
showQuickStart = false;
}
if (args->isSet("w")) {
showDataWizard = true;
showQuickStart = false;
}
if (printfile != "<none>") {
kst->forceUpdate();
kst->immediatePrintToFile(printfile, false);
}
if (pngfile != "<none>") {
kst->forceUpdate();
kst->immediatePrintToPng(pngfile);
}
kst->document()->setModified(false);
if (print_and_exit) {
delete kst;
return 0;
} else {
kst->updateDialogs();
if (showQuickStart) {
kst->showQuickStartDialog();
}
if (showDataWizard) {
kst->showDataWizardWithFile(wizardfile);
}
for (size_t i = 0; i < startupErrors.size(); ++i) {
KstDebug::self()->log(startupErrors[i], KstDebug::Error);
}
startupErrors.clear();
}
// LEAVE THIS HERE - causes crashes otherwise!
int rc = app.exec();
delete kst;
return rc;
}
return app.exec();
}
bool KstPsdDialog::editObject() {
KstPSDList psList;
//
// if the user selected no vector, treat it as non-dirty...
//
_vectorDirty = _w->_vector->_vector->currentIndex() != 0;
_apodizeFxnDirty = _w->_kstFFTOptions->ApodizeFxn->currentIndex() != 0;
_fFTLenDirty = _w->_kstFFTOptions->FFTLen->text() != " ";
_sampRateDirty = !_w->_kstFFTOptions->SampRate->text().isEmpty();
_vectorUnitsDirty = !_w->_kstFFTOptions->VectorUnits->text().isEmpty();
_rateUnitsDirty = !_w->_kstFFTOptions->RateUnits->text().isEmpty();
_outputDirty = _w->_kstFFTOptions->Output->currentIndex() != 0;
psList = kstObjectSubList<KstDataObject,KstPSD>(KST::dataObjectList);
if (_editMultipleMode) {
bool didEdit = false;
int i;
for (i = 0; i < _editMultipleWidget->_objectList->count(); i++) {
if (_editMultipleWidget->_objectList->item(i)->isSelected()) {
KstPSDList::Iterator psIter;
KstPSDPtr psPtr;
psIter = psList.findTag(_editMultipleWidget->_objectList->item(i)->text());
if (psIter == psList.end()) {
return false;
}
psPtr = *psIter;
if (!editSingleObject(psPtr)) {
return false;
}
didEdit = true;
}
}
if (!didEdit) {
QMessageBox::warning(this, QObject::tr("Kst"), QObject::tr("Select one or more objects to edit."));
return false;
}
} else {
KstPSDPtr pp;
QString tag_name;
pp = kst_cast<KstPSD>(_dp);
tag_name = _tagName->text();
if (!pp || (tag_name != pp->tagName() && KstData::self()->dataTagNameNotUnique(tag_name))) {
_tagName->setFocus();
return false;
}
pp->writeLock();
pp->setTagName(tag_name);
pp->unlock();
// then edit the object
_vectorDirty = true;
_apodizeDirty = true;
_apodizeFxnDirty = true;
_gaussianSigmaDirty = true;
_removeMeanDirty = true;
_interleavedDirty = true;
_sampRateDirty = true;
_vectorUnitsDirty = true;
_rateUnitsDirty = true;
_fFTLenDirty = true;
_outputDirty = true;
_interpolateHolesDirty = true;
if (!editSingleObject(pp)) {
return false;
}
}
emit modified();
return true;
}
void KstPsdDialog::fillFieldsForEdit() {
KstPSDPtr pp;
pp = kst_cast<KstPSD>(_dp);
if (pp) {
pp->readLock();
_tagName->setText(pp->tagName());
_w->_vector->setSelection(pp->vTag());
// set sample rate, Units, FFT len, and vector units
_w->_kstFFTOptions->FFTLen->setValue(pp->len());
_w->_kstFFTOptions->SampRate->setText(QString::number(pp->freq()));
_w->_kstFFTOptions->VectorUnits->setText(pp->vUnits());
_w->_kstFFTOptions->RateUnits->setText(pp->rUnits());
_w->_kstFFTOptions->Apodize->setChecked(pp->apodize());
_w->_kstFFTOptions->ApodizeFxn->setCurrentIndex(pp->apodizeFxn());
_w->_kstFFTOptions->Sigma->setValue(pp->gaussianSigma());
_w->_kstFFTOptions->RemoveMean->setChecked(pp->removeMean());
_w->_kstFFTOptions->Interleaved->setChecked(pp->average());
_w->_kstFFTOptions->Output->setCurrentIndex(pp->output());
_w->_kstFFTOptions->InterpolateHoles->setChecked(pp->interpolateHoles());
_w->_kstFFTOptions->synch();
pp->unlock();
_w->_curveAppearance->hide();
_w->_curvePlacement->hide();
_legendText->hide();
_legendLabel->hide();
adjustSize();
resize(minimumSizeHint());
setFixedHeight(height());
}
}
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();
}
