Merge branch 'check_stream_write' into 'master'

Add check whether writing to bitstream succeeded

See merge request !738

source/Lib/EncoderLib/AnnexBwrite.h

@@ -42,6 +42,15 @@
 //! \ingroup EncoderLib
 //! \{
 
+void checkWriteError(std::ostream& out)
+{
+  if (out.fail())
+  {
+    printf ("\nError writing bitstream file\n");
+    exit (EXIT_FAILURE);
+  }
+}
+
 #if JVET_R0294_SUBPIC_HASH
 uint32_t writeAnnexBNalUnit(std::ostream& out, const NALUnitEBSP& nalu, bool useLongStartcode)
 {
@@ -52,14 +61,17 @@ uint32_t writeAnnexBNalUnit(std::ostream& out, const NALUnitEBSP& nalu, bool use
   if (useLongStartcode)
   {
     out.write(reinterpret_cast<const char*>(startCodePrefix), 4);
+    checkWriteError(out);
     size += 4;
   }
   else
   {
     out.write(reinterpret_cast<const char*>(startCodePrefix+1), 3);
+    checkWriteError(out);
     size += 3;
   }
   out << nalu.m_nalUnitData.str();
+  checkWriteError(out);
   size += uint32_t(nalu.m_nalUnitData.str().size());
 
   return size;
@@ -101,11 +113,13 @@ static std::vector<uint32_t> writeAnnexB(std::ostream& out, const AccessUnit& au
        *    7.4.1.2.3.
        */
       out.write(reinterpret_cast<const char*>(start_code_prefix), 4);
+      checkWriteError(out);
       size += 4;
     }
     else
     {
       out.write(reinterpret_cast<const char*>(start_code_prefix+1), 3);
+      checkWriteError(out);
       size += 3;
     }
     out << nalu.m_nalUnitData.str();