OSDN Git Service

not exporting qdt file
[qcad/qcad.git] / compiler / qcrun.cpp
1 //----------------------------------------------------------------------------\r
2 //  qcrun.cpp\r
3 //  Interpreter for ntermediate code\r
4 //  $Date: 2003/02/20 00:24:16 $\r
5 //  $Revision: 1.8 $\r
6 //----------------------------------------------------------------------------\r
7 #include <iostream>\r
8 #include <iomanip>\r
9 #include <fstream>\r
10 #include <cstdlib>\r
11 #include <vector>\r
12 #include <string>\r
13 \r
14 #include "QBits.h"\r
15 #include "QCompilerCntl.h"\r
16 #include "QCalcUnit.h"\r
17 \r
18 typedef unsigned int                Appoptions;\r
19 typedef std::vector<QCalcUnit *>    QCalcUnits;\r
20 \r
21 const char * const  app_name        = "qcrun";\r
22 const char * const  app_version     = "1.0";\r
23 \r
24 const Appoptions    opt_version     = 1;\r
25 const Appoptions    opt_help        = 1 << 1;\r
26 const Appoptions    opt_verbose     = 1 << 2;\r
27 const Appoptions    opt_output      = 1 << 3;\r
28 const Appoptions    opt_input       = 1 << 4;\r
29 \r
30 //----------------------------------------------------------------------------\r
31 //  Function prototypes\r
32 //----------------------------------------------------------------------------\r
33 inline bool isSet(const Appoptions &o1, const Appoptions &o2) {\r
34   return (o2 == (o1 & o2));\r
35 }\r
36 \r
37 Appoptions  checkOption(const char * inOptions);\r
38 void        compile(QCompilerCntl * const , QCalcUnits * const);\r
39 void        forceQuit(void);\r
40 void        printUsage(void);\r
41 void        printVersion(void);\r
42 void        errorAndQuit(const char * const inMessage);\r
43 std::string getBaseFilename(const std::string &inString);\r
44 \r
45 //----------------------------------------------------------------------------\r
46 /**\r
47  *  Main\r
48  */\r
49 int main(int argc, char **argv) {\r
50   Appoptions      parameters_ready    = 0;\r
51   QBits           *qBits              = NULL;\r
52   QCompilerCntl   *qCCntl             = NULL;\r
53   QCalcUnits      *qCalcUnits         = NULL;\r
54 \r
55   std::string               tfilename = "";\r
56   std::string               ofilename = "";\r
57   std::string               ifilename = "";\r
58   std::vector<std::string>  args;\r
59 \r
60   if (argc <= 1) {\r
61     printVersion();\r
62     printUsage();\r
63     std::exit(0);\r
64   }\r
65 \r
66   for (int i = 0; i < argc; i++) {\r
67     std::string tmpstr = argv[i];\r
68     args.push_back(tmpstr);\r
69   }\r
70 \r
71   int vec_size = args.size();\r
72   int arg_at   = 1;\r
73 \r
74   while (arg_at < vec_size) {\r
75     if ('-' == args[arg_at][0] && args[arg_at].length() > 1) {\r
76       Appoptions tmpoptions = checkOption(args[arg_at].c_str());\r
77 \r
78       //_____ HELP _____\r
79       if (isSet(tmpoptions, opt_help)) {\r
80         if (isSet(parameters_ready, opt_help)) {\r
81           errorAndQuit("-h option can be used only once.");\r
82         } else {\r
83           parameters_ready |= opt_help;\r
84         }\r
85       }\r
86 \r
87       //_____ VERBOSE _____\r
88       if (isSet(tmpoptions, opt_verbose)) {\r
89         if (isSet(parameters_ready, opt_verbose)) {\r
90           errorAndQuit("-v option can be used only once.");\r
91         } else {\r
92           parameters_ready |= opt_verbose;\r
93         }\r
94       }\r
95 \r
96       //_____ VERSION _____\r
97       if (isSet(tmpoptions, opt_version)) {\r
98         if (isSet(parameters_ready, opt_version)) {\r
99           errorAndQuit("-V option can be used only once.");\r
100         } else {\r
101           parameters_ready |= opt_version;\r
102         }\r
103       }\r
104 \r
105       //_____ OUTPUT _____\r
106       if (isSet(tmpoptions, opt_output)) {\r
107         if (isSet(parameters_ready, opt_output)) {\r
108           errorAndQuit("-o option can be used only once.");\r
109         }\r
110 \r
111         if (arg_at < vec_size - 1) {\r
112           ofilename         = args[++arg_at];\r
113           parameters_ready |= opt_output;\r
114         } else {\r
115           errorAndQuit("too few parameters.");\r
116         }\r
117       }\r
118       arg_at++;\r
119       continue;\r
120     }\r
121 \r
122     if (isSet(parameters_ready, opt_input)) {\r
123       errorAndQuit("too many input file.");\r
124     } else {\r
125       ifilename = args[arg_at];\r
126       parameters_ready |= opt_input;\r
127     }\r
128     arg_at++;\r
129   }\r
130 \r
131   //_____ VERSION _____\r
132   if (isSet(parameters_ready, opt_version)) printVersion();\r
133 \r
134   //_____ HELP ________\r
135   if (isSet(parameters_ready, opt_help))    printUsage();\r
136 \r
137   //_____ OUTPUT ______\r
138   if (isSet(parameters_ready, opt_input)) {\r
139     if (false == isSet(parameters_ready, opt_output)) {\r
140       ofilename = getBaseFilename(ifilename)+".qdt";\r
141       tfilename = getBaseFilename(ifilename)+".txt";\r
142     }\r
143 \r
144     std::cerr << "Reading " << ifilename << std::endl;\r
145     std::ifstream ifs(ifilename.c_str(), std::ios::in);\r
146     if (!ifs) {\r
147       std::cerr << "Cannot open the input file.\n";\r
148       forceQuit();\r
149     }\r
150 \r
151     if (isSet(parameters_ready, opt_verbose)) {\r
152       std::cerr << "Input  file: " << ifilename << "\n"\r
153                 << "Output file: " << ofilename << "\n";\r
154     }\r
155 \r
156     try {\r
157       qCCntl      = new QCompilerCntl(ifs);\r
158       qCalcUnits  = new QCalcUnits(0);\r
159     } catch (const std::bad_alloc ex) {\r
160       std::cerr << "Failed in memory allocation.\n";\r
161       forceQuit();\r
162     }\r
163 \r
164     compile(qCCntl, qCalcUnits);\r
165 \r
166     try {\r
167       qBits       = new QBits(qCCntl->GetNumberOfQBits());\r
168     } catch (const std::bad_alloc &ex) {\r
169       std::cerr << "Failed in memory allocation.\n"\r
170                 << "This .mcd code needs at least "\r
171                 << (1 << (qCCntl->GetNumberOfQBits() + 1 + 3))\r
172                 << " bytes to allocate the qubits.\n";\r
173       forceQuit();\r
174     }\r
175 \r
176     for (unsigned int i = 0; i < qCalcUnits->size(); i++) {\r
177       if (isSet(parameters_ready, opt_verbose)) {\r
178         std::cerr << std::setw(15) << std::left\r
179                   << (*qCalcUnits)[i]->GetName()\r
180                   << " ... ";\r
181       }\r
182       (*qCalcUnits)[i]->Calc(qBits);\r
183       if (isSet(parameters_ready, opt_verbose)) {\r
184         std::cerr << "OK\n";\r
185       }\r
186     }\r
187 \r
188     //TODO: *.qdt file cannot be imported by QCAD.\r
189     //std::cerr << "Saving " << ofilename << std::endl;\r
190     //qBits->SaveToFile(ofilename.c_str());\r
191     std::cerr << "Saving " << tfilename << std::endl;\r
192     qBits->SaveAsText(tfilename.c_str());\r
193     delete qBits;\r
194     delete qCCntl;\r
195 \r
196     for (unsigned int i = 0; i < qCalcUnits->size(); i++) {\r
197       delete (*qCalcUnits)[i];\r
198     }\r
199     delete qCalcUnits;\r
200     ifs.close();\r
201 \r
202     std::cerr << "done." << std::endl;\r
203   }\r
204 \r
205   return 0;\r
206 }\r
207 \r
208 //----------------------------------------------------------------------------\r
209 /**\r
210  *  Check options\r
211  */\r
212 Appoptions checkOption(const char * inOptions) {\r
213   Appoptions retVal = 0;\r
214   inOptions++;\r
215   while ('\0' != *inOptions) {\r
216     switch (*inOptions) {\r
217     case 'h':\r
218       retVal |= opt_help;\r
219       break;\r
220     case 'o':\r
221       retVal |= opt_output;\r
222       break;\r
223     case 'v':\r
224       retVal |= opt_verbose;\r
225       break;\r
226     case 'V':\r
227       retVal |= opt_version;\r
228       break;\r
229     default:\r
230       errorAndQuit("there are some invalid options.");\r
231       break;\r
232     }\r
233     inOptions++;\r
234   }\r
235   return retVal;\r
236 }\r
237 \r
238 //----------------------------------------------------------------------------\r
239 /**\r
240  *\r
241  */\r
242 void compile(QCompilerCntl * const ioCompiler, QCalcUnits * const ioUnits) {\r
243   QCalcUnit *cu = NULL;\r
244   ioUnits->clear();\r
245 \r
246   if (true == ioCompiler->Compile()) {\r
247     unsigned int cntl_size = ioCompiler->GetNumberOfControl();\r
248     for (unsigned int i = 0; i < cntl_size; i++) {\r
249       cu = ioCompiler->AllocateControl(i);\r
250       if (NULL != cu) ioUnits->push_back(cu);\r
251     }\r
252   } else {\r
253     errorAndQuit("Parse error occurred.\n");\r
254   }\r
255 }\r
256 \r
257 //----------------------------------------------------------------------------\r
258 /**\r
259  *  Force quit\r
260  */\r
261 void forceQuit(void) {\r
262   std::cerr << "Quit forced.\n";\r
263   std::exit(1);\r
264 }\r
265 \r
266 //----------------------------------------------------------------------------\r
267 /**\r
268  *  Print usages\r
269  */\r
270 void printUsage(void) {\r
271   std::cout << "Usage: " << app_name\r
272             << " [-hvV] [-o output_file] input_file.mcd\n\n"\r
273             << "    -h  show this help\n"\r
274             << "    -v  verbose mode\n"\r
275             << "    -V  show version\n\n"\r
276             << std::flush;\r
277 }\r
278 \r
279 //----------------------------------------------------------------------------\r
280 /**\r
281  *  Print version\r
282  */\r
283 void printVersion(void) {\r
284   std::cout << "\nCopyright (C) 2002-2003 QCAD project\n"\r
285             << app_name << " version " << app_version << "\n\n"\r
286             << std::flush;\r
287 }\r
288 \r
289 //----------------------------------------------------------------------------\r
290 /**\r
291  *  Show message and quit\r
292  */\r
293 void errorAndQuit(const char * const inMessage) {\r
294   std::cerr << "Error: " << inMessage << "\n\n";\r
295   printUsage();\r
296   forceQuit();\r
297 }\r
298 \r
299 //----------------------------------------------------------------------------\r
300 /**\r
301  *  get basename of filename\r
302  */\r
303 std::string getBaseFilename(const std::string &inString) {\r
304   std::string retString = "";\r
305 \r
306   unsigned int posdot = inString.rfind(".", inString.length() - 1);\r
307 \r
308   if (posdot == std::string::npos) {\r
309     retString = inString;\r
310   } else {\r
311     retString.assign(inString, 0, posdot);\r
312   }\r
313   return retString;\r
314 }\r
315 //----------------------------------------------------------------------------\r