Compare commits
3 Commits
v0.2.13
...
556bb22d26
| Author | SHA1 | Date | |
|---|---|---|---|
| 556bb22d26 | |||
| c6581e1122 | |||
| fa39440b9e |
+1
-1
@@ -16,7 +16,7 @@ option(BUILD_SHARED_LIBS "Build using shared libraries" ON)
|
||||
option(USE_BUNDLED_LIBS "Use bundled LZ4 PugiXML and Zlib. You can still exclude some" ON)
|
||||
cmake_dependent_option(USE_BUNDLED_LZ4 "Use bundled LZ4" ON "USE_BUNDLED_LIBS" OFF)
|
||||
cmake_dependent_option(USE_BUNDLED_PUGIXML "Use bundled PugiXML" ON "USE_BUNDLED_LIBS" OFF)
|
||||
cmake_dependent_option(USE_BUNDLED_ZLIB "Use bundled Zlib" ON "USE_BUNDLED_LIBS" OFF)
|
||||
cmake_dependent_option(USE_BUNDLED_ZLIB "Use bundled Zlib" OFF "USE_BUNDLED_LIBS" OFF)
|
||||
|
||||
find_package(PkgConfig)
|
||||
|
||||
|
||||
+84
-4
@@ -101,6 +101,51 @@ static void byteShuffle(ByteArray &data, int itemSize)
|
||||
}
|
||||
}
|
||||
|
||||
#ifdef __SSE2__
|
||||
#include <x86intrin.h>
|
||||
|
||||
template<int itemSize>
|
||||
size_t byteUnshuffleSSE(const char *in, char *out, size_t num)
|
||||
{
|
||||
//size_t size = num;
|
||||
const __m128i *_in[4] = {(const __m128i*)in,
|
||||
(const __m128i*)(in + num),
|
||||
(const __m128i*)(in + num * 2),
|
||||
(const __m128i*)(in + num * 3)};
|
||||
__m128i *_out = (__m128i*)out;
|
||||
num /= 16;
|
||||
|
||||
for(size_t i = 0; i < num; i++)
|
||||
{
|
||||
if constexpr(itemSize == 4)
|
||||
{
|
||||
__m128i a = _mm_loadu_si128(_in[0] + i);
|
||||
__m128i b = _mm_loadu_si128(_in[1] + i);
|
||||
__m128i c = _mm_loadu_si128(_in[2] + i);
|
||||
__m128i d = _mm_loadu_si128(_in[3] + i);
|
||||
__m128i x = _mm_unpacklo_epi8(a, b);
|
||||
__m128i y = _mm_unpackhi_epi8(a, b);
|
||||
__m128i z = _mm_unpacklo_epi8(c, d);
|
||||
__m128i w = _mm_unpackhi_epi8(c, d);
|
||||
_mm_storeu_si128(_out + i * 4, _mm_unpacklo_epi16(x, z));
|
||||
_mm_storeu_si128(_out + i * 4 + 1, _mm_unpackhi_epi16(x, z));
|
||||
_mm_storeu_si128(_out + i * 4 + 2, _mm_unpacklo_epi16(y, w));
|
||||
_mm_storeu_si128(_out + i * 4 + 3, _mm_unpackhi_epi16(y, w));
|
||||
}
|
||||
if constexpr(itemSize == 2)
|
||||
{
|
||||
__m128i a = _mm_loadu_si128(_in[0] + i);
|
||||
__m128i b = _mm_loadu_si128(_in[1] + i);
|
||||
__m128i x = _mm_unpacklo_epi8(a, b);
|
||||
__m128i y = _mm_unpackhi_epi8(a, b);
|
||||
_mm_storeu_si128(_out + i * 2, x);
|
||||
_mm_storeu_si128(_out + i * 2 + 1, y);
|
||||
}
|
||||
}
|
||||
return num * 16 * itemSize;
|
||||
}
|
||||
#endif
|
||||
|
||||
static void byteUnshuffle(ByteArray &data, int itemSize)
|
||||
{
|
||||
if(itemSize > 1)
|
||||
@@ -108,11 +153,19 @@ static void byteUnshuffle(ByteArray &data, int itemSize)
|
||||
ByteArray &input = data;
|
||||
ByteArray output(input.size());
|
||||
size_t num = input.size() / itemSize;
|
||||
const char *s = input.constData();
|
||||
size_t off = 0;
|
||||
#ifdef __SSE2__
|
||||
if(itemSize == 4)
|
||||
off = byteUnshuffleSSE<4>(input.data(), output.data(), num);
|
||||
if(itemSize == 2)
|
||||
off = byteUnshuffleSSE<2>(input.data(), output.data(), num);
|
||||
#endif
|
||||
|
||||
const char *s = input.constData() + off;
|
||||
for(int i=0; i<itemSize; i++)
|
||||
{
|
||||
char *u = output.data() + i;
|
||||
for(size_t o=0; o<num; o++, s++, u += itemSize)
|
||||
char *u = output.data() + i + off;
|
||||
for(size_t o = off / itemSize; o < num; o++, s++, u += itemSize)
|
||||
*u = *s;
|
||||
}
|
||||
memcpy(output.data() + num * itemSize, s, input.size() % itemSize);
|
||||
@@ -981,11 +1034,11 @@ public:
|
||||
void save(std::ostream &io);
|
||||
void writeImage(const Image &image);
|
||||
static void writeFITSKeyword(pugi::xml_node &node, const FITSKeyword &keyword);
|
||||
static void writePropertyElement(pugi::xml_node &node, const Property &property);
|
||||
private:
|
||||
void writeHeader();
|
||||
void writeImageElement(pugi::xml_node &node, const Image &image);
|
||||
void writeDataBlockAttributes(pugi::xml_node &image_node, const DataBlock &dataBlock);
|
||||
void writePropertyElement(pugi::xml_node &node, const Property &property);
|
||||
void writeMetadata(pugi::xml_node &node);
|
||||
void updateImageAttachmentPos(pugi::xml_node &root, size_t offset);
|
||||
ByteArray _xisfHeader;
|
||||
@@ -1318,6 +1371,7 @@ public:
|
||||
void addFITSKeyword(uint32_t image, const FITSKeyword &keyword);
|
||||
void updateFITSKeyword(uint32_t image, const FITSKeyword &keyword, bool add);
|
||||
void removeFITSKeyword(uint32_t image, const String &name);
|
||||
void updateProperty(uint32_t image, const Property &property);
|
||||
private:
|
||||
void readXISFHeader();
|
||||
void parseAttachmentPos(pugi::xml_node &root);
|
||||
@@ -1498,6 +1552,27 @@ void XISFModifyPrivate::removeFITSKeyword(uint32_t image, const String &name)
|
||||
imageNode.remove_child(keywordNode);
|
||||
}
|
||||
|
||||
void XISFModifyPrivate::updateProperty(uint32_t image, const Property &property)
|
||||
{
|
||||
if(!_root)
|
||||
throw Error("No input file opened");
|
||||
|
||||
pugi::xpath_node_set images = _root.select_nodes("//Image");
|
||||
|
||||
if(image >= images.size())
|
||||
throw Error("Out of bounds");
|
||||
|
||||
pugi::xpath_variable_set variables;
|
||||
variables.set("id", property.id.c_str());
|
||||
pugi::xml_node imageNode = images[image].node();
|
||||
pugi::xml_node propertyNode = imageNode.select_node("Property[@id=string($id)]", &variables).node();
|
||||
|
||||
if(propertyNode)
|
||||
imageNode.remove_child(propertyNode);
|
||||
|
||||
XISFWriterPrivate::writePropertyElement(imageNode, property);
|
||||
}
|
||||
|
||||
void XISFModifyPrivate::readXISFHeader()
|
||||
{
|
||||
char signature[8];
|
||||
@@ -1616,6 +1691,11 @@ void XISFModify::removeFITSKeyword(uint32_t image, const String &name)
|
||||
p->removeFITSKeyword(image, name);
|
||||
}
|
||||
|
||||
void XISFModify::updateProperty(uint32_t image, const Property &property)
|
||||
{
|
||||
p->updateProperty(image, property);
|
||||
}
|
||||
|
||||
#define STRING_ENUM(map, map2, c, e) { map.insert({#e, c::e}); map2.insert({c::e, #e}); }
|
||||
|
||||
struct Init
|
||||
|
||||
@@ -443,6 +443,12 @@ public:
|
||||
* @param name of keyword that will be removed
|
||||
*/
|
||||
void removeFITSKeyword(uint32_t image, const String &name);
|
||||
/**
|
||||
* @brief updateProperty add new or update existing XISF Property
|
||||
* @param image index of image to update
|
||||
* @param property new value of property
|
||||
*/
|
||||
void updateProperty(uint32_t image, const Property &property);
|
||||
private:
|
||||
XISFModifyPrivate *p;
|
||||
};
|
||||
|
||||
@@ -17,7 +17,6 @@
|
||||
************************************************************************/
|
||||
|
||||
#include <charconv>
|
||||
#include <type_traits>
|
||||
#include <map>
|
||||
#include <regex>
|
||||
#include <iomanip>
|
||||
@@ -403,7 +402,6 @@ Variant variantFromString(Variant::Type type, const String &str)
|
||||
|
||||
Variant::Type Variant::type() const
|
||||
{
|
||||
int idx = _value.index();
|
||||
return (Variant::Type)_value.index();
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user