PlottingTools/Class_DtTable.py at master · cms-mual/PlottingTools · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
from Util_Html import *
from Util_Tex import *

class DtTable:

  def __init__(self):
    self.data       = []
    for wheel in -2, -1, 0, +1, +2:
      self.data.append([])
      for station in 1, 2, 3, 4:
        self.data[wheel+2].append([])
        if station != 4:
          for sector in 1,2,3,4,5,6,7,8,9,10,11,12:
            self.data[wheel+2][station-1].append(None)
        else:
          for sector in 1,2,3,4,5,6,7,8,9,10,11,12,13,14:
            self.data[wheel+2][station-1].append(None)

  def FillDt(self, wheel, station, sector, data):
    self.data[wheel+2][station-1][sector-1] = data

  def PrintData(self):
    print self.data

  def PrintHtml(self, htmlFileName, caption="DT Table Caption", isComplete=0):
    if isComplete != 0:
      PrintHtmlHeader(htmlFileName)

    htmlFile=open(htmlFileName, 'a')
    print >> htmlFile, "<table border=\"1\" cellpadding=\"5\">"
    print >> htmlFile, "<caption>%s</caption>" % caption
    print >> htmlFile, "<tr align=center>"
    print >> htmlFile, "<th rowspan=\"2\"><i>wheels</i></th> <th rowspan=\"2\"><i>stations</i></th> <th colspan=\"14\"><i>sectors in stations 1,2,3</i></th>"
    print >> htmlFile, "</tr>"
    print >> htmlFile, "<tr align=center>"
    for sector in 1,2,3,4,5,6,7,8,9,10,11,12:
      if sector == 4 or sector == 10:
        print >> htmlFile, "<th colspan=\"2\"><i>%s</i></th>" % sector
      else:
        print >> htmlFile, "<th><i>%s</i></th>" % sector
    print >> htmlFile, "</tr>"
    for wheel in -2, -1, 0, +1, +2:
      for station in 1, 2, 3, 4:
        print >> htmlFile, "<tr align=center>"
        if station == 1: print >> htmlFile, "<th rowspan=\"4\"><i>MB%s</i></th>" % wheel
        print >> htmlFile, "<th><i>MB%s/%s</i></th>" % (wheel, station)
        if station != 4:
          for sector in 1,2,3,4,5,6,7,8,9,10,11,12:
            if sector == 4 or sector == 10:
              print >> htmlFile, "<td colspan=\"2\">%s</td>" % self.data[wheel+2][station-1][sector-1]
            else:
              print >> htmlFile, "<td>%s</td>" % self.data[wheel+2][station-1][sector-1]
        else:
          for sector in 1,2,3,13,4,5,6,7,8,9,14,10,11,12:
            print >> htmlFile, "<td>%s</td>" % self.data[wheel+2][station-1][sector-1]
      print >> htmlFile, "</tr>"
    print >> htmlFile, "<tr align=center>"
    print >> htmlFile, "<th rowspan=\"2\"><i>wheels</i></th> <th rowspan=\"2\"><i>stations</i></th>"
    for sector in 1,2,3,13,4,5,6,7,8,9,14,10,11,12:
      print >> htmlFile, "<th><i>%s</i></th>" % sector
    print >> htmlFile, "</tr>"
    print >> htmlFile, "<tr align=center>"
    print >> htmlFile, "<th colspan=\"14\"><i>sectors in station 4</i></th>"
    print >> htmlFile, "</tr>"
    print >> htmlFile, "</table>"
    htmlFile.close()

    if isComplete != 0:
      PrintHtmlTrailer(htmlFileName)

  def PrintTex(self, texFileName, caption="DT Table Caption", label="dtTable", isComplete = 0):
    if isComplete != 0:
      PrintTexHeader(texFileName)

    texFile=open(texFileName, 'a')
    print >> texFile, "\\begin{landscape}"
    print >> texFile, "\\begin{table}[tbh]"
    print >> texFile, "\\cprotect\\caption{"+caption+" \\label{tab:"+label+"}}"
    print >> texFile, "\\begin{center}"
    print >> texFile, "\\begin{tabular}{|c|c|c|c|c|c|c|c|c|c|c|c|c|c|c|c|}"
    print >> texFile, "\\hline"
    print >> texFile, (" &  & \\multicolumn{14}{c|}{\\textit{\\textbf{sectors in stations 1,2,3}}} \\\\ \\cline{3-16}")
    print >> texFile, ("\\textit{\\textbf{wheels}} & \\textit{\\textbf{stations}}"),
    for sector in 1,2,3,4,5,6,7,8,9,10,11,12:
      if sector == 4 or sector == 10:
        print >> texFile, (" & \\multicolumn{2}{c|}{\\textit{\\textbf{%s}}}" % sector),
      else:
        print >> texFile, (" & \\textit{\\textbf{%s}}" % sector),
    print >> texFile, "\\\\ \\hline \\hline"
    for wheel in -2, -1, 0, +1, +2:
      for station in 1, 2, 3, 4:
        if station == 1: print >> texFile, ("\\multirow{4}{*}{\\textit{\\textbf{MB%s}}}" % wheel),
        print >> texFile, (" & \\textit{\\textbf{MB%s/%s}}" % (wheel, station)),
        if station != 4:
          for sector in 1,2,3,4,5,6,7,8,9,10,11,12:
            if sector == 4 or sector == 10:
              print >> texFile, (" & \\multicolumn{2}{c|}{$%s$}" % self.data[wheel+2][station-1][sector-1]),
            else:
              print >> texFile, (" & $%s$" % self.data[wheel+2][station-1][sector-1]),
        else:
          for sector in 1,2,3,13,4,5,6,7,8,9,14,10,11,12:
            print >> texFile, (" & $%s$" % self.data[wheel+2][station-1][sector-1]),
        if station != 4: print >> texFile, " \\\\ \cline{2-16}"
        else:            print >> texFile, " \\\\ \hline \hline"
    print >> texFile, ("\\textit{\\textbf{wheels}}"),
    print >> texFile, (" & \\textit{\\textbf{stations}}"),
    for sector in 1,2,3,13,4,5,6,7,8,9,14,10,11,12:
      print >> texFile, (" & \\textit{\\textbf{%s}}" % sector),
    print >> texFile, "\\\\ \cline{3-16}"
    print >> texFile, (" &  & \\multicolumn{14}{c|}{\\textit{\\textbf{sectors in station 4}}} \\\\")
    print >> texFile, "\\hline"
    print >> texFile, "\\end{tabular}"
    print >> texFile, "\\end{center}"
    print >> texFile, "\\end{table}"
    print >> texFile, "\\end{landscape}"
    texFile.close()

    if isComplete != 0:
      PrintTexTrailer(texFileName)