BS 0.0.1 EasySave完成
This commit is contained in:
@@ -3,299 +3,302 @@ using System.Collections.Generic;
|
||||
using System.IO;
|
||||
using Convention.EasySave.Internal;
|
||||
|
||||
public class EasySaveSpreadsheet
|
||||
namespace Convention.EasySave
|
||||
{
|
||||
private int cols = 0;
|
||||
private int rows = 0;
|
||||
private Dictionary<Index, string> cells = new Dictionary<Index, string>();
|
||||
|
||||
private const string QUOTE = "\"";
|
||||
private const char QUOTE_CHAR = '"';
|
||||
private const char COMMA_CHAR = ',';
|
||||
private const char NEWLINE_CHAR = '\n';
|
||||
private const string ESCAPED_QUOTE = "\"\"";
|
||||
private static char[] CHARS_TO_ESCAPE = { ',', '"', '\n', ' ' };
|
||||
|
||||
public int ColumnCount
|
||||
public class EasySaveSpreadsheet
|
||||
{
|
||||
get{ return cols; }
|
||||
}
|
||||
private int cols = 0;
|
||||
private int rows = 0;
|
||||
private Dictionary<Index, string> cells = new Dictionary<Index, string>();
|
||||
|
||||
public int RowCount
|
||||
{
|
||||
get{ return rows; }
|
||||
}
|
||||
private const string QUOTE = "\"";
|
||||
private const char QUOTE_CHAR = '"';
|
||||
private const char COMMA_CHAR = ',';
|
||||
private const char NEWLINE_CHAR = '\n';
|
||||
private const string ESCAPED_QUOTE = "\"\"";
|
||||
private static char[] CHARS_TO_ESCAPE = { ',', '"', '\n', ' ' };
|
||||
|
||||
public int GetColumnLength(int col)
|
||||
{
|
||||
if (col >= cols)
|
||||
return 0;
|
||||
|
||||
int maxRow = -1;
|
||||
|
||||
foreach(var index in cells.Keys)
|
||||
if (index.col == col && index.row > maxRow)
|
||||
maxRow = index.row;
|
||||
|
||||
return maxRow+1;
|
||||
}
|
||||
|
||||
public int GetRowLength(int row)
|
||||
{
|
||||
if (row >= rows)
|
||||
return 0;
|
||||
|
||||
int maxCol = -1;
|
||||
|
||||
foreach (var index in cells.Keys)
|
||||
if (index.row == row && index.col > maxCol)
|
||||
maxCol = index.col;
|
||||
|
||||
return maxCol + 1;
|
||||
}
|
||||
|
||||
public void SetCell(int col, int row, object value)
|
||||
{
|
||||
var type = value.GetType();
|
||||
|
||||
// If we're writing a string, add it without formatting.
|
||||
if (type == typeof(string))
|
||||
{
|
||||
SetCellString(col, row, (string)value);
|
||||
return;
|
||||
}
|
||||
|
||||
var settings = new EasySaveSettings();
|
||||
if (EasySaveReflection.IsPrimitive(type))
|
||||
SetCellString(col, row, value.ToString());
|
||||
else
|
||||
SetCellString(col, row, settings.encoding.GetString(EasySave.Serialize(value, EasySaveTypeMgr.GetOrCreateEasySaveType(type))));
|
||||
|
||||
// Expand the spreadsheet if necessary.
|
||||
if (col >= cols)
|
||||
cols = (col + 1);
|
||||
if (row >= rows)
|
||||
rows = (row + 1);
|
||||
}
|
||||
|
||||
private void SetCellString(int col, int row, string value)
|
||||
{
|
||||
cells [new Index (col, row)] = value;
|
||||
|
||||
// Expand the spreadsheet if necessary.
|
||||
if(col >= cols)
|
||||
cols = (col+1);
|
||||
if (row >= rows)
|
||||
rows = (row + 1);
|
||||
}
|
||||
|
||||
|
||||
// Don't create non-generic version of this. Generic parameter is necessary as no type data is stored in the CSV file.
|
||||
public T GetCell<T>(int col, int row)
|
||||
{
|
||||
var val = GetCell(typeof(T), col, row);
|
||||
|
||||
if (val == null)
|
||||
return default(T);
|
||||
return (T)val;
|
||||
}
|
||||
|
||||
public object GetCell(System.Type type, int col, int row)
|
||||
{
|
||||
string value;
|
||||
|
||||
if (col >= cols || row >= rows)
|
||||
throw new System.IndexOutOfRangeException("Cell (" + col + ", " + row + ") is out of bounds of spreadsheet (" + cols + ", " + rows + ").");
|
||||
|
||||
if (!cells.TryGetValue(new Index(col, row), out value) || value == null)
|
||||
return null;
|
||||
|
||||
// If we're loading a string, simply return the string value.
|
||||
if (type == typeof(string))
|
||||
{
|
||||
var str = (object)value;
|
||||
return str;
|
||||
}
|
||||
|
||||
var settings = new EasySaveSettings();
|
||||
return EasySave.Deserialize(EasySaveTypeMgr.GetOrCreateEasySaveType(type, true), settings.encoding.GetBytes(value), settings);
|
||||
}
|
||||
|
||||
public void Load(string filePath)
|
||||
{
|
||||
Load(new EasySaveSettings (filePath));
|
||||
}
|
||||
|
||||
public void Load(string filePath, EasySaveSettings settings)
|
||||
{
|
||||
Load(new EasySaveSettings (filePath, settings));
|
||||
}
|
||||
|
||||
public void Load(EasySaveSettings settings)
|
||||
{
|
||||
Load(EasySaveStream.CreateStream(settings, EasySaveFileMode.Read), settings);
|
||||
}
|
||||
|
||||
public void LoadRaw(string str)
|
||||
{
|
||||
Load(new MemoryStream (((new EasySaveSettings ()).encoding).GetBytes(str)), new EasySaveSettings());
|
||||
}
|
||||
|
||||
public void LoadRaw(string str, EasySaveSettings settings)
|
||||
{
|
||||
Load(new MemoryStream ((settings.encoding).GetBytes(str)), settings);
|
||||
}
|
||||
|
||||
private void Load(Stream stream, EasySaveSettings settings)
|
||||
{
|
||||
using (var reader = new StreamReader(stream))
|
||||
public int ColumnCount
|
||||
{
|
||||
int c_int;
|
||||
char c;
|
||||
string value = "";
|
||||
int col = 0;
|
||||
int row = 0;
|
||||
get { return cols; }
|
||||
}
|
||||
|
||||
// Read until the end of the stream.
|
||||
while(true)
|
||||
public int RowCount
|
||||
{
|
||||
get { return rows; }
|
||||
}
|
||||
|
||||
public int GetColumnLength(int col)
|
||||
{
|
||||
if (col >= cols)
|
||||
return 0;
|
||||
|
||||
int maxRow = -1;
|
||||
|
||||
foreach (var index in cells.Keys)
|
||||
if (index.col == col && index.row > maxRow)
|
||||
maxRow = index.row;
|
||||
|
||||
return maxRow + 1;
|
||||
}
|
||||
|
||||
public int GetRowLength(int row)
|
||||
{
|
||||
if (row >= rows)
|
||||
return 0;
|
||||
|
||||
int maxCol = -1;
|
||||
|
||||
foreach (var index in cells.Keys)
|
||||
if (index.row == row && index.col > maxCol)
|
||||
maxCol = index.col;
|
||||
|
||||
return maxCol + 1;
|
||||
}
|
||||
|
||||
public void SetCell(int col, int row, object value)
|
||||
{
|
||||
var type = value.GetType();
|
||||
|
||||
// If we're writing a string, add it without formatting.
|
||||
if (type == typeof(string))
|
||||
{
|
||||
c_int = reader.Read();
|
||||
c = (char)c_int;
|
||||
if(c == QUOTE_CHAR)
|
||||
{
|
||||
while (true)
|
||||
{
|
||||
c = (char)reader.Read();
|
||||
SetCellString(col, row, (string)value);
|
||||
return;
|
||||
}
|
||||
|
||||
if(c == QUOTE_CHAR)
|
||||
{
|
||||
// If this quote isn't escaped by another, it is the last quote, so we should stop parsing this value.
|
||||
if(((char)reader.Peek()) != QUOTE_CHAR)
|
||||
break;
|
||||
else
|
||||
c = (char)reader.Read();
|
||||
}
|
||||
value += c;
|
||||
}
|
||||
}
|
||||
// If this is the end of a column, row, or the stream, add the value to the spreadsheet.
|
||||
else if(c == COMMA_CHAR || c == NEWLINE_CHAR || c_int == -1)
|
||||
var settings = new EasySaveSettings();
|
||||
if (EasySaveReflection.IsPrimitive(type))
|
||||
SetCellString(col, row, value.ToString());
|
||||
else
|
||||
SetCellString(col, row, settings.encoding.GetString(EasySave.Serialize(value, EasySaveTypeMgr.GetOrCreateEasySaveType(type))));
|
||||
|
||||
// Expand the spreadsheet if necessary.
|
||||
if (col >= cols)
|
||||
cols = (col + 1);
|
||||
if (row >= rows)
|
||||
rows = (row + 1);
|
||||
}
|
||||
|
||||
private void SetCellString(int col, int row, string value)
|
||||
{
|
||||
cells[new Index(col, row)] = value;
|
||||
|
||||
// Expand the spreadsheet if necessary.
|
||||
if (col >= cols)
|
||||
cols = (col + 1);
|
||||
if (row >= rows)
|
||||
rows = (row + 1);
|
||||
}
|
||||
|
||||
|
||||
// Don't create non-generic version of this. Generic parameter is necessary as no type data is stored in the CSV file.
|
||||
public T GetCell<T>(int col, int row)
|
||||
{
|
||||
var val = GetCell(typeof(T), col, row);
|
||||
|
||||
if (val == null)
|
||||
return default(T);
|
||||
return (T)val;
|
||||
}
|
||||
|
||||
public object GetCell(System.Type type, int col, int row)
|
||||
{
|
||||
string value;
|
||||
|
||||
if (col >= cols || row >= rows)
|
||||
throw new System.IndexOutOfRangeException("Cell (" + col + ", " + row + ") is out of bounds of spreadsheet (" + cols + ", " + rows + ").");
|
||||
|
||||
if (!cells.TryGetValue(new Index(col, row), out value) || value == null)
|
||||
return null;
|
||||
|
||||
// If we're loading a string, simply return the string value.
|
||||
if (type == typeof(string))
|
||||
{
|
||||
var str = (object)value;
|
||||
return str;
|
||||
}
|
||||
|
||||
var settings = new EasySaveSettings();
|
||||
return EasySave.Deserialize(EasySaveTypeMgr.GetOrCreateEasySaveType(type, true), settings.encoding.GetBytes(value), settings);
|
||||
}
|
||||
|
||||
public void Load(string filePath)
|
||||
{
|
||||
Load(new EasySaveSettings(filePath));
|
||||
}
|
||||
|
||||
public void Load(string filePath, EasySaveSettings settings)
|
||||
{
|
||||
Load(new EasySaveSettings(filePath, settings));
|
||||
}
|
||||
|
||||
public void Load(EasySaveSettings settings)
|
||||
{
|
||||
Load(EasySaveStream.CreateStream(settings, EasySaveFileMode.Read), settings);
|
||||
}
|
||||
|
||||
public void LoadRaw(string str)
|
||||
{
|
||||
Load(new MemoryStream(((new EasySaveSettings()).encoding).GetBytes(str)), new EasySaveSettings());
|
||||
}
|
||||
|
||||
public void LoadRaw(string str, EasySaveSettings settings)
|
||||
{
|
||||
Load(new MemoryStream((settings.encoding).GetBytes(str)), settings);
|
||||
}
|
||||
|
||||
private void Load(Stream stream, EasySaveSettings settings)
|
||||
{
|
||||
using (var reader = new StreamReader(stream))
|
||||
{
|
||||
int c_int;
|
||||
char c;
|
||||
string value = "";
|
||||
int col = 0;
|
||||
int row = 0;
|
||||
|
||||
// Read until the end of the stream.
|
||||
while (true)
|
||||
{
|
||||
SetCell(col, row, value);
|
||||
value = "";
|
||||
if(c == COMMA_CHAR)
|
||||
col++;
|
||||
else if(c == NEWLINE_CHAR)
|
||||
c_int = reader.Read();
|
||||
c = (char)c_int;
|
||||
if (c == QUOTE_CHAR)
|
||||
{
|
||||
col = 0;
|
||||
row++;
|
||||
while (true)
|
||||
{
|
||||
c = (char)reader.Read();
|
||||
|
||||
if (c == QUOTE_CHAR)
|
||||
{
|
||||
// If this quote isn't escaped by another, it is the last quote, so we should stop parsing this value.
|
||||
if (((char)reader.Peek()) != QUOTE_CHAR)
|
||||
break;
|
||||
else
|
||||
c = (char)reader.Read();
|
||||
}
|
||||
value += c;
|
||||
}
|
||||
}
|
||||
// If this is the end of a column, row, or the stream, add the value to the spreadsheet.
|
||||
else if (c == COMMA_CHAR || c == NEWLINE_CHAR || c_int == -1)
|
||||
{
|
||||
SetCell(col, row, value);
|
||||
value = "";
|
||||
if (c == COMMA_CHAR)
|
||||
col++;
|
||||
else if (c == NEWLINE_CHAR)
|
||||
{
|
||||
col = 0;
|
||||
row++;
|
||||
}
|
||||
else
|
||||
break;
|
||||
}
|
||||
else
|
||||
break;
|
||||
value += c;
|
||||
}
|
||||
else
|
||||
value += c;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public void Save(string filePath)
|
||||
{
|
||||
Save(new EasySaveSettings (filePath), false);
|
||||
}
|
||||
|
||||
public void Save(string filePath, EasySaveSettings settings)
|
||||
{
|
||||
Save(new EasySaveSettings (filePath, settings), false);
|
||||
}
|
||||
|
||||
public void Save(EasySaveSettings settings)
|
||||
{
|
||||
Save(settings, false);
|
||||
}
|
||||
|
||||
public void Save(string filePath, bool append)
|
||||
{
|
||||
Save(new EasySaveSettings (filePath), append);
|
||||
}
|
||||
|
||||
public void Save(string filePath, EasySaveSettings settings, bool append)
|
||||
{
|
||||
Save(new EasySaveSettings (filePath, settings), append);
|
||||
}
|
||||
|
||||
public void Save(EasySaveSettings settings, bool append)
|
||||
{
|
||||
using (var writer = new StreamWriter(EasySaveStream.CreateStream(settings, append ? EasySaveFileMode.Append : EasySaveFileMode.Write)))
|
||||
public void Save(string filePath)
|
||||
{
|
||||
// If data already exists and we're appending, we need to prepend a newline.
|
||||
if(append && EasySave.FileExists(settings))
|
||||
writer.Write(NEWLINE_CHAR);
|
||||
Save(new EasySaveSettings(filePath), false);
|
||||
}
|
||||
|
||||
var array = ToArray();
|
||||
for(int row = 0; row < rows; row++)
|
||||
public void Save(string filePath, EasySaveSettings settings)
|
||||
{
|
||||
Save(new EasySaveSettings(filePath, settings), false);
|
||||
}
|
||||
|
||||
public void Save(EasySaveSettings settings)
|
||||
{
|
||||
Save(settings, false);
|
||||
}
|
||||
|
||||
public void Save(string filePath, bool append)
|
||||
{
|
||||
Save(new EasySaveSettings(filePath), append);
|
||||
}
|
||||
|
||||
public void Save(string filePath, EasySaveSettings settings, bool append)
|
||||
{
|
||||
Save(new EasySaveSettings(filePath, settings), append);
|
||||
}
|
||||
|
||||
public void Save(EasySaveSettings settings, bool append)
|
||||
{
|
||||
using (var writer = new StreamWriter(EasySaveStream.CreateStream(settings, append ? EasySaveFileMode.Append : EasySaveFileMode.Write)))
|
||||
{
|
||||
if(row != 0)
|
||||
// If data already exists and we're appending, we need to prepend a newline.
|
||||
if (append && EasySave.FileExists(settings))
|
||||
writer.Write(NEWLINE_CHAR);
|
||||
|
||||
for(int col = 0; col < cols; col++)
|
||||
var array = ToArray();
|
||||
for (int row = 0; row < rows; row++)
|
||||
{
|
||||
if(col != 0)
|
||||
writer.Write(COMMA_CHAR);
|
||||
if (row != 0)
|
||||
writer.Write(NEWLINE_CHAR);
|
||||
|
||||
writer.Write( Escape(array [col, row]) );
|
||||
for (int col = 0; col < cols; col++)
|
||||
{
|
||||
if (col != 0)
|
||||
writer.Write(COMMA_CHAR);
|
||||
|
||||
writer.Write(Escape(array[col, row]));
|
||||
}
|
||||
}
|
||||
}
|
||||
if (!append)
|
||||
EasySaveIO.CommitBackup(settings);
|
||||
}
|
||||
if(!append)
|
||||
EasySaveIO.CommitBackup(settings);
|
||||
}
|
||||
|
||||
private static string Escape(string str, bool isAlreadyWrappedInQuotes=false)
|
||||
{
|
||||
if (str == "")
|
||||
return "\"\"";
|
||||
else if(str == null)
|
||||
return null;
|
||||
|
||||
// Now escape any other quotes.
|
||||
if(str.Contains(QUOTE))
|
||||
str = str.Replace(QUOTE, ESCAPED_QUOTE);
|
||||
|
||||
// If there's chars to escape, wrap the value in quotes.
|
||||
if(str.IndexOfAny(CHARS_TO_ESCAPE) > -1)
|
||||
str = QUOTE + str + QUOTE;
|
||||
return str;
|
||||
}
|
||||
|
||||
private static string Unescape(string str)
|
||||
{
|
||||
if(str.StartsWith(QUOTE) && str.EndsWith(QUOTE))
|
||||
private static string Escape(string str, bool isAlreadyWrappedInQuotes = false)
|
||||
{
|
||||
str = str.Substring(1, str.Length-2);
|
||||
if(str.Contains(ESCAPED_QUOTE))
|
||||
str = str.Replace(ESCAPED_QUOTE, QUOTE);
|
||||
if (str == "")
|
||||
return "\"\"";
|
||||
else if (str == null)
|
||||
return null;
|
||||
|
||||
// Now escape any other quotes.
|
||||
if (str.Contains(QUOTE))
|
||||
str = str.Replace(QUOTE, ESCAPED_QUOTE);
|
||||
|
||||
// If there's chars to escape, wrap the value in quotes.
|
||||
if (str.IndexOfAny(CHARS_TO_ESCAPE) > -1)
|
||||
str = QUOTE + str + QUOTE;
|
||||
return str;
|
||||
}
|
||||
return str;
|
||||
}
|
||||
|
||||
private string[,] ToArray()
|
||||
{
|
||||
var array = new string[cols, rows];
|
||||
foreach (var cell in cells)
|
||||
array [cell.Key.col, cell.Key.row] = cell.Value;
|
||||
return array;
|
||||
}
|
||||
|
||||
protected struct Index
|
||||
{
|
||||
public int col;
|
||||
public int row;
|
||||
|
||||
public Index(int col, int row)
|
||||
private static string Unescape(string str)
|
||||
{
|
||||
this.col = col;
|
||||
this.row = row;
|
||||
if (str.StartsWith(QUOTE) && str.EndsWith(QUOTE))
|
||||
{
|
||||
str = str.Substring(1, str.Length - 2);
|
||||
if (str.Contains(ESCAPED_QUOTE))
|
||||
str = str.Replace(ESCAPED_QUOTE, QUOTE);
|
||||
}
|
||||
return str;
|
||||
}
|
||||
|
||||
private string[,] ToArray()
|
||||
{
|
||||
var array = new string[cols, rows];
|
||||
foreach (var cell in cells)
|
||||
array[cell.Key.col, cell.Key.row] = cell.Value;
|
||||
return array;
|
||||
}
|
||||
|
||||
protected struct Index
|
||||
{
|
||||
public int col;
|
||||
public int row;
|
||||
|
||||
public Index(int col, int row)
|
||||
{
|
||||
this.col = col;
|
||||
this.row = row;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user