190 lines
2.6 KiB
C++
190 lines
2.6 KiB
C++
|
|
#include <lib/lib.h>
|
|
|
|
FILE *open_file (const std::string &file, const char *mode)
|
|
{
|
|
FILE *fp = fopen (file.c_str (), mode);
|
|
if (fp == 0)
|
|
{
|
|
stderror ("fopen: %s", file.c_str ());
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
return fp;
|
|
}
|
|
|
|
void close_file (FILE *fp)
|
|
{
|
|
fclose (fp);
|
|
}
|
|
|
|
|
|
writer::writer (const std::string &file)
|
|
: fp(0)
|
|
{
|
|
fp = open_file (file, "w");
|
|
}
|
|
|
|
writer::~writer ()
|
|
{
|
|
if (fp)
|
|
{
|
|
fclose (fp);
|
|
fp = 0;
|
|
}
|
|
}
|
|
|
|
reader::reader (const std::string &file)
|
|
: fp(0)
|
|
{
|
|
fp = open_file (file, "r");
|
|
}
|
|
|
|
reader::~reader ()
|
|
{
|
|
if (fp)
|
|
{
|
|
fclose (fp);
|
|
fp = 0;
|
|
}
|
|
}
|
|
|
|
void
|
|
writer::write_char (char x)
|
|
{
|
|
if (fwrite (&x, sizeof x, 1, fp) != 1)
|
|
{
|
|
stderror ("fwrite");
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
void
|
|
writer::write_bool (bool x)
|
|
{
|
|
if (fwrite (&x, sizeof x, 1, fp) != 1)
|
|
{
|
|
stderror ("fwrite");
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
void
|
|
writer::write_int (int x)
|
|
{
|
|
if (fwrite (&x, sizeof x, 1, fp) != 1)
|
|
{
|
|
stderror ("fwrite");
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
void
|
|
writer::write_unsigned (unsigned x)
|
|
{
|
|
if (fwrite (&x, sizeof x, 1, fp) != 1)
|
|
{
|
|
stderror ("fwrite");
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
void
|
|
writer::write_uint64 (uint64 x)
|
|
{
|
|
if (fwrite (&x, sizeof x, 1, fp) != 1)
|
|
{
|
|
stderror ("fwrite");
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
bool
|
|
reader::read_bool ()
|
|
{
|
|
bool x;
|
|
if (fread (&x, sizeof x, 1, fp) != 1)
|
|
{
|
|
stderror ("fread");
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
return x;
|
|
}
|
|
|
|
char
|
|
reader::read_char ()
|
|
{
|
|
char x;
|
|
if (fread (&x, sizeof x, 1, fp) != 1)
|
|
{
|
|
stderror ("fread");
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
return x;
|
|
}
|
|
|
|
int
|
|
reader::read_int ()
|
|
{
|
|
int x;
|
|
if (fread (&x, sizeof x, 1, fp) != 1)
|
|
{
|
|
stderror ("fread");
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
return x;
|
|
}
|
|
|
|
unsigned
|
|
reader::read_unsigned ()
|
|
{
|
|
unsigned x;
|
|
if (fread (&x, sizeof x, 1, fp) != 1)
|
|
{
|
|
stderror ("fread");
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
return x;
|
|
}
|
|
|
|
uint64
|
|
reader::read_uint64 ()
|
|
{
|
|
uint64 x;
|
|
if (fread (&x, sizeof x, 1, fp) != 1)
|
|
{
|
|
stderror ("fread");
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
return x;
|
|
}
|
|
|
|
void
|
|
read (reader &r, std::string &s)
|
|
{
|
|
unsigned n = r.read_unsigned ();
|
|
|
|
char buf[n + 1];
|
|
unsigned k = fread (buf, sizeof (char), n + 1, r.fp);
|
|
if (k != n + 1)
|
|
{
|
|
stderror ("fread");
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
assert (buf[n] == 0);
|
|
|
|
s = std::string (buf);
|
|
}
|
|
|
|
void
|
|
write (writer &w, const std::string &s)
|
|
{
|
|
unsigned n = s.length ();
|
|
w.write_unsigned (n);
|
|
|
|
if (fwrite (s.c_str (), sizeof (char), n + 1, w.fp) != n + 1)
|
|
{
|
|
stderror ("fwrite");
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
}
|