00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039 #include <ArrayGraph.h>
00040 #include <Vertex.h>
00041 #include <AnalysisModel.h>
00042 #include <DOF_Group.h>
00043 #include <FE_Element.h>
00044
00045 ArrayGraph::ArrayGraph(int arraySize)
00046 :numVertex(0), numEdge(0), sizeVertices(arraySize), lastEmpty(0),
00047 theVertices(0), myIter(*this)
00048 {
00049
00050 theVertices = new Vertex *[arraySize];
00051 if (theVertices == 0) {
00052 opserr << "Warning ArrayGraph::ArrayGraph";
00053 opserr << " - no contiguous memory block big enough available\n";
00054 sizeVertices = 0;
00055 }
00056
00057
00058
00059 for (int i=0; i<arraySize; i++)
00060 theVertices[i] = 0;
00061 }
00062
00063 ArrayGraph::~ArrayGraph()
00064 {
00065
00066
00067 if (theVertices != 0) {
00068 for (int i=0; i<numVertex; i++)
00069 if (theVertices[i] != 0)
00070 delete theVertices[i];
00071 delete [] theVertices;
00072 }
00073 }
00074
00075
00076
00077
00078
00079
00080
00081
00082
00083
00084
00085
00086
00087
00088
00089
00090
00091 bool
00092 ArrayGraph::addVertex(Vertex *vertexPtr)
00093 {
00094
00095 if (vertexPtr == 0) {
00096 opserr << "WARNING ArrayGraph::addVertex";
00097 opserr << " - attempting to add a NULL vertex*\n";
00098 return false;
00099 }
00100
00101 if (vertexPtr->getDegree() != 0) {
00102 const ID &adjacency = vertexPtr->getAdjacency();
00103 int size = adjacency.Size();
00104 for (int i=0; i<size; i++) {
00105 Vertex *other = this->getVertexPtr(adjacency(i));
00106 if (other == 0) {
00107 opserr << "WARNING ArrayGraph::addVertex";
00108 opserr << " - vertex with adjacent vertex not in graph\n";
00109 return false;
00110 }
00111 }
00112 }
00113
00114
00115 if (numVertex == sizeVertices) {
00116
00117 int newSize = sizeVertices*2;
00118 Vertex **newVertices = new Vertex *[newSize];
00119
00120 if (newVertices == 0) {
00121 opserr << "WARNING ArrayGraph::addVertex";
00122 opserr << " - out of contiguous memory could not create a new array";
00123 delete vertexPtr;
00124 return false;
00125 }
00126
00127
00128 for (int i=0; i<sizeVertices; i++)
00129 newVertices[i] = theVertices[i];
00130 for (int j=sizeVertices; j<newSize; j++)
00131 newVertices[j] = 0;
00132
00133 delete [] theVertices;
00134
00135 theVertices = newVertices;
00136 sizeVertices = newSize;
00137 }
00138
00139
00140 int vertexTag = vertexPtr->getTag();
00141 if ((vertexTag >= 0) && (vertexTag < sizeVertices) &&
00142 (theVertices[vertexTag] == 0)) {
00143
00144 theVertices[vertexTag]= vertexPtr;
00145 numVertex++;
00146 return 0;
00147
00148 } else {
00149
00150
00151
00152 for (int i=0; i<sizeVertices; i++)
00153 if (theVertices[i] == 0) {
00154
00155 lastEmpty = i+1;
00156 theVertices[i] = vertexPtr;
00157 numVertex++;
00158 return true;
00159 }
00160 }
00161
00162
00163
00164 return false;
00165 }
00166
00167
00168
00169
00170
00171
00172
00173
00174 Vertex *
00175 ArrayGraph::getVertexPtr(int vertexTag)
00176 {
00177
00178 if ((vertexTag >= 0) && (vertexTag < sizeVertices) &&
00179 (theVertices[vertexTag] != 0) &&
00180 (theVertices[vertexTag]->getTag() == vertexTag)) {
00181
00182 return theVertices[vertexTag];
00183 }
00184
00185
00186
00187 else
00188 for (int i=0; i<sizeVertices; i++)
00189 if ((theVertices[i] != 0) &&
00190 (theVertices[i]->getTag() == vertexTag)){
00191
00192 return theVertices[i];
00193 }
00194
00195
00196
00197 return 0;
00198 }
00199
00200
00201
00202
00203
00204
00205
00206
00207
00208 int
00209 ArrayGraph::addEdge(int vertexTag, int otherVertexTag)
00210 {
00211
00212
00213 Vertex *vertex1 = this->getVertexPtr(vertexTag);
00214 Vertex *vertex2 = this->getVertexPtr(otherVertexTag);
00215 if ((vertex1 == 0) || (vertex2 == 0))
00216 return -1;
00217
00218
00219 int result;
00220 if ((result = vertex1->addEdge(otherVertexTag)) == 0)
00221 if ((result = vertex2->addEdge(vertexTag)) == 0)
00222 numEdge++;
00223
00224 return result;
00225 }
00226
00227
00228
00229
00230
00231 VertexIter &
00232 ArrayGraph::getVertices(void)
00233 {
00234
00235 myIter.reset();
00236 return myIter;
00237 }
00238
00239
00240 int
00241 ArrayGraph::getNumVertex(void) const
00242 {
00243 return numVertex;
00244 }
00245
00246
00247 int
00248 ArrayGraph::getNumEdge(void) const
00249 {
00250 return numEdge;
00251 }
00252
00253
00254 int
00255 ArrayGraph::getArraySize(void) const
00256 {
00257 return sizeVertices;
00258 }
00259
00260
00261 void
00262 ArrayGraph::Print(OPS_Stream &s) const
00263 {
00264 s << numVertex << " " << numEdge << endln;
00265
00266 Vertex *vertexPtr;
00267
00268
00269
00270 for (int i=0; i<sizeVertices; i++) {
00271 vertexPtr = theVertices[i];
00272 if (vertexPtr != 0)
00273 vertexPtr->Print(s);
00274 }
00275 }
00276
00277 OPS_Stream &operator<<(OPS_Stream &s, const ArrayGraph &M)
00278 {
00279 M.Print(s);
00280 return s;
00281 }
00282