CUDA (Проект)/Описание DDS формата — различия между версиями

#!/usr/bin/python
# -*- coding: UTF-8 -*-
 
import sys
import os
#import pickle
 
if __name__ == "__main__":
    if len (sys.argv) > 1:
        if len (sys.argv) == 4:
            X = int(sys.argv[1]);
            Y = int(sys.argv[2]);
            InFile = sys.argv[3];
        if len (sys.argv) == 5:
            X = int(sys.argv[1]);
            Y = int(sys.argv[2]);
            Type = int(sys.argv[3]);
            InFile = sys.argv[4];
            OutFile = sys.argv[4] + ".txt";
        if len (sys.argv) == 6:
            X = int(sys.argv[1]);
            Y = int(sys.argv[2]);
            Type = int(sys.argv[3]);
            InFile = sys.argv[4];
            OutFile = sys.argv[5];
    else:
        print ("{} Columns Rows [Type] InputFileName [OutputFileName]".format(sys.argv[0]))
        print ("Type: \n"
        "  0 (по умолчанию) — слева направо, сверху вниз;\n"
        "  1 — слева направо, снизу вверх;\n"
        "  2 — сверху вниз, слева направо;\n"
        "  3 — снизу вверх, слева направо")
        sys.exit()
 
#print X, Y, Type, InFile
 
if (int(Type) > 3) or (int(Type) < 0):
    print ("Type is wrong")
    sys.exit()
 
if (int(X) > 1000) or (int(X) < 0):
    print ("Numper of columns is wrong")
    sys.exit()
 
if (int(Y) > 1000) or (int(X) < 0):
    print ("Number of rows is wrong")
    sys.exit()
 
if os.path.isfile(InFile) :
    fi = open( InFile, 'r')
else:
    print ("File " + InFile + "is not exist!")
    sys.exit()
 
 
 
 
sstr = fi.readlines()
 
if (X * Y) != len(sstr) :
    print ("Col (%d) * Row (%d) != Lines number (%d) in File " % (X, Y, len(sstr))  + InFile + "!")
    sys.exit()
 
row_s = 0
row_e = Y
row_d = 1
col_s = 0
col_e = X
col_d = 1
 
 
if Type == 1 :
    row_s = Y-1
    row_e = -1
    row_d = -1
    col_s = 0
    col_e = X
    col_d = 1
elif Type == 2 :
    row_s = 0
    row_e = Y
    row_d = 1
    col_s = 0
    col_e = X
    col_d = 1
elif Type == 3 :
    row_s = Y-1
    row_e = -1
    row_d = -1
    col_s = 0
    col_e = X
    col_d = 1
 
 
fo = open( OutFile, 'w')
 
# Create matrix
m = [[] for i in range(2010)]
for i in range(Y):
    for j in range(X):
        m[i].append(-1)
 
# Init matrix
if (Type == 0) or (Type == 1) :
    row = row_s
    i=0
    while row != row_e:
        col = col_s
        while col != col_e:
            print(row, col, sstr[i])
            m[row][col] = sstr[i]
            i = i + 1
            col = col + col_d
        row = row + row_d
 
if (Type == 2) or (Type == 3) :
    col = col_s
    i=0
    while col != col_e:
        row = row_s
        while row != row_e:
            print(row, col, sstr[i])
            m[row][col] = sstr[i]
            i = i + 1
            row = row + row_d
        col = col + col_d
 
# Write matrix to file
for yy in range (Y) :
    for xx in range(X) :
        fo.write(m[yy][xx])
 
# Test print
for yy in range (Y):
    print m[yy]
 
fo.close()
fi.close()