nou/tenmon
1
0
Fork 1
Code Issues Pull Requests Projects Releases 33 Wiki Activity

Add RawImage class that handle images as byte array

Browse Source
This commit is contained in:
Dušan Poizl
2019-09-24 18:32:38 +02:00
parent 9c0c0d41c7
commit fa025dff0e
3 changed files with 292 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files
loadrunable.cpp
+108 -42
View File
@@ -5,6 +5,7 @@
#include <QFileInfo>
#include <libexif/exif-data.h>
#include <fitsio2.h>
#include "rawimage.h"
LoadRunable::LoadRunable(const QString &file, Image *receiver) :
m_file(file),
@@ -23,14 +24,86 @@ void loadExifEntry(ImageInfoData &info, ExifContent *content, ExifTag tag)
}
}
QImage loadFITS(QString path, ImageInfoData &info)
bool loadRAW(QString path, ImageInfoData &info, RawImageAbs **image, QImage *qimage)
{
QImage img;
if(!image && !qimage)
return false;
LibRaw raw;
raw.open_file(path.toLocal8Bit().data());
raw.imgdata.params.half_size = true;
raw.imgdata.params.use_camera_wb = true;
raw.imgdata.params.user_flip = 0;
if(raw.unpack())
return false;
libraw_rawdata_t rawdata = raw.imgdata.rawdata;
size_t size = rawdata.sizes.width*rawdata.sizes.height;
std::vector<uint16_t> out;
out.resize(size);
size_t d = 0;
uint h=rawdata.sizes.top_margin+rawdata.sizes.height;
uint w=rawdata.sizes.left_margin+rawdata.sizes.width;
size_t pitch = rawdata.sizes.raw_pitch/sizeof(uint16_t);
for(size_t i=rawdata.sizes.top_margin;i<h;i+=2)
{
for(size_t o=rawdata.sizes.left_margin;o<w;o+=2)
{
uint16_t p = rawdata.raw_image[i*pitch+o];
out[d++] = p;
}
}
if(image)
{
*image = new RawImage<uint16_t>(w, h, out);
}
if(qimage)
{
raw.dcraw_process();
libraw_processed_image_t *rawImg = raw.dcraw_make_mem_image();
QImage img(rawImg->width, rawImg->height, QImage::Format_RGB888);
uint scanLine = rawImg->width*rawImg->colors;
for(uint i=0; i<rawImg->height; i++)
{
memcpy(img.scanLine(i), rawImg->data+(i*scanLine), scanLine);
}
QString shutterSpeed = QString::number(raw.imgdata.other.shutter);
if(raw.imgdata.other.shutter < 1)
{
shutterSpeed = QString("1/%1s").arg(1.0f/raw.imgdata.other.shutter);
}
info.append(StringPair(QObject::tr("Width"), QString::number(rawImg->width)));
info.append(StringPair(QObject::tr("Height"), QString::number(rawImg->height)));
info.append(StringPair(QObject::tr("ISO"), QString::number(raw.imgdata.other.iso_speed)));
info.append(StringPair(QObject::tr("Shutter speed"), shutterSpeed));
#if LIBRAW_MINOR_VERSION>=19
info.append(StringPair(QObject::tr("Camera temperature"), QString::number(raw.imgdata.other.CameraTemperature)));
#endif
raw.dcraw_clear_mem(rawImg);
*qimage = img;
}
return true;
}
bool loadFITS(QString path, ImageInfoData &info, RawImageAbs **image, QImage *qimage)
{
if(!image && !qimage)
return false;
fitsfile *file;
int status = 0;
int type;
fits_open_image(&file, path.toLocal8Bit().data(), READONLY, &status);
fits_get_hdu_type(file, &type, &status);
if(type == IMAGE_HDU)
{
int imgtype;
@@ -38,7 +111,7 @@ QImage loadFITS(QString path, ImageInfoData &info)
long naxes[2];
fits_get_img_param(file, 2, &imgtype, &naxis, naxes, &status);
if(naxis == 2)
if(naxis == 2 && status == 0)
{
std::vector<uint8_t> bits8;
std::vector<uint16_t> bits16;
@@ -46,8 +119,12 @@ QImage loadFITS(QString path, ImageInfoData &info)
long fpixel[2] = {1,1};
size_t size = naxes[0]*naxes[1];
img = QImage(naxes[0], naxes[1], QImage::Format_Grayscale8);
uchar *data = img.bits();
uchar *data = nullptr;
if(qimage)
{
*qimage = QImage(naxes[0], naxes[1], QImage::Format_Grayscale8);
data = qimage->bits();
}
info.append(StringPair(QObject::tr("Width"), QString::number(naxes[0])));
info.append(StringPair(QObject::tr("Height"), QString::number(naxes[1])));
@@ -57,29 +134,45 @@ QImage loadFITS(QString path, ImageInfoData &info)
bits8.resize(size);
fits_read_pix(file, TBYTE, fpixel, size, NULL, &bits8[0], NULL, &status);
if(status)break;
for(size_t i=0; i<size; i++)data[i] = bits8[i];
if(data)
{
for(size_t i=0; i<size; i++)
data[i] = bits8[i];
}
if(image)*image = new RawImage<uint8_t>(naxes[0], naxes[1], bits8);
break;
case SHORT_IMG:
bits16.resize(size);
fits_read_pix(file, TUSHORT, fpixel, size, NULL, &bits16[0], NULL, &status);
if(status)break;
for(size_t i=0; i<size; i++)data[i] = bits16[i] >> 8;
if(data)
{
for(size_t i=0; i<size; i++)
data[i] = bits16[i] >> 8;
}
if(image)*image = new RawImage<uint16_t>(naxes[0], naxes[1], bits16);
break;
case LONG_IMG:
bits32.resize(size);
fits_read_pix(file, TUINT, fpixel, size, NULL, &bits32[0], NULL, &status);
if(status)break;
for(size_t i=0; i<size; i++)data[i] = bits32[i] >> 24;
if(data)
{
for(size_t i=0; i<size; i++)
data[i] = bits32[i] >> 24;
}
if(image)*image = new RawImage<uint32_t>(naxes[0], naxes[1], bits32);
break;
}
}
}
int nheader, more;
int nheader = 0, more;
fits_get_hdrspace(file, &nheader, &more, &status);
for(int i=1; i<=nheader; i++)
{
char key[FLEN_KEYWORD];
char val[FLEN_VALUE];
char comm[FLEN_COMMENT];
fits_read_keyn(file, i, key, val, comm, &status);
@@ -94,7 +187,7 @@ QImage loadFITS(QString path, ImageInfoData &info)
info.append(StringPair(QObject::tr("Error"), QString(err)));
}
return img;
return true;
}
void LoadRunable::run()
@@ -109,42 +202,15 @@ void LoadRunable::run()
if(m_file.endsWith(".CR2", Qt::CaseInsensitive))
{
LibRaw raw;
raw.open_file(m_file.toLocal8Bit().data());
raw.imgdata.params.half_size = true;
raw.imgdata.params.use_camera_wb = true;
raw.imgdata.params.user_flip = 0;
raw.unpack();
raw.dcraw_process();
libraw_processed_image_t *rawImg = raw.dcraw_make_mem_image();
QImage img(rawImg->width, rawImg->height, QImage::Format_RGB888);
QString shutterSpeed = QString::number(raw.imgdata.other.shutter);
if(raw.imgdata.other.shutter < 1)
{
shutterSpeed = QString("1/%1s").arg(1.0f/raw.imgdata.other.shutter);
}
uint scanLine = rawImg->width*rawImg->colors;
for(uint i=0; i<rawImg->height; i++)
{
memcpy(img.scanLine(i), rawImg->data+(i*scanLine), scanLine);
}
info.append(StringPair(QObject::tr("Width"), QString::number(rawImg->width)));
info.append(StringPair(QObject::tr("Height"), QString::number(rawImg->height)));
info.append(StringPair(QObject::tr("ISO"), QString::number(raw.imgdata.other.iso_speed)));
info.append(StringPair(QObject::tr("Shutter speed"), shutterSpeed));
#if LIBRAW_MINOR_VERSION>=19
info.append(StringPair(QObject::tr("Camera temperature"), QString::number(raw.imgdata.other.CameraTemperature)));
#endif
raw.dcraw_clear_mem(rawImg);
QImage img;
loadRAW(m_file, info, nullptr, &img);
QMetaObject::invokeMethod(m_receiver, "imageLoaded", Qt::QueuedConnection, Q_ARG(QImage, img), Q_ARG(ImageInfoData, info));
}
else if(m_file.endsWith(".FIT", Qt::CaseInsensitive))
{
QImage img = loadFITS(m_file, info);
QMetaObject::invokeMethod(m_receiver, "imageLoaded", Qt::QueuedConnection, Q_ARG(QImage, img), Q_ARG(ImageInfoData, info));
QImage img;
if(loadFITS(m_file, info, nullptr, &img))
QMetaObject::invokeMethod(m_receiver, "imageLoaded", Qt::QueuedConnection, Q_ARG(QImage, img), Q_ARG(ImageInfoData, info));
}
else
{