1 files changed, 133 insertions, 0 deletions
diff --git a/tools/XBMCTexXBT/XBTFWriter.cpp b/tools/XBMCTexXBT/XBTFWriter.cpp
new file mode 100644
index 0000000000..c196a0837a
--- /dev/null
+++ b/tools/XBMCTexXBT/XBTFWriter.cpp
@@ -0,0 +1,133 @@
+#include "XBTFWriter.h"
+#include "EndianSwap.h"
+
+#define TEMP_FILE "temp.xbt"
+#define TEMP_SIZE (10*1024*1024)
+
+#define WRITE_STR(str, size, file) fwrite(str, size, 1, file)
+#define WRITE_U32(i, file) { unsigned long _n = i; _n = Endian_SwapLE32(i); fwrite(&_n, 4, 1, file); }
+#define WRITE_U64(i, file) { unsigned long long _n = i; _n = Endian_SwapLE64(i); fwrite(&_n, 8, 1, file); }
+
+CXBTFWriter::CXBTFWriter(CXBTF& xbtf, const std::string outputFile) : m_xbtf(xbtf)
+{
+  m_outputFile = outputFile;
+  m_file = NULL;
+}
+
+bool CXBTFWriter::Create()
+{
+  m_file = fopen(m_outputFile.c_str(), "wb");
+  if (m_file == NULL)
+  {
+    return false;
+  }
+  
+  m_tempFile = fopen(TEMP_FILE, "wb");
+  if (m_tempFile == NULL)
+  {
+    return false;
+  }
+  
+  return true;
+}
+
+bool CXBTFWriter::Close()
+{
+  if (m_file == NULL || m_tempFile == NULL)
+  {
+    return false;
+  }
+
+  fclose(m_tempFile);
+  m_tempFile = fopen(TEMP_FILE, "rb");
+  if (m_tempFile == NULL)
+  {
+    return false;
+  }
+  
+  unsigned char* tmp = new unsigned char[10*1024*1024];
+  size_t bytesRead;
+  while ((bytesRead = fread(tmp, 1, TEMP_SIZE, m_tempFile)) > 0)
+  {
+    fwrite(tmp, bytesRead, 1, m_file);
+  }
+  
+  fclose(m_file);
+  fclose(m_tempFile);
+  unlink(TEMP_FILE);
+  
+  return true;
+}
+
+bool CXBTFWriter::AppendContent(unsigned char* data, size_t length)
+{
+  if (m_tempFile == NULL)
+  {
+    return false;
+  }
+  
+  fwrite(data, length, 1, m_tempFile);
+  
+  return true;
+}
+
+bool CXBTFWriter::UpdateHeader()
+{
+  if (m_file == NULL)
+  {
+    return false;
+  }
+  
+  unsigned long long offset = m_xbtf.GetHeaderSize();
+  
+  WRITE_STR(XBTF_MAGIC, 4, m_file);
+  WRITE_STR(XBTF_VERSION, 1, m_file);
+  
+  std::vector<CXBTFFile>& files = m_xbtf.GetFiles();
+  WRITE_U32(files.size(), m_file);
+  for (size_t i = 0; i < files.size(); i++)
+  {
+    CXBTFFile& file = files[i];
+    
+    // Convert path to lower case
+    char* ch = file.GetPath();
+    while (*ch)
+    {
+      *ch = tolower(*ch);
+      ch++;
+    }
+    
+    WRITE_STR(file.GetPath(), 256, m_file);
+    WRITE_U32(file.GetFormat(), m_file);
+    WRITE_U32(file.GetLoop(), m_file);
+    
+    std::vector<CXBTFFrame>& frames = file.GetFrames();
+    WRITE_U32(frames.size(), m_file);
+    for (size_t j = 0; j < frames.size(); j++)
+    {
+      CXBTFFrame& frame = frames[j];
+      frame.SetOffset(offset);
+      offset += frame.GetPackedSize();
+      
+      WRITE_U32(frame.GetWidth(), m_file);
+      WRITE_U32(frame.GetHeight(), m_file);
+      WRITE_U32(frame.GetX(), m_file);
+      WRITE_U32(frame.GetY(), m_file);
+      WRITE_U64(frame.GetPackedSize(), m_file);
+      WRITE_U64(frame.GetUnpackedSize(), m_file);
+      WRITE_U32(frame.GetDuration(), m_file);
+      WRITE_U64(frame.GetOffset(), m_file);
+    }
+  }
+  
+  // Sanity check
+  fpos_t pos;
+  fgetpos(m_file, &pos);
+  if ((unsigned int) pos != m_xbtf.GetHeaderSize())
+  {
+    printf("Expected header size (%llu) != actual size (%llu)\n", m_xbtf.GetHeaderSize(), pos);
+    return false;
+  }
+  
+  return true;
+}