Better handling of FITS records

imageinfo.cpp

@@ -4,6 +4,7 @@
 #include <QHeaderView>
 #include <wcslib/wcshdr.h>
 #include <wcslib/wcsfix.h>
+#include "pcl/FITSHeaderKeyword.h"
 
 static const QVector<QByteArray> noEditableKey = {"SIMPLE", "BITPIX", "NAXIS", "NAXIS1", "NAXIS2", "NAXIS3", "EXTEND", "BZERO", "BSCALE"};
 
@@ -12,6 +13,44 @@ bool FITSRecord::editable() const
     return noEditableKey.count(key);
 }
 
+FITSRecord::FITSRecord(const QByteArray &key, const QVariant &value, const QByteArray &comment) :
+    key(key), value(value), comment(comment)
+{
+}
+
+FITSRecord::FITSRecord(const pcl::FITSHeaderKeyword &record)
+{
+    key = record.name.c_str();
+    comment = record.comment.c_str();
+
+    QString string = record.value.c_str();
+    if(string.startsWith('\'') && string.endsWith('\''))
+    {
+        string.chop(1);
+        string.remove(0, 1);
+    }
+    bool isint;
+    bool isdouble;
+    double vald = string.toDouble(&isdouble);
+    long long vall = string.toLongLong(&isint);
+    if(isint)
+        value = vall;
+    else if(isdouble)
+        value = vald;
+    else if(string == "T" || string == "F")
+        value = string == "T";
+    else
+        value = string;
+}
+
+QByteArray FITSRecord::valueToByteArray() const
+{
+    if(value.type() == QVariant::Bool)
+        return value.toBool() ? "T" : "F";
+    else
+        return value.toString().toLatin1();
+}
+
 ImageInfo::ImageInfo(QWidget *parent) : QTreeWidget(parent)
 {
     setColumnCount(3);
@@ -86,16 +125,10 @@ WCSData::WCSData(int width, int height, const QVector<FITSRecord> &header) :
     {
         if(record.key.startsWith("PV"))continue;
 
-        QByteArray value = record.value.toString().toLatin1();
-        if(value.startsWith('\'') && value.endsWith('\''))
-        {
-            value.chop(1);
-            value = value.remove(0, 1);
-        }
         QByteArray rec;
         rec.append(record.key.leftJustified(8, ' '));
         rec.append("= ");
-        rec.append(value);
+        rec.append(record.value.toString().toLatin1());
         rec.append(" / ");
         rec.append(record.comment);
         str.append(rec.leftJustified(80, ' ', true));